update codes

app.py

@@ -4,28 +4,52 @@ import requests
 import inspect
 import pandas as pd
 import random
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
+import time
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# --- RPM Guard: 프리 티어 버스트 방어형 커스텀 모델 어댑터 ---
+class PacedGeminiModel(LiteLLMModel):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.last_call_time = 0.0
+        self.min_interval = 5  # 15 RPM 제한을 안전하게 회피하기 위한 최소 초 단위 간격 고정
+
+    def __call__(self, *args, **kwargs):
+        # 마지막 API 호출 시점으로부터 경과한 시간을 계산하여 정밀 페이싱 락(Lock) 가동
+        now = time.time()
+        elapsed = now - self.last_call_time
+        if elapsed < self.min_interval:
+            sleep_needed = self.min_interval - elapsed
+            time.sleep(sleep_needed)
+            
+        try:
+            response = super().__call__(*args, **kwargs)
+            return response
+        finally:
+            # 실패하든 성공하든 타임스탬프를 갱신하여 연속 리트라이로 인한 버스트 폭증 차단
+            self.last_call_time = time.time()
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         gemini_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
         
-        # 2026년 현재 v1/v1beta 엔드포인트에서 가장 안정적으로 서빙되는 gemini-2.5-flash 매핑
-        self.model = LiteLLMModel(
+        # 기본 LiteLLMModel 대신 정밀 감속기가 내장된 PacedGeminiModel로 바인딩
+        self.model = PacedGeminiModel(
             model_id="gemini/gemini-2.5-flash",
             api_key=gemini_key
         )
                 
         self.search_tool = DuckDuckGoSearchTool()
+        self.visit_tool = VisitWebpageTool()
                 
-        # 45점 베이스라인의 다단계 계획 수립 및 내장 인터프리터 툴 세팅 구조 유지
+        # 계획 주기 가동 및 툴 가방 바인딩
         self.alfred = CodeAgent(
-            tools=[self.search_tool], 
+            tools=[self.search_tool, self.visit_tool], 
             model=self.model,
             add_base_tools=True,  
             planning_interval=3   
@@ -33,6 +57,8 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         try:
+            # 연속된 문제 풀이 사이에도 프리 티어 게이트웨이가 휴식할 수 있도록 5초 전처리 대기
+            time.sleep(5.0)
             result = self.alfred.run(question)
             if result is None:
                 return "unknown"