alita-sdk 0.3.532__py3-none-any.whl → 0.3.602__py3-none-any.whl

alita_sdk/runtime/tools/tool.py

@@ -86,7 +86,9 @@ Answer must be JSON only extractable by JSON.LOADS."""
         else:
             input_[-1].content += self.unstructured_output
             completion = self.client.invoke(input_, config=config)
-            result = _extract_json(completion.content.strip())
+            from ..langchain.utils import extract_text_from_completion
+            content_text = extract_text_from_completion(completion)
+            result = _extract_json(content_text.strip())
             logger.info(f"ToolNode tool params: {result}")
         try:
             # handler for application added as a tool

alita_sdk/runtime/tools/vectorstore.py

@@ -12,9 +12,11 @@ from alita_sdk.tools.vector_adapters.VectorStoreAdapter import VectorStoreAdapte
 from logging import getLogger
 
 from ..utils.logging import dispatch_custom_event
+from ..langchain.utils import extract_text_from_completion
 
 logger = getLogger(__name__)
 
+
 class IndexDocumentsModel(BaseModel):
     documents: Any = Field(description="Generator of documents to index")
 
@@ -684,8 +686,10 @@ class VectorStoreWrapper(BaseToolApiWrapper):
                 ]
             )
         ])
+        # Extract text content safely (handles both string and list content from thinking models)
+        search_query = extract_text_from_completion(result)
         search_results = self.search_documents(
-            result.content, doctype, filter, cut_off, search_top, 
+            search_query, doctype, filter, cut_off, search_top, 
             full_text_search=full_text_search,
             reranking_config=reranking_config,
             extended_search=extended_search
@@ -714,7 +718,8 @@ class VectorStoreWrapper(BaseToolApiWrapper):
                 ]
             )
         ])
-        return result.content
+        # Extract text content safely (handles both string and list content from thinking models)
+        return extract_text_from_completion(result)
 
     def _log_data(self, message: str, tool_name: str = "index_data"):
         """Log data and dispatch custom event for indexing progress"""

alita_sdk/runtime/tools/vectorstore_base.py

@@ -13,9 +13,11 @@ from pydantic import BaseModel, model_validator, Field
 from alita_sdk.tools.elitea_base import BaseToolApiWrapper
 from alita_sdk.tools.vector_adapters.VectorStoreAdapter import VectorStoreAdapterFactory
 from ...runtime.utils.utils import IndexerKeywords
+from ...runtime.langchain.utils import extract_text_from_completion
 
 logger = getLogger(__name__)
 
+
 class IndexDocumentsModel(BaseModel):
     documents: Any = Field(description="Generator of documents to index")
 
@@ -625,8 +627,10 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
                 ]
             )
         ])
+        # Extract text content safely (handles both string and list content from thinking models)
+        search_query = extract_text_from_completion(result)
         search_results = self.search_documents(
-            result.content, doctype, filter, cut_off, search_top, 
+            search_query, doctype, filter, cut_off, search_top, 
             full_text_search=full_text_search,
             reranking_config=reranking_config,
             extended_search=extended_search
@@ -655,7 +659,8 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
                 ]
             )
         ])
-        return result.content
+        # Extract text content safely (handles both string and list content from thinking models)
+        return extract_text_from_completion(result)
 
     def get_available_tools(self):
         return [

alita_sdk/runtime/utils/constants.py

@@ -436,4 +436,8 @@ STYLES = r"""
             filter: hue-rotate(180deg) brightness(1.2);
         }
         </style>
-        """
+        """
+
+TOOLKIT_NAME_META = "toolkit_name"
+TOOL_NAME_META = "tool_name"
+TOOLKIT_TYPE_META = "toolkit_type"

alita_sdk/runtime/utils/mcp_client.py

@@ -192,7 +192,7 @@ class McpClient:
                     "sampling": {}
                 },
                 "clientInfo": {
-                    "name": "Alita MCP Client",
+                    "name": "ELITEA MCP Client",
                     "version": "1.0.0"
                 }
             }

alita_sdk/runtime/utils/mcp_sse_client.py

