import requests
import json

def parse_deepseek_response(response):
    try:
        data = response.json()
        
        # 兼容不同版本的响应格式
        if 'choices' in data:
            if isinstance(data['choices'], list) and len(data['choices']) > 0:
                choice = data['choices'][0]
                if 'message' in choice:
                    return choice['message']['content']
                elif 'text' in choice:  # 兼容旧版本
                    return choice['text']
        
        # 处理错误响应
        if 'error' in data:
            return f"API错误: {data['error'].get('message', '未知错误')}"
            
        return "无法解析响应格式"
        
    except json.JSONDecodeError:
        return "响应不是有效的JSON格式"

def handle_streaming_response(response):
    full_content = ""
    try:
        for line in response.iter_lines():
            if line:
                line_str = line.decode('utf-8')
                if line_str.startswith('data: '):
                    data_str = line_str[6:]  # 移除"data: "前缀
                    if data_str == '[DONE]':
                        break
                    try:
                        data = json.loads(data_str)
                        if 'choices' in data and len(data['choices']) > 0:
                            delta = data['choices'][0].get('delta', {})
                            if 'content' in delta:
                                full_content += delta['content']
                                print(delta['content'], end='', flush=True)
                    except json.JSONDecodeError:
                        continue
    except Exception as e:
        print(f"\n流式处理错误: {e}")
    
    return full_content

import requests
import json
import time

class DeepSeekAPIDebugger:
    def __init__(self, api_key, base_url="https://api.deepseek.com/v1"):
        self.api_key = api_key
        self.base_url = base_url
        self.session = requests.Session()
        self.session.headers.update({
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json"
        })
    
    def debug_api_call(self, prompt, model="deepseek-chat", max_tokens=1000):
        """执行API调用并返回详细的调试信息"""
        
        payload = {
            "model": model,
            "messages": [{"role": "user", "content": prompt}],
            "max_tokens": max_tokens,
            "temperature": 0.7
        }
        
        start_time = time.time()
        
        try:
            response = self.session.post(
                f"{self.base_url}/chat/completions",
                json=payload,
                timeout=30
            )
            
            response_time = time.time() - start_time
            
            debug_info = {
                "status_code": response.status_code,
                "response_time": f"{response_time:.2f}s",
                "headers": dict(response.headers),
                "raw_response": response.text
            }
            
            if response.status_code == 200:
                try:
                    data = response.json()
                    debug_info["parsed_response"] = data
                    debug_info["content"] = self._extract_content(data)
                    debug_info["token_usage"] = data.get("usage", {})
                except json.JSONDecodeError:
                    debug_info["parse_error"] = "Invalid JSON response"
            else:
                debug_info["error"] = f"HTTP Error: {response.status_code}"
                try:
                    debug_info["error_details"] = response.json()
                except:
                    debug_info["error_details"] = response.text
            
            return debug_info
            
        except requests.exceptions.Timeout:
            return {"error": "Request timeout"}
        except requests.exceptions.ConnectionError:
            return {"error": "Connection error"}
        except Exception as e:
            return {"error": f"Unexpected error: {str(e)}"}
    
    def _extract_content(self, data):
        """从响应数据中提取内容"""
        try:
            if 'choices' in data and len(data['choices']) > 0:
                choice = data['choices'][0]
                if 'message' in choice and 'content' in choice['message']:
                    return choice['message']['content']
                elif 'text' in choice:
                    return choice['text']
            return None
        except:
            return None

# 使用示例
if __name__ == "__main__":
    # 替换为您的API密钥
    debugger = DeepSeekAPIDebugger("your_api_key_here")
    
    # 测试调用
    result = debugger.debug_api_call("请用Python写一个快速排序算法")
    
    print(f"状态码: {result['status_code']}")
    print(f"响应时间: {result['response_time']}")
    
    if 'content' in result and result['content']:
        print("\n生成的内容:")
        print(result['content'])
    
    if 'error' in result:
        print(f"\n错误信息: {result['error']}")
        if 'error_details' in result:
            print(f"错误详情: {result['error_details']}")

# 错误示例
{"error": {"message": "Invalid API key", "type": "invalid_request_error"}}

# 解决方案：检查API密钥是否正确配置

# 错误示例  
{"error": {"message": "Rate limit exceeded", "type": "rate_limit_error"}}

# 解决方案：实现重试机制
def api_call_with_retry(self, prompt, max_retries=3):
    for attempt in range(max_retries):
        result = self.debug_api_call(prompt)
        if result['status_code'] != 429:  # 不是速率限制错误
            return result
        wait_time = 2 ** attempt  # 指数退避
        time.sleep(wait_time)
    return result

# 错误示例
{"error": {"message": "The model's maximum context length is exceeded"}}

# 解决方案：检查输入令牌数
def estimate_tokens(text):
    # 简单估算：中文大约1字=1.5token，英文1字=0.75token
    chinese_chars = sum(1 for c in text if '\u4e00' <= c <= '\u9fff')
    other_chars = len(text) - chinese_chars
    return int(chinese_chars * 1.5 + other_chars * 0.75)