thoth-dbmanager 0.4.1__tar.gz → 0.4.2__tar.gz

{thoth_dbmanager-0.4.1/thoth_dbmanager.egg-info → thoth_dbmanager-0.4.2}/PKG-INFO

@@ -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 → thoth_dbmanager-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "thoth_dbmanager"
-version = "0.4.1"
+version = "0.4.2"
 authors = [
   { name="Marco Pancotti", email="mp@tylconsulting.it" },
 ]
@@ -43,6 +43,7 @@ oracle = ["cx_Oracle>=8.3.0"]
 informix = ["informixdb>=2.2.0"]
 supabase = ["supabase>=2.0.0", "postgrest-py>=0.16.0", "gotrue-py>=2.0.0"]
 sqlite = []  # Built into Python
+qdrant = ["qdrant-client>=1.7.0"]
 
 # Convenience groups
 all = [
@@ -54,7 +55,8 @@ all = [
     "informixdb>=2.2.0",
     "supabase>=2.0.0",
     "postgrest-py>=0.16.0",
-    "gotrue-py>=2.0.0"
+    "gotrue-py>=2.0.0",
+    "qdrant-client>=1.7.0"
 ]
 
 # Development dependencies

thoth_dbmanager-0.4.2/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-0.4.2/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 → thoth_dbmanager-0.4.2/thoth_dbmanager.egg-info}/PKG-INFO

@@ -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 → thoth_dbmanager-0.4.2}/thoth_dbmanager.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ thoth_dbmanager/adapters/mariadb.py
 thoth_dbmanager/adapters/mysql.py
 thoth_dbmanager/adapters/oracle.py
 thoth_dbmanager/adapters/postgresql.py
+thoth_dbmanager/adapters/qdrant.py
 thoth_dbmanager/adapters/sqlite.py
 thoth_dbmanager/adapters/sqlserver.py
 thoth_dbmanager/adapters/supabase.py
@@ -51,6 +52,7 @@ thoth_dbmanager/plugins/mariadb.py
 thoth_dbmanager/plugins/mysql.py
 thoth_dbmanager/plugins/oracle.py
 thoth_dbmanager/plugins/postgresql.py
+thoth_dbmanager/plugins/qdrant.py
 thoth_dbmanager/plugins/sqlite.py
 thoth_dbmanager/plugins/sqlserver.py
 thoth_dbmanager/plugins/supabase.py

{thoth_dbmanager-0.4.1 → thoth_dbmanager-0.4.2}/thoth_dbmanager.egg-info/requires.txt

@@ -13,6 +13,7 @@ informixdb>=2.2.0
 supabase>=2.0.0
 postgrest-py>=0.16.0
 gotrue-py>=2.0.0
+qdrant-client>=1.7.0
 
 [dev]
 pytest>=7.0.0
@@ -37,6 +38,9 @@ cx_Oracle>=8.3.0
 [postgresql]
 psycopg2-binary>=2.9.0
 
+[qdrant]
+qdrant-client>=1.7.0
+
 [sqlite]
 
 [sqlserver]

{thoth_dbmanager-0.4.1 → thoth_dbmanager-0.4.2}/thoth_dbmanager.egg-info/top_level.txt

@@ -1 +1,2 @@
+dbmanager
 thoth_dbmanager