Update app.py

app.py

@@ -17,17 +17,6 @@ from langchain_huggingface.llms import HuggingFacePipeline
 from langchain_ollama import ChatOllama
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.rate_limiters import InMemoryRateLimiter
-from smolagents import (
-    InferenceClientModel, LiteLLMModel, OpenAIServerModel, TransformersModel,
-    CodeAgent,
-    DuckDuckGoSearchTool,
-    HfApiModel,
-    LiteLLMModel,
-    OpenAIServerModel,
-    PythonInterpreterTool,
-    tool,
-    InferenceClientModel, ToolCallingAgent
-)
 
 
 # (Keep Constants as is)
@@ -117,9 +106,11 @@ class BasicAgent:
             SystemMessage(content=system_prompt),
             HumanMessage(content=question)]
         response = self.graph.invoke({"messages": messages})
-        print(f"RESPONSE {response}")
         response = response['messages'][-1].content
         print(f"Agent returning answer: {response}")
+        final_answer_idx = response.find("FINAL ANSWER: ")
+        if final_answer_idx != -1:
+            return response[final_answer_idx + len("FINAL ANSWER: "):]
         return response