tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-13 00:14:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix roles in retriever layer

Browse Source
This commit is contained in:
Alex committed 2025-01-21 09:31:38 +00:00
1 parent 1086bfe1ba
commit 94f682e461
2 files changed
+6 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

application/retriever/brave_search.py
+4 -6
View File
@@ -74,12 +74,10 @@ class BraveRetSearch(BaseRetriever):
if len(self.chat_history) > 1:
for i in self.chat_history:
if "prompt" in i and "response" in i:
messages_combine.append(
{"role": "user", "content": i["prompt"]}
)
messages_combine.append(
{"role": "system", "content": i["response"]}
)
messages_combine.append({"role": "user", "content": i["prompt"]})
messages_combine.append(
{"role": "assistant", "content": i["response"]}
)
messages_combine.append({"role": "user", "content": self.question})
llm = LLMCreator.create_llm(
application/retriever/duckduck_search.py
+2 -4
View File
@@ -91,11 +91,9 @@ class DuckDuckSearch(BaseRetriever):
if len(self.chat_history) > 1:
for i in self.chat_history:
if "prompt" in i and "response" in i:
messages_combine.append({"role": "user", "content": i["prompt"]})
messages_combine.append(
{"role": "user", "content": i["prompt"]}
)
messages_combine.append(
{"role": "system", "content": i["response"]}
{"role": "assistant", "content": i["response"]}
)
messages_combine.append({"role": "user", "content": self.question})