PyCharm对接DeepSeek大模型的操作流程

pip install requests python-dotenv

DEEPSEEK_API_KEY=你的API_Key
DEEPSEEK_API_ENDPOINT=https://api.deepseek.com/v1/chat/completions  # 根据实际API文档调整

import os
import requests
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

class DeepSeekClient:
    def __init__(self):
        self.api_key = os.getenv("DEEPSEEK_API_KEY")
        self.endpoint = os.getenv("DEEPSEEK_API_ENDPOINT")
        self.headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }

    def generate_response(self, prompt, model="deepseek-chat", max_tokens=500):
        payload = {
            "model": model,
            "messages": [{"role": "user", "content": prompt}],
            "max_tokens": max_tokens,
            "temperature": 0.7
        }
        
        try:
            response = requests.post(self.endpoint, json=payload, headers=self.headers)
            response.raise_for_status()  # 检查HTTP错误
            return response.json()["choices"][0]["message"]["content"]
        except requests.exceptions.RequestException as e:
            return f"API请求失败: {str(e)}"
        except KeyError:
            return "解析响应时发生错误"

# 示例用法
if __name__ == "__main__":
    client = DeepSeekClient()
    prompt = "用Python写一个快速排序算法"
    response = client.generate_response(prompt)
    print("DeepSeek 响应：\n", response)

pip install aiohttp

import aiohttp
import asyncio

async def async_generate_response(self, prompt):
    async with aiohttp.ClientSession() as session:
        async with session.post(
            self.endpoint, 
            json=payload, 
            headers=self.headers
        ) as response:
            return await response.json()

import logging
logging.basicConfig(level=logging.INFO)

def stream_response(self, prompt):
    payload["stream"] = True
    response = requests.post(self.endpoint, json=payload, headers=self.headers, stream=True)
    for chunk in response.iter_lines():
        if chunk:
            print(chunk.decode("utf-8"))

POST {{DEEPSEEK_API_ENDPOINT}}
Content-Type: application/json
Authorization: Bearer {{DEEPSEEK_API_KEY}}

{
  "model": "deepseek-chat",
  "messages": [{"role": "user", "content": "你好"}]
}

pip install tenacity

from tenacity import retry, stop_after_attempt, wait_exponential

@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=2, max=10))
def generate_response(self, prompt):
    # 原有代码