github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 08:33:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix self ask w search (#206)

Browse Source
This commit is contained in:
Harrison Chase 2022-11-26 15:15:43 -08:00 committed by GitHub
parent eae358810b
commit 287f1857ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
langchain/agents/self_ask_with_search/base.py
View File

@ -1,5 +1,5 @@
"""Chain that does self ask with search."""
from typing import Any, ClassVar, List, Tuple
from typing import Any, ClassVar, List, Optional, Tuple
from langchain.agents.agent import Agent
from langchain.agents.self_ask_with_search.prompt import PROMPT
@ -25,7 +25,7 @@ class SelfAskWithSearchAgent(Agent):
f"Tool name should be Intermediate Answer, got {tool_names}"
)
def _extract_tool_and_input(self, text: str) -> Tuple[str, str]:
def _extract_tool_and_input(self, text: str) -> Optional[Tuple[str, str]]:
followup = "Follow up:"
if "\n" not in text:
last_line = text
@ -35,8 +35,8 @@ class SelfAskWithSearchAgent(Agent):
if followup not in last_line:
finish_string = "So the final answer is: "
if finish_string not in last_line:
raise ValueError("We should probably never get here")
return "Final Answer", text[len(finish_string) :]
return None
return "Final Answer", last_line[len(finish_string) :]
if ":" not in last_line:
after_colon = last_line
@ -50,6 +50,9 @@ class SelfAskWithSearchAgent(Agent):
return "Intermediate Answer", after_colon
def _fix_text(self, text: str) -> str:
return text + "\nSo the final answer is:"
@property
def observation_prefix(self) -> str:
"""Prefix to append the observation with."""