Update app.py

app.py

@@ -17,8 +17,11 @@ class Command:
 # Set the API key from Hugging Face secrets
 os.environ["ANTHROPIC_API_KEY"] = os.getenv("ANTHROPIC_API_KEY")
 
-# Claude 3.5 Sonnet model
-llm = ChatAnthropic(model="claude-3-5-sonnet-20240229")
+# Claude 3.5 Sonnet model - remove any default parameters that might be causing issues
+llm = ChatAnthropic(
+    model="claude-3-5-sonnet-20240229",
+    # No additional parameters that might cause conflicts
+)
 
 # Utility to build system prompts
 def make_system_prompt(suffix: str) -> str:
@@ -37,10 +40,8 @@ def research_node(state):
     )
     result = agent.invoke(state)
     last_msg = result["messages"][-1]
-
     # Determine next step
     goto = "chart_generator" if "FINAL ANSWER" not in last_msg.content else "__end__"
-
     result["messages"][-1] = HumanMessage(
         content=last_msg.content,
         name="researcher"
@@ -73,15 +74,13 @@ graph = workflow.compile()
 # LangGraph runner
 def run_langgraph(input_text):
     try:
-        events = graph.stream({"messages": [("user", input_text)]})
+        events = graph.stream({"messages": [HumanMessage(content=input_text)]})
         output = list(events)
         final_response = output[-1]["messages"][-1].content
-
         if "FINAL ANSWER" in final_response:
             # Dummy chart generation
             years = [2020, 2021, 2022, 2023, 2024]
             gdp = [21.4, 22.0, 23.1, 24.8, 26.2]
-
             plt.figure()
             plt.plot(years, gdp, marker="o")
             plt.title("USA GDP Over Last 5 Years")
@@ -90,7 +89,6 @@ def run_langgraph(input_text):
             plt.grid(True)
             plt.tight_layout()
             plt.savefig("gdp_chart.png")
-
             return "Chart generated based on FINAL ANSWER.", "gdp_chart.png"
         else:
             return final_response, None
@@ -110,4 +108,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()