linting litellm endpoint

litellm_endpoint.py

@@ -8,7 +8,6 @@ from smolagents import (
     DuckDuckGoSearchTool,
     load_tool,
     tool,
-    OpenAIServerModel,
     LiteLLMModel,
 )
 
@@ -47,18 +46,11 @@ def get_current_time_in_timezone(timezone: str) -> str:
 search_tool = DuckDuckGoSearchTool()
 final_answer = FinalAnswerTool()
 
-# 初始化 OpenAI 模型
-# model = LiteLLMModel(
-#     model_id="gpt-4o",
-#     api_base="https://api.openai.com/v1",
-#     api_key=os.environ["OPENAI_API_KEY"],
-# )
-
 # 官方文件寫法
 # https://github.com/samwit/smolagents_examples/blob/main/smolagent_gemini.py
 
 model = LiteLLMModel(
-    model_id="gemini/gemini-2.5-flash-preview-04-17",
+    model_id="gemini/gemini-2.0-flash-exp",
     api_key=os.getenv("GEMINI_API_KEY"))
 
 # 載入圖像產生工具