enterprise-kb-rag/2.1_Ollama大模型高效管理工具/05_requests_ollama_chat.py

from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parents[1]))

from _chapter2_config import env_value, print_dependency_hint


try:
    import requests
except ModuleNotFoundError as exc:
    print_dependency_hint(exc)
    raise SystemExit(1)


def main() -> None:
    base_url = env_value("OLLAMA_BASE_URL", default="http://127.0.0.1:11434").rstrip("/")
    url = f"{base_url}/api/chat"
    model = env_value("OLLAMA_MODEL", default="qwen2.5:7b")

    headers = {"Content-Type": "application/json"}
    data = {
        "model": model,
        "options": {
            "temperature": 0.0,
        },
        "stream": False,
        "messages": [
            {
                "role": "system",
                "content": "你是谁？",
            }
        ],
    }
    response = requests.post(url, json=data, headers=headers, timeout=60)
    response.raise_for_status()
    print(response.json())


if __name__ == "__main__":
    try:
        main()
    except Exception as exc:
        print(f"requests调用Ollama失败: {exc}")