Switch Back to GPT 4.1 Mini. QWEN overthinked stuff:D

app.py

@@ -263,7 +263,11 @@ async def run_example():
     ui_logger.info("--- Starting GradioCallback Example ---")
     # --- End Logging Setup ---
 
-    api_key = os.environ.get("NEBIUS_API_KEY")
+    #api_key = os.environ.get("NEBIUS_API_KEY")
+    #model = "openai/Qwen/Qwen3-235B-A22B"
+    #model = "openai/Qwen/Qwen3-30B-A3B-fast"
+    model = "gpt-4.1-mini"
+    api_key = os.environ.get("OPENAI_API_KEY")
     if not api_key:
         ui_logger.error("NEBIUS_API_KEY environment variable not set.")
         return
@@ -287,13 +291,14 @@ async def run_example():
         tools_meta_data[metadata["name"]] = metadata
     template_str = load_template("./prompts/code_agent.yaml")
     system_prompt = render_system_prompt(template_str, tools_meta_data, {}, ["tinyagent","gradio","requests","asyncio"]) + prompt_code_example + prompt_qwen_helper
-    agent = TinyAgent(model="openai/Qwen/Qwen3-235B-A22B", api_key=api_key,
-                      model_kwargs=dict(base_url="https://api.studio.nebius.com/v1/"),
+    agent = TinyAgent(model=model, api_key=api_key,
+                      #model_kwargs=dict(base_url="https://api.studio.nebius.com/v1/"),
             logger=agent_logger,
             system_prompt=system_prompt,
             
             )
-    python_interpreter = PythonCodeInterpreter(log_manager=log_manager,code_tools=tools,pip_packages=["tinyagent-py[all]","requests","cloudpickle"])
+    python_interpreter = PythonCodeInterpreter(log_manager=log_manager,code_tools=tools,pip_packages=["tinyagent-py[all]","requests","cloudpickle"],
+                                               default_python_codes=["import random","import requests","import cloudpickle","import tempfile","import shutil","import asyncio","import logging","import time"])
     agent.add_tool(python_interpreter.run_python)
 
 
@@ -343,6 +348,7 @@ async def run_example():
             share=False,
             prevent_thread_lock=True,  # Critical to not block our event loop
             show_error=True,
+            mcp_server=True,
         )
         ui_logger.info("Gradio interface launched (non-blocking).")
         

helper.py

@@ -24,7 +24,8 @@ prompt_qwen_helper = dedent("""
 Other tools calls and their responses are not visible to the user.
 - run_python is a capable tool, if you need to call a function with different arguments, you can do it in one take, just like you would do in a python code you developed to be executed in one cell of Jupyter Notebook Cell.
 - When Task is not resolvable using functions available in your python enviroment, first think about creating a new function to solve the task, then ask the user about your approach and your code, if user allowed you to use it, then define and execute your custom made function. [It is your super power, you can create functions to solve the task.]
-""")
+- When you are defining a new function, you need to add any needed imports inside the function.
+                            """)
 
 def load_template(path: str) -> str:
     """