本地模型的简洁的json输出

import requests
import json

# 定义 API 端点 URL
url = "http://localhost:11434/api/generate"

# 定义请求数据
data = {
    "model": "llama3.1",  # 使用 llama3.1 模型
    "prompt": "What color is the sky at different times of the day? Respond using JSON",
    "format": "json",
    "stream": False
}

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

# 检查响应状态码
if response.status_code == 200:
    # 提取聊天响应
    result = response.json()

    if 'response' in result and result['response']:
        # 解析 JSON 字符串
        chat_response = json.loads(result['response'])

        # 打印格式化的聊天响应
        print("Chat Response:")
        print(json.dumps(chat_response, indent=4, ensure_ascii=False))  # 格式化输出
    else:
        print("No chat response found in the result.")
else:
    print("Failed to generate response. Status code:", response.status_code)
    print("Response:", response.text)