Update app.py

app.py

@@ -235,12 +235,56 @@ def launch_interface():
             custom_search_url=None
         )
         
-        # Override the agent's system prompt for code generation if it supports it
-        # This is a simplification; `smolagents` might need more explicit prompt template modification.
-        if hasattr(agent, 'system_prompt'):
-            agent.system_prompt = SYSTEM_PROMPT_CODE_GEN
-        elif hasattr(agent, 'prompt_templates') and 'code_agent' in agent.prompt_templates:
-            agent.prompt_templates['code_agent']['system_prompt'] = SYSTEM_PROMPT_CODE_GEN
+# Override the agent's system prompt for code generation
+        # Based on the error, smolagents uses prompt_templates for system prompts.
+        # The specific key might vary, but 'user_agent' or 'managed_agent' are common.
+        # Assuming the CodeAgent also has a 'system_prompt' in its prompt_templates
+        # or we can modify the one for its internal 'managed_agent' if applicable.
+
+        # A common pattern for smolagents' CodeAgent might be to modify the 'user_agent'
+        # or 'managed_agent' template that it uses internally.
+        # Let's try modifying the 'managed_agent' task and prepending the system prompt.
+        # If this doesn't fully capture the 'system' role, you might need to inspect
+        # the exact structure of `agent.prompt_templates` in a debugger.
+
+        # Safely try to modify the prompt template used by the CodeAgent.
+        # The exact key for the system prompt might vary (e.g., 'system', 'default', 'user_agent').
+        # A common structure is `agent.prompt_templates["user_agent"]["system_prompt"]`
+        # or it might be directly injected in the `task` for agents that don't have a separate system role.
+        
+        # Let's assume the `CodeAgent` itself uses a "default" or similar template
+        # that allows injecting a system prompt, or that its primary instruction
+        # can be adjusted via its 'task' definition.
+        
+        # A more direct way to ensure the system prompt is used by LiteLLMModel
+        # within smolagents is to pass it during agent creation, or if the agent
+        # has an exposed way to set a 'system' message.
+        
+        # Given the error implies direct modification of `prompt_templates`,
+        # let's try to find a suitable place.
+        # Based on smolagents examples, `user_agent` or `managed_agent` templates are often customized.
+        # The `CodeAgent` itself doesn't typically expose a direct `system_prompt` property
+        # that it forwards to the LLM; instead, it uses the prompt templates.
+
+        # The error specifically points to `self.prompt_templates["system_prompt"]`.
+        # This implies there *is* a key "system_prompt" within prompt_templates.
+        # Let's apply SYSTEM_PROMPT_CODE_GEN to this specific key.
+        if hasattr(agent, 'prompt_templates') and "system_prompt" in agent.prompt_templates:
+            # If the agent explicitly exposes a "system_prompt" template key
+            agent.prompt_templates["system_prompt"] = SYSTEM_PROMPT_CODE_GEN
+            print("[DEBUG] Set agent.prompt_templates['system_prompt'] for code generation.")
+        elif hasattr(agent, 'prompt_templates') and 'user_agent' in agent.prompt_templates:
+            # If it's a ToolCallingAgent (which CodeAgent inherits from)
+            # and it has a 'user_agent' template, we can modify its system message.
+            # This is a common pattern for defining the agent's core persona.
+            agent.prompt_templates['user_agent']['system_message'] = SYSTEM_PROMPT_CODE_GEN
+            print("[DEBUG] Set agent.prompt_templates['user_agent']['system_message'] for code generation.")
+        else:
+            print("[WARNING] Could not set system prompt for CodeAgent using known patterns. "
+                  "Agent might not follow code generation instructions optimally.")
+            # Fallback: Prepend to the question for basic instruction if no proper system prompt mechanism
+            # This is not ideal but ensures the instruction is conveyed.
+            query = SYSTEM_PROMPT_CODE_GEN + "\n\n" + query
 
 
         reasoning_buffer = []