MindsDB 25.5.3.0__py3-none-any.whl → 25.5.4.1__py3-none-any.whl

mindsdb/integrations/handlers/hsqldb_handler/connection_args.py

@@ -27,13 +27,13 @@ connection_args = OrderedDict(
         'description': 'The password to authenticate.',
         'secret': True
     },
-    
+
 )
 
 connection_args_example = OrderedDict(
-    server_name = 'localhost',
-    port =  9001,
-    database_name = 'xdb',
-    username = 'SA',
-    password = 'password'
+    server_name='localhost',
+    port=9001,
+    database_name='xdb',
+    username='SA',
+    password='password'
 )

mindsdb/integrations/handlers/hsqldb_handler/hsqldb_handler.py

@@ -14,6 +14,7 @@ from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
 
 logger = log.getLogger(__name__)
 
+
 class HSQLDBHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the HyperSQL statements.
@@ -21,7 +22,7 @@ class HSQLDBHandler(DatabaseHandler):
 
     name = 'hsqldb'
 
-    def __init__(self, name: str,  **kwargs):
+    def __init__(self, name: str, **kwargs):
         """
         Initialize the handler.
         Args:
@@ -162,7 +163,7 @@ class HSQLDBHandler(DatabaseHandler):
         cursor = connection.cursor()
         cursor.execute("SELECT * FROM information_schema.tables WHERE table_schema NOT IN ('information_schema', 'pg_catalog') AND table_type='BASE TABLE'")
         results = cursor.fetchall()
-        df = pd.DataFrame([x[2] for x in results], columns=['table_name']) # Workaround since cursor.tables() wont work with postgres driver
+        df = pd.DataFrame([x[2] for x in results], columns=['table_name'])  # Workaround since cursor.tables() wont work with postgres driver
         response = Response(
             RESPONSE_TYPE.TABLE,
             df
@@ -181,7 +182,7 @@ class HSQLDBHandler(DatabaseHandler):
 
         connection = self.connect()
         cursor = connection.cursor()
-        query = f'SELECT * FROM information_schema.columns WHERE table_name ={table_name}' # Workaround since cursor.columns() wont work with postgres driver
+        query = f'SELECT * FROM information_schema.columns WHERE table_name ={table_name}'  # Workaround since cursor.columns() wont work with postgres driver
         cursor.execute(query)
         results = cursor.fetchall()
         df = pd.DataFrame(

mindsdb/integrations/handlers/huggingface_api_handler/exceptions.py

@@ -3,4 +3,4 @@ class UnsupportedTaskException(Exception):
 
 
 class InsufficientParametersException(Exception):
-    pass
+    pass

mindsdb/integrations/handlers/huggingface_api_handler/huggingface_api_handler.py

@@ -1,20 +1,13 @@
 import json
-import os
 from typing import Optional, Dict
-
 import pandas as pd
-
 from huggingface_hub import HfApi
 from huggingface_hub import hf_hub_download
-
 from hugging_py_face import NLP, ComputerVision, AudioProcessing, get_in_df_supported_tasks
 
-from mindsdb.utilities.config import Config
 from mindsdb.integrations.libs.base import BaseMLEngine
-
-from .exceptions import UnsupportedTaskException, InsufficientParametersException
-
 from mindsdb.integrations.utilities.handler_utils import get_api_key
+from .exceptions import UnsupportedTaskException, InsufficientParametersException
 
 
 class HuggingFaceInferenceAPIHandler(BaseMLEngine):

mindsdb/integrations/handlers/huggingface_handler/huggingface_handler.py

@@ -117,7 +117,7 @@ class HuggingFaceHandler(BaseMLEngine):
         try:
             pipeline = transformers.pipeline(task=args['task_proper'], model=hf_model_storage_path,
                                              tokenizer=hf_model_storage_path)
-            logger.debug(f'Model already downloaded!')
+            logger.debug('Model already downloaded!')
         ####
         # Otherwise download it
         except (ValueError, OSError):
@@ -154,10 +154,10 @@ class HuggingFaceHandler(BaseMLEngine):
                 labels_map[labels_default[num]] = labels_default[num]
             args["labels_map"] = labels_map
 
-        ###### store and persist in model folder
+        # store and persist in model folder
         self.model_storage.json_set("args", args)
 
-        ###### persist changes to handler folder
+        # persist changes to handler folder
         self.engine_storage.folder_sync(model_name)
 
     # todo move infer tasks to a seperate file
@@ -257,7 +257,7 @@ class HuggingFaceHandler(BaseMLEngine):
             "fill-mask": self.predict_fill_mask,
         }
 
-        ###### get stuff from model folder
+        # get stuff from model folder
         args = self.model_storage.json_get("args")
 
         task = args["task"]
@@ -310,11 +310,11 @@ class HuggingFaceHandler(BaseMLEngine):
                         continue
                     elif truncation_policy == "left":
                         tokens = tokens[
-                            -max_tokens + 1 : -1
+                            -max_tokens + 1: -1
                         ]  # cut 2 empty tokens from left and right
                     else:
                         tokens = tokens[
-                            1 : max_tokens - 1
+                            1: max_tokens - 1
                         ]  # cut 2 empty tokens from left and right
 
                     item = pipeline.tokenizer.decode(tokens)

mindsdb/integrations/handlers/huggingface_handler/requirements.txt

@@ -3,4 +3,4 @@ datasets==2.16.1
 evaluate==0.4.3
 nltk==3.9.1
 huggingface-hub==0.29.3
-torch==2.6.0
+torch==2.6.0

mindsdb/integrations/handlers/huggingface_handler/requirements_cpu.txt

@@ -3,4 +3,4 @@ datasets==2.16.1
 evaluate==0.4.3
 nltk==3.9.1
 huggingface-hub==0.29.3
-torch==2.6.0+cpu
+torch==2.6.0+cpu

mindsdb/integrations/handlers/ignite_handler/ignite_handler.py

@@ -18,6 +18,7 @@ from mindsdb.integrations.libs.response import (
 
 logger = log.getLogger(__name__)
 
+
 class IgniteHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the Apache Ignite statements.
@@ -105,7 +106,7 @@ class IgniteHandler(DatabaseHandler):
             self.connect()
             response.success = True
         except Exception as e:
-            logger.error(f'Error connecting to Apache Ignite!')
+            logger.error('Error connecting to Apache Ignite!')
             response.error_message = str(e)
         finally:
             if response.success is True and need_to_close:

mindsdb/integrations/handlers/impala_handler/impala_handler.py

@@ -1,7 +1,7 @@
 from typing import Optional
 
 import pandas as pd
-from impala import dbapi as db , sqlalchemy as SA
+from impala import dbapi as db, sqlalchemy as SA
 
 from mindsdb_sql_parser import parse_sql
 from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
@@ -29,7 +29,6 @@ class ImpalaHandler(DatabaseHandler):
     def __init__(self, name: str, connection_data: Optional[dict], **kwargs):
         super().__init__(name)
 
-        
         self.parser = parse_sql
         self.dialect = 'impala'
         self.kwargs = kwargs
@@ -38,20 +37,18 @@ class ImpalaHandler(DatabaseHandler):
         self.connection = None
         self.is_connected = False
 
-
-
     def connect(self):
         if self.is_connected is True:
             return self.connection
 
         config = {
             'host': self.connection_data.get('host'),
-            'port': self.connection_data.get('port',21050),
+            'port': self.connection_data.get('port', 21050),
             'user': self.connection_data.get('user'),
             'password': self.connection_data.get('password'),
             'database': self.connection_data.get('database'),
-            
-            
+
+
         }
 
         connection = db.connect(**config)
@@ -67,7 +64,7 @@ class ImpalaHandler(DatabaseHandler):
         return
 
     def check_connection(self) -> StatusResponse:
-       
+
         result = StatusResponse(False)
         need_to_close = self.is_connected is False
 
@@ -100,7 +97,7 @@ class ImpalaHandler(DatabaseHandler):
                 cur.execute(query)
                 result = cur.fetchall()
                 if cur.has_result_set:
-                    
+
                     response = Response(
                         RESPONSE_TYPE.TABLE,
                         pd.DataFrame(
@@ -137,7 +134,7 @@ class ImpalaHandler(DatabaseHandler):
         Get a list with all of the tabels in Impala
         """
         q = "SHOW TABLES;"
