github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-20 00:54:43 +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

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

@@ -142,7 +142,10 @@ class MultiAgents(BaseComponent, ABC):
).create()
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()
short_term_memory = EnhancedShortTermMemory(
embeddings, executor=executor, buffer_size=10