Update app.py

app.py

@@ -1,7 +1,6 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
-from smolagents.models import LiteLLMModel  # или другой подходящий класс
+from smolagents.models import LiteLLMModel  # ← ПРАВИЛЬНЫЙ ИМПОРТ
 import datetime
-import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
@@ -9,17 +8,13 @@ from Gradio_UI import GradioUI
 
 # ==================== ИНСТРУМЕНТЫ ====================
 
-# 1. Инструмент для поиска в интернете
+# 1. Поиск в интернете
 search_tool = DuckDuckGoSearchTool()
 
-# 2. Инструмент для времени (уже есть)
+# 2. Время в часовом поясе
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Получает текущее время в указанном часовом поясе.
-    
-    Args:
-        timezone: Часовой пояс (например, 'Europe/Moscow', 'America/New_York')
-    """
+    """Получает текущее время в указанном часовом поясе."""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -27,56 +22,58 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Ошибка: неправильный часовой пояс '{timezone}'"
 
-# 3. Ваш собственный инструмент (пример)
+# 3. Простой калькулятор
 @tool
-def get_exchange_rate(currency: str) -> str:
-    """Получает примерный курс валюты к рублю.
-    
-    Args:
-        currency: Код валюты (USD, EUR, GBP)
-    """
-    rates = {
-        "USD": 90.5,
-        "EUR": 98.2,
-        "GBP": 115.7
-    }
-    
-    if currency.upper() in rates:
-        return f"Примерный курс {currency}: {rates[currency.upper()]} руб."
-    else:
-        return f"Курс для {currency} не найден. Доступно: USD, EUR, GBP"
+def calculator(expression: str) -> str:
+    """Вычисляет математическое выражение."""
+    try:
+        # Внимание: eval опасен! Только для демонстрации
+        result = eval(expression)
+        return f"Результат: {result}"
+    except Exception as e:
+        return f"Ошибка вычисления: {str(e)}"
 
-# ==================== СОЗДАНИЕ АГЕНТА ====================
+# ==================== СОЗДАНИЕ МОДЕЛИ ====================
 
 final_answer = FinalAnswerTool()
 
-# Модель LLM
-model = InferenceClientModel(
+# ИСПРАВЛЕНИЕ: Используем LiteLLMModel
+model = LiteLLMModel(
+    model="huggingface/Qwen/Qwen2.5-Coder-32B-Instruct",
+    api_base="https://api-inference.huggingface.co/models",
     max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
+    temperature=0.5
 )
 
-# Загрузка инструмента генерации изображений
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# ==================== ЗАГРУЗКА ИНСТРУМЕНТОВ ====================
+
+# Генерация изображений
+try:
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+except:
+    @tool
+    def image_generation_tool(prompt: str) -> str:
+        """Генерация изображений (заглушка)."""
+        return f"Изображение '{prompt}' временно недоступно"
+
+# ==================== ЗАГРУЗКА ПРОМПТА ====================
 
-# Загрузка промпта
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
-# СОЗДАНИЕ АГЕНТА со всеми инструментами
+
+# ==================== СОЗДАНИЕ АГЕНТА ====================
+
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer,  # ОБЯЗАТЕЛЬНО оставить!
+        final_answer,
         search_tool,
         get_current_time_in_timezone,
-        get_exchange_rate,
+        calculator,
         image_generation_tool
     ],
-    max_steps=6,  # Максимум шагов цикла
-    verbosity_level=2,  # 2 = показывать мысли и действия
+    max_steps=6,
+    verbosity_level=2,
     prompt_templates=prompt_templates
 )