-        result= self.native_query(q)
+        result = self.native_query(q)
         df = result.data_frame.rename(columns={'name': 'TABLE_NAME'})
         result.data_frame = df
 
@@ -149,8 +146,8 @@ class ImpalaHandler(DatabaseHandler):
         """
         q = f"DESCRIBE {table_name};"
 
-        result= self.native_query(q)
-        df = result.data_frame.iloc[:,0:2].rename(columns={'name': 'COLUMN_NAME', 'type': 'Data_Type'})
+        result = self.native_query(q)
+        df = result.data_frame.iloc[:, 0:2].rename(columns={'name': 'COLUMN_NAME', 'type': 'Data_Type'})
         result.data_frame = df
 
         return result

mindsdb/integrations/handlers/impala_handler/tests/test_impala_handler.py

@@ -7,12 +7,12 @@ class ImpalaHandlerTest(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.kwargs = {
-            'user':'<UID>',
-            'password':'<P455w0rd>',
-            'host':'127.0.0.1',
-            'port':21050,
-            'database':'temp'
-    
+            'user': '<UID>',
+            'password': '<P455w0rd>',
+            'host': '127.0.0.1',
+            'port': 21050,
+            'database': 'temp'
+
         }
         cls.handler = ImpalaHandler('test_impala_handler', **cls.kwargs)
 
@@ -21,25 +21,25 @@ class ImpalaHandlerTest(unittest.TestCase):
 
     def test_1_connect(self):
         assert self.handler.connect()
-    
+
     def test_2_create_table(self):
         query = "CREATE Table Car(Name Varchar, Price Integer);"
         result = self.handler.query(query)
-        assert result.type is not RESPONSE_TYPE.ERROR 
+        assert result.type is not RESPONSE_TYPE.ERROR
 
     def test_3_insert(self):
         query = "INSERT INTO Car ('Tata SUV', 860000)"
         result = self.handler.query(query)
-        assert result.type is not RESPONSE_TYPE.ERROR 
+        assert result.type is not RESPONSE_TYPE.ERROR
 
     def test_4_native_query_select(self):
         query = "SELECT * FROM Car;"
         result = self.handler.query(query)
-        assert result.type is RESPONSE_TYPE.TABLE 
+        assert result.type is RESPONSE_TYPE.TABLE
 
     def test_5_get_tables(self):
         tables = self.handler.get_tables()
-        assert tables.type is  RESPONSE_TYPE.TABLE
+        assert tables.type is RESPONSE_TYPE.TABLE
 
     def test_6_get_columns(self):
         columns = self.handler.get_columns('Car')

mindsdb/integrations/handlers/influxdb_handler/influxdb_handler.py

@@ -8,9 +8,9 @@ from mindsdb.utilities import log
 from mindsdb_sql_parser import parse_sql
 
 
-
 logger = log.getLogger(__name__)
 
+
 class InfluxDBHandler(APIHandler):
     """InfluxDB handler implementation"""
 
@@ -34,7 +34,6 @@ class InfluxDBHandler(APIHandler):
 
         influxdb_tables_data = InfluxDBTables(self)
         self._register_table("tables", influxdb_tables_data)
-        
 
     def connect(self):
         """Set up the connection required by the handler.
