From e4bba2a58119894459cdabf585a3616234e78316 Mon Sep 17 00:00:00 2001 From: dtrckd Date: Wed, 2 Oct 2024 15:06:00 +0200 Subject: [PATCH] fix(rag): messages type. --- app/core_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/core_llm.py b/app/core_llm.py index 729f8f5..a9c8003 100755 --- a/app/core_llm.py +++ b/app/core_llm.py @@ -107,7 +107,7 @@ def make_rag_prompt(self, model_embedding: str, messages: list[dict]) -> list[di chunks = self.semantic_search(model_embedding, query, limit, collections) prompt = self.format_albert_template(query, chunks) messages[-1]["content"] = prompt - return prompt + return messages def semantic_search( self, model: str, query: str, limit: int, collections: list[str]