alita-sdk 0.3.465__py3-none-any.whl → 0.3.497__py3-none-any.whl

alita_sdk/cli/toolkit.py

@@ -10,6 +10,7 @@ import logging
 from typing import Optional, Dict, Any
 
 from .cli import get_client
+from .toolkit_loader import load_toolkit_config
 
 logger = logging.getLogger(__name__)
 
@@ -140,24 +141,15 @@ def toolkit_schema(ctx, toolkit_name: str):
 @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.
+    """Test a specific tool from a toolkit.
     
     TOOLKIT_TYPE: Type of toolkit (e.g., 'jira', 'github', 'confluence')
     
+    \b
     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
+      alita toolkit test jira --tool get_issue --config jira.json --params params.json
+      alita toolkit test jira --tool get_issue --config jira.json --param issue_key=PROJ-123
+      alita -o json toolkit test github --tool get_issue --config github.json
     """
     formatter = ctx.obj['formatter']
     client = get_client(ctx)
@@ -166,9 +158,15 @@ def toolkit_test(ctx, toolkit_type: str, tool: str, config_file, params,
         # Load toolkit configuration
         toolkit_config = {}
         if config_file:
-            toolkit_config = json.load(config_file)
+            toolkit_config = load_toolkit_config(config_file.name)
             logger.debug(f"Loaded toolkit config from {config_file.name}")
         
+        # Add the tool to selected_tools in the config
+        if 'selected_tools' not in toolkit_config:
+            toolkit_config['selected_tools'] = []
+        if tool not in toolkit_config['selected_tools']:
+            toolkit_config['selected_tools'].append(tool)
+        
         # Load tool parameters
         tool_params = {}
         if params:
@@ -204,7 +202,6 @@ def toolkit_test(ctx, toolkit_type: str, tool: str, config_file, params,
         llm_config = {
             'temperature': temperature,
             'max_tokens': max_tokens,
-            'top_p': 1.0
         }
         
         # Execute test
@@ -254,7 +251,7 @@ def toolkit_tools(ctx, toolkit_type: str, config_file):
         # Load toolkit configuration if provided
         toolkit_config = {}
         if config_file:
-            toolkit_config = json.load(config_file)
+            toolkit_config = load_toolkit_config(config_file.name)
         
         # Import and instantiate toolkit
         from alita_sdk.tools import AVAILABLE_TOOLS

alita_sdk/cli/toolkit_loader.py

@@ -12,6 +12,20 @@ from typing import Dict, Any, List
 from .config import substitute_env_vars
 
 
+# All available tools in the inventory toolkit
+INVENTORY_TOOLS = [
+    "search_graph",
+    "get_entity",
+    "get_entity_content",
+    "impact_analysis",
+    "get_related_entities",
+    "get_stats",
+    "get_citations",
+    "list_entities_by_type",
+    "list_entities_by_layer",
+]
+
+
 def load_toolkit_config(file_path: str) -> Dict[str, Any]:
     """Load toolkit configuration from JSON or YAML file with env var substitution."""
     path = Path(file_path)
@@ -33,7 +47,15 @@ def load_toolkit_config(file_path: str) -> Dict[str, Any]:
 
 
 def load_toolkit_configs(agent_def: Dict[str, Any], toolkit_config_paths: tuple) -> List[Dict[str, Any]]:
-    """Load all toolkit configurations from agent definition and CLI options."""
+    """Load all toolkit configurations from agent definition and CLI options.
+    
+    Args:
+        agent_def: Agent definition dictionary
+        toolkit_config_paths: Tuple of file paths or dict configs from CLI options
+        
+    Returns:
+        List of toolkit configuration dictionaries
+    """
     toolkit_configs = []
     
     # Load from agent definition if present
@@ -45,11 +67,19 @@ def load_toolkit_configs(agent_def: Dict[str, Any], toolkit_config_paths: tuple)
                     toolkit_configs.append(config)
                 elif 'config' in tk_config:
                     toolkit_configs.append(tk_config['config'])
+                elif 'type' in tk_config:
+                    # Direct toolkit config (e.g., {'type': 'inventory', ...})
+                    toolkit_configs.append(tk_config)
     
-    # Load from CLI options
+    # Load from CLI options - can be file paths (str) or dict configs
     if toolkit_config_paths:
-        for config_path in toolkit_config_paths:
-            config = load_toolkit_config(config_path)
-            toolkit_configs.append(config)
+        for config_item in toolkit_config_paths:
+            if isinstance(config_item, dict):
+                # Direct config dict (e.g., from /inventory command)
+                toolkit_configs.append(config_item)
+            elif isinstance(config_item, str):
+                # File path
+                config = load_toolkit_config(config_item)
+                toolkit_configs.append(config)
     
     return toolkit_configs

alita_sdk/cli/tools/__init__.py

@@ -4,7 +4,7 @@ CLI tools package.
 Contains specialized tools for CLI agents.
 """
 
-from .filesystem import get_filesystem_tools
+from .filesystem import get_filesystem_tools, FilesystemApiWrapper
 from .terminal import get_terminal_tools, create_default_blocked_patterns_file
 from .planning import (
     get_planning_tools, 
@@ -14,12 +14,16 @@ from .planning import (
     create_session_memory,
     save_session_metadata,
     load_session_metadata,
+    update_session_metadata,
     get_session_dir,
+    to_portable_path,
+    from_portable_path,
 )
 from .approval import create_approval_wrapper, ApprovalToolWrapper, prompt_approval
 
 __all__ = [
     'get_filesystem_tools',
+    'FilesystemApiWrapper',
     'get_terminal_tools',
     'create_default_blocked_patterns_file',
     'get_planning_tools',
@@ -29,7 +33,10 @@ __all__ = [
     'create_session_memory',
     'save_session_metadata',
     'load_session_metadata',
+    'update_session_metadata',
     'get_session_dir',
+    'to_portable_path',
+    'from_portable_path',
     'create_approval_wrapper',
     'ApprovalToolWrapper',
     'prompt_approval',