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)