import requests

# Ollama 服务器的地址
OLLAMA_API_URL = "http://localhost:11434/api/generate"

# 合同文本
contract_text = """
合同编号：20231001

甲方（委托方）：
名称：北京某某科技有限公司
地址：北京市海淀区中关村大街1号
联系人：张三
联系电话：138-1234-5678

乙方（服务方）：
名称：上海某某信息技术有限公司
地址：上海市浦东新区张江高科技园区
联系人：李四
联系电话：139-8765-4321
"""

# 构造请求数据
payload = {
    "model": "deepseek-r1:1.5b",  # 使用的模型
    "prompt": f"请从以下合同文本中提取甲方电话和乙方电话：\n{contract_text}",
    "temperature": 0,  # 温度设置为 0
    "max_tokens": 20,  # 最大生成长度
    "stream": False  # 关闭流式响应
}

# 发送 POST 请求
response = requests.post(OLLAMA_API_URL, json=payload)

# 检查请求是否成功
if response.status_code == 200:
    # 解析响应内容
    result = response.json()
    print("提取结果：", result["response"])
else:
    print("请求失败，状态码：", response.status_code)
    print("错误信息：", response.text)