MindsDB 1.3.1__py3-none-any.whl → 24.12.4.0__py3-none-any.whl

mindsdb/integrations/handlers/byom_handler/icon.svg

@@ -0,0 +1,13 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <g clip-path="url(#clip0_146_1000)">
+    <rect x="0.0292969" width="39.9609" height="39.9902" rx="5.1416" fill="#00A587"/>
+    <path d="M26.6851 20.0842C27.0527 22.9691 27.3939 25.6778 27.7481 28.3835C27.79 28.7076 27.7229 28.822 27.3599 28.8401C25.1739 28.9487 25.1751 28.9574 24.3849 26.9668C24.0559 26.1385 23.7353 25.307 23.3849 24.487C23.3077 24.3522 23.2793 24.1957 23.3041 24.0429C23.329 23.8902 23.4056 23.7503 23.5217 23.6459C24.5557 22.5162 25.5629 21.356 26.6851 20.0842Z" fill="white"/>
+    <path d="M13.8364 22.5021C14.1494 24.0901 14.4583 25.5827 14.7302 27.0819C14.8142 27.546 15.1919 28.1357 14.8687 28.4495C14.5901 28.7191 13.9715 28.3376 13.496 28.3549C12.2328 28.402 11.6449 27.8181 11.4397 26.5922C11.3192 25.8726 11.2659 25.3742 11.869 24.8168C12.5729 24.0861 13.23 23.313 13.8364 22.5021Z" fill="white"/>
+    <path d="M17.4031 25.3986C17.4031 24.5068 17.3892 23.6149 17.4098 22.7235C17.4178 22.3733 17.3187 22.2552 16.9406 22.2428C14.3611 22.1589 11.7821 22.057 9.20374 21.9373C8.79877 21.9187 8.63888 22.0319 8.55285 22.4258C8.15501 24.2499 7.70345 26.0634 7.31484 27.8896C7.20573 28.4025 7.00681 28.5916 6.46167 28.5333C5.8568 28.4717 5.24787 28.4578 4.64075 28.4921C4.18332 28.5148 4.09519 28.3269 4.10191 27.9351C4.14136 25.1959 4.19969 22.4567 4.17283 19.7184C4.16788 18.8757 4.4539 18.0564 4.98405 17.3946C6.4109 15.4437 7.78529 13.4576 9.16765 11.4765C9.24221 11.3409 9.35956 11.2327 9.50201 11.1684C9.64446 11.1041 9.80429 11.087 9.95745 11.1198C15.1117 11.6623 20.268 12.188 25.4262 12.6966C25.7488 12.7347 26.0569 12.8506 26.323 13.034C29.3924 14.9965 32.4564 16.966 35.5413 18.9046C36.0029 19.1936 36.0155 19.3588 35.5719 19.6201C35.4764 19.6709 35.3911 19.7383 35.3201 19.8191C34.4317 20.9484 33.3897 20.8976 32.1307 20.3972C30.609 19.7939 29.0042 19.3939 27.4439 18.881C27.2954 18.8218 27.1344 18.7994 26.975 18.8157C26.8157 18.8321 26.6628 18.8867 26.5299 18.9748C25.0946 19.8518 23.6528 20.718 22.2044 21.5735C22.074 21.6423 21.9626 21.7413 21.8798 21.8621C21.7971 21.9828 21.7456 22.1216 21.7298 22.2664C21.4121 24.1492 21.0487 26.0237 20.7566 27.9115C20.6727 28.4524 20.4767 28.6238 19.9303 28.5862C19.2364 28.5458 18.5407 28.5447 17.8467 28.5829C17.435 28.6011 17.3296 28.459 17.3431 28.073C17.3699 27.1815 17.3515 26.2888 17.3515 25.397L17.4031 25.3986Z" fill="white"/>
+  </g>
+  <defs>
+    <clipPath id="clip0_146_1000">
+      <rect width="40" height="40" fill="white"/>
+    </clipPath>
+  </defs>
+</svg>

mindsdb/integrations/handlers/byom_handler/proc_wrapper.py

@@ -0,0 +1,229 @@
+"""
+Utility functions used in the 'Bring Your Own Model' (BYOM) engine.
+
+These functions interact with interfaces (stdin, stdout), python files, and the actual BYOM engine.
+
+In particular, they:
+    - Wrap and run python code in separate python proceess.
+    - Communicate with parent process throughout stdin/out using pickle to serialize objects.
+
+
+The flow is as follows:
+
+    1. Receive module code, method with parameters and stored attributes from parent process
+    2. A python class object is created from the code
+    3. Class is instanced and filled with stored attributes
+    4. A calls to the chosen method of the class is performed with any relevant parameters that were passed
+    5. Response is generated, appropriately packaged and sent to stdout
+    6. Exit
+"""
+
+import io
+import sys
+import pickle
+import inspect
+from enum import Enum
+
+import pandas as pd
+
+
+class BYOM_METHOD(Enum):
+    CHECK = 1
+    TRAIN = 2
+    PREDICT = 3
+    FINETUNE = 4
+    DESCRIBE = 5
+    FUNC_CALL = 6
+
+
+def pd_encode(df):
+    return df.to_parquet(engine='pyarrow')
+
+
+def pd_decode(encoded):
+    fd = io.BytesIO()
+    fd.write(encoded)
+    fd.seek(0)
+    return pd.read_parquet(fd, engine='pyarrow')
+
+
+def encode(obj):
+    return pickle.dumps(obj, protocol=5)
+
+
+def decode(encoded):
+    return pickle.loads(encoded)
+
+
+def return_output(obj):
+    # read stdin
+    encoded = encode(obj)
+    with open(1, 'wb') as fd:
+        fd.write(encoded)
+    sys.exit(0)
+
+
+def get_input():
+    # write to stdout
+    with open(0, 'rb') as fd:
+        encoded = fd.read()
+        obj = decode(encoded)
+    return obj
+
+
+def import_string(code, module_name='model'):
+    # import string as python module
+
+    import types
+    module = types.ModuleType(module_name)
+
+    exec(code, module.__dict__)
+    # sys.modules['my_module'] = module
+    return module
+
+
+def find_model_class(module):
+    # find the first class that contains predict and train methods
+    for _, cls in inspect.getmembers(module, inspect.isclass):
+        if inspect.getmodule(cls) is not None:
+            # is imported class
+            continue
+
+        funcs = [
+            name
+            for name, _ in inspect.getmembers(cls, inspect.isfunction)
+        ]
+        if 'predict' in funcs and 'train' in funcs:
+            # found
+            return cls
+
+
+def get_methods_info(module):
+    # get all methods and their types
+    methods = {}
+    for method_name, method in inspect.getmembers(module, inspect.isfunction):
+
+        sig = inspect.signature(method)
+        input_params = [
+            {'name': name, 'type': param.annotation.__name__}
+            for name, param in sig.parameters.items()
+        ]
+        methods[method_name] = {
+            'input_params': input_params,
+            'output_type': sig.return_annotation.__name__
+        }
+    return methods
+
+
+def check_module(module, mode):
+    # checks module and returns info
+
+    methods = {}
+    if mode == 'custom_function':
+        methods = get_methods_info(module)
+
+    else:
+        # is BYOM, check it.
+        model_class = find_model_class(module)
+        if model_class is None:
+            raise RuntimeError('Unable to find model class (it has to have `train` and `predict` methods)')
+
+        # try to initialize
+        model_class()
+    return {'methods': methods}
+
+
+def main():
+    # replace print output to stderr
+    sys.stdout = sys.stderr
+
+    params = get_input()
+
+    method = BYOM_METHOD(params['method'])
+    code = params['code']
+
+    module = import_string(code)
+
+    if method == BYOM_METHOD.FUNC_CALL:
+        func_name = params['func_name']
+        args = params['args']
+
+        func = getattr(module, func_name)
+        return return_output(func(*args))
+
+    if method == BYOM_METHOD.CHECK:
+
+        mode = params['mode']
+        info = check_module(module, mode)
+
+        return return_output(info)
+
+    model_class = find_model_class(module)
+
+    if method == BYOM_METHOD.TRAIN:
+        df = params['df']
+        if df is not None:
+            df = pd_decode(df)
+        to_predict = params['to_predict']
+        args = params['args']
+        model = model_class()
+
+        call_args = [df, to_predict]
+        if args:
+            call_args.append(args)
+        model.train(*call_args)
+
+        # return model
+        data = model.__dict__
+
+        model_state = encode(data)
+        return_output(model_state)
+
+    elif method == BYOM_METHOD.PREDICT:
+        model_state = params['model_state']
+        df = pd_decode(params['df'])
+        args = params['args']
+
+        model = model_class()
+        model.__dict__ = decode(model_state)
+
+        call_args = [df]
+        if args:
+            call_args.append(args)
+        res = model.predict(*call_args)
+        return_output(pd_encode(res))
+
+    elif method == BYOM_METHOD.FINETUNE:
+        model_state = params['model_state']
+        df = pd_decode(params['df'])
+        args = params['args']
+
+        model = model_class()
+        model.__dict__ = decode(model_state)
+
+        call_args = [df]
+        if args:
+            call_args.append(args)
+
+        model.finetune(*call_args)
+
+        # return model
+        data = model.__dict__
+        model_state = encode(data)
+        return_output(model_state)
+
+    elif method == BYOM_METHOD.DESCRIBE:
+        model_state = params['model_state']
+        model = model_class()
+        model.__dict__ = decode(model_state)
+        try:
+            df = model.describe(params.get('attribute'))
+        except Exception:
+            return_output(pd_encode(pd.DataFrame()))
+        return_output(pd_encode(df))
+
+    raise NotImplementedError(method)
+
+
+if __name__ == '__main__':
+    main()

mindsdb/integrations/handlers/byom_handler/requirements.txt

@@ -0,0 +1,2 @@
+virtualenv
+pyarrow==14.0.1

mindsdb/integrations/handlers/cassandra_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = 'MindsDB Cassandra handler'
+__package_name__ = 'mindsdb_cassandra_handler'
+__version__ = '0.0.1'
+__description__ = "MindsDB handler for CassandraDB"
+__author__ = 'MindsDB Inc'
+__github__ = 'https://github.com/mindsdb/mindsdb'
+__pypi__ = 'https://pypi.org/project/mindsdb/'
+__license__ = 'MIT'
+__copyright__ = 'Copyright 2022- mindsdb'

mindsdb/integrations/handlers/cassandra_handler/__init__.py

@@ -0,0 +1,21 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .connection_args import connection_args
+try:
+    from .cassandra_handler import CassandraHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+from .__about__ import __version__ as version, __description__ as description
+
+
+title = 'Apache Cassandra'
+name = 'cassandra'
+type = HANDLER_TYPE.DATA
+icon_path = 'icon.svg'
+
+__all__ = [
+    'Handler', 'version', 'name', 'type', 'title', 'description',
+    'connection_args', 'import_error', 'icon_path'
+]

mindsdb/integrations/handlers/cassandra_handler/cassandra_handler.py

@@ -0,0 +1,27 @@
+from mindsdb.integrations.handlers.scylla_handler import Handler as ScyllaHandler
+from mindsdb.integrations.libs.response import (
+    HandlerResponse as Response)
+
+
+class CassandraHandler(ScyllaHandler):
+    """
+    This handler handles connection and execution of the Cassandra statements.
+    """
+
+    name = 'cassandra'
+
+    def __init__(self, name, **kwargs):
+        super().__init__(name, **kwargs)
+
+    def get_tables(self) -> Response:
+        """
+        Get the list of tables in the connected Cassandra database.
+
+        :return: List of table names.
+        """
+        sql = "DESCRIBE TABLES"
+        result = self.native_query(sql)
+        df = result.data_frame
+        df = df.rename(columns={'name': 'table_name'})
+        result.data_frame = df
+        return result

mindsdb/integrations/handlers/cassandra_handler/connection_args.py

@@ -0,0 +1,50 @@
+from collections import OrderedDict
+
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+
+
+connection_args = OrderedDict(
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'User name',
+        'required': True,
+        'label': 'User',
+    },
+    password={
+        'type': ARG_TYPE.PWD,
+        'description': 'Password',
+        'required': True,
+        'label': 'Password',
+        'secret': True
+    },
+    protocol_version={
+        'type': ARG_TYPE.INT,
+        'description': 'is not required and defaults to 4.',
+        'required': False,
+        'label': 'Protocol version',
+    },
+    host={
+        'type': ARG_TYPE.STR,
+        'description': ' is the host name or IP address of the Cassandra database.',
+        'required': True,
+        'label': 'Host',
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'Server port',
+        'required': True,
+        'label': 'Port',
+    },
+    keyspace={
+        'type': ARG_TYPE.STR,
+        'description': ' is the keyspace to connect, the top level container for tables.',
+        'required': True,
+        'label': 'Keyspace',
+    },
+    secure_connect_bundle={
+        'type': ARG_TYPE.STR,
+        'description': 'Path or URL to the secure connect bundle',
+        'required': False,
+        'label': 'Host',
+    },
+)

mindsdb/integrations/handlers/cassandra_handler/icon.svg