@@ -374,7 +374,7 @@ class McpSseClient:
                     "sampling": {}
                 },
                 "clientInfo": {
-                    "name": "Alita MCP Client",
+                    "name": "ELITEA MCP Client",
                     "version": "1.0.0"
                 }
             }

alita_sdk/runtime/utils/toolkit_utils.py

@@ -59,10 +59,12 @@ def instantiate_toolkit_with_client(toolkit_config: Dict[str, Any],
 
         # Create a tool configuration dict with required fields
         # Note: MCP toolkit always requires toolkit_name, other toolkits respect use_prefix flag
+        # Note: 'name' is always set for provider-based toolkits (used by provider_worker.utils.tools)
         tool_config = {
             'id': toolkit_config.get('id', random.randint(1, 1000000)),
             'type': toolkit_config.get('type', toolkit_type),
             'settings': settings,
+            'name': toolkit_name,  # Always pass name for provider toolkits
             'toolkit_name': toolkit_name if (use_prefix or toolkit_type == 'mcp') else None
         }
         

alita_sdk/tools/__init__.py

@@ -49,6 +49,9 @@ def _safe_import_tool(tool_name, module_path, get_tools_name=None, toolkit_class
         if hasattr(module, 'get_toolkit'):
             imported['get_toolkit'] = getattr(module, 'get_toolkit')
 
+        if hasattr(module, 'get_toolkit_available_tools'):
+            imported['get_toolkit_available_tools'] = getattr(module, 'get_toolkit_available_tools')
+
         if toolkit_class_name and hasattr(module, toolkit_class_name):
             imported['toolkit_class'] = getattr(module, toolkit_class_name)
             AVAILABLE_TOOLKITS[toolkit_class_name] = getattr(module, toolkit_class_name)
@@ -64,7 +67,7 @@ def _safe_import_tool(tool_name, module_path, get_tools_name=None, toolkit_class
 
 # Safe imports for all tools
 _safe_import_tool('github', 'github', 'get_tools', 'AlitaGitHubToolkit')
-_safe_import_tool('openapi', 'openapi', 'get_tools')
+_safe_import_tool('openapi', 'openapi', 'get_tools', 'AlitaOpenAPIToolkit')
 _safe_import_tool('jira', 'jira', 'get_tools', 'JiraToolkit')
 _safe_import_tool('confluence', 'confluence', 'get_tools', 'ConfluenceToolkit')
 _safe_import_tool('service_now', 'servicenow', 'get_tools', 'ServiceNowToolkit')
@@ -99,7 +102,7 @@ _safe_import_tool('k8s', 'cloud.k8s', None, 'KubernetesToolkit')
 _safe_import_tool('elastic', 'elastic', None, 'ElasticToolkit')
 _safe_import_tool('keycloak', 'keycloak', None, 'KeycloakToolkit')
 _safe_import_tool('localgit', 'localgit', None, 'AlitaLocalGitToolkit')
-_safe_import_tool('pandas', 'pandas', 'get_tools', 'PandasToolkit')
+# pandas toolkit removed - use Data Analysis internal tool instead
 _safe_import_tool('azure_search', 'azure_ai.search', 'get_tools', 'AzureSearchToolkit')
 _safe_import_tool('figma', 'figma', 'get_tools', 'FigmaToolkit')
 _safe_import_tool('salesforce', 'salesforce', 'get_tools', 'SalesforceToolkit')
@@ -183,7 +186,9 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
                 toolkit = tkitclass.get_toolkit(**get_toolkit_params)
                 toolkit_tools.extend(toolkit.get_tools())
             except Exception as e:
+                import traceback
                 logger.error(f"Error in getting custom toolkit: {e}")
+                logger.error(f"Traceback:\n{traceback.format_exc()}")
         else:
             if tool_type in FAILED_IMPORTS:
                 logger.warning(f"Tool '{tool_type}' is not available: {FAILED_IMPORTS[tool_type]}")
@@ -240,6 +245,42 @@ def get_available_toolkit_models():
     """Return dict with available toolkit classes."""
     return deepcopy(AVAILABLE_TOOLS)
 
+
+def get_toolkit_available_tools(toolkit_type: str, settings: dict) -> dict:
+    """Return dynamic available tools + per-tool JSON schemas for a toolkit instance.
+
+    This is the single SDK entrypoint used by backend services (e.g. indexer_worker)
+    when the UI needs spec/instance-dependent tool enumeration. Toolkits that don't
+    support dynamic enumeration should return an empty payload.
+
+    Args:
+        toolkit_type: toolkit type string (e.g. 'openapi')
+        settings: persisted toolkit settings
+
+    Returns:
+        {
+          "tools": [{"name": str, "description": str}],
+          "args_schemas": {"tool_name": <json schema dict>}
+        }
+    """
+    toolkit_type = (toolkit_type or '').strip().lower()
+    if not isinstance(settings, dict):
+        settings = {}
+
+    tool_module = AVAILABLE_TOOLS.get(toolkit_type) or {}
+    enumerator = tool_module.get('get_toolkit_available_tools')
+    if not callable(enumerator):
+        return {"tools": [], "args_schemas": {}}
+
+    try:
+        result = enumerator(settings)
+        if not isinstance(result, dict):
+            return {"tools": [], "args_schemas": {}, "error": "Invalid response from toolkit enumerator"}
+        return result
+    except Exception as e:  # pylint: disable=W0718
+        logger.exception("Failed to compute available tools for toolkit_type=%s", toolkit_type)
+        return {"tools": [], "args_schemas": {}, "error": str(e)}
+
 def diagnose_imports():
     """Print diagnostic information about tool imports."""
     available_count = len(AVAILABLE_TOOLS)
@@ -276,6 +317,7 @@ def diagnose_imports():
 __all__ = [
     'get_tools',
     'get_toolkits',
+    'get_toolkit_available_tools',
     'get_available_tools',
     'get_failed_imports',
     'get_available_toolkits',

alita_sdk/tools/ado/repos/__init__.py

@@ -11,6 +11,7 @@ from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
 from .repos_wrapper import ReposApiWrapper
 from ...utils import clean_string, get_max_toolkit_length, check_connection_response
+from ....runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "ado_repos"
 
@@ -52,13 +53,30 @@ class AzureDevOpsReposToolkit(BaseToolkit):
             embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
 
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__={'json_schema_extra': {'metadata':
-                {
-                    "label": "ADO repos",
-                    "icon_url": "ado-repos-icon.svg",
-                    "categories": ["code repositories"],
-                    "extra_categories": ["code", "repository", "version control"],
-                }}}
+            __config__={
+                'json_schema_extra': {
+                    'metadata': {
+                        "label": "ADO repos",
+                        "icon_url": "ado-repos-icon.svg",
+                        "categories": ["code repositories"],
+                        "extra_categories": ["code", "repository", "version control"],
+                        "sections": {
+                            "auth": {
+                                "required": True,
+                                "subsections": [
+                                    {
+                                        "name": "Token",
+                                        "fields": ["token"]
+                                    }
+                                ]
+                            }
+                        },
+                        "configuration_group": {
+                            "name": "ado",
+                        }
+                    }
+                }
+            }
         )
 
         @check_connection_response
@@ -107,7 +125,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
                     name=tool["name"],
                     description=description,
                     args_schema=tool["args_schema"],
-                    metadata={"toolkit_name": toolkit_name} if toolkit_name else {}
+                    metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
                 )
             )
         return cls(tools=tools)

alita_sdk/tools/ado/repos/repos_wrapper.py

@@ -23,7 +23,8 @@ from langchain_core.tools import ToolException
 from msrest.authentication import BasicAuthentication
 from pydantic import Field, PrivateAttr, create_model, model_validator, SecretStr
 
-from ..utils import extract_old_new_pairs, generate_diff, get_content_from_generator
+from ...elitea_base import BaseCodeToolApiWrapper
+from ..utils import generate_diff, get_content_from_generator
 from ...code_indexer_toolkit import CodeIndexerToolkit
 from ...utils.available_tools_decorator import extend_with_parent_available_tools
 
@@ -129,9 +130,30 @@ class ArgsSchema(Enum):
     )
     UpdateFile = create_model(
         "UpdateFile",
-        branch_name=(str, Field(description="The name of the branch, e.g. `my_branch`.")),
-        file_path=(str, Field(description="Path of a file to be updated.")),
-        update_query=(str, Field(description="Update query used to adjust target file.")),
+        branch_name=(
+            str,
+            Field(description="The name of the branch, e.g. `my_branch`.")
+        ),
+        file_path=(
+            str,
+            Field(description="Path of a file to be updated."),
+        ),
+        update_query=(
+            str,
+            Field(
+                description=(
+                    "The exact OLD text to be replaced and the NEW "
+                    "text to insert, using one or more block pairs like:"
+                    "OLD <<<<Hello Earth!>>>> OLD NEW <<<<Hello Mars!>>>> NEW"
+                    "Each OLD block must contain the exact text that will be replaced "
+                    "via string replacement (exact full match including non-written characters). Each corresponding NEW "
+                    "block contains the replacement text. For multi-line changes, it is "
+                    "preferred to provide several smaller OLD/NEW pairs rather than one "
+                    "large block, so that each OLD block closely matches a contiguous "
+                    "snippet from the file."
+                )
+            ),
+        ),
     )
     DeleteFile = create_model(
         "DeleteFile",
@@ -252,6 +274,9 @@ class ReposApiWrapper(CodeIndexerToolkit):
     token: Optional[SecretStr]
     _client: Optional[GitClient] = PrivateAttr()
 
+    # Reuse common file helpers from BaseCodeToolApiWrapper
+    edit_file = BaseCodeToolApiWrapper.edit_file
+
     class Config:
         arbitrary_types_allowed = True
 
@@ -835,22 +860,50 @@ class ReposApiWrapper(CodeIndexerToolkit):
             logger.error(msg)
             return ToolException(msg)
 
-    def update_file(self, branch_name: str, file_path: str, update_query: str) -> str:
+    def _write_file(self, file_path: str, content: str, branch: str = None, commit_message: str = None) -> str:
+        """Write content to a file in Azure DevOps by creating an edit commit.
+
+        This implementation follows the previous `update_file` behavior: it always
+        performs an 'edit' change (does not create the file), gets the latest
+        commit id for the branch and pushes a new commit containing the change.
         """
-        Updates a file with new content in Azure DevOps.
+        try:
+            # Get the latest commit ID of the target branch
+            branch_obj = self._client.get_branch(
+                repository_id=self.repository_id,
+                project=self.project,
+                name=branch,
+            )
+            if branch_obj is None or not hasattr(branch_obj, 'commit') or not hasattr(branch_obj.commit, 'commit_id'):
+                raise ToolException(f"Branch `{branch}` does not exist or has no commits.")
+
+            latest_commit_id = branch_obj.commit.commit_id
+
+            # Build edit change and push
+            change = GitChange("edit", file_path, content).to_dict()
+
+            ref_update = GitRefUpdate(name=f"refs/heads/{branch}", old_object_id=latest_commit_id)
+            new_commit = GitCommit(comment=commit_message or ("Update " + file_path), changes=[change])
+            push = GitPush(commits=[new_commit], ref_updates=[ref_update])
+
+            self._client.create_push(push=push, repository_id=self.repository_id, project=self.project)
+            return f"Updated file {file_path}"
+        except ToolException:
+            # Re-raise known tool exceptions
+            raise
+        except Exception as e:
+            logger.error(f"Unable to write file {file_path}: {e}")
+            raise ToolException(f"Unable to write file {file_path}: {str(e)}")
+
+    def update_file(self, branch_name: str, file_path: str, update_query: str) -> str:
+        """Updates a file with new content in Azure DevOps using OLD/NEW markers.
+
         Parameters:
             branch_name (str): The name of the branch where update the file.
             file_path (str): Path to the file for update.
-            update_query(str): Contains the file contents requried to be updated.
-                The old file contents is wrapped in OLD <<<< and >>>> OLD
-                The new file contents is wrapped in NEW <<<< and >>>> NEW
-                For example:
-                OLD <<<<
-                Hello Earth!
-                >>>> OLD
-                NEW <<<<
-                Hello Mars!
-                >>>> NEW
+            update_query(str): Contains the file contents required to be updated,
+                wrapped in Git-style OLD/NEW blocks.
+
         Returns:
             A success or failure message
         """
@@ -863,43 +916,16 @@ class ReposApiWrapper(CodeIndexerToolkit):
                 "Please create a new branch and try again."
             )
         try:
