API 成本优化完整指南

class TokenOptimizer:
    """Token优化工具类"""
    
    def optimize_prompt(self, prompt: str) -> str:
        """优化提示词，减少Token使用"""
        # 1. 移除多余空格和换行
        prompt = ' '.join(prompt.split())
        
        # 2. 使用简洁的指令
        replacements = {
            "Please provide a detailed explanation of": "Explain",
            "Could you please": "Please",
            "I would like you to": "Please",
            "Can you help me understand": "Explain",
        }
        
        for long_form, short_form in replacements.items():
            prompt = prompt.replace(long_form, short_form)
        
        # 3. 移除冗余信息
        prompt = self._remove_redundancy(prompt)
        
        return prompt
    
    def _remove_redundancy(self, text: str) -> str:
        """移除冗余信息"""
        # 实现去重逻辑
        lines = text.split('\n')
        unique_lines = []
        seen = set()
        
        for line in lines:
            line_hash = hash(line.strip().lower())
            if line_hash not in seen:
                seen.add(line_hash)
                unique_lines.append(line)
        
        return '\n'.join(unique_lines)
    
    def calculate_cost(self, text: str, model: str) -> dict:
        """计算预估成本"""
        import tiktoken
        
        # 获取对应模型的编码器
        encoding = tiktoken.encoding_for_model(model)
        tokens = len(encoding.encode(text))
        
        # 定价（示例价格，实际请查看官方定价）
        pricing = {
            "gpt-4o": {"input": 0.03, "output": 0.06},  # per 1K tokens
            "gpt-4o-mini": {"input": 0.0002, "output": 0.0006},
            "gpt-3.5-turbo": {"input": 0.0005, "output": 0.0015},
            "claude-3-sonnet": {"input": 0.003, "output": 0.015},
            "claude-3-haiku": {"input": 0.00025, "output": 0.00125}
        }
        
        model_price = pricing.get(model, pricing["gpt-3.5-turbo"])
        
        return {
            "tokens": tokens,
            "estimated_cost": (tokens / 1000) * model_price["input"],
            "model": model
        }

# 使用示例
optimizer = TokenOptimizer()

# 优化前
original_prompt = """
Could you please provide a detailed explanation of how machine learning works?
I would like you to explain the basic concepts, algorithms, and applications.
Please include examples if possible.
"""

# 优化后
optimized = optimizer.optimize_prompt(original_prompt)
print(f"优化前: {len(original_prompt)} 字符")
print(f"优化后: {len(optimized)} 字符")
print(f"节省: {(1 - len(optimized)/len(original_prompt))*100:.1f}%")

# 计算成本
cost_info = optimizer.calculate_cost(optimized, "gpt-4o-mini")
print(f"Token数: {cost_info['tokens']}")
print(f"预估成本: ${cost_info['estimated_cost']:.4f}")