fix: Fix retrieve error

packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/neo4j_store.py

@@ -1,17 +1,18 @@
 """Neo4j store."""
 
 import logging
+from dataclasses import dataclass
 
-from dbgpt._private.pydantic import ConfigDict
 from dbgpt.storage.graph_store.base import GraphStoreBase, GraphStoreConfig
 
 logger = logging.getLogger(__name__)
 
 
+@dataclass
 class Neo4jStoreConfig(GraphStoreConfig):
     """Neo4j store config."""
 
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    __type__ = "neo4j"
 
 
 class Neo4jStore(GraphStoreBase):

packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/tugraph_store.py

@@ -18,7 +18,7 @@ logger = logging.getLogger(__name__)
 class TuGraphStoreConfig(GraphStoreConfig):
     """TuGraph store config."""
 
-    __type__ = "TuGraph"
+    __type__ = "tugraph"
 
     host: str = field(
         default="127.0.0.1",

packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/knowledge_graph.py

@@ -103,7 +103,7 @@ GRAPH_PARAMETERS = [
 class BuiltinKnowledgeGraphConfig(KnowledgeGraphConfig):
     """Builtin knowledge graph config."""
 
-    __type__ = "TuGraph"
+    __type__ = "tugraph"
 
     llm_model: Optional[str] = field(
         default=None, metadata={"description": "llm model name."}

packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/open_spg.py

@@ -1,17 +1,18 @@
 """OpenSPG class."""
 
 import logging
+from dataclasses import dataclass
 
-from dbgpt._private.pydantic import ConfigDict
 from dbgpt.storage.knowledge_graph.base import KnowledgeGraphBase, KnowledgeGraphConfig
 
 logger = logging.getLogger(__name__)
 
 
+@dataclass
 class OpenSPGConfig(KnowledgeGraphConfig):
     """OpenSPG config."""
 
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    __type__ = "openspg"
 
 
 class OpenSPG(KnowledgeGraphBase):

packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py

@@ -40,7 +40,7 @@ logger = logging.getLogger(__name__)
 class ChromaVectorConfig(VectorStoreConfig):
     """Chroma vector store config."""
 
-    __type__ = "Chroma"
+    __type__ = "chroma"
 
     persist_path: Optional[str] = field(
         default=os.getenv("CHROMA_PERSIST_PATH", None),
@@ -249,16 +249,6 @@ class ChromaStore(VectorStoreBase):
             # Delete collection if it exists
             self._chroma_client.delete_collection(self._collection.name)
             SharedSystemClient.clear_system_cache()
-
-            # Clean persist folder if it exists
-            if os.path.exists(self.persist_dir):
-                try:
-                    self._clean_persist_folder()
-                except Exception as e:
-                    logger.error(f"Failed to clean persist folder: {e}")
-                    # Even if folder cleanup fails, collection deletion succeeded
-                    return True
-
             return True
 
         except Exception as e:

packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/elastic_store.py

@@ -77,7 +77,7 @@ logger = logging.getLogger(__name__)
 class ElasticsearchStoreConfig(VectorStoreConfig):
     """Elasticsearch vector store config."""
 
-    __type__ = "ElasticSearch"
+    __type__ = "elasticsearch"
 
     uri: str = field(
         default="localhost",

packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/milvus_store.py

@@ -98,7 +98,7 @@ logger = logging.getLogger(__name__)
 class MilvusVectorConfig(VectorStoreConfig):
     """Milvus vector store config."""
 
-    __type__ = "Milvus"
+    __type__ = "milvus"
 
     uri: str = field(
         default=None,

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

@@ -126,7 +126,7 @@ def _normalize(vector: List[float]) -> List[float]:
 class OceanBaseConfig(VectorStoreConfig):
     """OceanBase vector store config."""
 
-    __type__ = "OceanBase"
+    __type__ = "oceanbase"
 
     ob_host: Optional[str] = field(
         default=None,

packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/pgvector_store.py

@@ -41,6 +41,8 @@ logger = logging.getLogger(__name__)
 class PGVectorConfig(VectorStoreConfig):
     """PG vector store config."""
 
+    __type__ = "pgvector"
+
     connection_string: str = field(
         default=None,
         metadata={
@@ -50,7 +52,7 @@ class PGVectorConfig(VectorStoreConfig):
     )
 
     def create_store(self, **kwargs) -> "PGVectorStore":
-        """Create Milvus Store."""
+        """Create a PGVectorStore instance."""
         return PGVectorStore(vector_store_config=self, **kwargs)
 
 

packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/weaviate_store.py

@@ -49,6 +49,8 @@ logger = logging.getLogger(__name__)
 class WeaviateVectorConfig(VectorStoreConfig):
     """Weaviate vector store config."""
 
+    __type__ = "weaviate"
+
     weaviate_url: str = field(
         default=os.getenv("WEAVIATE_URL", None),
         metadata={