From 4c6c5f0391bf62208d1083cc8093b5e5a37e3799 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Thu, 12 Jan 2023 06:09:12 -0800 Subject: [PATCH] wolfram alpha improvements (#591) Co-authored-by: Nicolas --- langchain/utilities/wolfram_alpha.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/langchain/utilities/wolfram_alpha.py b/langchain/utilities/wolfram_alpha.py index 085ee5924c1..d64ce410e75 100644 --- a/langchain/utilities/wolfram_alpha.py +++ b/langchain/utilities/wolfram_alpha.py @@ -52,8 +52,15 @@ class WolframAlphaAPIWrapper(BaseModel): def run(self, query: str) -> str: """Run query through WolframAlpha and parse result.""" 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 == "": + # We don't want to return the assumption alone if answer is empty return "No good Wolfram Alpha Result was found" - return answer + else: + return f"Assumption: {assumption} \nAnswer: {answer}"