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

from pathlib import Path
import sys

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

from _chapter2_config import get_ollama_llm, print_dependency_hint


def main() -> None:
    try:
        llm = get_ollama_llm(temperature=0)
    except ModuleNotFoundError as exc:
        print_dependency_hint(exc)
        raise SystemExit(1)

    result = llm.invoke("你是谁")
    print(result)


if __name__ == "__main__":
    try:
        main()
    except Exception as exc:
        print(f"LangChain调用Ollama失败: {exc}")
        print("请确认 Ollama 服务、模型名称，以及 langchain-ollama 依赖都可用。")