2. 调用 生成文本
任务目标
本案例将使用 Python SDK 调用 -turbo 模型,实现一个命令行文本生成工具。用户输入提示词后,程序调用 Chat Completions API 返回模型生成的回复,并在终端打印结果。
环境准备
需要安装 库(v1.x),并准备一个有效的 。通过环境变量 _ 传入密钥,避免硬编码在代码中。
pip install ==1.30.0
完整实现代码
import os
from import
client = (=os.getenv("_"))
def generate_text(prompt: str, model: str = "-turbo", max_tokens: int = 200) -> str:
"""调用 生成文本回复"""
try:
response = client.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": "你是一个有用的助手。"},
{"role": "user", "content": prompt}
],
max_tokens=max_tokens,
temperature=0.7
)
return response.choices[0].message.content.strip()
except Exception as e:
return f"调用失败: {e}"
if __name__ == "__main__":
user_input = input("请输入提示词: ")
result = generate_text(user_input)
print("\n--- 回复 ---")
print(result)
运行方式
export _="sk-你的密钥"
python _generate.py
验证方法
程序启动后输入提示词,例如 用一句话介绍Python,若 有效且网络通畅,终端会打印模型生成的文本回复。判定通过的标准:
- 程序无报错退出,正常打印回复内容;
- 回复内容与提示词相关,且为完整自然语言句子;
- 若密钥无效或网络异常,程序会打印以“调用失败”开头的错误信息,此时应检查密钥或网络配置。
常见问题
- 认证错误:确认环境变量
_已正确设置且未过期。 - 模型不可用:默认使用
-turbo,若账号无权限可改用-turbo-0125等版本。 - 响应截断:若生成内容过长,可调大
max_tokens参数(上限受模型限制)。