Ai 1343 Add Clients for DataScience, Encryption Services And Refactor #235

mariankrotil · 2025-08-15T12:45:00Z

In this PR

Refactor clients to submodules in the clients
Divide individual clients to the corresponding files and update imports
add a base client module within clients: clients/base.py containing the Raw and Service clients
Introduce patch method to the base clients
Create a Data Science client in clients/data_science.py calling data science service and using its endpoints for data apps.
Create an Encryption client in clients/encryption.py working over encryption service and using encrypt endpoint

…ubpackage

…se client preventing from circ imports

…uspend data app methods

…endpoint to the storage client

…ting given the corresponding endpoints

…ts for data science client methods

src/keboola_mcp_server/clients/client.py

…table due to long db sync

integtests/clients/test_client.py

src/keboola_mcp_server/clients/__init__.py

src/keboola_mcp_server/clients/base.py

src/keboola_mcp_server/clients/data_science.py

…t, improve docs

… when no token provided as encryption client does not use storage token in the headers

src/keboola_mcp_server/clients/data_science.py

src/keboola_mcp_server/clients/encryption.py

src/keboola_mcp_server/clients/data_science.py

integtests/clients/test_client.py

integtests/clients/test_data_science.py

integtests/clients/test_encryption.py

…turn type

vita-stejskal · 2025-08-19T13:02:22Z

This is looking good in general. There are going to be many conflicts with AI-1349-branch-id-http-header branch, but I'll wait for you to merge this PR first and then update AI-1349-branch-id-http-header branch.

…handle edge cases for None values

… naming

integtests/clients/test_data_science.py

src/keboola_mcp_server/clients/data_science.py

src/keboola_mcp_server/clients/encryption.py

mariankrotil added 11 commits August 12, 2025 17:36

AI-1343 feat: add data app component id

6a03cbd

AI-1343 feat: create client for data science service, create client s…

a7b25c7

…ubpackage

AI-1343 refactor: move client file to client subpackage and create ba…

1c03526

…se client preventing from circ imports

AI-1343 refactor: update imports wrt client submodule changes

d18d013

AI-1343 feat: add encryption client to encrpyt values

971cd97

AI-1343 style: update code wrt flake8 using black and isort

cea8f3f

AI-1343 fix: update endpoints, docs, refactor

a3e98de

AI-1343 feat: update data science client, add password, deploy, and s…

6dfc84d

…uspend data app methods

AI-1343 feat: add patch method to the base client

4146e06

AI-1343 feat: add method getting versions of component from versions …

f9a9e22

…endpoint to the storage client

AI-1343 style: apply flake8 changes

4ecb6d1

mariankrotil changed the title ~~Ai 1343 mcp add support for data apps~~ Ai 1343 Add Clients for DataScience, Encryption Services And Refactor Aug 15, 2025

Merge branch 'main' into AI-1343-mcp-add-support-for-data-apps

4bb1f9f

mariankrotil requested review from vita-stejskal, radcliffkey and ncapek and removed request for vita-stejskal August 15, 2025 13:54

mariankrotil added 6 commits August 18, 2025 09:14

Merge branch 'main' into AI-1343-mcp-add-support-for-data-apps

a95af68

AI-1343 feat(client): add handlers for listing, getting logs and dele…

92042e3

…ting given the corresponding endpoints

AI-1343 feat: add get method returning text value instead of json

2eab15a

AI-1343 test: Move tests for client to the corresponding dir, add tes…

9b08f89

…ts for data science client methods

AI-1343 refactor: move client tests to the cor. dir

ad226b5

AI-1343 style: apply black, isort and flake8 changes

4c9a436

radcliffkey reviewed Aug 18, 2025

View reviewed changes

src/keboola_mcp_server/clients/client.py Outdated Show resolved Hide resolved

mariankrotil added 4 commits August 18, 2025 17:22

AI-1343 test: add client ecryption integ tests

9a31c75

AI-1343 refactor: move clients to separate submodules

86489a9

AI-1343 test: skip global serarch test for client method as it is uns…

892b024

…table due to long db sync

AI-1343 style: apply isort

51f2bfb

radcliffkey reviewed Aug 19, 2025

View reviewed changes

integtests/clients/test_client.py Outdated Show resolved Hide resolved

radcliffkey reviewed Aug 19, 2025

View reviewed changes

src/keboola_mcp_server/clients/data_science.py Outdated Show resolved Hide resolved

src/keboola_mcp_server/clients/data_science.py Show resolved Hide resolved

mariankrotil added 6 commits August 19, 2025 14:35

AI-1343 refactor: validate parameters in encryption client for encryp…

ad951c7

…t, improve docs

AI-1343 fix: make clients token parameter optional and handle headers…

7ddeea2

… when no token provided as encryption client does not use storage token in the headers

AI-1343 style: apply black

0fb290f

AI-1343 refactor: update description, rename classes wrt reviews

a7b124e

Merge branch 'main' into AI-1343-mcp-add-support-for-data-apps

5efd70b

AI-1343 style: apply tox

7cda4d8

vita-stejskal reviewed Aug 19, 2025

View reviewed changes

AI-1343 docs: improve client method get_text description to follow re…

008f229

…turn type

mariankrotil added 8 commits August 19, 2025 15:41

AI-1343 refactor: update parameters for client methods - structured, …

5e89017

…handle edge cases for None values

AI-1343 test: add unit tests for data science client tail_app_log method

8247f12

AI-1343 test: improve tests with asynccontext for initial data app tests

b503441

AI-1343 fix: increase waiting time for global search

b48ff37

AI-1343 refactor: update parameter types for encrypt method wrt reviews

4457d0c

AI-1343 test: unskip the global search test

812a341

AI-1343 refactor: remove async annot for non-async test, improve test…

a9359bf

… naming

AI-1343 refactor: add shortcut imports

a16b0de

vita-stejskal reviewed Aug 19, 2025

View reviewed changes

integtests/clients/test_data_science.py Outdated Show resolved Hide resolved

src/keboola_mcp_server/clients/data_science.py Show resolved Hide resolved

vita-stejskal approved these changes Aug 19, 2025

View reviewed changes

src/keboola_mcp_server/clients/encryption.py Outdated Show resolved Hide resolved

AI-1343 refactor: change Optional[s] to s | None in clients, ref code

558a387

mariankrotil merged commit 960648f into main Aug 20, 2025
19 checks passed

mariankrotil deleted the AI-1343-mcp-add-support-for-data-apps branch August 20, 2025 07:41

mariankrotil mentioned this pull request Aug 29, 2025

Release 1.20.1 #248

Merged

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Ai 1343 Add Clients for DataScience, Encryption Services And Refactor #235

Ai 1343 Add Clients for DataScience, Encryption Services And Refactor #235

Uh oh!

mariankrotil commented Aug 15, 2025 •

edited

Loading

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

vita-stejskal commented Aug 19, 2025

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Ai 1343 Add Clients for DataScience, Encryption Services And Refactor #235

Ai 1343 Add Clients for DataScience, Encryption Services And Refactor #235

Uh oh!

Conversation

mariankrotil commented Aug 15, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

vita-stejskal commented Aug 19, 2025

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

mariankrotil commented Aug 15, 2025 •

edited

Loading