alita-sdk 0.3.145__py3-none-any.whl → 0.3.147__py3-none-any.whl

alita_sdk/runtime/tools/function.py

@@ -7,8 +7,6 @@ from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
 from typing import Any, Optional, Union, Annotated
 from langchain_core.utils.function_calling import convert_to_openai_tool
-from langgraph.prebuilt import InjectedStore
-from langgraph.store.base import BaseStore
 from pydantic import ValidationError
 from ..langchain.utils import propagate_the_input_mapping
 

alita_sdk/runtime/tools/indexer_tool.py

@@ -4,7 +4,7 @@ from traceback import format_exc
 
 from langchain_core.callbacks import dispatch_custom_event
 from langchain_core.runnables import RunnableConfig
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, ToolException
 from typing import Any, Optional, Union
 from langchain_core.messages import ToolCall
 from langchain_core.utils.function_calling import convert_to_openai_tool
@@ -72,6 +72,8 @@ class IndexerNode(BaseTool):
                     logger.error(f"Chunking error: {format_exc()}")
                     return {"messages": [{"role": "assistant", "content": f"""Chunking tool {self.chunking_tool} raised an error.\n\nError: {format_exc()}"""}]}
             index_results = self.index_tool.invoke({"documents": chunks if chunks else result}, config=config, kwargs=kwargs)
+            if isinstance(index_results, ToolException):
+                raise index_results
             logger.info(f"IndexNode response: {index_results}")
             total_time = round((time() - start_time), 2)
             index_results['total_time'] = total_time
