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 参数(上限受模型限制)。