From f8c81fc8b45fdcb80b7735edbe2a2addb1c5a8f3 Mon Sep 17 00:00:00 2001 From: wish2023 <60290739+wish2023@users.noreply.github.com> Date: Mon, 20 May 2024 23:22:43 +0800 Subject: [PATCH] Remove extra variable --- summariser/app/llm/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/summariser/app/llm/llm.py b/summariser/app/llm/llm.py index 3719c52..1a44b00 100644 --- a/summariser/app/llm/llm.py +++ b/summariser/app/llm/llm.py @@ -17,10 +17,10 @@ def __init__(self, config): summarise_prompt = ChatPromptTemplate.from_template(self.config["summariser_prompt"]) output_parser = StrOutputParser() - self.chain = chain = summarise_prompt | model | output_parser + self.chain = summarise_prompt | model | output_parser def query(self, groupchat_history: str) -> str: return self.chain.invoke({"groupchat_history": groupchat_history}) -llamabot = LlamaBot(llama_config) \ No newline at end of file +llamabot = LlamaBot(llama_config)