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.""" """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.agent import Agent
from langchain.agents.self_ask_with_search.prompt import PROMPT 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}" 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:" followup = "Follow up:"
if "\n" not in text: if "\n" not in text:
last_line = text last_line = text
@ -35,8 +35,8 @@ class SelfAskWithSearchAgent(Agent):
if followup not in last_line: if followup not in last_line:
finish_string = "So the final answer is: " finish_string = "So the final answer is: "
if finish_string not in last_line: if finish_string not in last_line:
raise ValueError("We should probably never get here") return None
return "Final Answer", text[len(finish_string) :] return "Final Answer", last_line[len(finish_string) :]
if ":" not in last_line: if ":" not in last_line:
after_colon = last_line after_colon = last_line
@ -50,6 +50,9 @@ class SelfAskWithSearchAgent(Agent):
return "Intermediate Answer", after_colon return "Intermediate Answer", after_colon
def _fix_text(self, text: str) -> str:
return text + "\nSo the final answer is:"
@property @property
def observation_prefix(self) -> str: def observation_prefix(self) -> str:
"""Prefix to append the observation with.""" """Prefix to append the observation with."""