update codes

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 import random
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -13,10 +13,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        # model_id를 지정하지 않고 프레임워크 기본 빌트인 모델을 사용하며, 
-        # 안전하게 Secrets의 HF_TOKEN만 인스턴스에 직접 연동
-        model = InferenceClientModel(
-            token=os.getenv("HF_TOKEN")
+        # Google AI Studio에서 발급받은 API 키를 환경 변수에서 로드
+        gemini_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
+        
+        # smolagents의 LiteLLM 커넥터를 사용하여 Google AI Studio의 고성능 무료 모델 바인딩
+        model = LiteLLMModel(
+            model_id="gemini/gemini-1.5-flash",
+            api_key=gemini_key
         )
                 
         search_tool = DuckDuckGoSearchTool()
@@ -179,10 +182,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    space_host_startup = os.getenv("SPACE_HOST")
-    space_id_startup = os.getenv("SPACE_ID")
-    if space_host_startup:
-        print(f"✅ SPACE_HOST found: {space_host_startup}")
-    if space_id_startup:
-        print(f"✅ SPACE_ID found: {space_id_startup}")
     demo.launch(debug=True, share=False)

requirements.txt

@@ -1,5 +1,6 @@
 smolagents>=1.1.0
 duckduckgo-search>=6.2.1
+litellm
 pandas
 requests
 gradio