Enhance AI chat functionality by increasing maximum tool calling rounds from 4 to 25 and adding a warning for reaching the maximum limit. Updated system message to emphasize schema discovery before querying.

Author: robin-rpr

utils/ai.py

@@ -62,15 +62,16 @@ def chat(messages: List[Dict[str, str]], model: str = "openrouter/auto") -> Gene
             'You are a network analysis assistant with access to routing database tools. '
             'When useful, call the available tools to fetch precise data before answering. '
             'If the user asks for ASN, prefixes, neighbors, flaps, or stats, use tools. '
-            'Think through problems step by step and use tools when needed to get accurate data.'
+            'Think through problems step by step and use tools when needed to get accurate data. '
+            'Always start with schema discovery tools to understand the database structure before querying.'
         )
     }
 
     conversation = [system_message] + list(messages)
     tools = _build_tools_schema()
 
-    # Maximum 4 rounds of tool calling
-    for round_num in range(4):
+    # Maximum rounds of tool calling
+    for round_num in range(25):
         try:
             # Create chat completion with tools
             response = client.chat.completions.create(
@@ -176,6 +177,11 @@ def chat(messages: List[Dict[str, str]], model: str = "openrouter/auto") -> Gene
             yield {'type': 'error', 'message': str(e)}
             return
 
+    # Check if we hit the maximum rounds limit
+    if round_num >= max_rounds:
+        app.logger.warning(f"Reached maximum tool calling rounds ({max_rounds})")
+        yield {'type': 'assistant_delta', 'delta': 'I\'ve reached the maximum number of tool calling rounds. Let me provide a response based on the information gathered so far.'}
+    
     # Fallback if no response
     yield {'type': 'assistant_delta', 'delta': 'I apologize, but I encountered an issue processing your request.'}
     yield {'type': 'assistant_done'}