github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-12 12:37:14 +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:
"""Create the collection."""
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]:
"""Load document in vector database."""