cr

langchain/schema/__init__.py

@@ -16,6 +16,7 @@ from langchain.schema.messages import (
     messages_to_dict,
 )
 from langchain.schema.output import (
+    RUN_KEY,
     ChatGeneration,
     ChatResult,
     Generation,
@@ -32,7 +33,6 @@ from langchain.schema.prompt import PromptValue
 from langchain.schema.prompt_template import BasePromptTemplate, format_document
 from langchain.schema.retriever import BaseRetriever
 
-RUN_KEY = "__run"
 Memory = BaseMemory
 
 __all__ = [

langchain/schema/chain.py

@@ -13,7 +13,6 @@ from pydantic import Field, root_validator, validator
 import langchain
 from langchain.load.dump import dumpd
 from langchain.load.serializable import Serializable
-from langchain.schema import RUN_KEY, BaseMemory, RunInfo
 from langchain.schema.callbacks.base import BaseCallbackManager
 from langchain.schema.callbacks.manager import (
     AsyncCallbackManager,
@@ -22,6 +21,8 @@ from langchain.schema.callbacks.manager import (
     CallbackManagerForChainRun,
     Callbacks,
 )
+from langchain.schema.memory import BaseMemory
+from langchain.schema.output import RUN_KEY, RunInfo
 
 logger = logging.getLogger(__name__)
 

langchain/schema/output.py

@@ -9,6 +9,8 @@ from pydantic import BaseModel, root_validator
 from langchain.load.serializable import Serializable
 from langchain.schema.messages import BaseMessage
 
+RUN_KEY = "__run"
+
 
 class Generation(Serializable):
     """A single text generation output."""