alita-sdk 0.3.302__tar.gz → 0.3.304__tar.gz

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.302
+Version: 0.3.304
 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

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/configurations/figma.py

@@ -28,4 +28,3 @@ class FigmaConfiguration(BaseModel):
         }
     )
     token: Optional[SecretStr] = Field(description="Figma Token", json_schema_extra={"secret": True}, default=None)
-    oauth2: Optional[SecretStr] = Field(description="OAuth2 Token", json_schema_extra={"secret": True}, default=None)

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/runtime/langchain/interfaces/llm_processor.py

@@ -180,6 +180,8 @@ def add_documents(vectorstore, documents):
     texts = []
     metadata = []
     for document in documents:
+        if not document.page_content:
+            continue
         texts.append(document.page_content)
         for key in document.metadata:
             if isinstance(document.metadata[key], list):

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/runtime/tools/vectorstore.py

@@ -746,7 +746,7 @@ class VectorStoreWrapper(BaseToolApiWrapper):
 
         try:
             dispatch_custom_event(
-                name="thinking_step_update",
+                name="thinking_step",
                 data={
                     "message": message,
                     "tool_name": tool_name,

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/runtime/tools/vectorstore_base.py

@@ -133,9 +133,9 @@ How did you come up with the answer?
 
 class VectorStoreWrapperBase(BaseToolApiWrapper):
     llm: Any
-    embedding_model: str
-    vectorstore_type: str
-    vectorstore_params: dict
+    embedding_model: Optional[str] = None
+    vectorstore_type: Optional[str]  = None
+    vectorstore_params: Optional[dict]  = None
     max_docs_per_add: int = 100
     dataset: str = None
     embedding: Any = None
@@ -150,28 +150,22 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
     @model_validator(mode='before')
     @classmethod
     def validate_toolkit(cls, values):
-        from ..langchain.interfaces.llm_processor import get_embeddings, get_vectorstore
+        from ..langchain.interfaces.llm_processor import get_vectorstore
         logger.debug(f"Validating toolkit: {values}")
-        if not values.get('vectorstore_type'):
-            raise ValueError("Vectorstore type is required.")
-        if not values.get('embedding_model'):
-            raise ValueError("Embedding model is required.")
-        if not values.get('vectorstore_params'):
-            raise ValueError("Vectorstore parameters are required.")
-        values["dataset"] = values.get('vectorstore_params').get('collection_name')
-        if not values["dataset"]:
-            raise ValueError("Collection name is required.")
-        if not values.get('embeddings'):
+        if 'vectorstore_params' in values:
+            values["dataset"] = values.get('vectorstore_params').get('collection_name')
+        if values.get('embedding_model'):
             values['embeddings'] = values['alita'].get_embeddings(values['embedding_model'])
-        values['vectorstore'] = get_vectorstore(values['vectorstore_type'], values['vectorstore_params'], embedding_func=values['embeddings'])
-        values['vectoradapter'] = VectorAdapter(
-            vectorstore=values['vectorstore'],
-            embeddings=values['embeddings'],
-            quota_params=None,
-        )
-        # Initialize the new vector adapter
-        values['vector_adapter'] = VectorStoreAdapterFactory.create_adapter(values['vectorstore_type'])
-        logger.debug(f"Vectorstore wrapper initialized: {values}")
+        if values.get('vectorstore_type') and values.get('vectorstore_params') and values.get('embedding_model'):
+            values['vectorstore'] = get_vectorstore(values['vectorstore_type'], values['vectorstore_params'], embedding_func=values['embeddings'])
+            values['vectoradapter'] = VectorAdapter(
+                vectorstore=values['vectorstore'],
+                embeddings=values['embeddings'],
+                quota_params=None,
+            )
+            # Initialize the new vector adapter
+            values['vector_adapter'] = VectorStoreAdapterFactory.create_adapter(values['vectorstore_type'])
+            logger.debug(f"Vectorstore wrapper initialized: {values}")
         return values
 
     def _init_pg_helper(self, language='english'):
@@ -587,7 +581,7 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
 
         try:
             dispatch_custom_event(
-                name="thinking_step_update",
+                name="thinking_step",
                 data={
                     "message": message,
                     "tool_name": tool_name,

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/ado/repos/__init__.py

@@ -5,6 +5,7 @@ from pydantic import BaseModel, Field, create_model
 
 import requests
 
+from ...elitea_base import filter_missconfigured_index_tools
 from ....configurations.ado import AdoReposConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
@@ -26,6 +27,7 @@ def _get_toolkit(tool) -> BaseToolkit:
         embedding_model=tool['settings'].get('embedding_model'),
         collection_name=tool['toolkit_name'],
         alita=tool['settings'].get('alita', None),
+        llm=tool['settings'].get('llm', None),
     )
 
 def get_toolkit():
@@ -77,6 +79,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         from os import environ
 

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/ado/test_plan/__init__.py

@@ -4,6 +4,8 @@ from langchain_core.tools import BaseTool, BaseToolkit
 from pydantic import create_model, BaseModel, Field
 
 import requests
+
+from ...elitea_base import filter_missconfigured_index_tools
 from ....configurations.ado import AdoConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from .test_plan_wrapper import TestPlanApiWrapper
@@ -79,6 +81,7 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         from os import environ
         if not environ.get('AZURE_DEVOPS_CACHE_DIR', None):

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/ado/wiki/__init__.py

@@ -4,6 +4,8 @@ from langchain_core.tools import BaseTool, BaseToolkit
 from pydantic import create_model, BaseModel, Field
 
 import requests
+
+from ...elitea_base import filter_missconfigured_index_tools
 from ....configurations.ado import AdoConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
@@ -77,6 +79,7 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         from os import environ
         if not environ.get('AZURE_DEVOPS_CACHE_DIR', None):

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/ado/work_item/__init__.py

@@ -4,6 +4,8 @@ from langchain_core.tools import BaseTool, BaseToolkit
 from pydantic import create_model, BaseModel, Field
 
 import requests
+
+from ...elitea_base import filter_missconfigured_index_tools
 from ....configurations.ado import AdoConfiguration
 from ....configurations.pgvector import PgVectorConfiguration
 from ...base.tool import BaseAction
@@ -77,6 +79,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         from os import environ
         if not environ.get('AZURE_DEVOPS_CACHE_DIR', None):

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/advanced_jira_mining/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, Field, SecretStr
 
 from .data_mining_wrapper import AdvancedJiraMiningWrapper
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "advanced_jira_mining"
@@ -59,6 +60,7 @@ class AdvancedJiraMiningToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/azure_ai/search/__init__.py

@@ -4,6 +4,8 @@ from .api_wrapper import AzureSearchApiWrapper
 from ...base.tool import BaseAction
 from langchain_core.tools import BaseToolkit, BaseTool
 from pydantic import create_model, BaseModel, ConfigDict, Field
+
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, check_connection_response
 from ....configurations.azure_search import AzureSearchConfiguration
 import requests
@@ -66,6 +68,7 @@ class AzureSearchToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/base_indexer_toolkit.py

@@ -110,13 +110,13 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         connection_string = conn.get_secret_value() if isinstance(conn, SecretStr) else conn
         collection_name = kwargs.get('collection_name')
         
-        if not kwargs.get('embedding_model'):
-            kwargs['embedding_model'] = 'text-embedding-ada-002'
         if 'vectorstore_type' not in kwargs:
             kwargs['vectorstore_type'] = 'PGVector'
         vectorstore_type = kwargs.get('vectorstore_type')
-        kwargs['vectorstore_params'] = VectorStoreAdapterFactory.create_adapter(vectorstore_type).get_vectorstore_params(collection_name, connection_string)
-        kwargs['_embedding'] = kwargs.get('alita').get_embeddings(kwargs.get('embedding_model'))
+        if connection_string:
+            # Initialize vectorstore params only if connection string is provided
+            kwargs['vectorstore_params'] = VectorStoreAdapterFactory.create_adapter(vectorstore_type).get_vectorstore_params(collection_name, connection_string)
+            kwargs['_embedding'] = kwargs.get('alita').get_embeddings(kwargs.get('embedding_model'))
         super().__init__(**kwargs)
 
     def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/bitbucket/__init__.py

@@ -8,6 +8,7 @@ from langchain_core.tools import BaseTool
 from pydantic import BaseModel, Field, ConfigDict, create_model
 
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, check_connection_response
 from ...configurations.bitbucket import BitbucketConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
@@ -85,6 +86,7 @@ class AlitaBitbucketToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/carrier/__init__.py

@@ -6,6 +6,7 @@ from functools import lru_cache
 
 from .api_wrapper import CarrierAPIWrapper
 from .tools import __all__
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.carrier import CarrierConfiguration
 
@@ -46,6 +47,7 @@ class AlitaCarrierToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(
             cls,
             selected_tools: Optional[List[str]] = None,

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/cloud/aws/__init__.py

@@ -5,6 +5,7 @@ from langchain_core.tools import BaseToolkit, BaseTool
 
 from .api_wrapper import AWSToolConfig
 from ...base.tool import BaseAction
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "aws"
@@ -46,6 +47,7 @@ class AWSToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/cloud/azure/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 from .api_wrapper import AzureApiWrapper
 from ...base.tool import BaseAction
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "azure"
@@ -39,6 +40,7 @@ class AzureToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/cloud/gcp/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 from .api_wrapper import GCPApiWrapper
 from ...base.tool import BaseAction
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "gcp"
@@ -33,6 +34,7 @@ class GCPToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/cloud/k8s/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 from .api_wrapper import KubernetesApiWrapper
 from ...base.tool import BaseAction
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "kubernetes"
@@ -43,6 +44,7 @@ class KubernetesToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/code/sonar/__init__.py

@@ -4,6 +4,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field
 
 from .api_wrapper import SonarApiWrapper
 from ...base.tool import BaseAction
+from ...elitea_base import filter_missconfigured_index_tools
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ....configurations.sonar import SonarConfiguration
 
@@ -43,6 +44,7 @@ class SonarToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/confluence/__init__.py

@@ -4,6 +4,8 @@ from .api_wrapper import ConfluenceAPIWrapper
 from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field
+
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, parse_list, check_connection_response
 from ...configurations.confluence import ConfluenceConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
@@ -103,6 +105,7 @@ class ConfluenceToolkit(BaseToolkit):
         return model
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/elitea_base.py

@@ -17,6 +17,9 @@ from ..runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
 
+INDEX_TOOL_NAMES = ['index_data', 'remove_index', 'list_collections', 'search_index', 'stepback_search_index',
+                            'stepback_summary_index']
+
 LoaderSchema = create_model(
     "LoaderSchema",
     branch=(Optional[str], Field(
@@ -137,7 +140,7 @@ class BaseToolApiWrapper(BaseModel):
 
             logger.info(message)
             dispatch_custom_event(
-                name="thinking_step_update",
+                name="thinking_step",
                 data={
                     "message": message,
                     "tool_name": tool_name,
@@ -699,4 +702,24 @@ def extend_with_vector_tools(method):
         #
         return tools
 
-    return wrapper
+    return wrapper
+
+
+def filter_missconfigured_index_tools(method):
+    def wrapper(self, *args, **kwargs):
+        toolkit = method(self, *args, **kwargs)
+
+        # Validate index tools misconfiguration and exclude them if necessary
+        is_index_toolkit = any(tool.name.rsplit(TOOLKIT_SPLITTER)[1]
+                               if TOOLKIT_SPLITTER in tool.name else tool.name
+                                                                     in INDEX_TOOL_NAMES for tool in toolkit.tools)
+        is_index_configuration_missing = not (kwargs.get('embedding_model')
+                                              and kwargs.get('pgvector_configuration'))
+
+        if is_index_toolkit and is_index_configuration_missing:
+            toolkit.tools = [tool for tool in toolkit.tools if (tool.name.rsplit(TOOLKIT_SPLITTER, 1)[
+                                                                    1] if TOOLKIT_SPLITTER in tool.name else tool.name) not in INDEX_TOOL_NAMES]
+
+        return toolkit
+
+    return wrapper

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/figma/__init__.py

@@ -5,6 +5,7 @@ from pydantic import BaseModel, ConfigDict, Field, create_model
 
 from ..base.tool import BaseAction
 from .api_wrapper import FigmaApiWrapper, GLOBAL_LIMIT
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.figma import FigmaConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
@@ -74,6 +75,7 @@ class FigmaToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/github/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field
 
 from .api_wrapper import AlitaGitHubAPIWrapper
 from .tool import GitHubAction
+from ..elitea_base import filter_missconfigured_index_tools
 
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.github import GithubConfiguration
@@ -72,6 +73,7 @@ class AlitaGitHubToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []
@@ -85,7 +87,8 @@ class AlitaGitHubToolkit(BaseToolkit):
         github_api_wrapper = AlitaGitHubAPIWrapper(**wrapper_payload)
         available_tools: List[Dict] = github_api_wrapper.get_available_tools()
         tools = []
-        prefix = clean_string(toolkit_name, AlitaGitHubToolkit.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
+        prefix = clean_string(toolkit_name) + TOOLKIT_SPLITTER if toolkit_name else ''
+        # prefix = clean_string(toolkit_name, AlitaGitHubToolkit.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         for tool in available_tools:
             if selected_tools:
                 if tool["name"] not in selected_tools:

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/gitlab/__init__.py

@@ -7,6 +7,7 @@ from pydantic import create_model, BaseModel, ConfigDict
 from pydantic.fields import Field
 
 from .api_wrapper import GitLabAPIWrapper
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.gitlab import GitlabConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
@@ -64,6 +65,7 @@ class AlitaGitlabToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/gitlab_org/__init__.py

@@ -4,6 +4,8 @@ from langchain_core.tools import BaseToolkit
 from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
+
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.gitlab import GitlabConfiguration
 
@@ -51,6 +53,7 @@ class AlitaGitlabSpaceToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/google_places/__init__.py

@@ -5,6 +5,7 @@ from pydantic.fields import Field
 
 from .api_wrapper import GooglePlacesAPIWrapper
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.google_places import GooglePlacesConfiguration
 
@@ -45,6 +46,7 @@ class GooglePlacesToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/jira/__init__.py

@@ -5,6 +5,7 @@ from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field
 import requests
 
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, parse_list, check_connection_response
 from ...configurations.jira import JiraConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
@@ -16,6 +17,7 @@ def get_tools(tool):
         selected_tools=tool['settings'].get('selected_tools', []),
         base_url=tool['settings'].get('base_url'),
         cloud=tool['settings'].get('cloud', True),
+        api_version=tool['settings'].get('api_version', '2'),
         jira_configuration=tool['settings']['jira_configuration'],
         limit=tool['settings'].get('limit', 5),
         labels=parse_list(tool['settings'].get('labels', [])),
@@ -64,8 +66,14 @@ class JiraToolkit(BaseToolkit):
 
         model = create_model(
             name,
+            name=(str, Field(description="Toolkit name",
+                             json_schema_extra={
+                                 'toolkit_name': True,
+                                 'max_toolkit_length': JiraToolkit.toolkit_max_length})
+                  ),
             cloud=(bool, Field(description="Hosting Option", json_schema_extra={'configuration': True})),
             limit=(int, Field(description="Limit issues. Default is 5", gt=0, default=5)),
+            api_version=(Optional[str], Field(description="Rest API version: optional. Default is 2", default="2")),
             labels=(Optional[str], Field(
                 description="List of comma separated labels used for labeling of agent's created or updated entities",
                 default=None,
@@ -95,6 +103,7 @@ class JiraToolkit(BaseToolkit):
         return model
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/qtest/__init__.py

@@ -6,6 +6,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 from .api_wrapper import QtestApiWrapper
 from .tool import QtestAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, get_max_toolkit_length, TOOLKIT_SPLITTER, check_connection_response
 from ...configurations.qtest import QtestConfiguration
 
@@ -61,6 +62,7 @@ class QtestToolkit(BaseToolkit):
         return m
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/rally/__init__.py

@@ -4,6 +4,7 @@ from pydantic import BaseModel, ConfigDict, Field, create_model, SecretStr
 from .api_wrapper import RallyApiWrapper
 from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.rally import RallyConfiguration
 
@@ -45,6 +46,7 @@ class RallyToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/report_portal/__init__.py

@@ -6,6 +6,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field
 
 from .api_wrapper import ReportPortalApiWrapper
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.report_portal import ReportPortalConfiguration
 
@@ -37,6 +38,7 @@ class ReportPortalToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/salesforce/__init__.py

@@ -3,6 +3,8 @@ from .api_wrapper import SalesforceApiWrapper
 from langchain_core.tools import BaseTool, BaseToolkit
 from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field
+
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER,get_max_toolkit_length
 from ...configurations.salesforce import SalesforceConfiguration
 
@@ -35,6 +37,7 @@ class SalesforceToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/servicenow/__init__.py

@@ -5,6 +5,8 @@ from .api_wrapper import ServiceNowAPIWrapper
 from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field
+
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.service_now import ServiceNowConfiguration
 
@@ -66,6 +68,7 @@ class ServiceNowToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []

{alita_sdk-0.3.302 → alita_sdk-0.3.304}/alita_sdk/tools/sharepoint/__init__.py

@@ -4,6 +4,7 @@ from langchain_core.tools import BaseToolkit, BaseTool
 from pydantic import create_model, BaseModel, ConfigDict, Field
 from .api_wrapper import SharepointApiWrapper
 from ..base.tool import BaseAction
+from ..elitea_base import filter_missconfigured_index_tools
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.pgvector import PgVectorConfiguration
 from ...configurations.sharepoint import SharepointConfiguration
@@ -53,6 +54,7 @@ class SharepointToolkit(BaseToolkit):
         )
 
     @classmethod
+    @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []