openrouter_api_handler.py

"""
OpenRouter API Handler Module
Handles all OpenRouter API interactions for Claude
"""
import os
import json
import httpx
from typing import Dict, Any, Optional


class OpenRouterHandler:
    """Handles all OpenRouter API interactions for Claude"""
    
    def __init__(self, api_key: Optional[str] = None):
        self.api_key = api_key or os.getenv("OPENROUTER_API_KEY")
        self.base_url = "https://openrouter.ai/api/v1/chat/completions"
        self.model = "anthropic/claude-sonnet-4"
        
        if not self.api_key:
            raise ValueError("OpenRouter API key required")
    
    async def generate_token_params(self, user_prompt: str) -> Dict[str, Any]:
        """
        Send prompt to Claude via OpenRouter and get structured token parameters
        
        Args:
            user_prompt: User's description of the token they want to create
            
        Returns:
            Dict with token parameters (name, symbol, description, image_url, twitter, telegram, website)
        """
        system_prompt = """You are a Pump.fun token creation assistant. Generate token parameters based on user ideas.

Respond ONLY with valid JSON in this exact format (no markdown, no backticks):
{
  "name": "Token Name (max 32 chars)",
  "symbol": "TICKER (max 8 chars, uppercase)"
}

Rules:
- Be creative but appropriate
- Symbol should be memorable and related to concept
- Description should be catchy for memecoin audience
- Only include social links if user mentions them
- If no image URL provided, use empty string"""

        payload = {
            "model": self.model,
            "messages": [
                {"role": "system", "content": system_prompt},
                {"role": "user", "content": user_prompt}
            ],
            "temperature": 0.7,
            "max_tokens": 500
        }
        
        headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json",
            "HTTP-Referer": "https://github.com/yourusername/pumpfun-ai-launcher",
            "X-Title": "Pump.fun AI Launcher"
        }
        
        async with httpx.AsyncClient(timeout=30.0) as client:
            response = await client.post(
                self.base_url,
                json=payload,
                headers=headers
            )
            response.raise_for_status()
            result = response.json()
            
        content = result["choices"][0]["message"]["content"]
        
        # Parse JSON response from Claude
        token_params = json.loads(content.strip())
        
        # Validate required fields
        required = ["name", "symbol"]
        for field in required:
            if field not in token_params:
                raise ValueError(f"Missing required field: {field}")
        
        return token_params