api_agent.py

import openai
import requests
import logging
import os
from typing import Optional, Dict, Any

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

class APIAgent:
    def __init__(self):
        self.config = {}
    
    def set_config(self, config: Dict[str, Any]):
        """Configura las claves API desde Gradio"""
        self.config = config or {}
    
    def call_openai(self, prompt: str, is_code: bool = False) -> Optional[str]:
        """Intenta llamar a OpenAI API"""
        api_key = self.config.get("openai_api_key")
        
        if not api_key:
            return None
        
        try:
            openai.api_key = api_key
            
            model = "gpt-3.5-turbo"
            max_tokens = self.config.get("max_tokens", 400)  # Reducido para HF
            temperature = self.config.get("temperature", 0.7)
            
            response = openai.ChatCompletion.create(
                model=model,
                messages=[{"role": "user", "content": prompt}],
                max_tokens=max_tokens,
                temperature=temperature,
                timeout=15  # Timeout reducido
            )
            
            return response.choices[0].message.content.strip()
            
        except Exception as e:
            logger.error(f"❌ Error llamando a OpenAI: {e}")
            return None
    
    def call_deepseek(self, prompt: str, is_code: bool = False) -> Optional[str]:
        """Intenta llamar a DeepSeek API"""
        api_key = self.config.get("deepseek_api_key")
        
        if not api_key:
            return None
        
        try:
            url = "https://api.deepseek.com/v1/chat/completions"
            headers = {
                "Content-Type": "application/json",
                "Authorization": f"Bearer {api_key}"
            }
            
            max_tokens = self.config.get("max_tokens", 400)
            temperature = self.config.get("temperature", 0.7)
            
            data = {
                "model": "deepseek-chat",
                "messages": [{"role": "user", "content": prompt}],
                "max_tokens": max_tokens,
                "temperature": temperature,
                "stream": False
            }
            
            response = requests.post(url, json=data, headers=headers, timeout=20)
            response.raise_for_status()
            
            result = response.json()
            return result["choices"][0]["message"]["content"].strip()
            
        except Exception as e:
            logger.error(f"❌ Error llamando a DeepSeek: {e}")
            return None
    
    def generate_response(self, prompt: str, is_code: bool = False) -> Dict[str, Any]:
        """
        Intenta generar respuesta usando APIs en orden de preferencia
        Returns: dict con response y source
        """
        # Verificar si hay claves configuradas
        has_deepseek = bool(self.config.get("deepseek_api_key"))
        has_openai = bool(self.config.get("openai_api_key"))
        
        if has_deepseek:
            deepseek_response = self.call_deepseek(prompt, is_code)
            if deepseek_response:
                return {"response": deepseek_response, "source": "deepseek"}
        
        if has_openai:
            openai_response = self.call_openai(prompt, is_code)
            if openai_response:
                return {"response": openai_response, "source": "openai"}
        
        # Si no hay APIs disponibles o fallan, retornar None para usar modelo local
        return {"response": None, "source": "none"}