注册

python 连续对话

import openai
import time

client = openai.OpenAI(
    base_url="https://api.quickrouter.ai/v1",
    api_key="YOUR_API_KEY"
)

messages = [{"role": "system", "content": "You are a helpful assistant."}]

while True:
    user_input = input("You: ")
    messages.append({"role": "user", "content": user_input})
    response = client.chat.completions.create(
        model="gpt-4o",
        messages=messages,
        temperature=1
    )
    assistant_response = response.choices[0].message.content
    print(f"Assistant: {assistant_response}")
    messages.append({"role": "assistant", "content": assistant_response})
    time.sleep(1)