blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'ankush/callbackhandler' into harrison/callback-updates

Browse Source
This commit is contained in:
Harrison Chase 2023-01-03 12:31:45 -08:00
commit 3086b752a3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/agents/agent.py
View File

@ -73,7 +73,7 @@ class Agent(BaseModel):
parsed_output = self._extract_tool_and_input(full_output) parsed_output = self._extract_tool_and_input(full_output)
tool, tool_input = parsed_output tool, tool_input = parsed_output
if tool == self.finish_tool_name: if tool == self.finish_tool_name:
return AgentFinish(full_output, {"output": tool_input}) return AgentFinish({"output": tool_input}, full_output)
return AgentAction(tool, tool_input, full_output) return AgentAction(tool, tool_input, full_output)
def prepare_for_new_call(self) -> None: def prepare_for_new_call(self) -> None:

2
langchain/schema.py
View File

@ -14,8 +14,8 @@ class AgentAction(NamedTuple):
class AgentFinish(NamedTuple): class AgentFinish(NamedTuple):
"""Agent's return value.""" """Agent's return value."""
log: str
return_values: dict return_values: dict
log: str
class Generation(NamedTuple): class Generation(NamedTuple):