-            file_content = self._read_file(file_path, branch_name)
-
-            if isinstance(file_content, ToolException):
-                return file_content
-
-            updated_file_content = file_content
-            for old, new in extract_old_new_pairs(update_query):
-                if not old.strip():
-                    continue
-                updated_file_content = updated_file_content.replace(old, new)
-
-            if file_content == updated_file_content:
-                return (
-                    "File content was not updated because old content was not found or empty. "
-                    "It may be helpful to use the read_file action to get "
-                    "the current file contents."
-                )
-
-            # Get the latest commit ID of the active branch to use as oldObjectId
-            branch = self._client.get_branch(
-                repository_id=self.repository_id,
-                project=self.project,
-                name=self.active_branch,
-            )
-            latest_commit_id = branch.commit.commit_id
-
-            change = GitChange("edit", file_path, updated_file_content).to_dict()
-
-            ref_update = GitRefUpdate(
-                name=f"refs/heads/{self.active_branch}", old_object_id=latest_commit_id
-            )
-            new_commit = GitCommit(comment=f"Update {file_path}", changes=[change])
-            push = GitPush(commits=[new_commit], ref_updates=[ref_update])
-            self._client.create_push(
-                push=push, repository_id=self.repository_id, project=self.project
+            # Let edit_file handle parsing and content updates; this will call _read_file and _write_file.
+            # For ADO, branch_name is used as branch; commit message is derived from file_path.
+            return self.edit_file(
+                file_path=file_path,
+                file_query=update_query,
+                branch=self.active_branch,
+                commit_message=f"Update {file_path}",
             )
-            return "Updated file " + file_path
+        except ToolException as e:
+            return str(e)
         except Exception as e:
             msg = f"Unable to update file due to error:\n{str(e)}"
             logger.error(msg)

alita_sdk/tools/ado/test_plan/__init__.py

@@ -11,6 +11,7 @@ from ....configurations.pgvector import PgVectorConfiguration
 from .test_plan_wrapper import TestPlanApiWrapper
 from ...base.tool import BaseAction
 from ...utils import clean_string, get_max_toolkit_length, check_connection_response
+from ....runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 
 name = "azure_devops_plans"
@@ -40,7 +41,7 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
         m = create_model(
             name_alias,
             ado_configuration=(AdoConfiguration, Field(description="Ado configuration", json_schema_extra={'configuration_types': ['ado']})),
-            limit=(Optional[int], Field(description="ADO plans limit used for limitation of the list with results", default=5)),
+            limit=(Optional[int], Field(description="ADO plans limit used for limitation of the list with results", default=5, gt=0)),
             # indexer settings
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
@@ -122,7 +123,7 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
                 name=tool["name"],
                 description=description,
                 args_schema=tool["args_schema"],
-                metadata={"toolkit_name": toolkit_name} if toolkit_name else {}
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -180,7 +180,29 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
             connection = Connection(base_url=values['organization_url'], creds=credentials)
             cls._client = connection.clients.get_test_plan_client()
         except Exception as e:
-            raise ImportError(f"Failed to connect to Azure DevOps: {e}")
+            error_msg = str(e).lower()
+            if "expired" in error_msg or "token" in error_msg and ("invalid" in error_msg or "unauthorized" in error_msg):
+                raise ValueError(
+                    "Azure DevOps connection failed: Your access token has expired or is invalid. "
+                    "Please refresh your token in the toolkit configuration."
+                )
+            elif "401" in error_msg or "unauthorized" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Authentication failed. "
+                    "Please check your credentials in the toolkit configuration."
+                )
+            elif "404" in error_msg or "not found" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Organization or project not found. "
+                    "Please verify your organization URL and project name."
+                )
+            elif "timeout" in error_msg or "timed out" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Connection timed out. "
+                    "Please check your network connection and try again."
+                )
+            else:
+                raise ValueError(f"Azure DevOps connection failed: {e}")
         return super().validate_toolkit(values)
 
     def create_test_plan(self, test_plan_create_params: str):

