Closed hamutama closed 11 months ago
去看content文件夹中的Notebook文件,应该就是把一些功能函数封装了,比如:
# 一个封装 OpenAI 接口的函数,参数为 Prompt,返回对应结果
def get_completion(prompt, model="gpt-3.5-turbo"):
'''
prompt: 对应的提示词
model: 调用的模型,默认为 gpt-3.5-turbo(ChatGPT),有内测资格的用户可以选择 gpt-4
'''
messages = [{"role": "user", "content": prompt}]
response = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=0, # 模型输出的温度系数,控制输出的随机程度
)
# 调用 OpenAI 的 ChatCompletion 接口
return response.choices[0].message["content"]
建议最好阅读 content 目录下的 Notebook 文件,可以直接复现;我们在在线阅读为保证阅读便利性,将一些重复使用的函数封装为了本地 tool 包
比如from tool import get_completion,这东西貌似书里没提