github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 00:48:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

wolfram alpha improvements (#591)

Browse Source 
Co-authored-by: Nicolas <nicolascamara29@gmail.com>
This commit is contained in:
Harrison Chase 2023-01-12 06:09:12 -08:00 committed by GitHub
parent a5ee7de650
commit 4c6c5f0391
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
langchain/utilities/wolfram_alpha.py
View File

@ -52,8 +52,15 @@ class WolframAlphaAPIWrapper(BaseModel):
def run(self, query: str) -> str: def run(self, query: str) -> str:
"""Run query through WolframAlpha and parse result.""" """Run query through WolframAlpha and parse result."""
res = self.wolfram_client.query(query) res = self.wolfram_client.query(query)
# Includes only text from the response
answer = next(res.results).text try:
assumption = next(res.pods).text
answer = next(res.results).text
except StopIteration:
return "Wolfram Alpha wasn't able to answer it"
if answer is None or answer == "": if answer is None or answer == "":
# We don't want to return the assumption alone if answer is empty
return "No good Wolfram Alpha Result was found" return "No good Wolfram Alpha Result was found"
return answer else:
return f"Assumption: {assumption} \nAnswer: {answer}"