@@ -84,7 +86,7 @@ class IndexerNode(BaseTool):
             return {
                 "messages": [{"role": "assistant", "content": f"""Tool input to the {self.tool.name} with value {result} raised ValidationError. 
         \n\nTool schema is {dumps(params)} \n\nand the input to LLM was 
-        {func_args}"""}]}
+        {func_args}\n{e}"""}]}
 
     def _run(self, *args, **kwargs):
         return self.invoke(**kwargs)

alita_sdk/tools/ado/repos/__init__.py

@@ -72,7 +72,9 @@ class AzureDevOpsReposToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["code repositories"],
+                    "extra_categories": ["code", "repository", "version control"]
                 }}}
         )
 

alita_sdk/tools/ado/test_plan/__init__.py

@@ -31,6 +31,8 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
                 {
                     "label": "ADO plans",
                     "icon_url": "ado-plans.svg",
+                    "categories": ["test management"],
+                    "extra_categories": ["test case management", "qa"],
                     "sections": {
                         "auth": {
                             "required": True,

alita_sdk/tools/ado/wiki/__init__.py

@@ -31,22 +31,23 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={
                 'json_schema_extra': {
-                    'metadata':
-                        {
-                            "label": "ADO wiki",
-                            "icon_url": None,
-                            "sections": {
-                                "auth": {
-                                    "required": True,
-                                    "subsections": [
-                                        {
-                                            "name": "Token",
-                                            "fields": ["token"]
-                                        }
-                                    ]
-                                }
+                    'metadata': {
+                        "label": "ADO wiki",
+                        "icon_url": None,
+                        "categories": ["documentation"],
+                        "extra_categories": ["knowledge base", "documentation management", "wiki"],
+                        "sections": {
+                            "auth": {
+                                "required": True,
+                                "subsections": [
+                                    {
+                                        "name": "Token",
+                                        "fields": ["token"]
+                                    }
+                                ]
                             }
                         }
+                    }
                 }
             }
         )
@@ -75,4 +76,5 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self):
-        return self.tools
+        return self.tools
+

alita_sdk/tools/ado/work_item/__init__.py

@@ -34,6 +34,8 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
                     'metadata': {
                         "label": "ADO boards",
                         "icon_url": None,
+                        "categories": ["project management"],
+                        "extra_categories": ["work item management", "issue tracking", "agile boards"],
                         "sections": {
                             "auth": {
                                 "required": True,
@@ -75,4 +77,5 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self):
-        return self.tools
+        return self.tools
+

alita_sdk/tools/advanced_jira_mining/__init__.py

@@ -46,7 +46,16 @@ class AdvancedJiraMiningToolkit(BaseToolkit):
             is_jira_cloud=(bool, Field(default=True, title="Cloud", description="JIRA Cloud")),
             verify_ssl=(bool, Field(default=True, title="Verify SSL", description="Verify SSL")),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__={'json_schema_extra': {'metadata': {"label": "Advanced JIRA mining", "icon_url": "jira-icon.svg", "hidden": True}}}
+            __config__={
+                'json_schema_extra':
+                    {'metadata':
+                         {
+                             "label": "Advanced JIRA mining", "icon_url": "jira-icon.svg", "hidden": True,
+                             "categories": ["project management"],
+                             "extra_categories": ["jira", "confluence", "issue tracking", "agile management"],
+                         }
+                    }
+            }
         )
 
     @classmethod

alita_sdk/tools/azure_ai/search/__init__.py

@@ -45,7 +45,13 @@ class AzureSearchToolkit(BaseToolkit):
             openai_api_key=(Optional[str], Field(description="Azure OpenAI API Key", default=None, json_schema_extra={'secret': True})),
             model_name=(str, Field(description="Model name for Embeddings model", default=None)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Azure Search", "icon_url": None, "hidden": True}})
+            __config__=ConfigDict(json_schema_extra={
+                'metadata': {
+                    "label": "Azure Search", "icon_url": None, "hidden": True,
+                    "categories": ["other"],
+                    "extra_categories": ["azure cognitive search", "vector database", "knowledge base"]
+                }
+            })
         )
 
     @classmethod

alita_sdk/tools/bitbucket/__init__.py

@@ -45,7 +45,15 @@ class AlitaBitbucketToolkit(BaseToolkit):
             password=(SecretStr, Field(description="GitLab private token", json_schema_extra={'secret': True})),
             cloud=(Optional[bool], Field(description="Hosting Option", default=None)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Bitbucket", "icon_url": "bitbucket-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra=
+            {
+                'metadata':
+                    {
+                        "label": "Bitbucket", "icon_url": "bitbucket-icon.svg",
+                        "categories": ["code repositories"],
+                        "extra_categories": ["bitbucket", "git", "repository", "code", "version control"],
+                    }
+            })
         )
 
     @classmethod

alita_sdk/tools/browser/__init__.py

@@ -50,7 +50,7 @@ class BrowserToolkit(BaseToolkit):
 
         return create_model(
             name,
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Browser", "icon_url": None}}),
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Browser", "icon_url": None, "categories": ["testing"], "extra_categories": ["web scraping", "search", "crawler"]}}),
             google_cse_id=(Optional[str], Field(description="Google CSE id", default=None)),
             google_api_key=(Optional[SecretStr],
                             Field(description="Google API key", default=None, json_schema_extra={'secret': True})),

alita_sdk/tools/carrier/__init__.py

@@ -42,10 +42,8 @@ class AlitaCarrierToolkit(BaseToolkit):
                     "label": "Carrier",
                     "version": "2.0.1",
                     "icon_url": "carrier.svg",
-                    "capabilities": {
-                        "total_tools": len(selected_tools),
-                        "tool_categories": ["Ticket Management", "Reporting", "Audit Logs"]
-                    }
+                    "categories": ["testing"],
+                    "extra_categories": ["carrier", "ticket management", "log management"],
                 }
             })
         )

alita_sdk/tools/cloud/aws/__init__.py

@@ -33,7 +33,16 @@ class AWSToolkit(BaseToolkit):
             access_key_id=(Optional[str], Field(default=None, title="Access Key ID", description="AWS access key ID")),
             secret_access_key=(Optional[SecretStr], Field(default=None, title="Secret Access Key", description="AWS secret access key", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Cloud AWS", "icon_url": None, "hidden": True}})
+            __config__=ConfigDict(json_schema_extra=
+            {
+                'metadata':
+                    {
+                        "label": "Cloud AWS", "icon_url": None, "hidden": True,
+                        "categories": ["cloud"],
+                        "extra_categories": ["aws", "amazon web services", "cloud computing",
+                                             "infrastructure as a service"]
+                    }
+            })
         )
 
     @classmethod

alita_sdk/tools/code/linter/__init__.py

@@ -27,6 +27,9 @@ class PythonLinterToolkit(BaseToolkit):
         return create_model(
             name,
             error_codes=(str, Field(description="Error codes to be used by the linter")),
+            __config__={'json_schema_extra': {'metadata': {"label": "Python Linter", "icon_url": None, "hidden": True,
+                                                           "categories": ["development"],
+                                                           "extra_categories": ["code linter", "python linter"]}}}
         )
 
     @classmethod
@@ -50,4 +53,5 @@ class PythonLinterToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/code/sonar/__init__.py

@@ -32,7 +32,15 @@ class SonarToolkit(BaseToolkit):
             sonar_token=(SecretStr, Field(description="SonarQube user token for authentication", json_schema_extra={'secret': True})),
             sonar_project_name=(str, Field(description="Project name of the desired repository")),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Sonar", "icon_url": "sonar-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra=
+                                  {
+                                      'metadata':
+                                          {
+                                              "label": "Sonar", "icon_url": "sonar-icon.svg",
+                                              "categories": ["development"],
+                                              "extra_categories": ["code quality", "code security", "code coverage", "quality", "sonarqube"]
+                                          }
+                                  })
         )
 
     @classmethod
@@ -55,4 +63,5 @@ class SonarToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/confluence/__init__.py

@@ -86,7 +86,9 @@ class ConfluenceToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                         "categories": ["documentation"],
+                    "extra_categories": ["confluence", "wiki", "knowledge base", "documentation", "atlassian"]
                 }
             })
         )

alita_sdk/tools/custom_open_api/__init__.py

@@ -30,7 +30,10 @@ class OpenApiToolkit(BaseToolkit):
             spec=(str, Field(default="", title="Specification", description="OpenAPI specification")),
             api_key=(str, Field(default="", title="API key", description="API key", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "OpenAPI", "icon_url": None}})
+            __config__=ConfigDict(json_schema_extra={'metadata':
+                                                         {"label": "OpenAPI", "icon_url": None,
+                                                          "categories": ["testing"],
+                                                          "extra_categories": [ "openapi", "swagger"]}})
         )
 
     @classmethod

alita_sdk/tools/elastic/__init__.py

@@ -61,4 +61,5 @@ class ElasticToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/figma/__init__.py

@@ -65,7 +65,9 @@ class FigmaToolkit(BaseToolkit):
                                      }
                                  ]
                              }
-                         }
+                         },
+                         "categories": ["other"],
+                         "extra_categories": ["figma", "design", "ui/ux", "prototyping", "collaboration"],
                      }
                  }
             ),

alita_sdk/tools/github/__init__.py

@@ -71,7 +71,9 @@ class AlitaGitHubToolkit(BaseToolkit):
                                     }
                                 ]
                             },
-                        }
+                        },
+                        "categories": ["code repositories"],
+                        "extra_categories": ["github", "git", "repository", "code", "version control"],
                     },
                 }
             ),
@@ -119,4 +121,5 @@ class AlitaGitHubToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self):
-        return self.tools
+        return self.tools
+

alita_sdk/tools/gitlab/__init__.py

@@ -53,7 +53,9 @@ class AlitaGitlabToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["code repositories"],
+                    "extra_categories": ["gitlab", "git", "repository", "code", "version control"],
                 }
             })
         )

alita_sdk/tools/gitlab_org/__init__.py

@@ -50,7 +50,9 @@ class AlitaGitlabSpaceToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["code repositories"],
+                    "extra_categories": ["gitlab", "git", "repository", "code", "version control"],
                 }
             })
         )

alita_sdk/tools/google_places/__init__.py

@@ -31,7 +31,16 @@ class GooglePlacesToolkit(BaseToolkit):
             api_key=(SecretStr, Field(description="Google Places API key", json_schema_extra={'secret': True, 'max_toolkit_length': GooglePlacesToolkit.toolkit_max_length})),
             results_count=(Optional[int], Field(description="Results number to show", default=None)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Google Places", "icon_url": "gplaces-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra=
+                                  {
+                                      'metadata':
+                                          {
+                                              "label": "Google Places", "icon_url": "gplaces-icon.svg",
+                                              "categories": ["other"],
+                                              "extra_categories": ["google", "places", "maps", "location",
+                                                                   "geolocation"],
+                                          }
+                                  })
         )
 
     @classmethod
@@ -54,4 +63,4 @@ class GooglePlacesToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools

alita_sdk/tools/jira/__init__.py

@@ -51,7 +51,7 @@ class JiraToolkit(BaseToolkit):
             api_key=(Optional[SecretStr], Field(description="API key", default=None, json_schema_extra={'secret': True, 'configuration': True})),
             username=(Optional[str], Field(description="Jira Username", default=None, json_schema_extra={'configuration': True})),
             token=(Optional[SecretStr], Field(description="Jira token", default=None, json_schema_extra={'secret': True, 'configuration': True})),
-            limit=(int, Field(description="Limit issues", default=5)),
+            limit=(int, Field(description="Limit issues")),
             labels=(Optional[str], Field(
                 description="List of comma separated labels used for labeling of agent's created or updated entities",
                 default=None,
@@ -78,7 +78,9 @@ class JiraToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["project management"],
+                    "extra_categories": ["jira", "atlassian", "issue tracking", "project management", "task management"],
                 }
             })
         )

alita_sdk/tools/keycloak/__init__.py

@@ -34,7 +34,7 @@ class KeycloakToolkit(BaseToolkit):
             client_id=(str, Field(default="", title="Client ID", description="Keycloak client ID")),
             client_secret=(SecretStr, Field(default="", title="Client sercet", description="Keycloak client secret", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Keycloak", "icon_url": None, "hidden": True}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Keycloak", "icon_url": None, "hidden": True, "categories": ["authentication", "identity management"]}})
         )
 
     @classmethod
@@ -57,4 +57,5 @@ class KeycloakToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/ocr/__init__.py

@@ -36,7 +36,10 @@ class OCRToolkit(BaseToolkit):
             structured_output=(bool, Field(description="Whether to return structured JSON output", default=False)),
             expected_fields=(dict, Field(description="Expected fields for structured output", default={})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "OCR", "icon_url": None, "hidden": True}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "OCR", "icon_url": None, "hidden": True,
+                                                                  "categories": ["analysis"],
+                                                                    "extra_categories": ["optical character recognition", "text extraction"]
+                                                                  }})
         )
 
     @classmethod
@@ -59,4 +62,5 @@ class OCRToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/pandas/__init__.py

@@ -31,7 +31,9 @@ class PandasToolkit(BaseToolkit):
             name,
             bucket_name=(str, Field(default=None, title="Bucket name", description="Bucket where the content file is stored", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': PandasToolkit.toolkit_max_length})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Pandas", "icon_url": "pandas-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Pandas", "icon_url": "pandas-icon.svg",
+                                                                  "categories": ["analysis"],
+                                                                    "extra_categories": ["data science", "data manipulation", "dataframes"]}})
         )
 
     @classmethod
@@ -54,4 +56,5 @@ class PandasToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self):
-        return self.tools
+        return self.tools
+

alita_sdk/tools/pptx/__init__.py

@@ -51,7 +51,9 @@ class PPTXToolkit(BaseToolkit):
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "PPTX",
-                    "icon_url": None
+                    "icon_url": None,
+                    "categories": ["office"],
+                    "extra_categories": ["presentation", "office automation", "document"]
                 }
             })
         )
@@ -94,4 +96,5 @@ class PPTXToolkit(BaseToolkit):
         """
         Return all tools in the toolkit.
         """
-        return self.tools
+        return self.tools
+

alita_sdk/tools/qtest/__init__.py

@@ -36,7 +36,9 @@ class QtestToolkit(BaseToolkit):
             qtest_project_id=(int, Field(description="QTest project id", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': QtestToolkit.toolkit_max_length})),
             qtest_api_token=(SecretStr, Field(description="QTest API token", json_schema_extra={'secret': True, 'configuration': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "QTest", "icon_url": "qtest.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "QTest", "icon_url": "qtest.svg",
+                                                                  "categories": ["test management"],
+                                                                  "extra_categories": ["quality assurance", "test case management", "test planning"]}})
         )
 
         @check_connection_response

alita_sdk/tools/rally/__init__.py

@@ -55,7 +55,9 @@ class RallyToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["project management"],
+                    "extra_categories": ["agile management", "test management", "scrum", "kanban"]
                 }
             })
         )
@@ -81,4 +83,4 @@ class RallyToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self):
-        return self.tools
+        return self.tools

alita_sdk/tools/report_portal/__init__.py

@@ -34,7 +34,9 @@ class ReportPortalToolkit(BaseToolkit):
             project=(str, Field(description="Report Portal project")),
             api_key=(SecretStr, Field(description="User API key", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Report Portal", "icon_url": "reportportal-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Report Portal", "icon_url": "reportportal-icon.svg",
+                                                                  "categories": ["testing"],
+                                                                  "extra_categories": ["test reporting", "test automation"]}})
         )
 
     @classmethod

alita_sdk/tools/salesforce/__init__.py

@@ -30,7 +30,11 @@ class SalesforceToolkit(BaseToolkit):
             client_secret=(SecretStr, Field(description="Salesforce Connected App Client Secret", json_schema_extra={'secret': True})),
             api_version=(str, Field(description="Salesforce API Version", default='v59.0')),
             selected_tools=(List[Literal[tuple(available_tools)]], Field(default=[], json_schema_extra={'args_schemas': available_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Salesforce", "icon_url": "salesforce-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {
+                "label": "Salesforce", "icon_url": "salesforce-icon.svg",
+                "categories": ["other"],
+                "extra_categories": ["customer relationship management", "cloud computing", "marketing automation", "salesforce"]
+                                                                  }})
         )
 
     @classmethod

alita_sdk/tools/servicenow/__init__.py

@@ -58,7 +58,9 @@ class ServiceNowToolkit(BaseToolkit):
                                 }
                             ]
                         }
-                    }
+                    },
+                    "categories": ["other"],
+                    "extra_categories": ["incident management", "problem management", "change management", "service catalog"]
                 }
             })
         )

alita_sdk/tools/sharepoint/__init__.py

@@ -32,7 +32,12 @@ class SharepointToolkit(BaseToolkit):
             client_id=(str, Field(description="Client ID")),
             client_secret=(SecretStr, Field(description="Client Secret", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Sharepoint", "icon_url": "sharepoint.svg"}})
+            __config__=ConfigDict(json_schema_extra={
+                'metadata': {
+                    "label": "Sharepoint", "icon_url": "sharepoint.svg",
+                    "categories": ["office"],
+                    "extra_categories": ["microsoft", "cloud storage", "team collaboration", "content management"]
+        }})
         )
 
     @classmethod

alita_sdk/tools/sql/__init__.py

@@ -41,7 +41,13 @@ class SQLToolkit(BaseToolkit):
             password=(SecretStr, Field(description="Database password", json_schema_extra={'secret': True})),
             database_name=(str, Field(description="Database name", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': SQLToolkit.toolkit_max_length})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "SQL", "icon_url": "sql-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra=
+                                  {
+                                      'metadata':
+                                          {
+                                              "label": "SQL", "icon_url": "sql-icon.svg",
+                                              "categories": ["development"],
+                                              "extra_categories": ["sql", "data management", "data analysis"]}})
         )
 
     @classmethod
@@ -64,4 +70,5 @@ class SQLToolkit(BaseToolkit):
         return cls(tools=tools)
 
     def get_tools(self) -> list[BaseTool]:
-        return self.tools
+        return self.tools
+

alita_sdk/tools/testio/__init__.py

@@ -32,7 +32,9 @@ class TestIOToolkit(BaseToolkit):
             endpoint=(str, Field(description="TestIO endpoint", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': TOOLKIT_MAX_LENGTH})),
             api_key=(SecretStr, Field(description="API key", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "TestIO", "icon_url": "testio-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "TestIO", "icon_url": "testio-icon.svg",
+                                                                  "categories": ["testing"],
+                                                                  "extra_categories": ["test automation", "test case management", "test planning"]}})
         )
 
     @classmethod

alita_sdk/tools/testrail/__init__.py

@@ -45,7 +45,11 @@ class TestrailToolkit(BaseToolkit):
             email=(str, Field(description="User's email", json_schema_extra={'configuration': True})),
             password=(SecretStr, Field(description="User's password", json_schema_extra={'secret': True, 'configuration': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Testrail", "icon_url": "testrail-icon.svg"}})
+            __config__=ConfigDict(json_schema_extra={'metadata':
+                                                         {"label": "Testrail", "icon_url": "testrail-icon.svg",
+                                                          "categories": ["test management"],
+                                                          "extra_categories": ["quality assurance", "test case management", "test planning"]
+                                                          }})
         )
 
         @check_connection_response

alita_sdk/tools/xray/__init__.py

@@ -39,7 +39,15 @@ class XrayToolkit(BaseToolkit):
             client_secret=(SecretStr, Field(description="Client secret", json_schema_extra={'secret': True})),
             limit=(Optional[int], Field(description="Limit", default=100)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__={'json_schema_extra': {'metadata': {"label": "XRAY cloud", "icon_url": "xray.svg"}}}
+            __config__={'json_schema_extra':
+                            {
+                                'metadata': {
+                                    "label": "XRAY cloud", "icon_url": "xray.svg",
+                                "categories": ["test management"],
+                                    "extra_categories": ["test automation", "test case management", "test planning"]
+                                }
+                            }
+            }
         )
 
     @classmethod