LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-03-04 12:54:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/arc53/DocsGPT

Browse Source
This commit is contained in:
GH Action - Upstream Sync
2025-06-06 01:42:12 +00:00
parent 3bae30c70c 31baf181a3
commit ece5c8bb31
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
application/retriever/classic_rag.py
View File

@@ -44,8 +44,8 @@ class ClassicRAG(BaseRetriever):
user_api_key=self.user_api_key, user_api_key=self.user_api_key,
decoded_token=decoded_token, decoded_token=decoded_token,
) )
self.question = self._rephrase_query()
self.vectorstore = source["active_docs"] if "active_docs" in source else None self.vectorstore = source["active_docs"] if "active_docs" in source else None
self.question = self._rephrase_query()
self.decoded_token = decoded_token self.decoded_token = decoded_token
def _rephrase_query(self): def _rephrase_query(self):
@@ -53,6 +53,8 @@ class ClassicRAG(BaseRetriever):
not self.original_question not self.original_question
or not self.chat_history or not self.chat_history
or self.chat_history == [] or self.chat_history == []
or self.chunks == 0
or self.vectorstore is None
): ):
return self.original_question return self.original_question
@@ -77,7 +79,7 @@ class ClassicRAG(BaseRetriever):
return self.original_question return self.original_question
def _get_data(self): def _get_data(self):
if self.chunks == 0: if self.chunks == 0 or self.vectorstore is None:
docs = [] docs = []
else: else:
docsearch = VectorCreator.create_vectorstore( docsearch = VectorCreator.create_vectorstore(