Migrate to OpenAI

app.py

@@ -1,6 +1,7 @@
 import yaml
 import os
 from smolagents import GradioUI, CodeAgent, InferenceClientModel
+from smolagents import OpenAIServerModel
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -14,9 +15,16 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
 
-model = InferenceClientModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-)
+#model = InferenceClientModel(
+#model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+#)
+
+OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
+model = OpenAIServerModel(
+        model_id = "gpt-4o-mini",
+        api_key = OPENAI_API_KEY
+        )
+
 
 web_search = WebSearch()
 visit_webpage = VisitWebpage()

requirements.txt

@@ -1,4 +1,4 @@
 duckduckgo_search
 markdownify
 requests
-smolagents
+smolagents[openai]