Update agent_langchain.py

agent_langchain.py

@@ -29,15 +29,6 @@ ROUTING_URL = os.environ.get("ROUTING_URL")
 SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
 FIREBASE_CREDS_PATH = os.environ.get("FIREBASE_CREDS_PATH")
 
-import json
-
-# If Firebase credentials are stored as a secret string, write them to /tmp
-if os.environ.get("FIREBASE_CREDS_JSON"):
-    creds_path = "/tmp/firebase_creds.json"
-    with open(creds_path, "w") as f:
-        f.write(os.environ["FIREBASE_CREDS_JSON"])
-    os.environ["FIREBASE_CREDS_PATH"] = creds_path
-
 # Initialize Firebase
 db = None
 if FIREBASE_CREDS_PATH and os.path.exists(FIREBASE_CREDS_PATH):
@@ -159,7 +150,7 @@ def save_ticket_to_firestore(ticket_data: Dict[str, Any]):
 
 # Gemini LLM
 llm = ChatGoogleGenerativeAI(
-    model="gemini-2.5-flash",
+    model="gemini-2.0-flash-exp",
     temperature=0.3,
     google_api_key=GEMINI_API_KEY
 )
@@ -229,7 +220,6 @@ GUIDING PRINCIPLES:
 4. **Escalate wisely** - Only escalate when truly necessary (low KB confidence, complex issues, or failed solutions)
 5. **Maintain context** - Remember previous conversation history when handling follow-ups
 6. **Be empathetic** - Users are frustrated when things break; be professional and supportive
-7. **Be Useful** - If low KB confidence , then try answering the query using general Knowledge and escalate the issue.
 
 DECISION FRAMEWORK:
 - For a NEW ticket: You might need to classify, route, search KB - but decide based on what's needed
@@ -266,10 +256,10 @@ agent_executor = AgentExecutor(
     agent=agent,
     tools=tools,
     verbose=True,
-    max_iterations=8,  # Increased for more autonomy
-    handle_parsing_errors=True,
+    max_iterations=6,  # Reduced to prevent loops
+    handle_parsing_errors="Check your output and make sure it conforms to the format instructions!",
     return_intermediate_steps=True,
-    early_stopping_method="generate"  # Allow agent to decide when to stop
+    early_stopping_method="force"  # Force stop if max iterations reached
 )
 
 # Main Processing Function