alita_sdk/tools/ado/utils.py

@@ -1,24 +1,6 @@
-import re
 import difflib
 
 
-def extract_old_new_pairs(file_query: str):
-    """
-    Extracts old and new content pairs from a file query.
-    Parameters:
-        file_query (str): The file query containing old and new content.
-    Returns:
-        list of tuples: A list where each tuple contains (old_content, new_content).
-    """
-    old_pattern = re.compile(r"OLD <<<<\s*(.*?)\s*>>>> OLD", re.DOTALL)
-    new_pattern = re.compile(r"NEW <<<<\s*(.*?)\s*>>>> NEW", re.DOTALL)
-
-    old_contents = old_pattern.findall(file_query)
-    new_contents = new_pattern.findall(file_query)
-
-    return list(zip(old_contents, new_contents))
-
-
 def generate_diff(base_text, target_text, file_path):
     base_lines = base_text.splitlines(keepends=True)
     target_lines = target_text.splitlines(keepends=True)
@@ -28,6 +10,7 @@ def generate_diff(base_text, target_text, file_path):
 
     return "".join(diff)
 
+
 def get_content_from_generator(content_generator):
     def safe_decode(chunk):
         try:

alita_sdk/tools/ado/wiki/__init__.py

@@ -10,6 +10,7 @@ from ....configurations.ado import AdoConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
 from ...utils import clean_string, get_max_toolkit_length, check_connection_response
