Skip to content

Commit

Permalink
fix(rag): messages type.
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrckd committed Oct 2, 2024
1 parent 52e584a commit e4bba2a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion app/core_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit e4bba2a

Please sign in to comment.