@@ -0,0 +1,33 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <g clip-path="url(#clip0_141_393)">
+    <path d="M31.6925 17.3973C31.9821 20.7406 26.717 23.926 19.9514 24.5052C13.1857 25.0843 7.44679 22.8467 7.15721 19.5033C6.86763 16.16 12.1327 12.9746 18.8984 12.3955C25.6903 11.8163 31.4029 14.054 31.6925 17.3973Z" fill="#BBE6FB"/>
+    <g opacity="0.35">
+      <path d="M19.1105 12.8005C16.5306 12.7479 13.5032 13.7746 11.2392 14.7486C11.0023 15.4331 10.8707 16.1439 10.8707 16.881C10.8707 20.8561 14.4509 24.0678 18.8736 24.0678C23.2963 24.0678 26.8765 20.8561 26.8765 16.881C26.8765 15.6173 26.508 14.459 25.8762 13.4323C24.086 13.1427 21.6114 12.8269 19.1105 12.8005Z" fill="white"/>
+    </g>
+    <path d="M18.8916 12.802C17.47 12.723 14.9954 13.4601 12.9421 14.1972C12.6262 14.9343 12.4682 15.7504 12.4682 16.5928C12.4682 20.0678 15.6273 22.8846 18.9443 22.8846C22.2613 22.8846 25.1834 20.0678 25.4203 16.5928C25.4993 15.4608 25.1044 14.4078 24.5779 13.4864C22.4192 13.0915 20.1552 12.8809 18.8916 12.802Z" fill="white"/>
+    <path d="M15.2136 16.1364C15.3979 15.7415 15.6611 15.5572 15.9244 15.2413C15.8981 15.1623 15.8454 14.8991 15.8454 14.8201C15.8454 14.4252 16.1613 14.1093 16.5562 14.1093C16.6615 14.1093 16.7405 14.1356 16.8195 14.162C17.9778 13.2669 19.531 12.872 21.0842 13.2142C21.2158 13.2406 21.3738 13.2932 21.5054 13.3195C19.9785 13.1089 18.241 13.6618 17.0827 14.6095C17.109 14.6885 17.2143 14.9254 17.2143 15.0307C17.2143 15.4256 16.9511 15.5045 16.5825 15.5045C16.5036 15.5045 16.3982 15.4782 16.3193 15.4519C16.056 15.8205 15.7401 16.4786 15.5822 16.8998C16.2403 17.479 16.8195 17.6896 17.5829 17.9791C17.5829 17.9528 17.6092 17.9002 17.6092 17.8738C17.6092 16.6892 18.6359 15.7415 19.8206 15.7415C20.8736 15.7415 21.7423 16.5049 21.9266 17.5053C22.4531 17.1631 22.9533 16.9261 23.3482 16.4259C23.2692 16.3206 23.1639 16.0047 23.1639 15.8468C23.1639 15.5046 23.4535 15.215 23.7957 15.215C23.8484 15.215 23.901 15.215 23.9537 15.2413C24.1379 14.8464 24.2959 14.3989 24.4012 13.9514C22.5584 12.5824 19.0045 12.7667 19.0045 12.7667C19.0045 12.7667 16.1613 12.6088 14.2396 13.5565C14.2922 14.5042 14.6608 15.3992 15.2136 16.1364Z" fill="#373535"/>
+    <path d="M24.0087 15.5925C24.1404 15.6978 24.2193 15.8032 24.272 15.9085C24.4563 16.2244 24.0877 16.5139 23.7455 16.5139C23.7191 16.5139 23.6665 16.5139 23.6402 16.5139C23.2453 17.1457 22.5345 17.9355 21.929 18.3304C22.8767 18.4884 23.7718 18.8569 24.5089 19.4624C25.0354 18.541 25.4303 17.1721 25.4303 16.0401C25.4303 15.1713 25.0354 14.5395 24.4563 14.0657C24.4036 14.5132 24.1667 15.1713 24.0087 15.5925Z" fill="#373535"/>
+    <path d="M16.2673 21.509C16.5832 20.6403 17.373 19.9558 18.1101 19.3503C18.0048 19.1924 17.8468 18.8238 17.7678 18.6659C16.8728 18.4026 16.0567 17.8761 15.4512 17.1917C15.4512 17.218 15.4512 17.218 15.4249 17.2443C15.3196 17.7708 15.2933 18.2973 15.3459 18.7975C15.7671 18.8765 16.1093 19.245 16.1093 19.7189C16.1093 19.9821 15.8724 20.377 15.6881 20.5613C15.8987 20.9825 16.0304 21.1931 16.2673 21.509Z" fill="#373535"/>
+    <path d="M16.2124 22.2556C15.8439 21.887 15.5806 21.1236 15.3174 20.6497C15.2647 20.6497 15.2121 20.6497 15.1857 20.6497C14.6592 20.6497 14.1327 20.2285 14.2643 19.7283C14.3433 19.4651 14.5276 19.2808 14.8435 19.1228C14.7645 18.5437 14.7645 17.7276 14.8962 17.1221C14.9225 16.9641 15.0278 16.8852 15.0804 16.7009C14.5013 15.8585 14.159 14.7791 14.159 13.6998C14.159 13.6471 14.159 13.6208 14.159 13.5945C13.2113 14.0683 12.4742 14.8318 12.4742 16.0164C12.4479 18.6753 13.9484 21.2552 16.2124 22.2556Z" fill="#373535"/>
+    <path d="M21.7817 18.5999C21.4922 19.4423 20.7024 20.0214 19.781 20.0214C19.2808 20.0214 18.8333 19.8372 18.4647 19.5739C17.7276 20.1531 17.0432 21.1797 16.7009 22.0748C16.8589 22.2328 16.9642 22.3907 17.1485 22.5224C17.675 22.654 18.3858 22.5224 18.9649 22.5224C21.2552 22.5224 23.256 21.522 24.4143 19.7318C23.5982 19.2053 22.7031 18.6788 21.7817 18.5999Z" fill="#373535"/>
+    <path d="M15.1551 16.426C15.3394 16.0311 15.55 15.6889 15.8132 15.3466C15.7869 15.2676 15.7606 15.1887 15.7606 15.1097C15.7606 14.7148 16.0765 14.3989 16.4714 14.3989C16.5767 14.3989 16.6557 14.4252 16.7346 14.4516C17.893 13.5565 19.4462 13.1616 20.9994 13.5038C21.131 13.5302 21.2889 13.5828 21.4206 13.6091C19.8937 13.3985 18.3142 13.8724 17.1295 14.8201C17.1558 14.8991 17.1822 15.0044 17.1822 15.0834C17.1822 15.4783 16.8663 15.7942 16.4714 15.7942C16.3924 15.7942 16.2871 15.7678 16.2081 15.7415C15.9449 16.1101 15.7343 16.4786 15.5763 16.9261C16.1292 17.5316 16.8136 18.0055 17.5771 18.2951C17.5771 18.2688 17.5771 18.2161 17.5771 18.1898C17.5771 17.0051 18.5248 16.0574 19.7094 16.0574C20.7624 16.0574 21.6312 16.8208 21.8154 17.8212C22.342 17.479 22.8158 17.0578 23.2107 16.5576C23.1317 16.4523 23.0527 16.3207 23.0527 16.1627C23.0527 15.8205 23.3423 15.5309 23.6846 15.5309C23.7372 15.5309 23.7899 15.5309 23.8425 15.5572C24.0268 15.1623 24.1847 14.7148 24.29 14.2673C22.4473 12.8983 18.8933 13.0826 18.8933 13.0826C18.8933 13.0826 16.0502 12.9247 14.1284 13.8724C14.2337 14.8201 14.6023 15.6888 15.1551 16.426Z" fill="#1287B1"/>
+    <path d="M24.1269 15.7034C24.2585 15.8087 24.3375 15.993 24.3375 16.1773C24.3375 16.5195 24.0479 16.8091 23.7057 16.8091C23.6794 16.8091 23.6267 16.8091 23.6004 16.8091C23.2055 17.4409 22.679 17.9674 22.0472 18.3623C22.9686 18.4939 23.8373 18.8888 24.5481 19.4943C25.0746 18.5729 25.3642 17.4936 25.3642 16.3352C25.3642 15.4665 24.9693 14.8347 24.3901 14.3608C24.3638 14.8347 24.2585 15.2822 24.1269 15.7034Z" fill="#1287B1"/>
+    <path d="M16.3835 21.7294C16.6994 20.8607 17.2522 20.0709 17.9894 19.4654C17.8841 19.3075 17.8051 19.1495 17.7261 18.9916C16.831 18.7283 16.0149 18.2018 15.4095 17.5173C15.4095 17.5437 15.4095 17.5437 15.3831 17.57C15.2778 18.0965 15.2515 18.623 15.3042 19.1232C15.7254 19.2022 16.0676 19.5707 16.0676 20.0446C16.0676 20.3078 15.9623 20.5448 15.778 20.729C15.936 21.0713 16.1466 21.4135 16.3835 21.7294Z" fill="#1287B1"/>
+    <path d="M16.2087 22.2448C15.8402 21.8763 15.5243 21.4287 15.261 20.9549C15.2084 20.9549 15.1557 20.9549 15.1294 20.9549C14.6029 20.9549 14.208 20.5337 14.208 20.0335C14.208 19.6649 14.4186 19.349 14.7345 19.1911C14.6555 18.6119 14.6819 18.0327 14.8135 17.4273C14.8398 17.2693 14.8925 17.085 14.9451 16.9271C14.366 16.0847 14.0501 15.0843 14.0501 14.005C14.0501 13.9523 14.0501 13.926 14.0501 13.8997C13.1023 14.3735 12.3652 15.137 12.3652 16.3216C12.3916 18.9805 13.9448 21.2445 16.2087 22.2448Z" fill="#1287B1"/>
+    <path d="M21.7233 18.9158C21.4337 19.7582 20.6439 20.3373 19.7225 20.3373C19.2223 20.3373 18.7748 20.1531 18.4062 19.8898C17.6691 20.469 17.09 21.2587 16.7477 22.1538C16.9057 22.3118 17.0636 22.4697 17.2479 22.6277C17.7744 22.7593 18.3273 22.8383 18.8801 22.8383C21.1704 22.8383 23.1711 21.6536 24.3031 19.8898C23.5397 19.337 22.6446 18.9947 21.7233 18.9158Z" fill="#1287B1"/>
+    <path d="M21.8057 18.3722L23.7801 17.8983L21.8057 17.793L23.4116 16.5557L21.5951 17.2139L22.8324 15.5027L21.1213 16.6347L21.7267 14.6076L20.5421 16.2661L20.5158 14.1601L19.7523 16.0292L19.0942 14.1338L19.1205 16.3715L18.0149 14.3707L18.6203 16.5031L16.8039 15.1868L18.2255 16.9769L15.9878 16.1609L17.7516 17.4771L15.4876 17.5561L17.8042 18.1616L15.4876 18.6618L17.7779 18.8197L15.8298 20.057L17.8306 19.3462L16.4353 21.11L18.1728 19.6885L17.541 22.0314L18.8836 20.1097L18.6993 22.3736L19.4628 20.3203L20.1209 22.479L20.2788 20.3203L21.2529 22.163L20.6737 20.057L22.4375 21.4259L21.2792 19.6621L23.2273 20.4256L21.7267 18.9513L23.7275 19.1093L21.8057 18.3722Z" fill="white"/>
+    <g opacity="0.35">
+      <path d="M0.819336 22.1322C6.92684 16.0247 11.8234 14.1292 16.246 13.2605C16.8252 13.1552 16.9305 11.8652 16.9305 11.8652C16.9305 11.8652 17.0095 12.8129 17.4307 12.9446C17.8519 13.0762 18.3784 11.4704 18.3784 11.4704C18.3784 11.4704 17.8782 12.9183 18.3784 12.9972C18.8786 13.0762 19.7737 11.602 19.7737 11.602C19.7737 11.602 19.4051 12.8656 19.642 12.9446C19.879 13.0236 21.1162 11.1018 21.1162 11.1018C21.1162 11.1018 20.3791 12.3654 21.0636 12.497C21.7744 12.6287 22.7748 11.6283 22.7748 11.6283C22.7748 11.6283 21.985 12.5234 22.3272 12.655C23.8541 13.1815 25.0651 11.2071 25.0651 11.2071C25.0651 11.2071 24.7755 12.0758 24.0384 13.0235C25.6442 13.4448 26.8289 11.0228 26.8289 11.0228L25.6442 13.4448C26.2497 13.7607 28.7243 10.9438 28.7243 10.9438C28.7243 10.9438 27.4607 13.1552 26.7236 13.6027C27.1448 13.9186 28.619 12.6287 28.619 12.6287C28.619 12.6287 27.4344 14.1029 27.8819 14.1819C28.54 14.7084 30.962 11.444 30.962 11.444C30.962 11.444 30.0142 13.3658 28.619 14.8927C29.7773 15.4718 32.6731 11.7336 32.6731 11.7336C32.6731 11.7336 32.5941 12.9182 30.4881 14.8137C32.0413 14.6031 34.0684 11.6546 34.0684 11.6546C34.0684 11.6546 33.3313 13.866 31.5674 15.4192C33.0943 15.2612 35.5426 11.4704 35.5426 11.4704C35.5426 11.4704 34.5949 14.1029 32.6731 15.6034C34.8055 16.0773 37.8856 12.8919 37.8856 12.8919C37.8856 12.8919 36.6746 15.1559 35.0951 16.0773C36.8589 16.7618 39.2808 13.9713 39.2808 13.9713C39.2808 13.9713 36.8589 17.7095 33.0153 17.5515C31.7517 17.4989 27.9345 13.708 19.721 14.0239C8.90125 14.4188 6.58461 19.0258 0.819336 22.1322Z" fill="#373535"/>
+    </g>
+    <path d="M1.0517 21.6749C7.1592 15.5674 12.0557 13.672 16.4784 12.8032C17.0576 12.6979 17.1629 11.408 17.1629 11.408C17.1629 11.408 17.2418 12.3557 17.663 12.4873C18.0843 12.6189 18.6108 11.0131 18.6108 11.0131C18.6108 11.0131 18.1106 12.461 18.6108 12.54C19.1109 12.6189 20.006 11.1447 20.006 11.1447C20.006 11.1447 19.6375 12.4083 19.8744 12.4873C20.1113 12.5663 21.3486 10.6445 21.3486 10.6445C21.3486 10.6445 20.6115 11.9082 21.296 12.0398C21.9804 12.1714 23.0071 11.171 23.0071 11.171C23.0071 11.171 22.2174 12.0661 22.5596 12.1977C24.0865 12.7242 25.2974 10.7498 25.2974 10.7498C25.2974 10.7498 25.0078 11.6186 24.2707 12.5663C25.8766 12.9875 27.0612 10.5656 27.0612 10.5656L25.8766 12.9875C26.4821 13.3034 28.9567 10.4866 28.9567 10.4866C28.9567 10.4866 27.693 12.6979 26.9559 13.1455C27.3771 13.4614 28.8514 12.1714 28.8514 12.1714C28.8514 12.1714 27.6667 13.6456 28.1142 13.7246C28.7724 14.2511 31.1943 10.9868 31.1943 10.9868C31.1943 10.9868 30.2466 12.9085 28.8514 14.4354C30.0097 15.0145 32.9055 11.2763 32.9055 11.2763C32.9055 11.2763 32.8265 12.461 30.7205 14.3564C32.2737 14.1458 34.3007 11.1974 34.3007 11.1974C34.3007 11.1974 33.5636 13.4087 31.7998 14.9619C33.3267 14.8039 35.775 11.0131 35.775 11.0131C35.775 11.0131 34.8272 13.6456 32.9055 15.1462C35.0378 15.62 38.1179 12.4347 38.1179 12.4347C38.1179 12.4347 36.9069 14.6986 35.3274 15.62C37.0912 16.3045 39.5132 13.514 39.5132 13.514C39.5132 13.514 37.0912 17.2522 33.2477 17.0943C31.9841 17.0416 28.1669 13.2507 19.9534 13.5667C9.13361 13.9615 6.84329 18.5422 1.0517 21.6749Z" fill="#373535"/>
+    <g opacity="0.35">
+      <path d="M0.639404 22.7927C3.87743 24.4249 6.82588 22.5558 10.5378 23.6088C13.3809 24.3986 16.8032 25.399 21.4892 24.7145C26.1751 24.03 30.1765 22.003 32.072 18.3701C32.6775 16.7379 35.0994 18.7386 35.0994 18.7386C35.0994 18.7386 33.3356 18.0015 33.3883 18.3437C33.4409 18.686 36.0471 19.8969 36.0471 19.8969C36.0471 19.8969 33.6778 19.1598 33.8358 19.7127C33.9674 20.2655 36.7842 22.4768 36.7842 22.4768C36.7842 22.4768 33.5989 20.0549 33.3356 20.3971C33.0723 20.713 34.6782 22.0556 34.6782 22.0556C34.6782 22.0556 31.993 20.3971 31.0453 20.871C30.3608 21.2132 33.8358 23.5035 33.8358 23.5035C33.8358 23.5035 31.5718 21.7924 31.0453 22.2399C30.5188 22.6874 33.5989 26.373 33.5989 26.373C33.5989 26.373 30.1239 22.6084 29.7553 22.7927C29.4131 22.977 30.7294 24.7935 30.7294 24.7935C30.7294 24.7935 29.0182 22.898 28.5444 23.2403C28.0705 23.5562 30.9926 28.479 30.9926 28.479C30.9926 28.479 27.8073 23.4509 26.491 24.2933C28.5444 28.8213 27.7809 29.4531 27.7809 29.4531C27.7809 29.4531 27.5177 24.8461 24.9378 24.9514C23.6478 25.0041 25.5433 28.4001 25.5433 28.4001C25.5433 28.4001 23.7795 25.32 22.6211 25.3726C24.7798 29.4531 24.0164 31.1116 24.0164 31.1116C24.0164 31.1116 24.385 27.6366 21.4628 25.6885C22.5685 26.6626 20.7784 31.1116 20.7784 31.1116C20.7784 31.1116 22.3579 24.5829 19.62 25.8202C19.1725 26.0044 19.5937 29.0845 19.5937 29.0845C19.5937 29.0845 19.0409 25.5832 18.1985 25.7675C17.7246 25.8728 15.1184 30.6377 15.1184 30.6377C15.1184 30.6377 17.2508 25.5043 16.5926 25.7149C16.1188 25.8728 15.25 27.9788 15.25 27.9788C15.25 27.9788 15.6712 25.8991 15.25 25.7675C14.8288 25.6359 12.4069 27.9788 12.4069 27.9788C12.4069 27.9788 14.197 25.8991 13.8811 25.4516C13.4072 24.7935 12.8807 24.7671 12.2226 25.0304C11.2749 25.3989 10.0113 27.1101 10.0113 27.1101C10.0113 27.1101 11.2749 25.3726 11.0643 24.7145C10.6957 23.6615 7.06281 26.7415 7.06281 26.7415C7.06281 26.7415 9.2215 24.7145 8.72131 24.267C8.22113 23.8458 5.56226 23.9774 4.71985 23.9774C2.24526 23.9774 1.40284 23.3982 0.639404 22.7927Z" fill="#373535"/>
+    </g>
+    <path d="M0.539612 22.4346C3.77764 24.0668 6.72609 22.1976 10.438 23.2507C13.2811 24.0404 16.7034 25.0408 21.3894 24.3563C26.0753 23.6719 30.0768 21.6448 31.9722 18.0119C32.5777 16.3797 34.9996 18.3805 34.9996 18.3805C34.9996 18.3805 33.2358 17.6433 33.2885 17.9856C33.3411 18.3278 35.9473 19.5388 35.9473 19.5388C35.9473 19.5388 33.578 18.8017 33.736 19.3545C33.8676 19.9073 36.6844 22.1187 36.6844 22.1187C36.6844 22.1187 33.4991 19.6967 33.2358 20.039C32.9726 20.3549 34.5784 21.6975 34.5784 21.6975C34.5784 21.6975 31.8932 20.039 30.9455 20.5128C30.261 20.8551 33.736 23.1454 33.736 23.1454C33.736 23.1454 31.472 21.4342 30.9455 21.8817C30.419 22.3293 33.4991 26.0148 33.4991 26.0148C33.4991 26.0148 30.0241 22.2503 29.6556 22.4346C29.3133 22.6189 30.6296 24.4353 30.6296 24.4353C30.6296 24.4353 28.9184 22.5399 28.4446 22.8821C27.9707 23.198 30.8928 28.1209 30.8928 28.1209C30.8928 28.1209 27.7075 23.0927 26.3912 23.9351C28.4446 28.4631 27.6811 29.0949 27.6811 29.0949C27.6811 29.0949 27.4179 24.488 24.838 24.5933C23.548 24.6459 25.4435 28.0419 25.4435 28.0419C25.4435 28.0419 23.6797 24.9618 22.5214 25.0145C24.68 29.0949 23.9166 30.7534 23.9166 30.7534C23.9166 30.7534 24.2852 27.2785 21.363 25.3304C22.4687 26.3044 20.6786 30.7534 20.6786 30.7534C20.6786 30.7534 22.2581 24.2247 19.5203 25.462C19.0727 25.6463 19.4939 28.7263 19.4939 28.7263C19.4939 28.7263 18.9411 25.2251 18.0987 25.4093C17.6248 25.5146 15.0186 30.2796 15.0186 30.2796C15.0186 30.2796 17.151 25.1461 16.4928 25.3567C16.019 25.5146 15.1502 27.6207 15.1502 27.6207C15.1502 27.6207 15.5714 25.541 15.1502 25.4093C14.729 25.2777 12.3071 27.6207 12.3071 27.6207C12.3071 27.6207 14.0972 25.541 13.7813 25.0934C13.3075 24.4353 12.7809 24.409 12.1228 24.6722C11.1751 25.0408 9.91147 26.7519 9.91147 26.7519C9.91147 26.7519 11.1751 25.0145 10.9645 24.3563C10.5959 23.3033 6.96302 26.3834 6.96302 26.3834C6.96302 26.3834 9.12171 24.3563 8.62152 23.9351C8.12134 23.5139 5.46247 23.6455 4.62006 23.6455C2.11914 23.5929 1.27672 23.0137 0.539612 22.4346Z" fill="#373535"/>
+  </g>
+  <defs>
+    <clipPath id="clip0_141_393">
+      <rect width="40" height="40" fill="white"/>
+    </clipPath>
+  </defs>
+</svg>

