alita-sdk 0.3.458__py3-none-any.whl → 0.3.460__py3-none-any.whl

alita_sdk/cli/toolkit.py

@@ -0,0 +1,330 @@
+"""
+Toolkit testing commands for Alita CLI.
+
+Provides commands to list, inspect, and test toolkits directly from the command line.
+"""
+
+import click
+import json
+import logging
+from typing import Optional, Dict, Any
+
+from .cli import get_client
+
+logger = logging.getLogger(__name__)
+
+
+@click.group()
+def toolkit():
+    """Toolkit testing commands."""
+    pass
+
+
+@toolkit.command('list')
+@click.option('--failed', is_flag=True, help='Show failed imports')
+@click.pass_context
+def toolkit_list(ctx, failed: bool):
+    """List all available toolkits."""
+    formatter = ctx.obj['formatter']
+    
+    try:
+        # Import toolkit registry
+        from alita_sdk.tools import AVAILABLE_TOOLS, AVAILABLE_TOOLKITS, FAILED_IMPORTS
+        
+        if failed:
+            # Show failed imports
+            if FAILED_IMPORTS:
+                click.echo("\nFailed toolkit imports:\n")
+                for name, error in FAILED_IMPORTS.items():
+                    click.echo(f"  - {name}: {error}")
+                click.echo(f"\nTotal failed: {len(FAILED_IMPORTS)}")
+            else:
+                click.echo("\nNo failed imports")
+            return
+        
+        # Build toolkit list
+        toolkits = []
+        for name, toolkit_dict in AVAILABLE_TOOLS.items():
+            toolkit_class_name = None
+            if 'toolkit_class' in toolkit_dict:
+                toolkit_class_name = toolkit_dict['toolkit_class'].__name__
+            
+            toolkits.append({
+                'name': name,
+                'class_name': toolkit_class_name,
+                'has_get_tools': 'get_tools' in toolkit_dict
+            })
+        
+        # Format and display
+        output = formatter.format_toolkit_list(toolkits)
+        click.echo(output)
+        
+    except Exception as e:
+        logger.exception("Failed to list toolkits")
+        click.echo(formatter.format_error(str(e)), err=True)
+        raise click.Abort()
+
+
+@toolkit.command('schema')
+@click.argument('toolkit_name')
+@click.pass_context
+def toolkit_schema(ctx, toolkit_name: str):
+    """
+    Show configuration schema for a toolkit.
+    
+    TOOLKIT_NAME: Name of the toolkit (e.g., 'jira', 'github', 'confluence')
+    """
+    formatter = ctx.obj['formatter']
+    
+    try:
+        # Import toolkit registry
+        from alita_sdk.tools import AVAILABLE_TOOLKITS
+        
+        # Find toolkit class
+        toolkit_class = None
+        for name, cls in AVAILABLE_TOOLKITS.items():
+            if name.lower().replace('toolkit', '').replace('alita', '').strip() == toolkit_name.lower():
+                toolkit_class = cls
+                break
+        
+        if not toolkit_class:
+            # Try direct match
+            for name, cls in AVAILABLE_TOOLKITS.items():
+                if toolkit_name.lower() in name.lower():
+                    toolkit_class = cls
+                    break
+        
+        if not toolkit_class:
+            available = [name.lower().replace('toolkit', '').replace('alita', '').strip() 
+                        for name in AVAILABLE_TOOLKITS.keys()]
+            raise click.ClickException(
+                f"Toolkit '{toolkit_name}' not found.\n"
+                f"Available toolkits: {', '.join(sorted(set(available)))}"
+            )
+        
+        # Get schema
+        schema_model = toolkit_class.toolkit_config_schema()
+        schema = schema_model.model_json_schema()
+        
+        # Format and display
+        if formatter.__class__.__name__ == 'JSONFormatter':
+            output = formatter.format_toolkit_schema(toolkit_name, schema)
+        else:
+            output = formatter.format_toolkit_schema(toolkit_name, schema)
+        
+        click.echo(output)
+        
+    except click.ClickException:
+        raise
+    except Exception as e:
+        logger.exception(f"Failed to get schema for toolkit '{toolkit_name}'")
+        click.echo(formatter.format_error(str(e)), err=True)
+        raise click.Abort()
+
+
+@toolkit.command('test')
+@click.argument('toolkit_type')
+@click.option('--tool', required=True, help='Tool name to execute')
+@click.option('--config', 'config_file', type=click.File('r'), 
+              help='Toolkit configuration JSON file')
+@click.option('--params', type=click.File('r'), 
+              help='Tool parameters JSON file')
+@click.option('--param', multiple=True, 
+              help='Tool parameter as key=value (can be used multiple times)')
+@click.option('--llm-model', default='gpt-4o-mini', 
+              help='LLM model to use (default: gpt-4o-mini)')
+@click.option('--temperature', default=0.1, type=float,
+              help='LLM temperature (default: 0.1)')
+@click.option('--max-tokens', default=1000, type=int,
+              help='LLM max tokens (default: 1000)')
+@click.pass_context
+def toolkit_test(ctx, toolkit_type: str, tool: str, config_file, params, 
+                param, llm_model: str, temperature: float, max_tokens: int):
+    """
+    Test a specific tool from a toolkit.
+    
+    TOOLKIT_TYPE: Type of toolkit (e.g., 'jira', 'github', 'confluence')
+    
+    Examples:
+    
+        # Test with config and params from files
+        alita-cli toolkit test jira --tool get_issue \\
+            --config jira-config.json --params params.json
+        
+        # Test with inline parameters
+        alita-cli toolkit test jira --tool get_issue \\
+            --config jira-config.json --param issue_key=PROJ-123
+        
+        # Test with JSON output for scripting
+        alita-cli --output json toolkit test github --tool get_issue \\
+            --config github-config.json --param owner=user --param repo=myrepo
+    """
+    formatter = ctx.obj['formatter']
+    client = get_client(ctx)
+    
+    try:
+        # Load toolkit configuration
+        toolkit_config = {}
+        if config_file:
+            toolkit_config = json.load(config_file)
+            logger.debug(f"Loaded toolkit config from {config_file.name}")
+        
+        # Load tool parameters
+        tool_params = {}
+        if params:
+            tool_params = json.load(params)
+            logger.debug(f"Loaded tool params from {params.name}")
+        
+        # Parse inline parameters
+        if param:
+            for param_pair in param:
+                if '=' not in param_pair:
+                    raise click.ClickException(
+                        f"Invalid parameter format: '{param_pair}'. "
+                        "Use --param key=value"
+                    )
+                key, value = param_pair.split('=', 1)
+                
+                # Try to parse as JSON for complex values
+                try:
+                    tool_params[key] = json.loads(value)
+                except json.JSONDecodeError:
+                    tool_params[key] = value
+            
+            logger.debug(f"Parsed {len(param)} inline parameters")
+        
+        # Prepare full toolkit configuration
+        full_config = {
+            'toolkit_name': toolkit_type,
+            'type': toolkit_type,
+            'settings': toolkit_config
+        }
+        
+        # LLM configuration
+        llm_config = {
+            'temperature': temperature,
+            'max_tokens': max_tokens,
+            'top_p': 1.0
+        }
+        
+        # Execute test
+        logger.info(f"Testing tool '{tool}' from toolkit '{toolkit_type}'")
+        result = client.test_toolkit_tool(
+            toolkit_config=full_config,
+            tool_name=tool,
+            tool_params=tool_params,
+            llm_model=llm_model,
+            llm_config=llm_config
+        )
+        
+        # Format and display result
+        output = formatter.format_toolkit_result(result)
+        click.echo(output)
+        
+        # Exit with error code if test failed
+        if not result.get('success', False):
+            raise click.Abort()
+        
+    except click.ClickException:
+        raise
+    except Exception as e:
+        logger.exception("Failed to execute toolkit test")
+        click.echo(formatter.format_error(str(e)), err=True)
+        raise click.Abort()
+
+
+@toolkit.command('tools')
+@click.argument('toolkit_type')
+@click.option('--config', 'config_file', type=click.File('r'),
+              help='Toolkit configuration JSON file (required for some toolkits)')
+@click.pass_context
+def toolkit_tools(ctx, toolkit_type: str, config_file):
+    """
+    List available tools for a specific toolkit.
+    
+    TOOLKIT_TYPE: Type of toolkit (e.g., 'jira', 'github', 'confluence')
+    
+    Some toolkits require configuration to determine available tools.
+    Use --config to provide configuration if needed.
+    """
+    formatter = ctx.obj['formatter']
+    client = get_client(ctx)
+    
+    try:
+        # Load toolkit configuration if provided
+        toolkit_config = {}
+        if config_file:
+            toolkit_config = json.load(config_file)
+        
+        # Import and instantiate toolkit
+        from alita_sdk.tools import AVAILABLE_TOOLS
+        
+        if toolkit_type not in AVAILABLE_TOOLS:
+            raise click.ClickException(
+                f"Toolkit '{toolkit_type}' not found. "
+                f"Use 'alita-cli toolkit list' to see available toolkits."
+            )
+        
+        toolkit_entry = AVAILABLE_TOOLS[toolkit_type]
+        
+        if 'toolkit_class' not in toolkit_entry:
+            raise click.ClickException(
+                f"Toolkit '{toolkit_type}' does not support tool listing"
+            )
+        
+        # Get toolkit class and instantiate
+        toolkit_class = toolkit_entry['toolkit_class']
+        
+        # Create minimal configuration
+        full_config = {
+            'toolkit_name': toolkit_type,
+            'type': toolkit_type,
+            'settings': toolkit_config
+        }
+        
+        # Try to get available tools via API wrapper
+        try:
+            # Instantiate API wrapper if possible
+            api_wrapper_class = None
+            
+            # Find API wrapper class by inspecting toolkit
+            import inspect
+            for name, obj in inspect.getmembers(toolkit_class):
+                if inspect.isclass(obj) and 'ApiWrapper' in obj.__name__:
+                    api_wrapper_class = obj
+                    break
+            
+            if api_wrapper_class:
+                try:
+                    api_wrapper = api_wrapper_class(**toolkit_config)
+                    available_tools = api_wrapper.get_available_tools()
+                    
+                    # Format tools list
+                    click.echo(f"\nAvailable tools for {toolkit_type}:\n")
+                    for tool_info in available_tools:
+                        tool_name = tool_info.get('name', 'unknown')
+                        description = tool_info.get('description', '')
+                        click.echo(f"  - {tool_name}")
+                        if description:
+                            click.echo(f"      {description}")
+                    
+                    click.echo(f"\nTotal: {len(available_tools)} tools")
+                    return
+                    
+                except Exception as e:
+                    logger.debug(f"Could not instantiate API wrapper: {e}")
+            
+            # Fallback: show general info
+            click.echo(f"\n{toolkit_type} toolkit is available")
+            click.echo("Use 'alita-cli toolkit schema {toolkit_type}' to see configuration options")
+            
+        except Exception as e:
+            logger.exception("Failed to list tools")
+            raise click.ClickException(str(e))
+        
+    except click.ClickException:
+        raise
+    except Exception as e:
+        logger.exception("Failed to list toolkit tools")
+        click.echo(formatter.format_error(str(e)), err=True)
+        raise click.Abort()

alita_sdk/runtime/tools/function.py

@@ -16,6 +16,18 @@ from ..langchain.utils import propagate_the_input_mapping
 logger = logging.getLogger(__name__)
 
 
+def replace_escaped_newlines(data):
+    """
+        Replace \\n with \n in all string values recursively.
+        Required for sanitization of state variables in code node
+    """
+    if isinstance(data, dict):
+        return {key: replace_escaped_newlines(value) for key, value in data.items()}
+    elif isinstance(data, str):
+        return data.replace('\\n', '\n')
+    else:
+        return data
+
 class FunctionTool(BaseTool):
     name: str = 'FunctionalTool'
     description: str = 'This is direct call node for tools'
@@ -30,11 +42,13 @@ class FunctionTool(BaseTool):
     def _prepare_pyodide_input(self, state: Union[str, dict, ToolCall]) -> str:
         """Prepare input for PyodideSandboxTool by injecting state into the code block."""
         # add state into the code block here since it might be changed during the execution of the code
-        state_copy = deepcopy(state)
+        state_copy = replace_escaped_newlines(deepcopy(state))
 
         del state_copy['messages']  # remove messages to avoid issues with pickling without langchain-core
         # inject state into the code block as alita_state variable
-        pyodide_predata = f"#state dict\nalita_state = {state_copy}\n"
+        state_json = json.dumps(state_copy, ensure_ascii=False)
+        pyodide_predata = f'#state dict\nimport json\nalita_state = json.loads({json.dumps(state_json)})\n'
+
         return pyodide_predata
 
     def _handle_pyodide_output(self, tool_result: Any) -> dict:
