thoth-dbmanager 0.4.1__py3-none-any.whl → 0.4.2__py3-none-any.whl

thoth_dbmanager/adapters/qdrant.py

@@ -0,0 +1,189 @@
+"""
+Qdrant adapter for Thoth SQL Database Manager.
+"""
+
+from typing import Any, Dict, List, Optional, Union
+from ..core.interfaces import DbAdapter
+
+
+class QdrantAdapter(DbAdapter):
+    """
+    Qdrant vector database adapter implementation.
+    """
+    
+    def __init__(self, **kwargs):
+        """Initialize Qdrant adapter with connection parameters."""
+        super().__init__()
+        self.host = kwargs.get('host', 'localhost')
+        self.port = kwargs.get('port', 6333)
+        self.api_key = kwargs.get('api_key')
+        self.collection_name = kwargs.get('collection_name', 'thoth_documents')
+        self._client = None
+    
+    def connect(self) -> bool:
+        """Establish connection to Qdrant."""
+        try:
+            # Import qdrant_client here to avoid dependency issues
+            from qdrant_client import QdrantClient
+            
+            if self.api_key:
+                self._client = QdrantClient(
+                    host=self.host,
+                    port=self.port,
+                    api_key=self.api_key
+                )
+            else:
+                self._client = QdrantClient(
+                    host=self.host,
+                    port=self.port
+                )
+            
+            # Test connection
+            self._client.get_collections()
+            return True
+        except Exception as e:
+            print(f"Failed to connect to Qdrant: {e}")
+            return False
+    
+    def disconnect(self) -> None:
+        """Disconnect from Qdrant."""
+        if self._client:
+            self._client.close()
+            self._client = None
+    
+    def execute_query(self, query: str, params: Optional[Dict] = None, 
+                     fetch: Union[str, int] = "all", timeout: int = 60) -> Any:
+        """
+        Execute a query against Qdrant.
+        Note: Qdrant doesn't use SQL, so this adapts the interface.
+        """
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        # This is a placeholder - adapt based on your specific needs
+        # Qdrant uses vector search, not SQL queries
+        return {"message": "Qdrant uses vector search, not SQL queries"}
+    
+    def get_tables(self) -> List[Dict[str, str]]:
+        """Get collections (equivalent to tables in Qdrant)."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        try:
+            collections = self._client.get_collections()
+            return [
+                {
+                    "table_name": collection.name,
+                    "table_type": "COLLECTION"
+                }
+                for collection in collections.collections
+            ]
+        except Exception as e:
+            print(f"Error getting collections: {e}")
+            return []
+    
+    def get_columns(self, table_name: str) -> List[Dict[str, Any]]:
+        """Get collection info (equivalent to columns in Qdrant)."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        try:
+            collection_info = self._client.get_collection(table_name)
+            return [
+                {
+                    "column_name": "id",
+                    "data_type": "UUID",
+                    "is_nullable": False
+                },
+                {
+                    "column_name": "vector",
+                    "data_type": f"VECTOR({collection_info.config.params.vectors.size})",
+                    "is_nullable": False
+                },
+                {
+                    "column_name": "payload",
+                    "data_type": "JSON",
+                    "is_nullable": True
+                }
+            ]
+        except Exception as e:
+            print(f"Error getting collection info: {e}")
+            return []
+    
+    def get_foreign_keys(self) -> List[Dict[str, str]]:
+        """Get foreign keys (not applicable for Qdrant)."""
+        return []
+    
+    def get_unique_values(self) -> Dict[str, Dict[str, List[str]]]:
+        """Get unique values from collections."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        # This is a simplified implementation
+        # In practice, you'd need to scroll through points and extract unique payload values
+        return {}
+    
+    def add_documentation(self, doc_type: str, content: Dict[str, Any]) -> str:
+        """Add documentation to Qdrant collection."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        try:
+            from qdrant_client.models import PointStruct
+            import uuid
+            
+            # Generate a unique ID for the document
+            doc_id = str(uuid.uuid4())
+            
+            # Create a point with the documentation content
+            point = PointStruct(
+                id=doc_id,
+                vector=content.get('vector', [0.0] * 384),  # Default vector size
+                payload={
+                    "doc_type": doc_type,
+                    "content": content
+                }
+            )
+            
+            # Upsert the point
+            self._client.upsert(
+                collection_name=self.collection_name,
+                points=[point]
+            )
+            
+            return doc_id
+        except Exception as e:
+            print(f"Error adding documentation: {e}")
+            raise
+    
+    def delete_collection(self, collection_name: str) -> bool:
+        """Delete a collection from Qdrant."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        try:
+            self._client.delete_collection(collection_name)
+            return True
+        except Exception as e:
+            print(f"Error deleting collection: {e}")
+            return False
+    
+    def create_collection(self, collection_name: str, vector_size: int = 384) -> bool:
+        """Create a new collection in Qdrant."""
+        if not self._client:
+            raise RuntimeError("Not connected to Qdrant")
+        
+        try:
+            from qdrant_client.models import VectorParams, Distance
+            
+            self._client.create_collection(
+                collection_name=collection_name,
+                vectors_config=VectorParams(
+                    size=vector_size,
+                    distance=Distance.COSINE
+                )
+            )
+            return True
+        except Exception as e:
+            print(f"Error creating collection: {e}")
+            return False

thoth_dbmanager/plugins/qdrant.py

@@ -0,0 +1,41 @@
+"""
+Qdrant plugin for Thoth SQL Database Manager.
+"""
+
+from typing import List
+from ..core.interfaces import DbPlugin
+from ..adapters.qdrant import QdrantAdapter
+
+
+class QdrantPlugin(DbPlugin):
+    """Plugin for Qdrant vector database."""
+    
+    plugin_name = "qdrant"
+    plugin_version = "1.0.0"
+    supported_db_types = ["qdrant"]
+    required_dependencies = ["qdrant-client"]
+    
+    def create_adapter(self, **kwargs) -> QdrantAdapter:
+        """Create and return a QdrantAdapter instance."""
+        return QdrantAdapter(**kwargs)
+    
+    def validate_connection_params(self, **kwargs) -> bool:
+        """Validate Qdrant connection parameters."""
+        required_params = ['host']
+        
+        for param in required_params:
+            if param not in kwargs:
+                return False
+        
+        # Validate host format
+        host = kwargs.get('host')
+        if not isinstance(host, str) or not host.strip():
+            return False
+        
+        # Validate port if provided
+        port = kwargs.get('port')
+        if port is not None:
+            if not isinstance(port, int) or port <= 0 or port > 65535:
+                return False
+        
+        return True

{thoth_dbmanager-0.4.1.dist-info → thoth_dbmanager-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thoth_dbmanager
-Version: 0.4.1
+Version: 0.4.2
 Summary: A Python library for managing SQL databases with support for multiple database types, LSH-based similarity search, and a modern plugin architecture.
 Author-email: Marco Pancotti <mp@tylconsulting.it>
 License: MIT
@@ -45,6 +45,8 @@ Requires-Dist: supabase>=2.0.0; extra == "supabase"
 Requires-Dist: postgrest-py>=0.16.0; extra == "supabase"
 Requires-Dist: gotrue-py>=2.0.0; extra == "supabase"
 Provides-Extra: sqlite
+Provides-Extra: qdrant
+Requires-Dist: qdrant-client>=1.7.0; extra == "qdrant"
 Provides-Extra: all
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "all"
 Requires-Dist: mysql-connector-python>=8.0.0; extra == "all"
@@ -55,6 +57,7 @@ Requires-Dist: informixdb>=2.2.0; extra == "all"
 Requires-Dist: supabase>=2.0.0; extra == "all"
 Requires-Dist: postgrest-py>=0.16.0; extra == "all"
 Requires-Dist: gotrue-py>=2.0.0; extra == "all"
+Requires-Dist: qdrant-client>=1.7.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

{thoth_dbmanager-0.4.1.dist-info → thoth_dbmanager-0.4.2.dist-info}/RECORD

@@ -7,6 +7,7 @@ thoth_dbmanager/adapters/mariadb.py,sha256=LTsf0gORiwqZkd6WtKcOsYLHyDgysxdqNesBs
 thoth_dbmanager/adapters/mysql.py,sha256=TrFbxoMMNWbmUcgkKQYOIfsstmMUmuLlGB7R4ZFEIYI,5698
 thoth_dbmanager/adapters/oracle.py,sha256=JSrsgohjz5PbVc8nI188MZ4QGBQls4ieNmwWfAKA7II,21468
 thoth_dbmanager/adapters/postgresql.py,sha256=qxdlxOV7Nvn8U4Lhat50w87Z2S8AzBfmLfEwKfz7dis,17299
+thoth_dbmanager/adapters/qdrant.py,sha256=sL8ldUbAczENb01MmcrkJBKw0dieWPAG2aj97jznj3k,6557
 thoth_dbmanager/adapters/sqlite.py,sha256=RTDszgnAtkE14LKFeoe9lBHgsqXqkmDk6jDCTmVpnoM,14659
 thoth_dbmanager/adapters/sqlserver.py,sha256=V555kUH54Fb1Atow0BfvbSHmoSwGnrB_RJGn718VQSI,23880
 thoth_dbmanager/adapters/supabase.py,sha256=bl2C6LpOpykPF3vIbdNRDk43aXLADzSk0wQuwTcEHZA,10348
@@ -29,11 +30,12 @@ thoth_dbmanager/plugins/mariadb.py,sha256=ElYa4Rexwrofcjcs0UQKan8fZpbU6-n9zghYR9
 thoth_dbmanager/plugins/mysql.py,sha256=mbDsIDV2H_BWYANU4JHMsUkxLQICuGtjKTTPbig2Ngs,16546
 thoth_dbmanager/plugins/oracle.py,sha256=k4Yxvz5MdsH3Sfty9lxbhr8igSnHvGbGujz3bLpNcHo,5230
 thoth_dbmanager/plugins/postgresql.py,sha256=GF6k4K0t7-Y08THWJzS0eWJkrQ1e4GfoKIcanC0Z5Ng,5401
+thoth_dbmanager/plugins/qdrant.py,sha256=XWmc4K6ILiwvB1BhyJBBtA7ohRzQmk1MiZ1N7S-YF78,1199
 thoth_dbmanager/plugins/sqlite.py,sha256=esgJqDp2aSu4a32WnmynfFyY9JfW5W8VjNTkaA-hZhM,6402
 thoth_dbmanager/plugins/sqlserver.py,sha256=mMb3F5FmSWV02FZwj-Ult-2TjuyeVA4Fl1iME1dbgLU,5289
 thoth_dbmanager/plugins/supabase.py,sha256=mWlaGAdpywx4-pU4Ffpmn24ze8sg0sM5kc6bFDoeYRg,8645
-thoth_dbmanager-0.4.1.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
-thoth_dbmanager-0.4.1.dist-info/METADATA,sha256=NirV8lH7JsaZiFhogxnBUQuJqqKzq8WD_9U3ALpeGCk,12053
-thoth_dbmanager-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thoth_dbmanager-0.4.1.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
-thoth_dbmanager-0.4.1.dist-info/RECORD,,
+thoth_dbmanager-0.4.2.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
+thoth_dbmanager-0.4.2.dist-info/METADATA,sha256=iMFzfRMoh0BqUahLBLt2Pny1oA7L6F1tB7iF8W4D9no,12183
+thoth_dbmanager-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thoth_dbmanager-0.4.2.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
+thoth_dbmanager-0.4.2.dist-info/RECORD,,