KAB-871 refactor(tools): use KeboolaClient.from_state with context in tools

Author: mariankrotil

src/keboola_mcp_server/component_tools.py

@@ -193,7 +193,6 @@ async def list_component_configurations(
 ) -> List[ComponentConfigurationListItem]:
     client = KeboolaClient.from_state(ctx.session.state)
 
-
     component = await get_component_details(component_id, ctx)
     r_configs = client.storage_client.configurations.list(component_id)
     logger.info(f"Found {len(r_configs)} configurations for component {component_id}.")
@@ -238,8 +237,7 @@ async def get_component_configuration_details(
     """
     if isinstance(config_id, int):
         config_id = str(config_id)
-    client = ctx.session.state["sapi_client"]
-    assert isinstance(client, KeboolaClient)
+    client = KeboolaClient.from_state(ctx.session.state)
 
     component = await get_component_details(component_id, ctx)
     r_config = client.storage_client.configurations.detail(component_id, config_id)

src/keboola_mcp_server/jobs_tools.py

@@ -193,8 +193,7 @@ async def list_jobs(
         ascending order.
     :return: A list of job list items, if empty then no jobs were found.
     """
-    client = ctx.session.state["sapi_client"]
-    assert isinstance(client, KeboolaClient)
+    client = KeboolaClient.from_state(ctx.session.state)
 
     status = handle_status_param(status=status)
 
@@ -218,8 +217,7 @@ async def get_job_details(
     :param job_id: The unique identifier of the job whose details should be retrieved.
     :return: A job detail object
     """
-    client = ctx.session.state["sapi_client"]
-    assert isinstance(client, KeboolaClient)
+    client = KeboolaClient.from_state(ctx.session.state)
 
     r_job = client.jobs_queue.detail(job_id)
     logger.info(f"Found job details for {job_id}." if r_job else f"Job {job_id} not found.")
@@ -274,8 +272,7 @@ async def list_component_config_jobs(
         ascending order.
     :return: A list of job list items.
     """
-    client = ctx.session.state["sapi_client"]
-    assert isinstance(client, KeboolaClient)
+    client = KeboolaClient.from_state(ctx.session.state)
 
     status = handle_status_param(status=status)
     params = {
@@ -336,8 +333,7 @@ async def list_component_jobs(
         order.
     :return: A list of job list items.
     """
-    client = ctx.session.state["sapi_client"]
-    assert isinstance(client, KeboolaClient)
+    client = KeboolaClient.from_state(ctx.session.state)
 
     status = handle_status_param(status=status)
     params = {