Upload agent.py

agent.py

@@ -21,7 +21,7 @@ def web_search(query: str) -> str:
     search_docs = TavilySearchResults(max_results=3).invoke(input=query)
     formatted_search_docs = "\n\n---\n\n".join(
     [
-        f'<Document source="{doc["title"]}" page="{doc.get("url", "")}"/>\n{doc['content']}\n</Document>'
+        f'<Document source="{doc["title"]}" page="{doc.get("url", "")}"/>\n{doc["content"]}\n</Document>'
         for doc in search_docs
     ])
     return {"web_results": formatted_search_docs}
@@ -137,7 +137,7 @@ def build_graph():
 
 if __name__=="__main__":
     system_prompt = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
-    question = "whats 2 + 15?"
+    question = "whats 2 + 30?"
     question = f"{system_prompt} \
     The question: \
     {question} \