埃隆马斯克X-AI发布Grok-2大模型，快来体验~

curl https://api.x.ai/v1/chat/completions -H "Content-Type: application/json" -H "Authorization: Bearer 《your API Key》" -d '{
  "messages": [
    {
      "role": "system",
      "content": "You are a test assistant."
    },
    {
      "role": "user",
      "content": "Testing. Just say hi and hello world and nothing else."
    }
  ],
  "model": "grok-beta",
  "stream": false,
  "temperature": 0
}'

import requests
import json

# 设置API的URL和Headers
url = "https://api.x.ai/v1/chat/completions"
headers = {
    "Content-Type": "application/json",
    "Authorization": "Bearer 《your API Key》"
}

# 初始消息
conversation_history = [
    {"role": "system", "content": "You are a helpful assistant."}
]

while True:
    # 获取用户输入
    user_input = input("You: ")

    if user_input.lower() in ["exit", "quit"]:
        print("Exiting the chat. Goodbye!")
        break

    # 添加用户输入到对话历史
    conversation_history.append({"role": "user", "content": user_input})

    # 构建请求数据
    payload = {
        "messages": conversation_history,
        "model": "grok-beta",
        "stream": False,
        "temperature": 0
    }

    # 发送POST请求
    response = requests.post(url, headers=headers, data=json.dumps(payload))

    if response.status_code == 200:
        # 获取助手回复
        assistant_reply = response.json()["choices"][0]["message"]["content"]
        print(f"Assistant: {assistant_reply}")

        # 将助手回复添加到对话历史
        conversation_history.append({"role": "assistant", "content": assistant_reply})
    else:
        print(f"Error: {response.status_code}")
        print(response.text)