enterprise-kb-rag/2.2_LangChain基础知识/03_models_chat_messages.py

from pathlib import Path
import sys

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

from _chapter2_config import as_text, get_chat_model, print_dependency_hint


try:
    from langchain_core.messages import HumanMessage, SystemMessage
except ModuleNotFoundError as exc:
    print_dependency_hint(exc)
    raise SystemExit(1)


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

    messages = [
        SystemMessage(content="现在你是一个著名的诗人"),
        HumanMessage(content="给我写一首唐诗"),
    ]

    result = model.invoke(messages)
    print(as_text(result))


if __name__ == "__main__":
    main()