BBIT-Kai
27 lines
749 B
Python
27 lines
749 B
Python
|
|
from config.llm import llm
|
|
from langchain.prompts import PromptTemplate
|
|
|
|
chatPrompt = PromptTemplate(
|
|
input_variables=["aiRole", "history", "userInput"],
|
|
template = """
|
|
你的用户画像为：{aiRole}。
|
|
你需要基于你的角色性格，使用中文回答用户。
|
|
|
|
聊天历史：
|
|
{history}
|
|
|
|
用户最新输入：
|
|
{userInput}
|
|
|
|
最后，请注意，不要编造数据，不知道就说不知道，现在，请生成你的回复：
|
|
"""
|
|
)
|
|
chatChain = chatPrompt | llm
|
|
|
|
def get_chat_response(aiRole: str,history: str, userInput: str) -> str:
|
|
return chatChain.invoke({
|
|
"aiRole": aiRole,
|
|
"history": history,
|
|
"userInput": userInput
|
|
}) |