From ea6fa901c8a566f53faeb4a1704a301875ec11fc Mon Sep 17 00:00:00 2001
From: Harrison Chase <hw.chase.17@gmail.com>
Date: Thu, 13 Jul 2023 11:43:14 -0700
Subject: [PATCH] cr

---
 langchain/schema/__init__.py | 2 +-
 langchain/schema/chain.py    | 3 ++-
 langchain/schema/output.py   | 2 ++
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/langchain/schema/__init__.py b/langchain/schema/__init__.py
index 1e660248211..83def34c446 100644
--- a/langchain/schema/__init__.py
+++ b/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__ = [
diff --git a/langchain/schema/chain.py b/langchain/schema/chain.py
index 6ec7c4d9554..9ddb72f4946 100644
--- a/langchain/schema/chain.py
+++ b/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__)
 
diff --git a/langchain/schema/output.py b/langchain/schema/output.py
index c085a495249..49ace2ea4ae 100644
--- a/langchain/schema/output.py
+++ b/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."""