alita-sdk 0.3.243__tar.gz → 0.3.245__tar.gz

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.243
+Version: 0.3.245
 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 <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

alita_sdk-0.3.245/alita_sdk/configurations/figma.py

@@ -0,0 +1,38 @@
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field, SecretStr
+
+
+class FigmaConfiguration(BaseModel):
+    model_config = ConfigDict(
+        json_schema_extra={
+            "metadata": {
+                "label": "Figma",
+                "icon_url": "figma-icon.svg",
+                "sections": {
+                    "auth": {
+                        "required": True,
+                        "subsections": [
+                            {
+                                "name": "Token",
+                                "fields": ["token"]
+                            },
+                            {
+                                "name": "Oath2",
+                                "fields": ["oauth2"]
+                            }
+                        ]
+                    }
+                },
+                "section": "credentials",
+                "type": "figma",
+                "categories": ["other"],
+                "extra_categories": ["figma", "design", "ui/ux", "prototyping", "collaboration"],
+            }
+        }
+    )
+    url: str = Field(description="Testrail URL")
+    email: str = Field(description="TestRail Email")
+    password: SecretStr = Field(description="TestRail Password")
+    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.245/alita_sdk/configurations/rally.py

@@ -0,0 +1,39 @@
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field, SecretStr
+
+
+class RallyConfiguration(BaseModel):
+    model_config = ConfigDict(
+        json_schema_extra={
+            "metadata": {
+                "label": "Rally",
+                "icon_url": "rally.svg",
+                "sections": {
+                    "auth": {
+                        "required": True,
+                        "subsections": [
+                            {
+                                "name": "Password",
+                                "fields": ["username", "password"]
+                            },
+                            {
+                                "name": "API Key",
+                                "fields": ["api_key"]
+                            }
+                        ]
+                    }
+                },
+                "section": "credentials",
+                "type": "rally",
+                "categories": ["project management"],
+                "extra_categories": ["agile management", "test management", "scrum", "kanban"]
+            }
+        }
+    )
+
+    server: str = Field(description="Rally server url")
+    api_key: Optional[SecretStr] = Field(default=None, description="User's API key", json_schema_extra={'secret': True})
+    username: Optional[str] = Field(default=None, description="Username")
+    password: Optional[SecretStr] = Field(default=None, description="User's password",
+                                          json_schema_extra={'secret': True})

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk/tools/elitea_base.py

@@ -301,7 +301,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
         collection_suffix = kwargs.get("collection_suffix")
         progress_step = kwargs.get("progress_step")
         clean_index = kwargs.get("clean_index")
-        vs = self._init_vector_store(embeddings=self._embedding)
+        vs = self._init_vector_store()
         #
         return vs.index_documents(docs, collection_suffix=collection_suffix, progress_step=progress_step, clean_index=clean_index)
 
@@ -337,7 +337,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
 
 
     # TODO: init store once and re-use the instance
-    def _init_vector_store(self, embeddings: Optional[Any] = None):
+    def _init_vector_store(self):
         """Initializes the vector store wrapper with the provided parameters."""
         try:
             from alita_sdk.runtime.tools.vectorstore import VectorStoreWrapper
@@ -353,7 +353,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                 embedding_model=self.embedding_model,
                 embedding_model_params=self.embedding_model_params,
                 vectorstore_params=vectorstore_params,
-                embeddings=embeddings,
+                embeddings=self._embedding,
                 process_document_func=self._process_documents,
             )
         return self._vector_store
@@ -411,7 +411,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      extended_search: Optional[List[str]] = None,
                      **kwargs):
         """ Searches indexed documents in the vector store."""
-        vectorstore = self._init_vector_store(collection_suffix)
+        vectorstore = self._init_vector_store()
         found_docs = vectorstore.stepback_search(
             query,
             messages,
@@ -436,7 +436,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      extended_search: Optional[List[str]] = None,
                      **kwargs):
         """ Generates a summary of indexed documents using stepback technique."""
