Langchain-Chatchat

1
from server.utils import get_ChatOpenAI
2
from configs.model_config import LLM_MODELS, TEMPERATURE
3
from langchain.chains import LLMChain
4
from langchain.prompts.chat import (
5
    ChatPromptTemplate,
6
    HumanMessagePromptTemplate,
7
)
8

9
model = get_ChatOpenAI(model_name=LLM_MODELS[0], temperature=TEMPERATURE)
10

11

12
human_prompt = "{input}"
13
human_message_template = HumanMessagePromptTemplate.from_template(human_prompt)
14

15
chat_prompt = ChatPromptTemplate.from_messages(
16
    [("human", "我们来玩成语接龙，我先来，生龙活虎"),
17
     ("ai", "虎头虎脑"),
18
     ("human", "{input}")])
19

20

21
chain = LLMChain(prompt=chat_prompt, llm=model, verbose=True)
22
print(chain({"input": "恼羞成怒"}))