+from ....runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "azure_devops_wiki"
 name_alias = 'ado_wiki'
@@ -116,7 +117,7 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
                 name=tool["name"],
                 description=description,
                 args_schema=tool["args_schema"],
-                metadata={"toolkit_name": toolkit_name} if toolkit_name else {}
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/ado/wiki/ado_wrapper.py

@@ -105,7 +105,29 @@ class AzureDevOpsApiWrapper(NonCodeIndexerToolkit):
             cls._core_client = connection.clients.get_core_client()
 
         except Exception as e:
-            return ImportError(f"Failed to connect to Azure DevOps: {e}")
+            error_msg = str(e).lower()
+            if "expired" in error_msg or "token" in error_msg and ("invalid" in error_msg or "unauthorized" in error_msg):
+                raise ValueError(
+                    "Azure DevOps connection failed: Your access token has expired or is invalid. "
+                    "Please refresh your token in the toolkit configuration."
+                )
+            elif "401" in error_msg or "unauthorized" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Authentication failed. "
+                    "Please check your credentials in the toolkit configuration."
+                )
+            elif "404" in error_msg or "not found" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Organization or project not found. "
+                    "Please verify your organization URL and project name."
+                )
+            elif "timeout" in error_msg or "timed out" in error_msg:
+                raise ValueError(
+                    "Azure DevOps connection failed: Connection timed out. "
+                    "Please check your network connection and try again."
+                )
+            else:
+                raise ValueError(f"Azure DevOps connection failed: {e}")
 
         return super().validate_toolkit(values)
 

alita_sdk/tools/ado/work_item/__init__.py

@@ -10,6 +10,7 @@ from ....configurations.ado import AdoConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
 from ...utils import clean_string, get_max_toolkit_length, check_connection_response
+from ....runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "ado_boards"
 
@@ -37,7 +38,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
         m = create_model(
             name,
             ado_configuration=(AdoConfiguration, Field(description="Ado Work Item configuration", json_schema_extra={'configuration_types': ['ado']})),
-            limit=(Optional[int], Field(description="ADO plans limit used for limitation of the list with results", default=5)),
+            limit=(Optional[int], Field(description="Default ADO boards result limit (can be overridden by agent instructions)", default=5, gt=0)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             # indexer settings
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default = None,
@@ -117,7 +118,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
                 name=tool["name"],
                 description=description,
                 args_schema=tool["args_schema"],
-                metadata={"toolkit_name": toolkit_name} if toolkit_name else {}
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)