@@ -94,9 +108,7 @@ class FunctionTool(BaseTool):
         # special handler for PyodideSandboxTool
         if self._is_pyodide_tool():
             code = func_args['code']
-            func_args['code'] = (f"{self._prepare_pyodide_input(state)}\n{code}"
-                                # handle new lines in the code properly
-                                 .replace('\\n','\\\\n'))
+            func_args['code'] = f"{self._prepare_pyodide_input(state)}\n{code}"
         try:
             tool_result = self.tool.invoke(func_args, config, **kwargs)
             dispatch_custom_event(

{alita_sdk-0.3.458.dist-info → alita_sdk-0.3.460.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.458
+Version: 0.3.460
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0
@@ -25,6 +25,10 @@ Requires-Dist: httpcore==1.0.7
 Requires-Dist: urllib3>=2
 Requires-Dist: certifi==2024.8.30
 Requires-Dist: aiohttp>=3.9.0
+Provides-Extra: cli
+Requires-Dist: click>=8.1.0; extra == "cli"
+Requires-Dist: rich>=13.0.0; extra == "cli"
+Requires-Dist: pyyaml>=6.0; extra == "cli"
 Provides-Extra: runtime
 Requires-Dist: streamlit>=1.28.0; extra == "runtime"
 Requires-Dist: langchain_core<0.4.0,>=0.3.76; extra == "runtime"
@@ -138,6 +142,7 @@ Provides-Extra: all
 Requires-Dist: alita-sdk[runtime]; extra == "all"
 Requires-Dist: alita-sdk[tools]; extra == "all"
 Requires-Dist: alita-sdk[community]; extra == "all"
+Requires-Dist: alita-sdk[cli]; extra == "all"
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"

{alita_sdk-0.3.458.dist-info → alita_sdk-0.3.460.dist-info}/RECORD

@@ -1,4 +1,11 @@
 alita_sdk/__init__.py,sha256=fxeNiqiVpIFAJls31Oomifyrtd5gT9iPUTdkWjDOB2Y,656
+alita_sdk/cli/__init__.py,sha256=HiHHouMh0EHmK1qMVzvz6vfIOu8Pm6XXw0dd_hXZ288,278
+alita_sdk/cli/__main__.py,sha256=dAhgWWWEOG1ZiwCs6wQoJwnD62cFRnxiyiAoiFqU8K4,420
+alita_sdk/cli/agents.py,sha256=uMlKiWs85zExE7_Q6NBbw6frnUVK-Ei2KynIddMHXn0,42419
+alita_sdk/cli/cli.py,sha256=k8jwaTKt3At9NfopOcp-BqXitqO0jcei1ACuyoUA9eY,4812
+alita_sdk/cli/config.py,sha256=GiWe7K2RHC_PnH1nEUBodgi0d14oPbIdQhYl69Gbx1o,3978
+alita_sdk/cli/formatting.py,sha256=ip95brhcFcx3Ub7PRLv1aDGniGW13RCGDgc1_8i9ZwM,6566
+alita_sdk/cli/toolkit.py,sha256=IC1zCXiIRUh-swcBC5VhdREIaXGCu5-OfSWyGSUOx0Q,11693
 alita_sdk/community/__init__.py,sha256=8N7wWwPhoyOq3p8wlV3-pb3l3nJCR8TUrtV9iIPLU88,2523
 alita_sdk/community/utils.py,sha256=lvuCJaNqVPHOORJV6kIPcXJcdprVW_TJvERtYAEgpjM,249
 alita_sdk/configurations/__init__.py,sha256=3_MlzyzAi1dog6MFQD_ICOZvaPfXf2UfhSZwlF1knnw,3983
@@ -114,7 +121,7 @@ alita_sdk/runtime/tools/application.py,sha256=RCGe-mRfj8372gTFkEX2xBvcYhw7IKdU1t
 alita_sdk/runtime/tools/artifact.py,sha256=u3szFwZqguHrPZ3tZJ7S_TiZl7cxlT3oHYd6zbdpRDE,13842
 alita_sdk/runtime/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
-alita_sdk/runtime/tools/function.py,sha256=8KTMwut7BnruaVhgO2zFn9YvCU5bT7REQdpPV-SbHh8,6605
+alita_sdk/runtime/tools/function.py,sha256=1kBVMgLNjoE2yQb5rX8O6MR82rDaylJSrj2Hd-JdkN8,7002
 alita_sdk/runtime/tools/graph.py,sha256=7jImBBSEdP5Mjnn2keOiyUwdGDFhEXLUrgUiugO3mgA,3503
 alita_sdk/runtime/tools/image_generation.py,sha256=Kls9D_ke_SK7xmVr7I9SlQcAEBJc86gf66haN0qIj9k,7469
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
@@ -361,8 +368,9 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=Gl1YAFaMMufTNjIpnuo_c_lbDsQkmproYdZz2ZppgJ4,3007
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.458.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.458.dist-info/METADATA,sha256=2rKUTBUSXzFWP9316T-IRjMtsAuwy-bvNtOOlkxOPLg,19101
-alita_sdk-0.3.458.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.458.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.458.dist-info/RECORD,,
+alita_sdk-0.3.460.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.460.dist-info/METADATA,sha256=TEpptFsgT4y0uXNkzXHpOpFagAU_xZ8lCiU6sbqVGbo,19298
+alita_sdk-0.3.460.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.460.dist-info/entry_points.txt,sha256=sSxsIZu9YjVGsgf-5mHC631FeBcT6f2vUjm0A_L9tcU,53
+alita_sdk-0.3.460.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.460.dist-info/RECORD,,

alita_sdk-0.3.460.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+alita-cli = alita_sdk.cli.cli:main