mindsdb/integrations/handlers/cassandra_handler/requirements.txt

@@ -0,0 +1 @@
+-r mindsdb/integrations/handlers/scylla_handler/requirements.txt

mindsdb/integrations/handlers/cassandra_handler/tests/test_cassandra_handler.py

@@ -0,0 +1,39 @@
+import unittest
+from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
+from mindsdb.integrations.handlers.cassandra_handler.cassandra_handler import CassandraHandler
+
+
+class CassandraHandlerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.kwargs = {
+            "connection_data": {
+                "host": "127.0.0.1",
+                "port": "9043",
+                "user": "cassandra",
+                "password": "",
+                "keyspace": "test_data",
+                "protocol_version": 4
+            }
+        }
+        cls.handler = CassandraHandler('test_cassandra_handler', **cls.kwargs)
+
+    def test_0_connect(self):
+        self.handler.check_connection()
+
+    def test_1_native_query_show_keyspaces(self):
+        dbs = self.handler.native_query("DESC KEYSPACES;")
+        assert dbs.type is not RESPONSE_TYPE.ERROR
+
+    def test_2_get_tables(self):
+        tbls = self.handler.get_tables()
+        assert tbls.type is not RESPONSE_TYPE.ERROR
+
+    def test_3_describe_table(self):
+        described = self.handler.get_columns("home_rentals")
+        assert described.type is RESPONSE_TYPE.TABLE
+
+    def test_4_select_query(self):
+        query = "SELECT * FROM home_rentals WHERE 'id'='3712'"
+        result = self.handler.query(query)
+        assert result.type is RESPONSE_TYPE.TABLE

mindsdb/integrations/handlers/chromadb_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = "MindsDB ChromaDB handler"
+__package_name__ = "mindsdb_chromadb_handler"
+__version__ = "0.0.1"
+__description__ = "MindsDB handler for ChromaDB"
+__author__ = "Daniel Usvyat"
+__github__ = "https://github.com/mindsdb/mindsdb"
+__pypi__ = "https://pypi.org/project/mindsdb/"
+__license__ = "MIT"
+__copyright__ = "Copyright 2023 - mindsdb"

mindsdb/integrations/handlers/chromadb_handler/__init__.py

@@ -0,0 +1,29 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __description__ as description
+from .__about__ import __version__ as version
+from .connection_args import connection_args, connection_args_example
+try:
+    from .chromadb_handler import ChromaDBHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+title = "ChromaDB"
+name = "chromadb"
+type = HANDLER_TYPE.DATA
+icon_path = "icon.png"
+
+__all__ = [
+    "Handler",
+    "version",
+    "name",
+    "type",
+    "title",
+    "description",
+    "connection_args",
+    "connection_args_example",
+    "import_error",
+    "icon_path",
+]