BaseTemplateChatBot

import dotenv
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_openai import ChatOpenAI

dotenv.load_dotenv()

# 创建提示模板
prompt = ChatPromptTemplate.from_messages([
    (
        "system", "你是强大的ChantBot,请根据对应的上下文回答用户问题。"
    ),
    MessagesPlaceholder("history"),
    (
        "human", "请根据上下文,回答用户问题:{query}"
    )
])

# 创建大语言模型
llm = ChatOpenAI(model="qwen3.6-plus")


# build chain

chain = prompt | llm | StrOutputParser()

# 死循环构建对话命令行
while True:
    query = input("Human: ")
    if query == "q":
        exit(0)

    chain_input = {"query": query, "history": []}

    resp = chain.stream(chain_input)
    print("AI: ", flush=True, end="")
    for chunk in resp:
        print(chunk, end="", flush=True)
    print("")

posted @ 2026-04-11 21:38  Felix_Openmind  阅读(1)  评论(0)    收藏  举报
*{cursor: url(https://files-cdn.cnblogs.com/files/morango/fish-cursor.ico),auto;}