MindsDB 25.3.1.0__py3-none-any.whl → 25.3.3.0__py3-none-any.whl

mindsdb/integrations/handlers/pgvector_handler/pgvector_handler.py

@@ -27,7 +27,7 @@ logger = log.getLogger(__name__)
 
 
 # todo Issue #7316 add support for different indexes and search algorithms e.g. cosine similarity or L2 norm
-class PgVectorHandler(VectorStoreHandler, PostgresHandler):
+class PgVectorHandler(PostgresHandler, VectorStoreHandler):
     """This handler handles connection and execution of the PostgreSQL with pgvector extension statements."""
 
     name = "pgvector"
@@ -64,11 +64,11 @@ class PgVectorHandler(VectorStoreHandler, PostgresHandler):
             return Response(RESPONSE_TYPE.OK)
         return super().get_tables()
 
-    def native_query(self, query) -> Response:
+    def native_query(self, query, params=None) -> Response:
         # Prevent execute native queries
         if self._is_shared_db:
             return Response(RESPONSE_TYPE.OK)
-        return super().native_query(query)
+        return super().native_query(query, params=params)
 
     def raw_query(self, query, params=None) -> Response:
         resp = super().native_query(query, params)

mindsdb/integrations/handlers/redshift_handler/redshift_handler.py

@@ -57,6 +57,7 @@ class RedshiftHandler(PostgresHandler):
                 connection.commit()
             except Exception as e:
                 logger.error(f"Error inserting data into {table_name}, {e}!")
+                connection.rollback()
                 response = Response(
                     RESPONSE_TYPE.ERROR,
                     error_code=0,

mindsdb/integrations/handlers/salesforce_handler/requirements.txt

@@ -1 +1 @@
-salesforce_api
+salesforce_api==0.1.45

mindsdb/integrations/handlers/salesforce_handler/salesforce_handler.py

@@ -39,31 +39,8 @@ class SalesforceHandler(APIHandler):
 
         self.connection = None
         self.is_connected = False
-
-        # Register Salesforce tables.
-        self.resource_names = {
-            'Account',
-            'Contact',
-            'Opportunity',
-            'Lead',
-            'Task',
-            'Event',
-            'User',
-            'Product2',
-            'Pricebook2',
-            'PricebookEntry',
-            'Order',
-            'OrderItem',
-            'Case',
-            'Campaign',
-            'CampaignMember',
-            'Contract',
-            'Asset'
-        }
-
-        for resource_name in self.resource_names:
-            table_class = create_table_class(resource_name, resource_name)
-            self._register_table(resource_name, table_class(self))
+        self.thread_safe = True
+        self.resource_names = []
 
     def connect(self) -> salesforce_api.client.Client:
         """
@@ -92,6 +69,12 @@ class SalesforceHandler(APIHandler):
                 is_sandbox=self.connection_data.get('is_sandbox', False)
             )
             self.is_connected = True
+
+            # Register Salesforce tables.
+            for resource_name in self._get_resource_names():
+                table_class = create_table_class(resource_name)
+                self._register_table(resource_name.lower(), table_class(self))
+
             return self.connection
         except AuthenticationError as auth_error:
             logger.error(f"Authentication error connecting to Salesforce, {auth_error}!")
@@ -179,3 +162,15 @@ class SalesforceHandler(APIHandler):
             )
 
         return response
+
+    def _get_resource_names(self) -> None:
+        """
+        Retrieves the names of the Salesforce resources.
+
+        Returns:
+            None
+        """
+        if not self.resource_names:
+            self.resource_names = [resource['name'] for resource in self.connection.sobjects.describe()['sobjects']]
+
+        return self.resource_names

mindsdb/integrations/handlers/salesforce_handler/salesforce_tables.py

@@ -11,7 +11,7 @@ from mindsdb.utilities import log
 logger = log.getLogger(__name__)
 
 
-def create_table_class(table_name: Text, resource_name: Text) -> APIResource:
+def create_table_class(resource_name: Text) -> APIResource:
     """
     Creates a table class for the given Salesforce resource.
     """
@@ -31,7 +31,7 @@ def create_table_class(table_name: Text, resource_name: Text) -> APIResource:
             Returns:
                 pd.DataFrame: A DataFrame containing the data retrieved from the Salesforce resource.
             """
-            query.from_table = table_name
+            query.from_table = resource_name
 
             # SOQL does not support * in SELECT queries. Replace * with column names.
             if isinstance(query.targets[0], Star):

mindsdb/integrations/handlers/slack_handler/slack_handler.py

@@ -267,7 +267,8 @@ class SlackHandler(APIChatHandler):
         user_info = web_connection.auth_test().data
         return user_info['bot_id']
 
-    def subscribe(self, stop_event: threading.Event, callback: Callable, table_name: Text, columns: List = None, **kwargs: Any) -> None:
+    def subscribe(self, stop_event: threading.Event, callback: Callable, table_name: Text = 'messages',
+                  columns: List = None, **kwargs: Any) -> None:
         """
         Subscribes to the Slack API using the Socket Mode for real-time responses to messages.
 

mindsdb/integrations/handlers/timescaledb_handler/timescaledb_handler.py

@@ -10,8 +10,6 @@ class TimeScaleDBHandler(PostgresHandler):
         super().__init__(name, **kwargs)
 
 
-
-
 connection_args = OrderedDict(
     host={
         'type': ARG_TYPE.STR,
@@ -31,6 +29,12 @@ connection_args = OrderedDict(
         'type': ARG_TYPE.STR,
         'description': 'The password to authenticate the user with the TimeScaleDB server.'
     },
+    schema={
+        'type': ARG_TYPE.STR,
+        'description': 'The schema in which objects are searched first.',
+        'required': False,
+        'label': 'Schema'
+    },
     port={
         'type': ARG_TYPE.INT,
         'description': 'Specify port to connect TimeScaleDB '
@@ -39,8 +43,9 @@ connection_args = OrderedDict(
 
 connection_args_example = OrderedDict(
     host='127.0.0.1',
-    port=36806,
-    password='P455W0rD',
-    user='tsdbadmin',
-    database="tsdb"
+    port=5432,
+    password='password',
+    user='root',
+    database="timescaledb",
+    schema='public'
 )