Update app.py

app.py

@@ -7,7 +7,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langgraph.graph import StateGraph, MessagesState, START
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_community.document_loaders import WikipediaLoader
-from langchain_community.tools import TavilySearchResults, DuckDuckGoSearchRun
+from langchain_community.tools import TavilySearchResults
 import operator
 from typing import Annotated
 from typing_extensions import TypedDict
@@ -36,7 +36,7 @@ class BasicAgent:
         input_state = {"question": question}
         # Run the graph
         answer = self.graph.invoke(input_state)
-        print(f"Agent returning fixed answer: {answer.content}")
+        print(f"Agent returning fixed answer: {answer}")
         return answer
 
 class State(TypedDict):
@@ -96,7 +96,7 @@ def generate_answer(state):
     answer = llm.invoke([SystemMessage(content=final_instruction)] + [HumanMessage(content=f"Answer the question: {question}")])
 
     # Append it to state
-    return {"answer": answer.content}
+    return {"answer": answer}
 
 
 builder = StateGraph(State)