diff --git a/mindsdb_sdk/__about__.py b/mindsdb_sdk/__about__.py index 9c3a3d6..54f8c9b 100755 --- a/mindsdb_sdk/__about__.py +++ b/mindsdb_sdk/__about__.py @@ -1,6 +1,6 @@ __title__ = 'mindsdb_sdk' __package_name__ = 'mindsdb_sdk' -__version__ = '3.4.8' +__version__ = '3.4.9' __description__ = "MindsDB Python SDK, provides an SDK to use a remote mindsdb instance" __email__ = "jorge@mindsdb.com" __author__ = 'MindsDB Inc' diff --git a/mindsdb_sdk/databases.py b/mindsdb_sdk/databases.py index 011be4f..8e5732d 100644 --- a/mindsdb_sdk/databases.py +++ b/mindsdb_sdk/databases.py @@ -1,6 +1,6 @@ from typing import Dict, List, Union -from mindsdb_sql_parser.ast.mindsdb import CreateDatabase +from mindsdb_sql_parser.ast.mindsdb import AlterDatabase, CreateDatabase from mindsdb_sql_parser.ast import DropDatabase, Identifier from mindsdb_sdk.utils.objects_collection import CollectionBase @@ -175,3 +175,19 @@ def get(self, name: str) -> Database: if name not in databases: raise AttributeError("Database doesn't exist") return databases[name] + + def update(self, name: str, connection_args: Dict) -> Database: + """ + Update integration connection parameters + + :param name: name of integration + :param connection_args: new connection parameters + :return: updated Database object + """ + ast_query = AlterDatabase( + name=Identifier(name), + altered_params={ + "parameters": connection_args}, + ) + self.api.sql_query(ast_query.to_string()) + return self.get(name) diff --git a/requirements.txt b/requirements.txt index c76a188..17f62dd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ requests pandas >= 1.3.5 -mindsdb-sql-parser ~= 0.0.0 +mindsdb-sql-parser == 0.10.0 docstring-parser >= 0.7.3 tenacity >= 8.0.1 openai >= 1.74.1