github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-20 03:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve chat error (#1632)

Browse Source
This commit is contained in:
Harrison Chase 2023-03-13 07:43:44 -07:00 committed by GitHub
parent cb646082ba
commit 72b461e257
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/agents/chat/base.py
View File

@ -44,11 +44,14 @@ class ChatAgent(Agent):
def _extract_tool_and_input(self, text: str) -> Optional[Tuple[str, str]]: def _extract_tool_and_input(self, text: str) -> Optional[Tuple[str, str]]:
if FINAL_ANSWER_ACTION in text: if FINAL_ANSWER_ACTION in text:
return "Final Answer", text.split(FINAL_ANSWER_ACTION)[-1].strip() return "Final Answer", text.split(FINAL_ANSWER_ACTION)[-1].strip()
try:
_, action, _ = text.split("```") _, action, _ = text.split("```")
response = json.loads(action.strip()) response = json.loads(action.strip())
return response["action"], response["action_input"] return response["action"], response["action_input"]
except Exception:
raise ValueError(f"Could not parse LLM output: {text}")
@property @property
def _stop(self) -> List[str]: def _stop(self) -> List[str]:
return ["Observation:"] return ["Observation:"]