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

mindsdb/integrations/handlers/altibase_handler/tests/test_altibase_handler_dsn.py

@@ -2,24 +2,25 @@ import unittest
 from mindsdb.integrations.handlers.altibase_handler.altibase_handler import AltibaseHandler
 from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
 
+
 class AltibaseHandlerTest(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.kwargs = {
             "connection_data": {
                 "dsn": "altiodbc",
-                #Override
-                #"host": "127.0.0.1",
-                #"port": 20300,
-                #"user": "sys",
-                #"password": "manager"
+                # Override
+                # "host": "127.0.0.1",
+                # "port": 20300,
+                # "user": "sys",
+                # "password": "manager"
             }
         }
         cls.handler = AltibaseHandler('test_altibase_handler', **cls.kwargs)
 
     def test_0_connect(self):
         assert self.handler.connect()
-    
+
     def test_1_drop_table(self):
         # Not supported 'IF EXISTS' syntax
         res = self.handler.query("DROP TABLE TEST_TABLE")
@@ -28,20 +29,20 @@ class AltibaseHandlerTest(unittest.TestCase):
     def test_2_create_table(self):
         res = self.handler.query(
             '''CREATE TABLE TEST_TABLE (
-                ID INT PRIMARY KEY, 
+                ID INT PRIMARY KEY,
                 NAME VARCHAR(14)
                 )'''
-            )
+        )
         assert res.type is RESPONSE_TYPE.OK
 
     def test_3_insert(self):
         res = self.handler.query(
-            """INSERT INTO TEST_TABLE 
-            VALUES 
+            """INSERT INTO TEST_TABLE
+            VALUES
                 (100,'ONE HUNDRED'),
                 (200,'TWO HUNDRED'),
                 (300,'THREE HUNDRED')"""
-            )
+        )
         assert res.type is RESPONSE_TYPE.OK
 
     def test_4_select(self):
@@ -61,8 +62,7 @@ class AltibaseHandlerTest(unittest.TestCase):
 
     def test_8_disconnect(self):
         assert self.handler.disconnect()
-    
 
-        
+
 if __name__ == '__main__':
     unittest.main()

mindsdb/integrations/handlers/anthropic_handler/__init__.py

@@ -1,10 +1,10 @@
+from .__about__ import __version__ as version
+from .__about__ import __description__ as description
 from mindsdb.integrations.libs.const import HANDLER_TYPE
 from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
-from .__about__ import __description__ as description
-from .__about__ import __version__ as version
 
 try:
     from .anthropic_handler import AnthropicHandler as Handler

mindsdb/integrations/handlers/anthropic_handler/anthropic_handler.py

@@ -1,12 +1,10 @@
-import os
 from typing import Dict, Optional
 
 import pandas as pd
-from anthropic import AI_PROMPT, HUMAN_PROMPT, Anthropic
+from anthropic import Anthropic
 
 from mindsdb.integrations.libs.base import BaseMLEngine
 from mindsdb.utilities import log
-from mindsdb.utilities.config import Config
 
 from mindsdb.integrations.utilities.handler_utils import get_api_key
 

mindsdb/integrations/handlers/aurora_handler/aurora_handler.py

@@ -14,6 +14,7 @@ from mindsdb.integrations.handlers.postgres_handler.postgres_handler import Post
 
 logger = log.getLogger(__name__)
 
+
 class AuroraHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the Amazon Aurora statements.

mindsdb/integrations/handlers/autosklearn_handler/autosklearn_handler.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict
+from typing import Optional
 
 import dill
 import pandas as pd

mindsdb/integrations/handlers/autosklearn_handler/config.py

@@ -1,6 +1,5 @@
 import sys
 from dataclasses import dataclass
-from autosklearn.metrics import *
 from autosklearn.metrics import Scorer
 
 

mindsdb/integrations/handlers/bigquery_handler/bigquery_handler.py

@@ -7,7 +7,7 @@ from mindsdb.utilities import log
 from mindsdb_sql_parser.ast.base import ASTNode
 from mindsdb.integrations.libs.base import DatabaseHandler
 from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
-from mindsdb.integrations.utilities.handlers.auth_utilities import GoogleServiceAccountOAuth2Manager
+from mindsdb.integrations.utilities.handlers.auth_utilities.google import GoogleServiceAccountOAuth2Manager
 from mindsdb.integrations.libs.response import (
     HandlerStatusResponse as StatusResponse,
     HandlerResponse as Response,

mindsdb/integrations/handlers/bigquery_handler/tests/test_bigquery_handler.py

@@ -4,7 +4,7 @@ import pandas as pd
 from google.cloud import bigquery
 
 from mindsdb.integrations.handlers.bigquery_handler.bigquery_handler import BigQueryHandler
-from mindsdb.integrations.utilities.handlers.auth_utilities import GoogleServiceAccountOAuth2Manager
+from mindsdb.integrations.utilities.handlers.auth_utilities.google import GoogleServiceAccountOAuth2Manager
 from mindsdb.integrations.libs.response import (
     HandlerResponse as Response,
     RESPONSE_TYPE,

mindsdb/integrations/handlers/binance_handler/binance_handler.py

@@ -16,6 +16,7 @@ _BASE_BINANCE_US_URL = 'https://api.binance.us'
 
 logger = log.getLogger(__name__)
 
+
 class BinanceHandler(APIHandler):
     """A class for handling connections and interactions with the Binance API.
 

mindsdb/integrations/handlers/binance_handler/binance_tables.py

@@ -1,5 +1,4 @@
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.libs.response import HandlerResponse as Response
 from mindsdb.integrations.utilities.date_utils import interval_str_to_duration_ms, utc_date_str_to_timestamp_ms
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 from mindsdb_sql_parser import ast
@@ -21,7 +20,7 @@ class BinanceAggregatedTradesTable(APITable):
 
     def _get_batch_klines(self, executor: ThreadPoolExecutor, total_results: int, params: Dict) -> pd.DataFrame:
         """Gets aggregate trade data in batches and combines the results together.
-        
+
         Returns all results as a pandas DataFrame.
 
         Args:
@@ -70,7 +69,7 @@ class BinanceAggregatedTradesTable(APITable):
 
     def _get_kline_params_from_conditions(self, conditions: List) -> Dict:
         """Gets aggregate trade data API params from SQL WHERE conditions.
-        
+
         Returns params to use for Binance API call to klines.
 
         Args:
@@ -111,7 +110,7 @@ class BinanceAggregatedTradesTable(APITable):
 
     def select(self, query: ast.Select) -> pd.DataFrame:
         """Selects data from the Binance API and returns it as a pandas DataFrame.
-        
+
         Returns dataframe representing the Binance API results.
 
         Args:

mindsdb/integrations/handlers/byom_handler/__init__.py

@@ -18,4 +18,3 @@ icon_path = "icon.svg"
 __all__ = [
     'Handler', 'version', 'name', 'type', 'title', 'description', 'connection_args'
 ]
-

mindsdb/integrations/handlers/ckan_handler/ckan_handler.py

@@ -19,6 +19,7 @@ class DatasetsTable(APITable):
     Datasets table contains information about CKAN datasets.
     This table is used to list all datasets available in CKAN that have datastore active resources.
     '''
+
     def select(self, query: ast.Select) -> pd.DataFrame:
         conditions = extract_comparison_conditions(query.where) if query.where else []
         limit = query.limit.value if query.limit else 1000
@@ -63,6 +64,7 @@ class ResourceIDsTable(APITable):
     ResourceIDs table contains information about CKAN resources.
     This table is used to list all resources available in CKAN that are datastore active.
     '''
+
     def select(self, query: ast.Select) -> pd.DataFrame:
         conditions = extract_comparison_conditions(query.where) if query.where else []
         limit = query.limit.value if query.limit else 1000
@@ -114,6 +116,7 @@ class DatastoreTable(APITable):
     This table is used to query data from CKAN datastore resources.
     It is using the datastore_search_sql API to execute SQL queries on CKAN datastore resources.
     '''
+
     def select(self, query: ast.Select) -> pd.DataFrame:
         conditions = extract_comparison_conditions(query.where) if query.where else []
         resource_id = self.extract_resource_id(conditions)

mindsdb/integrations/handlers/clickhouse_handler/__init__.py

@@ -17,4 +17,4 @@ icon_path = 'icon.svg'
 __all__ = [
     'Handler', 'version', 'name', 'type', 'title', 'description',
     'connection_args', 'connection_args_example', 'import_error', 'icon_path'
-]
+]

mindsdb/integrations/handlers/cloud_spanner_handler/tests/test_cloud_spanner_handler.py

@@ -13,7 +13,6 @@ class CloudSpannerHandlerTest(unittest.TestCase):
         cls.kwargs = {'connection_data': {'database_id': 'example-db', 'instance_id': 'test-instance', 'project': 'your-project-id'}}
         cls.handler = CloudSpannerHandler('test_cloud_spanner_handler', **cls.kwargs)
 
-
     def test_0_connect(self):
         self.handler.connect()
 
@@ -24,7 +23,6 @@ class CloudSpannerHandlerTest(unittest.TestCase):
         res = self.handler.query('CREATE TABLE integers(i INT64) PRIMARY KEY (i)')
         assert res.type is not RESPONSE_TYPE.ERROR
 
-
     def test_3_insert_into_table(self):
         res = self.handler.query('INSERT INTO integers (i) VALUES (42)')
         assert res.type is not RESPONSE_TYPE.ERROR

mindsdb/integrations/handlers/cloud_sql_handler/cloud_sql_handler.py

@@ -2,7 +2,6 @@ from typing import Optional
 
 from mindsdb_sql_parser.ast.base import ASTNode
 
-from mindsdb.utilities import log
 from mindsdb.integrations.libs.base import DatabaseHandler
 from mindsdb.integrations.libs.response import (
     HandlerStatusResponse as StatusResponse

mindsdb/integrations/handlers/cohere_handler/__init__.py

@@ -1,9 +1,9 @@
+from .__about__ import __version__ as version, __description__ as description
 from mindsdb.integrations.libs.const import HANDLER_TYPE
 from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
-from .__about__ import __version__ as version, __description__ as description
 try:
     from .cohere_handler import CohereHandler as Handler
     import_error = None

mindsdb/integrations/handlers/cohere_handler/cohere_handler.py

@@ -1,10 +1,8 @@
-import os
 from typing import Optional, Dict
 
 import cohere
 import pandas as pd
 
-from mindsdb.utilities.config import Config
 from mindsdb.integrations.libs.base import BaseMLEngine
 
 from mindsdb.utilities import log
@@ -14,6 +12,7 @@ from mindsdb.integrations.utilities.handler_utils import get_api_key
 
 logger = log.getLogger(__name__)
 
+
 class CohereHandler(BaseMLEngine):
     """
     Integration with the Cohere Python Library
@@ -39,28 +38,27 @@ class CohereHandler(BaseMLEngine):
 
         if input_column not in df.columns:
             raise RuntimeError(f'Column "{input_column}" not found in input data')
-        
-        result_df = pd.DataFrame() 
+
+        result_df = pd.DataFrame()
 
         if args['using']['task'] == 'text-summarization':
-            result_df['predictions'] = df[input_column].apply(self.predict_text_summary)     
+            result_df['predictions'] = df[input_column].apply(self.predict_text_summary)
 
         elif args['using']['task'] == 'text-generation':
             result_df['predictions'] = df[input_column].apply(self.predict_text_generation)
-      
+
         else:
             raise Exception(f"Task {args['using']['task']} is not supported!")
 
         result_df = result_df.rename(columns={'predictions': args['target']})
-        
-        return result_df
 
+        return result_df
 
-    def predict_text_summary(self,text):
-        """ 
+    def predict_text_summary(self, text):
+        """
         connects with cohere api to predict the summary of the input text
 
-        """ 
+        """
 
         args = self.model_storage.json_get('args')
 
@@ -72,8 +70,8 @@ class CohereHandler(BaseMLEngine):
 
         return text_summary
 
-    def predict_text_generation(self,text):
-        """    
+    def predict_text_generation(self, text):
+        """
         connects with cohere api to predict the next prompt of the input text
 
         """

mindsdb/integrations/handlers/confluence_handler/confluence_tables.py

@@ -19,6 +19,7 @@ class ConfluenceSpacesTable(APIResource):
     """
     The table abstraction for the 'spaces' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,
@@ -145,6 +146,7 @@ class ConfluencePagesTable(APIResource):
     """
     The table abstraction for the 'pages' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,
@@ -281,6 +283,7 @@ class ConfluenceBlogPostsTable(APIResource):
     """
     The table abstraction for the 'blogposts' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,
@@ -412,6 +415,7 @@ class ConfluenceWhiteboardsTable(APIResource):
     """
     The table abstraction for the 'whiteboards' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,
@@ -488,6 +492,7 @@ class ConfluenceDatabasesTable(APIResource):
     """
     The table abstraction for the 'databases' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,
@@ -564,6 +569,7 @@ class ConfluenceTasksTable(APIResource):
     """
     The table abstraction for the 'tasks' resource of the Confluence API.
     """
+
     def list(
         self,
         conditions: List[FilterCondition] = None,

mindsdb/integrations/handlers/databend_handler/connection_args.py

@@ -33,4 +33,4 @@ connection_args_example = OrderedDict(
     user='root',
     password='password',
     database='test_db'
-)
+)

mindsdb/integrations/handlers/databend_handler/databend_handler.py

@@ -69,7 +69,7 @@ class DatabendHandler(DatabaseHandler):
         self.is_connected = True
 
         return self.connection
-    
+
     def disconnect(self):
         """
         Close any existing connections.
@@ -105,7 +105,7 @@ class DatabendHandler(DatabaseHandler):
                 self.is_connected = False
 
         return response
-    
+
     def native_query(self, query: str) -> StatusResponse:
         """
         Receive raw query and act upon it somehow.
@@ -146,7 +146,7 @@ class DatabendHandler(DatabaseHandler):
             self.disconnect()
 
         return response
-    
+
     def query(self, query: ASTNode) -> StatusResponse:
         """
         Receive query as AST (abstract syntax tree) and act upon it somehow.
@@ -178,7 +178,7 @@ class DatabendHandler(DatabaseHandler):
             result.data_frame = df.rename(columns={f'Tables_in_{self.connection_data["database"]}': 'table_name'})
 
         return result
-    
+
     def get_columns(self, table_name: str) -> StatusResponse:
         """
         Returns a list of entity columns.

mindsdb/integrations/handlers/databend_handler/tests/__init__.py

@@ -1 +0,0 @@
- 

mindsdb/integrations/handlers/databend_handler/tests/test_databend_handler.py

@@ -34,4 +34,4 @@ class DatabendHandlerTest(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

mindsdb/integrations/handlers/derby_handler/connection_args.py

@@ -47,4 +47,4 @@ connection_args_example = OrderedDict(
     database="testdb",
     jdbcClass='org.apache.derby.jdbc.ClientDriver',
     jdbcJarLocation='/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derby.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbyclient.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbynet.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbyoptionaltools.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbyrun.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbyshared.jar,/opt/homebrew/Cellar/derby/10.16.1.1/libexec/lib/derbytools.jar',
-)
+)

mindsdb/integrations/handlers/derby_handler/derby_handler.py

@@ -13,11 +13,10 @@ import jaydebeapi as jdbcconnector
 
 logger = log.getLogger(__name__)
 
-class DerbyHandler(DatabaseHandler):
-
 
-    name= 'derby'
+class DerbyHandler(DatabaseHandler):
 
+    name = 'derby'
 
     def __init__(self, name: str, connection_data: Optional[dict], **kwargs):
         """ Initialize the handler
@@ -27,18 +26,17 @@ class DerbyHandler(DatabaseHandler):
             **kwargs: arbitrary keyword arguments.
         """
         super().__init__(name)
-        
+
         self.kwargs = kwargs
         self.parser = parse_sql
         self.database = connection_data['database']
         self.connection_config = connection_data
         self.host = connection_data['host']
         self.port = connection_data['port']
-        self.schema =  'APP'
+        self.schema = 'APP'
         self.connection = None
         self.is_connected = False
-          
-    
+
     def connect(self):
         """ Set up any connections required by the handler
         Should return output of check_connection() method after attempting
@@ -56,18 +54,18 @@ class DerbyHandler(DatabaseHandler):
 
         jdbc_url = "jdbc:derby://" + self.host + ":" + self.port + "/" + self.database + ";"
 
-        if not jdbc_class: 
+        if not jdbc_class:
             jdbc_class = "org.apache.derby.jdbc.ClientDriver"
 
-        if user: 
+        if user:
             self.schema = user
 
         try:
-            if user and password and jar_location: 
+            if user and password and jar_location:
                 self.connection = jdbcconnector.connect(jclassname=jdbc_class, url=jdbc_url, driver_args=[user, password], jars=jar_location.split(","))
-            elif user and password: 
+            elif user and password:
                 self.connection = jdbcconnector.connect(jclassname=jdbc_class, url=jdbc_url, driver_args=[user, password])
-            elif jar_location: 
+            elif jar_location:
                 self.connection = jdbcconnector.connect(jclassname=jdbc_class, url=jdbc_url, jars=jar_location.split(","))
             else:
                 self.connection = jdbcconnector.connect(jdbc_class, jdbc_url)
@@ -76,7 +74,6 @@ class DerbyHandler(DatabaseHandler):
 
         return self.connection
 
-
     def disconnect(self):
         """ Close any existing connections
         Should switch self.is_connected.
@@ -85,12 +82,11 @@ class DerbyHandler(DatabaseHandler):
             return
         try:
             self.connection.close()
-            self.is_connected=False
+            self.is_connected = False
         except Exception as e:
             logger.error(f"Error while disconnecting to {self.database}, {e}")
 
-        return 
-
+        return
 
     def check_connection(self) -> StatusResponse:
         """ Check connection to the handler
@@ -114,7 +110,6 @@ class DerbyHandler(DatabaseHandler):
 
         return responseCode
 
-
     def native_query(self, query: str) -> StatusResponse:
         """Receive raw query and act upon it somehow.
         Args:
@@ -129,7 +124,7 @@ class DerbyHandler(DatabaseHandler):
             try:
                 cur.execute(query)
                 if cur.description:
-                    result = cur.fetchall() 
+                    result = cur.fetchall()
                     response = Response(
                         RESPONSE_TYPE.TABLE,
                         data_frame=pd.DataFrame(
@@ -153,7 +148,6 @@ class DerbyHandler(DatabaseHandler):
 
         return response
 
-    
     def query(self, query: ASTNode) -> StatusResponse:
         """Render and execute a SQL query.
 
@@ -173,7 +167,6 @@ class DerbyHandler(DatabaseHandler):
 
         return self.native_query(query_str)
 
-
     def get_tables(self) -> StatusResponse:
         """Get a list of all the tables in the database.
 
@@ -182,13 +175,12 @@ class DerbyHandler(DatabaseHandler):
         """
         query = f'''
         SELECT st.tablename FROM sys.systables st LEFT OUTER JOIN sys.sysschemas ss ON (st.schemaid = ss.schemaid) WHERE ss.schemaname ='{self.schema}' '''
-    
+
         result = self.native_query(query)
         df = result.data_frame
         result.data_frame = df.rename(columns={df.columns[0]: 'table_name'})
         return result
 
-    
     def get_columns(self, table_name: str) -> StatusResponse:
         """Get details about a table.
 

mindsdb/integrations/handlers/derby_handler/tests/test_derby_handler.py

@@ -2,6 +2,7 @@ import unittest
 from mindsdb.integrations.handlers.derby_handler.derby_handler import DerbyHandler
 from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
 
+
 class DerbyHandlerTest(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
@@ -9,14 +10,14 @@ class DerbyHandlerTest(unittest.TestCase):
             "connection_data": {
                 "host": "localhost",
                 "port": "1527",
-                "database": "seconddb",                
+                "database": "seconddb",
             }
         }
         cls.handler = DerbyHandler('test_derby_handler', **cls.kwargs)
 
     def test_0_connect(self):
         self.handler.connect()
-    
+
     def test_1_check_connection(self):
         self.handler.check_connection()
 
@@ -31,7 +32,7 @@ class DerbyHandlerTest(unittest.TestCase):
     def test_4_select(self):
         res = self.handler.query('SELECT * FROM TESTTABLEX')
         assert res.type is RESPONSE_TYPE.TABLE
-    
+
     def test_5_get_tables(self):
         res = self.handler.get_tables()
         assert res.type is RESPONSE_TYPE.TABLE
@@ -39,8 +40,7 @@ class DerbyHandlerTest(unittest.TestCase):
     def test_6_get_columns(self):
         res = self.handler.get_columns("TESTTABLEX")
         assert res.type is RESPONSE_TYPE.TABLE
-    
 
-        
+
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

mindsdb/integrations/handlers/discord_handler/discord_handler.py

@@ -1,5 +1,4 @@
 import requests
-import asyncio
 import pandas as pd
 
 from mindsdb.integrations.handlers.discord_handler.discord_tables import MessagesTable
@@ -18,6 +17,7 @@ from mindsdb.integrations.libs.response import (
 discord_bot = None
 logger = log.getLogger(__name__)
 
+
 class DiscordHandler(APIHandler):
     """
     The Discord handler implementation.
@@ -55,7 +55,7 @@ class DiscordHandler(APIHandler):
         if self.is_connected:
             return StatusResponse(True)
 
-        url = f'https://discord.com/api/v10/applications/@me'
+        url = 'https://discord.com/api/v10/applications/@me'
         result = requests.get(
             url,
             headers={
@@ -105,7 +105,7 @@ class DiscordHandler(APIHandler):
         df = self.call_discord_api(operation, params)
 
         return Response(RESPONSE_TYPE.TABLE, data_frame=df)
-    
+
     def utc_to_snowflake(self, utc_date: str) -> int:
         """
         Convert a UTC date to a Snowflake date.
@@ -120,8 +120,8 @@ class DiscordHandler(APIHandler):
         )
 
     def call_discord_api(
-            self, operation: str, params: dict = None, filters: list = None
-        ):
+        self, operation: str, params: dict = None, filters: list = None
+    ):
         """
         Call a Discord API method.
         Args: