OpenAI 重磅更新：Assistant API 功能尝鲜

from openai import OpenAI

client = OpenAI()

assistant = client.beta.assistants.create(
    name="test",
    instructions="You are a customer support chatbot. Use your knowledge base to best respond to customer queries.",
    model="gpt-3.5-turbo-1106",
)

thread = client.beta.threads.create() # （可选）在创建时指定对话内容

message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="Compose a poem that explains the concept of recursion in programming.",
)

run = client.beta.threads.runs.create(
    thread_id=thread.id,
    assistant_id=assistant.id,
)

while True:
    run = client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
    if run.status not in ["queued", "in_progress"]:
        break
    time.sleep(1)

messages = client.beta.threads.messages.list(thread_id=thread.id)

# 上传 PDF 文件
file = client.files.create(
    file=open("Flexera-State-of-the-Cloud-Report-2023.pdf", "rb"),  # 我用来测试的 PDF
    purpose='assistants',
)

# 创建 Assistant
assistant = client.beta.assistants.create(
    name="test-retrieval",
    instructions="You are a customer support chatbot. Use your knowledge base to best respond to customer queries.",
    model="gpt-3.5-turbo-1106",
    tools=[{"type": "retrieval"}],
    file_ids=[file.id],
)

message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="According to the PDF report, what are the most popular container tools in companies?",
    file_ids=[file.id],
)

User:
According to the PDF report, what are the most popular container tools in companies?

Assistant:
According to the Flexera 2023 State of the Cloud Report, the most popular container tools in companies are as follows:
- AWS ECS/EKS: 45% currently use, 22% plan to use
- Kubernetes (not via a cloud provider service): 40% currently use, 22% plan to use
- Azure Kubernetes Service (AKS): 38% currently use, 23% plan to use
- Docker: 42% currently use, 15% plan to use
- Google Kubernetes Engine (GKE): 23% currently use, 22% plan to use
- Red Hat OpenShift: 17% currently use, 18% plan to use
- Docker Swarm: 13% currently use, 18% plan to use
- SUSE Rancher: 8% currently use, 15% plan to use【19†source】.

# 上传脚本文件
file = client.files.create(
  file=open("speech.py", "rb"),
  purpose='assistants'
)

# 创建 Assistant
assistant = client.beta.assistants.create(
    name="test-code-interpreter",
    instructions="You are a personal math tutor. When asked a math question, write and run code to answer the question.",
    model="gpt-3.5-turbo-1106",
    tools=[{"type": "code_interpreter"}],
)

message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="I need to solve the equation `3x + 11 = 14`. Can you help me?",
    file_ids=[file.id],
)

from sympy import symbols, Eq, solve

# 创建符号变量
x = symbols('x')

# 输入方程
equation_str = input("请输入一元一次方程: ")

# 解析方程字符串
equation_parts = equation_str.split('=')
left_side = eval(equation_parts[0].strip())
right_side = eval(equation_parts[1].strip())

# 创建方程式
equation = Eq(left_side, right_side)

# 解方程
solution = solve(equation, x)

# 输出解
print("方程的解为 x =", solution[0])

User:
I need to solve the equation `3x + 11 = 14`. Can you help me?

Assistant:
Certainly! I will solve the equation `3x + 11 = 14` to find the value of x. Let me start by solving this equation for you.

Assistant:
The solution to the equation 3x + 11 = 14 is x = 1.

run_steps = client.beta.threads.runs.steps.list(thread_id=thread.id, run_id=run.id)
print(run_steps)

SyncCursorPage[RunStep](
    data=[
        RunStep(
            id="step_xxx",
            status="completed",
            step_details=MessageCreationStepDetails(
                message_creation=MessageCreation(
                    message_id="msg_xxx"
                ),
                type="message_creation",
            ),
            ... # 省略
        ),
        RunStep(
            id="step_xxx",
            status="completed",
            step_details=ToolCallsStepDetails(
                tool_calls=[
                    CodeToolCall(
                        id="call_xxx",
                        code_interpreter=CodeInterpreter(
                            input="from sympy import symbols, Eq, solve\n\n# Define the variable\nx = symbols('x')\n\n# Define the equation\nequation = Eq(3*x + 11, 14)\n\n# Solve the equation for x\nsolution = solve(equation, x)\nsolution",
                            outputs=[
                                CodeInterpreterOutputLogs(logs="[1]", type="logs")
                            ],
                        ),
                        type="code_interpreter",
                    )
                ],
                type="tool_calls",
            ),
            type="tool_calls",
            ... # 省略
        ),
        RunStep(
            ... # 省略
        ),
    ],
)

from sympy import symbols, Eq, solve

# Define the variable
x = symbols('x')

# Define the equation
equation = Eq(3*x + 11, 14)

# Solve the equation for x
solution = solve(equation, x)
solution

# 定义两个 function
# getCurrentWeather 获取当前天气
# getNickname 获取城市昵称
assistant = client.beta.assistants.create(
  instructions="You are a weather bot. Use the provided functions to answer questions.",
  model="gpt-4-1106-preview",
  tools=[{
    "type": "function",
    "function": {
      "name": "getCurrentWeather",
      "description": "Get the weather in location",
      "parameters": {
        "type": "object",
        "properties": {
          "location": {"type": "string", "description": "The city and state e.g. San Francisco, CA"},
          "unit": {"type": "string", "enum": ["c", "f"]}
        },
        "required": ["location"]
      }
    }
  }, {
    "type": "function",
    "function": {
      "name": "getNickname",
      "description": "Get the nickname of a city",
      "parameters": {
        "type": "object",
        "properties": {
          "location": {"type": "string", "description": "The city and state e.g. San Francisco, CA"},
        },
        "required": ["location"]
      }
    }
  }]
)

{
  "id": "run_3HV7rrQsagiqZmYynKwEdcxS",
  "object": "thread.run",
  "status": "requires_action",
  "required_action": {
    "type": "submit_tool_outputs",
    "submit_tool_outputs": {
      "tool_calls": [
        {
          "tool_call_id": "call_Vt5AqcWr8QsRTNGv4cDIpsmA",
          "type": "function",
          "function": {
            "name": "getCurrentWeather",
            "arguments": "{\"location\":\"San Francisco\"}"
          }
        }
      ]
    }
  }
}

run = client.beta.threads.runs.submit_tool_outputs(
  thread_id=thread.id,
  run_id=run.id,
  tool_outputs=[
      {
        "tool_call_id": call_ids[0],
        "output": "22C",
      },
    ]
)