提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


一、前言

在 AI 开发场景中,调用大模型 API 是核心环节之一。本文将以一段基于OpenAI客户端库的 API 调用代码为例,详细拆解从环境配置、请求构建到异常处理的完整流程,帮助开发者快速掌握同类接口的调用逻辑,规避常见问题。

二、代码功能总览

通过指定的第三方 API 服务(此处为火山引擎方舟平台的 DeepSeek 模型),发送用户 prompt 并获取模型返回结果。​
过程:API 客户端初始化 → 发送请求 → 验证连接成功并获取响应。

三、详细代码

1.引入库

from openai import OpenAI

2.API调用

API_KEY = "你的API密钥"  # 从火山引擎方舟控制台获取​
BASE_URL = "https://ark.cn-beijing.volces.com/api/v3"  # 服务地址​
MODEL = "deepseek-v3-2-251201"  # 调用的模型名称

3.模型配置

#初始化客户端(核心:建立连接)
client = OpenAI(base_url=BASE_URL, api_key=API_KEY)

resp = client.chat.completions.create(
        model=MODEL,#指定要调用的 AI 模型名称(此处是deepseek-v3-2-251201,必须与平台支持的模型名完全一致,否则会提示 “模型不存在”)
        messages=[{"role": "user", "content": prompt}], #role: "user" 表示这是 “用户” 发送的内容 ;content: prompt 就是你要让 AI 处理的提示词(比如测试用的 “请返回一句话证明连接成功”)
        temperature=0.7, #AI 生成内容的 “随机性”(取值 0~2):= 答案固定,不灵活- 0.7 = 平衡随机性和准确性(默认推荐值)>- 2 = 答案天马行空
        max_tokens=1024, #单次请求的 “最大字符容量”(含你的提问 + AI 的回答),1024 是通用配置,足够测试连接使用
        stream=False #是否 “分块返回” 结果:=一次性获取完整回答(适合测试连接,简单直接)True= 分段落返回(适合长文本,复杂)
    )

4.完整代码

from openai import OpenAI
def api_connection(prompt):
    API_KEY = "你的APIkey"
    BASE_URL = "https://ark.cn-beijing.volces.com/api/v3" 
    MODEL = "deepseek-v3-2-251201"
    client = OpenAI(base_url=BASE_URL, api_key=API_KEY)
    resp = client.chat.completions.create(
            model=MODEL,
            messages=[{"role": "user", "content": prompt}],
            temperature=0.7,
            max_tokens=1024,
            stream=False
        )
    return f"接口连接成功!\n模型返回:{resp.choices[0].message.content.strip()}"

if __name__ == "__main__":
    test_prompt = "请返回一句话证明连接成功"
    print(api_connection(test_prompt))

四、注意事项

1.必须将 API_KEY 替换为你的有效密钥(从火山引擎方舟控制台申请)
2.模型名称必须和平台保持一致
3.刚注册时会有免费额度,可根据偏好选择模型

Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