-        vectorstore = self._init_vector_store(collection_suffix)
+        vectorstore = self._init_vector_store()
         return vectorstore.stepback_summary(
             query, 
             messages, 

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk/tools/figma/__init__.py

@@ -7,6 +7,7 @@ from ..base.tool import BaseAction
 from .api_wrapper import FigmaApiWrapper, GLOBAL_LIMIT
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 from ...configurations.embedding import EmbeddingConfiguration
+from ...configurations.figma import FigmaConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
 
 name = "figma"
@@ -16,8 +17,7 @@ def get_tools(tool):
         FigmaToolkit()
         .get_toolkit(
             selected_tools=tool["settings"].get("selected_tools", []),
-            token=tool["settings"].get("token", None),
-            oauth2=tool["settings"].get("oauth2", None),
+            figma_configuration=tool['settings']['figma_configuration'],
             global_limit=tool["settings"].get("global_limit", GLOBAL_LIMIT),
             global_regexp=tool["settings"].get("global_regexp", None),
             toolkit_name=tool.get('toolkit_name'),
@@ -47,14 +47,15 @@ class FigmaToolkit(BaseToolkit):
         FigmaToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            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)),
             global_limit=(Optional[int], Field(description="Global limit", default=GLOBAL_LIMIT)),
             global_regexp=(Optional[str], Field(description="Global regex pattern", default=None)),
             selected_tools=(
                 List[Literal[tuple(selected_tools)]],
                 Field(default=[], json_schema_extra={"args_schemas": selected_tools}),
             ),
+            # Figma configuration
+            figma_configuration=(Optional[FigmaConfiguration], Field(description="Figma configuration", json_schema_extra={'configuration_types': ['figma']})),
+
             # indexer settings
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
 
@@ -68,21 +69,6 @@ class FigmaToolkit(BaseToolkit):
                          "label": "Figma",
                          "icon_url": "figma-icon.svg",
                          "max_length": FigmaToolkit.toolkit_max_length,
-                         "sections": {
-                             "auth": {
-                                 "required": True,
-                                 "subsections": [
-                                     {
-                                         "name": "Token",
-                                         "fields": ["token"]
-                                     },
-                                     {
-                                         "name": "Oath2",
-                                         "fields": ["oauth2"]
-                                     }
-                                 ]
-                             }
-                         },
                          "categories": ["other"],
                          "extra_categories": ["figma", "design", "ui/ux", "prototyping", "collaboration"],
                      }
@@ -96,6 +82,7 @@ class FigmaToolkit(BaseToolkit):
             selected_tools = []
         wrapper_payload = {
             **kwargs,
+            **kwargs.get('figma_configuration'),
             **(kwargs.get('pgvector_configuration') or {}),
             **(kwargs.get('embedding_configuration') or {}),
         }

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk/tools/gitlab/__init__.py

@@ -58,17 +58,6 @@ class AlitaGitlabToolkit(BaseToolkit):
                 'metadata': {
                     "label": "GitLab",
                     "icon_url": None,
-                    "sections": {
-                        "auth": {
-                            "required": True,
-                            "subsections": [
-                                {
-                                    "name": "GitLab private token",
-                                    "fields": ["private_token"]
-                                }
-                            ]
-                        }
-                    },
                     "categories": ["code repositories"],
                     "extra_categories": ["gitlab", "git", "repository", "code", "version control"],
                 }

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk/tools/gitlab_org/__init__.py

@@ -5,16 +5,16 @@ from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
 from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ...configurations.gitlab import GitlabConfiguration
 
 name = "gitlab_org"
 
 def get_tools(tool):
     return AlitaGitlabSpaceToolkit().get_toolkit(
         selected_tools=tool['settings'].get('selected_tools', []),
-        url=tool['settings']['url'],
+        gitlab_configuration=tool['settings']['gitlab_configuration'],
         repositories=tool['settings'].get('repositories', ''),
         branch=tool['settings']['branch'],
-        private_token=tool['settings']['private_token'],
         toolkit_name=tool.get('toolkit_name')
     ).get_tools()
 
