创建双提示词的 get_response() 函数

接下来的练习将基于向 OpenAI API 的 chat.completions 端点传入两个提示词（系统提示词和用户提示词）。为此准备，在本练习中，您将创建一个双提示词的 get_response() 函数，该函数接收两个输入提示词（system_prompt 和 user_prompt），并返回模型的回复作为输出。然后，您可以将此函数应用到任意您选择的示例。

OpenAI 包已为您预先加载。

本练习是课程的一部分

使用 OpenAI API 的 Prompt Engineering

查看课程

练习说明

在 messages 列表中为每条消息指定相应的角色和内容。
通过传入您自定义的 system_prompt 和 user_prompt 来试用该函数。

交互式实操练习

通过完成这段示例代码来试试这个练习。

client = OpenAI(api_key="")

def get_response(system_prompt, user_prompt):
  # Assign the role and content for each message
  messages = [{"role": ____, "content": ____},
      		  {"role": ____, "content": ____}]  
  response = client.chat.completions.create(
      model="gpt-4o-mini", messages= messages, temperature=0)
  
  return response.choices[0].message.content

# Try the function with a system and user prompts of your choice 
response = get_response("____", "____")
print(response)

编辑并运行代码