github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-01 17:16:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:memory create collection bug

Browse Source
This commit is contained in:
aries_ckt
2025-04-14 21:29:41 +08:00
parent 4c6386ad3f
commit 2991c7e437
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/oceanbase_store.py
View File

@@ -311,7 +311,7 @@ class OceanBaseStore(VectorStoreBase):
def create_collection(self, collection_name: str, **kwargs) -> Any: def create_collection(self, collection_name: str, **kwargs) -> Any:
"""Create the collection.""" """Create the collection."""
embeddings = self.embedding_function.embed_documents([collection_name]) embeddings = self.embedding_function.embed_documents([collection_name])
raise self._create_table_with_index(embeddings) return self._create_table_with_index(embeddings)
def load_document(self, chunks: List[Chunk]) -> List[str]: def load_document(self, chunks: List[Chunk]) -> List[str]:
"""Load document in vector database.""" """Load document in vector database."""

5
packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py
View File

@@ -142,7 +142,10 @@ class MultiAgents(BaseComponent, ABC):
).create() ).create()
storage_manager = StorageManager.get_instance(self.system_app) storage_manager = StorageManager.get_instance(self.system_app)
vector_store = storage_manager.create_vector_store(index_name="_agent_memory_") index_name = "_agent_memory_"
vector_store = storage_manager.create_vector_store(index_name=index_name)
if not vector_store.vector_name_exists():
vector_store.create_collection(collection_name=index_name)
embeddings = EmbeddingFactory.get_instance(self.system_app).create() embeddings = EmbeddingFactory.get_instance(self.system_app).create()
short_term_memory = EnhancedShortTermMemory( short_term_memory = EnhancedShortTermMemory(
embeddings, executor=executor, buffer_size=10 embeddings, executor=executor, buffer_size=10