@@ -28,29 +28,22 @@ class AlitaGitlabSpaceToolkit(BaseToolkit):
         AlitaGitlabSpaceToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            url=(str, Field(description="GitLab URL", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': AlitaGitlabSpaceToolkit.toolkit_max_length})),
+            name=(str, Field(description="Toolkit name", json_schema_extra={'toolkit_name': True,
+                                                                            'max_toolkit_length': AlitaGitlabSpaceToolkit.toolkit_max_length})),
+            gitlab_configuration=(Optional[GitlabConfiguration], Field(description="GitLab configuration",
+                                                                       json_schema_extra={
+                                                                           'configuration_types': ['gitlab']})),
             repositories=(str, Field(
                 description="List of comma separated repositories user plans to interact with. Leave it empty in case you pass it in instruction.",
                 default=''
             )),
-            private_token=(SecretStr, Field(description="GitLab private token", json_schema_extra={'secret': True})),
             branch=(str, Field(description="Main branch", default="main")),
-            selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
+            selected_tools=(List[Literal[tuple(selected_tools)]],
+                            Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "GitLab Org",
                     "icon_url": None,
-                    "sections": {
-                        "auth": {
-                            "required": True,
-                            "subsections": [
-                                {
-                                    "name": "GitLab private token",
-                                    "fields": ["private_token"]
-                                }
-                            ]
-                        }
-                    },
                     "categories": ["code repositories"],
                     "extra_categories": ["gitlab", "git", "repository", "code", "version control"],
                 }
@@ -61,7 +54,12 @@ class AlitaGitlabSpaceToolkit(BaseToolkit):
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []
-        gitlab_wrapper = GitLabWorkspaceAPIWrapper(**kwargs)
+        wrapper_payload = {
+            **kwargs,
+            # TODO use gitlab_configuration fields
+            **kwargs['gitlab_configuration'],
+        }
+        gitlab_wrapper = GitLabWorkspaceAPIWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, AlitaGitlabSpaceToolkit.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = gitlab_wrapper.get_available_tools()
         tools = []

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk/tools/rally/__init__.py

@@ -5,16 +5,14 @@ from .api_wrapper import RallyApiWrapper
 from langchain_core.tools import BaseTool
 from ..base.tool import BaseAction
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ...configurations.rally import RallyConfiguration
 
 name = "rally"
 
 def get_tools(tool):
     return RallyToolkit().get_toolkit(
         selected_tools=tool['settings'].get('selected_tools', []),
-        server=tool['settings']['server'],
-        api_key=tool['settings'].get('api_key'),
-        username=tool['settings'].get('username'),
-        password=tool['settings'].get('password'),
+        rally_configuration=tool['settings']['rally_configuration'],
         workspace=tool['settings'].get('workspace', None),
         project=tool['settings'].get('project', None),
         toolkit_name=tool.get('toolkit_name')
@@ -30,10 +28,9 @@ class RallyToolkit(BaseToolkit):
         RallyToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            server=(str, Field(description="Rally server url", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': RallyToolkit.toolkit_max_length})),
-            api_key=(Optional[SecretStr], Field(default=None, description="User's API key", json_schema_extra={'secret': True})),
-            username=(Optional[str], Field(default=None, description="Username")),
-            password=(Optional[SecretStr], Field(default=None, description="User's password", json_schema_extra={'secret': True})),
+            name=(str, Field(description="Toolkit name", json_schema_extra={'toolkit_name': True,
+                                                                            'max_toolkit_length': RallyToolkit.toolkit_max_length})),
+            rally_configuration=(Optional[RallyConfiguration], Field(description="Rally configuration", json_schema_extra={'configuration_types': ['rally']})),
             workspace=(Optional[str], Field(default=None, description="Rally workspace")),
             project=(Optional[str], Field(default=None, description="Rally project")),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
@@ -41,21 +38,6 @@ class RallyToolkit(BaseToolkit):
                 'metadata': {
                     "label": "Rally",
                     "icon_url": "rally.svg",
-                    "sections": {
-                        "auth": {
-                            "required": True,
-                            "subsections": [
-                                {
-                                    "name": "Password",
-                                    "fields": ["username", "password"]
-                                },
-                                {
-                                    "name": "API Key",
-                                    "fields": ["api_key"]
-                                }
-                            ]
-                        }
-                    },
                     "categories": ["project management"],
                     "extra_categories": ["agile management", "test management", "scrum", "kanban"]
                 }
@@ -66,7 +48,11 @@ class RallyToolkit(BaseToolkit):
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         if selected_tools is None:
             selected_tools = []
-        rally_api_wrapper = RallyApiWrapper(**kwargs)
+        wrapper_payload = {
+            **kwargs,
+            **kwargs.get('rally_configuration'),
+        }
+        rally_api_wrapper = RallyApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = rally_api_wrapper.get_available_tools()
         tools = []

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.243
+Version: 0.3.245
 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 <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/alita_sdk.egg-info/SOURCES.txt

@@ -17,12 +17,14 @@ alita_sdk/configurations/bitbucket.py
 alita_sdk/configurations/confluence.py
 alita_sdk/configurations/delta_lake.py
 alita_sdk/configurations/embedding.py
+alita_sdk/configurations/figma.py
 alita_sdk/configurations/github.py
 alita_sdk/configurations/gitlab.py
 alita_sdk/configurations/jira.py
 alita_sdk/configurations/pgvector.py
 alita_sdk/configurations/postman.py
 alita_sdk/configurations/qtest.py
+alita_sdk/configurations/rally.py
 alita_sdk/configurations/service_now.py
 alita_sdk/configurations/slack.py
 alita_sdk/configurations/testrail.py

{alita_sdk-0.3.243 → alita_sdk-0.3.245}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.243"
+version = "0.3.245"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"