thoth-dbmanager 0.5.2__py3-none-any.whl → 0.5.3__py3-none-any.whl

thoth_dbmanager/core/interfaces.py

@@ -278,85 +278,4 @@ class DbPlugin(ABC):
         """Get unique values (backward compatibility)"""
         if not self.adapter:
             raise RuntimeError("Plugin not initialized")
-        return self.adapter.get_unique_values()
-    
-    def get_embedding_function(self):
-        """
-        Get the embedding function for similarity computations.
-        
-        Returns:
-            SafeSentenceTransformer: An embedding function with encode method
-        """
-        try:
-            # Import SafeSentenceTransformer
-            try:
-                from sentence_transformers import SentenceTransformer
-                import logging
-                
-                logger = logging.getLogger(__name__)
-                
-                class SafeSentenceTransformer:
-                    """
-                    Wrapper for SentenceTransformer that handles PyTorch meta tensor issues.
-                    """
-                    def __init__(self, model_name_or_path: str):
-                        self.model_name_or_path = model_name_or_path
-                        self._model = None
-
-                    def _get_model(self):
-                        """Lazy initialization of the SentenceTransformer model."""
-                        if self._model is None:
-                            try:
-                                logger.info(f"Initializing SentenceTransformer with model: {self.model_name_or_path}")
-                                self._model = SentenceTransformer(
-                                    model_name_or_path=self.model_name_or_path,
-                                    device='cpu'  # Explicitly set device to CPU to avoid meta tensor issues
-                                )
-                                logger.info("SentenceTransformer initialized successfully")
-                            except Exception as e:
-                                logger.error(f"Failed to initialize SentenceTransformer: {e}")
-                                # Try alternative initialization approach
-                                try:
-                                    logger.info("Trying alternative initialization approach...")
-                                    self._model = SentenceTransformer(self.model_name_or_path)
-                                    # Move to CPU explicitly after initialization
-                                    self._model = self._model.to('cpu')
-                                    logger.info("Alternative initialization successful")
-                                except Exception as e2:
-                                    logger.error(f"Alternative initialization also failed: {e2}")
-                                    raise e2
-                        return self._model
-
-                    def encode(self, sentences, **kwargs):
-                        """Encode sentences using the underlying SentenceTransformer model."""
-                        model = self._get_model()
-                        return model.encode(sentences, **kwargs)
-                
-                return SafeSentenceTransformer(
-                    model_name_or_path="paraphrase-multilingual-MiniLM-L12-v2"
-                )
-                
-            except ImportError:
-                import logging
-                logger = logging.getLogger(__name__)
-                logger.warning("sentence_transformers not available, creating dummy embedding function")
-                # Create a dummy embedding function for testing
-                class DummyEmbeddingFunction:
-                    def encode(self, sentences, **kwargs):
-                        import numpy as np
-                        # Return dummy embeddings - same shape for all sentences
-                        return np.random.rand(len(sentences), 384)  # 384 is typical embedding size
-                
-                return DummyEmbeddingFunction()
-                
-        except Exception as e:
-            import logging
-            logger = logging.getLogger(__name__)
-            logger.error(f"Failed to create embedding function: {e}")
-            # Return a basic dummy function as fallback
-            class BasicDummyEmbeddingFunction:
-                def encode(self, sentences, **kwargs):
-                    import numpy as np
-                    return np.random.rand(len(sentences), 384)
-            
-            return BasicDummyEmbeddingFunction()
+        return self.adapter.get_unique_values()

