change to openai

app.py

@@ -18,6 +18,34 @@ import yaml
 #with open("prompts.yaml", 'r') as stream:
 #    prompt_templates = yaml.safe_load(stream)
 
+import os
+from dotenv import load_dotenv
+
+# 1) load your .env (skip if you set it in the shell)
+load_dotenv()
+
+# 2) grab the key
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+# 3) build the Smolagents model
+from smolagents import OpenAIServerModel
+
+model = OpenAIServerModel(
+    model_id="gpt-4.1",                    # or "gpt-3.5-turbo", etc.
+    api_base="https://api.openai.com/v1",      # OpenAI’s standard endpoint
+    api_key=openai_api_key                     # your secret key
+)                                            # :contentReference[oaicite:1]{index=1}
+
+# 4) create a CodeAgent powered by that model
+from smolagents import CodeAgent, DuckDuckGoSearchTool
+
+OpenAIAgent = CodeAgent(
+    tools=[DuckDuckGoSearchTool()],            # any tools you want
+    model=model,
+    # you can allow extra imports if needed:
+    # additional_authorized_imports=["requests", "bs4"],
+)
+
 MyAgent = CodeAgent(
     model= InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
     tools=[],
@@ -70,7 +98,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = OpenAIAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None