LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-12-02 18:13:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: use list instead of string parsing

Browse Source
This commit is contained in:
Ankit Matth
2025-08-23 20:25:29 +05:30
parent bd73fa9ae7
commit 07d59b6640
6 changed files with 68 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
application/retriever/classic_rag.py
View File

@@ -46,17 +46,10 @@ class ClassicRAG(BaseRetriever):
user_api_key=self.user_api_key,
decoded_token=decoded_token,
)
if "active_docs" in source:
if "active_docs" in source and source["active_docs"] is not None:
if isinstance(source["active_docs"], list):
self.vectorstores = source["active_docs"]
elif (
isinstance(source["active_docs"], str) and "," in source["active_docs"]
):
self.vectorstores = [
doc_id.strip()
for doc_id in source["active_docs"].split(",")
if doc_id.strip()
]
else:
self.vectorstores = [source["active_docs"]]
else: