Update gemini_agent.py

gemini_agent.py

@@ -3,6 +3,10 @@ import time
 from dotenv import load_dotenv
 from typing import TypedDict, Annotated, Optional
 
+# Set USER_AGENT early to identify requests (fixes warning)
+os.environ["USER_AGENT"] = "Hennessy2025-Final-Agent"
+print(f"USER_AGENT set to: {os.getenv('USER_AGENT')}")
+
 from langgraph.prebuilt import ToolNode, tools_condition
 from langgraph.graph import StateGraph, START
 from langgraph.graph.message import add_messages
@@ -98,7 +102,7 @@ class GEMINI_AGENT:
         keyword = "FINAL ANSWER: "
         index = text.find(keyword)
         if index != -1:
-            return text[index + len(keyword):]
+            return text[index + len(keyword):].strip()
         else:
             return ""
 
@@ -118,7 +122,7 @@ class GEMINI_AGENT:
             try:
                 response = self.app.invoke({"messages": messages, "input_file": None})
                 final_ans = self.extract_after_final_answer(response['messages'][-1].content)
-                time.sleep(60) # avoid rate limit
+                time.sleep(60)  # avoid rate limit
                 return final_ans
             except Exception as e:
                 sleep_time = base_sleep * (attempt + 1)
@@ -128,4 +132,4 @@ class GEMINI_AGENT:
                     time.sleep(sleep_time)
                     continue
                 return f"Error processing query after {max_retries} attempts: {str(e)}"
-        return "This is a default answer."
+        return "This is a default answer."