@@ -44,17 +43,15 @@ class InfluxDBHandler(APIHandler):
 
         Raises Expection if ping check fails
         """
-        
+
         if self.is_connected is True:
             return self.connection
-        
-        self.connection=InfluxDBClient3(host=self.connection_data['influxdb_url'],token=self.connection_data['influxdb_token'],org=self.connection_data.get('org'))
 
+        self.connection = InfluxDBClient3(host=self.connection_data['influxdb_url'], token=self.connection_data['influxdb_token'], org=self.connection_data.get('org'))
 
         self.is_connected = True
 
         return self.connection
-        
 
     def check_connection(self) -> StatusResponse:
         """Check connection to the handler.
@@ -90,16 +87,16 @@ class InfluxDBHandler(APIHandler):
         ast = parse_sql(query)
         return self.query(ast)
 
-    def call_influxdb_tables(self,query):
+    def call_influxdb_tables(self, query):
         """Pulls all the records from the given  InfluxDB table and returns it select()
-    
+
         Returns
         -------
-        pd.DataFrame of all the records of the particular InfluxDB 
+        pd.DataFrame of all the records of the particular InfluxDB
         """
-        influx_connection=self.connect()
+        influx_connection = self.connect()
         if query is None:
-            query='SELECT * FROM '+ f"{self.connection_data['influxdb_table_name']}"
-        
-        table=influx_connection.query(query=query,database=self.connection_data['influxdb_db_name'], language='sql')
+            query = 'SELECT * FROM ' + f"{self.connection_data['influxdb_table_name']}"
+
+        table = influx_connection.query(query=query, database=self.connection_data['influxdb_db_name'], language='sql')
         return table.to_pandas()

mindsdb/integrations/handlers/influxdb_handler/influxdb_tables.py

@@ -3,7 +3,6 @@ import pandas as pd
 from typing import List
 
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 from mindsdb.utilities import log
 
 from mindsdb_sql_parser import ast
@@ -12,6 +11,7 @@ from mindsdb.integrations.utilities.handlers.query_utilities.select_query_utilit
 
 logger = log.getLogger(__name__)
 
+
 class InfluxDBTables(APITable):
     """InfluxDB Tables implementation"""
 
