如何用Python的OpenAI包或Langchain实现文本自动补全？

一、使用OpenAI官方包实现 #

步骤1：安装依赖包 #

执行以下命令安装OpenAI包：

pip install openai

步骤2：编写补全代码 #

设置API密钥后，调用接口生成补全内容：

import openai

# 替换为你的OpenAI API密钥
openai.api_key = "你的API密钥"

# 待补全的文本
prompt = "This is a"

# 调用补全接口
response = openai.Completion.create(
    model="text-davinci-003",  # 可替换为gpt-3.5-turbo-instruct等适配模型
    prompt=prompt,
    max_tokens=20,  # 控制生成文本的长度
    temperature=0.7,  # 0更严谨，1更发散，按需调整
    stop=None
)

# 提取并输出结果
completed_text = response.choices[0].text.strip()
print(f"补全结果：{prompt} {completed_text}")

执行后会生成类似This is a question in Stackoverflow的内容。

二、使用Langchain实现（更灵活的框架） #

Langchain便于管理prompt、模型调用及后续扩展，适合复杂场景。

步骤1：安装依赖包 #

pip install langchain openai

步骤2：编写补全代码 #

from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate

# 初始化LLM模型
llm = OpenAI(
    openai_api_key="你的API密钥",
    model_name="text-davinci-003",
    temperature=0.7,
    max_tokens=20
)

# 定义prompt模板
prompt_template = PromptTemplate(
    input_variables=["partial_text"],
    template="{partial_text}"
)

# 生成补全内容
partial_text = "This is a"
prompt = prompt_template.format(partial_text=partial_text)
completed_text = llm(prompt).strip()
print(f"补全结果：{partial_text} {completed_text}")

注意事项 #

• 确保API密钥有足够额度，且网络可正常访问OpenAI服务。
• 调整temperature参数可控制生成风格：值越低越贴合常规表述，值越高越具创造性。
• 若使用GPT-3.5或GPT-4系列模型，可改用ChatOpenAI类适配对话式补全逻辑。

内容的提问来源于stack exchange，提问作者tobias