GPT vs Claude 全面对比分析

import time
import openai
import anthropic

# 性能测试对比
class ModelBenchmark:
    def __init__(self, openai_key, anthropic_key):
        self.openai_client = openai.OpenAI(
            api_key=openai_key,
            base_url="https://api.n1n.ai/v1"
        )
        self.anthropic_client = anthropic.Anthropic(
            api_key=anthropic_key,
            base_url="https://api.n1n.ai/v1"
        )
    
    def test_response_time(self, prompt):
        # 测试 GPT-4o
        start = time.time()
        gpt_response = self.openai_client.chat.completions.create(
            model="gpt-4o",
            messages=[{"role": "user", "content": prompt}],
            max_tokens=500
        )
        gpt_time = time.time() - start
        
        # 测试 Claude
        start = time.time()
        claude_response = self.anthropic_client.messages.create(
            model="claude-3-5-sonnet-20241022",
            messages=[{"role": "user", "content": prompt}],
            max_tokens=500
        )
        claude_time = time.time() - start
        
        return {
            "gpt-4o": {"time": gpt_time, "response": gpt_response},
            "claude-3.5": {"time": claude_time, "response": claude_response}
        }

# 场景智能选择
def select_model(task_type, requirements):
    """根据任务类型推荐最佳模型"""
    
    if task_type == "coding":
        if requirements.get("quality") == "high":
            return "gpt-4o"  # 最强代码能力
        else:
            return "gpt-4o-mini"  # 性价比高
    
    elif task_type == "creative_writing":
        if requirements.get("context_length", 0) > 100000:
            return "claude-3-5-sonnet"  # 200K上下文
        else:
            return "gpt-4o"  # 均衡选择
    
    elif task_type == "customer_service":
        if requirements.get("cost") == "low":
            return "claude-3-5-haiku"  # 最低成本
        else:
            return "gpt-4o-mini"  # 快速响应
    
    elif task_type == "data_analysis":
        return "gpt-4o"  # 强大的数学推理
    
    elif task_type == "research":
        return "claude-3-5-sonnet"  # 深度分析能力
    
    return "gpt-4o-mini"  # 默认选择

# 成本计算器
class CostOptimizer:
    # 价格表 (每1M tokens)
    PRICING = {
        "gpt-4o": {"input": 2.50, "output": 10.00},
        "gpt-4o-mini": {"input": 0.15, "output": 0.60},
        "claude-3-5-sonnet": {"input": 3.00, "output": 15.00},
        "claude-3-5-haiku": {"input": 0.25, "output": 1.25}
    }
    
    @classmethod
    def estimate_cost(cls, model, input_tokens, output_tokens):
        """估算API调用成本"""
        pricing = cls.PRICING[model]
        input_cost = (input_tokens / 1_000_000) * pricing["input"]
        output_cost = (output_tokens / 1_000_000) * pricing["output"]
        
        return {
            "model": model,
            "input_cost": round(input_cost, 4),
            "output_cost": round(output_cost, 4),
            "total_cost": round(input_cost + output_cost, 4)
        }