Final_Assignment_Template

Sleeping

AksharaSachin commited on May 2

Commit

3e008e6

verified ·

1 Parent(s): 08b792c

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -36,7 +36,7 @@ def use_wikipedia_tool(query: str) -> str:
     if result:
         return {"Wikipedia_summary": result}
     else:
-        return f"Sorry, I couldn't find any information on '{query}' in Wikipedia."
 def build_agent():
     # llm = ChatOllama(model="llama3.1")

     if result:
         return {"Wikipedia_summary": result}
     else:
+        return {"response" : f"Sorry, I couldn't find any information on '{query}' in Wikipedia."}
 def build_agent():
     # llm = ChatOllama(model="llama3.1")