fix(storage): Fix load big documents error

2025-09-20 00:54:43 +00:00 · 2025-04-11 20:46:48 +08:00
parent 12170e2504
commit c04e3c7cb0
11 changed files with 88 additions and 17 deletions
--- a/packages/dbgpt-serve/src/dbgpt_serve/rag/storage_manager.py
+++ b/packages/dbgpt-serve/src/dbgpt_serve/rag/storage_manager.py
@@ -68,7 +68,10 @@ class StorageManager(BaseComponent):
        embedding_fn = embedding_factory.create()
        vector_store_config: VectorStoreConfig = storage_config.vector
        return vector_store_config.create_store(
-            name=index_name, embedding_fn=embedding_fn
+            name=index_name,
+            embedding_fn=embedding_fn,
+            max_chunks_once_load=vector_store_config.max_chunks_once_load,
+            max_threads=vector_store_config.max_threads,
        )

    def create_kg_store(