BJQA/DeepSeekAPI/test.py

import json
import requests

# === 读取 JSON 文件 ===
with open("9b6349887dc54a9e9b8e0a192ed64e52.json", "r", encoding="utf-8") as f:
    large_data = json.load(f)

large_data_subset = large_data[:5]

with open("G04数据字段解释.json", "r", encoding="utf-8") as f:
    explanation = json.load(f)

with open("省间关系映射.json", "r", encoding="utf-8") as f:
    province_mapping = json.load(f)

# === 构建完整 prompt ===
large_prompt = json.dumps(large_data_subset, ensure_ascii=False, indent=2)
explanation_str = json.dumps(explanation, ensure_ascii=False, indent=2)
province_mapping_str = json.dumps(province_mapping, ensure_ascii=False, indent=2)

prompt = f"""
以下是用户数据
{large_prompt}

以下是上面数据的字段解释：
{explanation_str}

以下是省份映射信息：
{province_mapping_str}

请将所有的json数据完整输出
"""

# === 发送请求 ===
OLLAMA_URL = 'http://192.168.1.207:11434/api/generate'
model = "deepseek-r1:32b"

payload = {
    "model": model,
    "prompt": prompt,
    "stream": False
}

print("[发送请求]")
response = requests.post(OLLAMA_URL, json=payload)
result = response.json()
print("\n[模型响应]")
print(result["response"])