deepseek_api.py

deepseek_api.py — DeepSeek 接口封装

OpenAI 兼容调用方式（call_gpt），目前作为备用模型或实验性接口存在。模块分工见 api 主篇。

代码

# Please install OpenAI SDK first: `pip3 install openai`

from openai import OpenAI

api_url = "https://api.deepseek.com"
api_key = "sk-2d7613aaa1be41d580b5406b3b05b059"
#当前使用的模型
model_name = "deepseek-chat"


def call_gpt(prompt) -> str: 
    #调用接口
    client = OpenAI(api_key=api_key, base_url=api_url)

    response = client.chat.completions.create(
        model= model_name,
        messages=[
            {"role": "user", "content": prompt},
        ],
        stream=False
    )

    print(response.choices[0].message.content)


#获取模型的列表
def getModels():
    client = OpenAI(api_key=api_key, base_url=api_url)
    models = client.models.list()
    for model in models.data:
        print(model.id)

#版本模型
def getVers():
    return model_name


if __name__ == "__main__":
    print(getModels())

项目分区导航：api ⬅️ | 01-deepseek_api | ➡️ doubao_api