Update app.py

app.py

@@ -712,10 +712,41 @@ system_prompt = f"""
 {instruction}
 """
 
-model = smolagents.OpenAIServerModel(
-    model_id="gpt-4.1-mini",
-    api_key=OPENAI_API,
-)
+client = openai.OpenAI(api_key=OPENAI_API)
+
+class MyOpenAIChatModel:
+    def __init__(self, model_name: str = "gpt-4.1-mini"):
+        self.model_name = model_name
+
+    def __call__(self, messages, stop_sequences=None, **kwargs):
+        """
+        Adapter so smolagents' CodeAgent can call the OpenAI client.
+
+        CodeAgent passes `messages` as a list of either:
+        - dicts with 'role' and 'content'
+        - or Message-like objects with .role and .content
+        """
+        normalized = []
+        for m in messages:
+            if isinstance(m, dict):
+                normalized.append(m)
+            else:
+                role = getattr(m, "role", None)
+                content = getattr(m, "content", None)
+                if role is None or content is None:
+                    raise ValueError(
+                        f"Unexpected message type: {type(m)}; cannot extract role/content"
+                    )
+                normalized.append({"role": role, "content": content})
+
+        resp = client.chat.completions.create(
+            model=self.model_name,
+            messages=normalized,
+            stop=stop_sequences,
+        )
+        return resp.choices[0].message.content
+
+model = MyOpenAIChatModel("gpt-4.1-mini")
 
 agent = smolagents.CodeAgent(
     tools=[
@@ -769,7 +800,8 @@ with gr.Blocks() as demo:
         examples=[
             "Who is Ed?",
             "Who to talk to to create a wooden table?",
-            "How can I access laser cutter?"
+            "How can I access the laser cutters?",
+            "How can I go to the welding space?"
         ],
         inputs=[inp],
     )