{thoth_dbmanager-0.5.2.dist-info → thoth_dbmanager-0.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thoth_dbmanager
-Version: 0.5.2
+Version: 0.5.3
 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>
 Project-URL: Homepage, https://github.com/mptyl/thoth_dbmanager
@@ -26,7 +26,6 @@ Requires-Dist: datasketch>=1.5.0
 Requires-Dist: tqdm>=4.60.0
 Requires-Dist: SQLAlchemy>=1.4.0
 Requires-Dist: pydantic>=2.0.0
-Requires-Dist: pandas>=1.3.0
 Requires-Dist: requests>=2.25.0
 Provides-Extra: postgresql
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "postgresql"
@@ -35,15 +34,10 @@ Requires-Dist: mariadb>=1.1.0; extra == "mariadb"
 Provides-Extra: sqlserver
 Requires-Dist: pyodbc>=4.0.0; extra == "sqlserver"
 Provides-Extra: sqlite
-Provides-Extra: embeddings
-Requires-Dist: sentence-transformers>=2.0.0; extra == "embeddings"
-Requires-Dist: numpy>=1.21.0; extra == "embeddings"
 Provides-Extra: all
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "all"
 Requires-Dist: mariadb>=1.1.0; extra == "all"
 Requires-Dist: pyodbc>=4.0.0; extra == "all"
-Requires-Dist: sentence-transformers>=2.0.0; extra == "all"
-Requires-Dist: numpy>=1.21.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.5.2.dist-info → thoth_dbmanager-0.5.3.dist-info}/RECORD

@@ -9,7 +9,7 @@ thoth_dbmanager/adapters/sqlite.py,sha256=RTDszgnAtkE14LKFeoe9lBHgsqXqkmDk6jDCTm
 thoth_dbmanager/adapters/sqlserver.py,sha256=V555kUH54Fb1Atow0BfvbSHmoSwGnrB_RJGn718VQSI,23880
 thoth_dbmanager/core/__init__.py,sha256=FlqNW0GZNv1rnwNgyXGzveLqaw0Z90y5AKhR_1DvHBE,269
 thoth_dbmanager/core/factory.py,sha256=84EeZYRoH7y7b19EFHqN4X0CSA6dv-0yxUmlX2zHETk,8840
-thoth_dbmanager/core/interfaces.py,sha256=MRtNWvsXgtvK7_vdngSe8mNCM0hmRRHyTKnI78uxZ6o,13725
+thoth_dbmanager/core/interfaces.py,sha256=s6t-8w4QWu_4Dl654LAU2p3Ao34wjeNaGEsUOJwYHaM,9575
 thoth_dbmanager/core/registry.py,sha256=url4qpQMoMw4rDrdAAvV6L7-NdO4z86xSJPSwTH_l5g,8624
 thoth_dbmanager/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thoth_dbmanager/helpers/multi_db_generator.py,sha256=frN0SZtWAfeojoJFLs4XLR3ri6h9pHYc-2O4aLAOlbo,23238
@@ -26,8 +26,8 @@ thoth_dbmanager/plugins/mariadb.py,sha256=ElYa4Rexwrofcjcs0UQKan8fZpbU6-n9zghYR9
 thoth_dbmanager/plugins/postgresql.py,sha256=pI1W9oHpQty8tHMoEDcsOT-Msv6S4aoFcArOGFxLR7Q,5518
 thoth_dbmanager/plugins/sqlite.py,sha256=gkgZ6-Vjkab0IP3ffHOg4bbpDHsjO_N4DesUnSDNAmQ,8857
 thoth_dbmanager/plugins/sqlserver.py,sha256=mMb3F5FmSWV02FZwj-Ult-2TjuyeVA4Fl1iME1dbgLU,5289
-thoth_dbmanager-0.5.2.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
-thoth_dbmanager-0.5.2.dist-info/METADATA,sha256=kjNeZj0QtbNG4zG1cynSzBGf9NYjSfPU3Bh6rNL99go,15032
-thoth_dbmanager-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thoth_dbmanager-0.5.2.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
-thoth_dbmanager-0.5.2.dist-info/RECORD,,
+thoth_dbmanager-0.5.3.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
+thoth_dbmanager-0.5.3.dist-info/METADATA,sha256=ex4_W1vKNmcoFo0ZDHa4Hq-uaoO2YM6T_2a0VvdChvw,14752
+thoth_dbmanager-0.5.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thoth_dbmanager-0.5.3.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
+thoth_dbmanager-0.5.3.dist-info/RECORD,,