@@ -29,23 +29,23 @@ class InfluxDBTables(APITable):
         ValueError
             If the query contains an unsupported condition
         """
-        
-        table_name=self.handler.connection_data['influxdb_table_name']
+
+        table_name = self.handler.connection_data['influxdb_table_name']
         select_statement_parser = SELECTQueryParser(
             query,
             "tables",
             self.get_columns()
         )
         selected_columns, where_conditions, order_by_conditions, _ = select_statement_parser.parse_query()
-        
+
         try:
             selected_columns.remove("name")
             selected_columns.remove("tags")
         except Exception as e:
             logger.warn(e)
 
-        formatted_query=self.get_select_query(table_name,selected_columns,where_conditions,order_by_conditions,query.limit)
-        influxdb_tables_df  = self.handler.call_influxdb_tables(formatted_query)
+        formatted_query = self.get_select_query(table_name, selected_columns, where_conditions, order_by_conditions, query.limit)
+        influxdb_tables_df = self.handler.call_influxdb_tables(formatted_query)
 
         return influxdb_tables_df
 
@@ -56,26 +56,26 @@ class InfluxDBTables(APITable):
         List[str]
             List of columns
         """
-        
+
         dataframe = self.handler.call_influxdb_tables(f"SELECT * FROM {self.handler.connection_data['influxdb_table_name']} LIMIT 1")
 
         return list(dataframe.columns)
-    def get_select_query(self,table_name,selected_columns, where_conditions, order_by_conditions, result_limit):
+
+    def get_select_query(self, table_name, selected_columns, where_conditions, order_by_conditions, result_limit):
         """Gets Well formed Query
         Returns
         -------
         str
         """
-        columns=", ".join([f'"{column}"' for column in selected_columns])
-        query=f'SELECT {columns} FROM "{table_name}"'
-        if(where_conditions is not None and len(where_conditions)>0):
-            query+=" WHERE "
-            query+=" AND ".join([f"{i[1]} {i[0]} {i[2]}" for i in where_conditions])
-        if (order_by_conditions!={} and order_by_conditions['columns'] is not None and len(order_by_conditions['columns'])>0):
-            query+=" ORDER BY "
-            query+=", ".join([f'{column_name} {"ASC"if asc else "DESC"}' for column_name,asc in zip(order_by_conditions['columns'],order_by_conditions['ascending'])] )
-        if(result_limit is not None):
-            query+=f" LIMIT {result_limit}"
-        query+=";"
+        columns = ", ".join([f'"{column}"' for column in selected_columns])
+        query = f'SELECT {columns} FROM "{table_name}"'
+        if (where_conditions is not None and len(where_conditions) > 0):
+            query += " WHERE "
+            query += " AND ".join([f"{i[1]} {i[0]} {i[2]}" for i in where_conditions])
+        if (order_by_conditions != {} and order_by_conditions['columns'] is not None and len(order_by_conditions['columns']) > 0):
+            query += " ORDER BY "
+            query += ", ".join([f'{column_name} {"ASC"if asc else "DESC"}' for column_name, asc in zip(order_by_conditions['columns'], order_by_conditions['ascending'])])
+        if (result_limit is not None):
+            query += f" LIMIT {result_limit}"
+        query += ";"
         return query
-

mindsdb/integrations/handlers/informix_handler/__about__.py

@@ -1,9 +1,9 @@
-__title__ = 'MindsDB IBM Informix handler'
-__package_name__ = 'mindsdb_informix_handler'
-__version__ = '0.0.1'
+__title__ = "MindsDB IBM Informix handler"
+__package_name__ = "mindsdb_informix_handler"
+__version__ = "0.0.1"
 __description__ = "MindsDB handler for IBM Informix"
-__author__ = 'Parthiv Makwana'
-__github__ = 'https://github.com/mindsdb/mindsdb'
-__pypi__ = 'https://pypi.org/project/mindsdb/'
-__license__ = 'MIT'
-__copyright__ = 'Copyright 2022- mindsdb'
+__author__ = "Parthiv Makwana"
+__github__ = "https://github.com/mindsdb/mindsdb"
+__pypi__ = "https://pypi.org/project/mindsdb/"
+__license__ = "MIT"
+__copyright__ = "Copyright 2022- mindsdb"

mindsdb/integrations/handlers/informix_handler/__init__.py

@@ -2,6 +2,7 @@ from mindsdb.integrations.libs.const import HANDLER_TYPE
 
 try:
     from .informix_handler import InformixHandler as Handler
+
     import_error = None
 except Exception as e:
     Handler = None
@@ -9,12 +10,18 @@ except Exception as e:
 from .__about__ import __version__ as version, __description__ as description
 
 
-title = 'IBM Informix'
-name = 'informix'
+title = "IBM Informix"
+name = "informix"
 type = HANDLER_TYPE.DATA
-icon_path = 'icon.svg'
+icon_path = "icon.svg"
 
 __all__ = [
-    'Handler', 'version', 'name', 'type', 'title', 'description',
-    'import_error', 'icon_path'
+    "Handler",
+    "version",
+    "name",
+    "type",
+    "title",
+    "description",
+    "import_error",
+    "icon_path",
 ]