initial commit

venv/Lib/site-packages/langchain_classic/agents/schema.py

@@ -0,0 +1,37 @@
+from typing import Any
+
+from langchain_core.agents import AgentAction
+from langchain_core.prompts.chat import ChatPromptTemplate
+from typing_extensions import override
+
+
+class AgentScratchPadChatPromptTemplate(ChatPromptTemplate):
+    """Chat prompt template for the agent scratchpad."""
+
+    @classmethod
+    @override
+    def is_lc_serializable(cls) -> bool:
+        return False
+
+    def _construct_agent_scratchpad(
+        self,
+        intermediate_steps: list[tuple[AgentAction, str]],
+    ) -> str:
+        if len(intermediate_steps) == 0:
+            return ""
+        thoughts = ""
+        for action, observation in intermediate_steps:
+            thoughts += action.log
+            thoughts += f"\nObservation: {observation}\nThought: "
+        return (
+            f"This was your previous work "
+            f"(but I haven't seen any of it! I only see what "
+            f"you return as final answer):\n{thoughts}"
+        )
+
+    def _merge_partial_and_user_variables(self, **kwargs: Any) -> dict[str, Any]:
+        intermediate_steps = kwargs.pop("intermediate_steps")
+        kwargs["agent_scratchpad"] = self._construct_agent_scratchpad(
+            intermediate_steps,
+        )
+        return kwargs