alita-sdk 0.3.244__py3-none-any.whl → 0.3.245__py3-none-any.whl

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/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/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/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/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/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.244.dist-info → alita_sdk-0.3.245.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.244
+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.244.dist-info → alita_sdk-0.3.245.dist-info}/RECORD

@@ -9,12 +9,14 @@ alita_sdk/configurations/bitbucket.py,sha256=iX0HAV_OJbgxitCVw5T_gJHOg7EAqHV2Ty8
 alita_sdk/configurations/confluence.py,sha256=LFFjhp0whiWcAN-2DnOVSUnQmjURBmV4C4iDyKs7vys,1423
 alita_sdk/configurations/delta_lake.py,sha256=ADWcjabi7Krq2yxIpoc_tmhdncdgot2GBphE7ziDeTY,1133
 alita_sdk/configurations/embedding.py,sha256=8GSC8Feh8CH7bT_6cQhNqlS6raE91S2YRAtb2N9bUA8,552
+alita_sdk/configurations/figma.py,sha256=UmnIKvC8G3hVvX47O-w_syiJwJAMn1LCmoZMWJTK40w,1473
 alita_sdk/configurations/github.py,sha256=GSj6sA4f6SfW0ZpoHXKi5FzbPDC6wE1AlscwWqIPj14,1832
 alita_sdk/configurations/gitlab.py,sha256=15tXlnFM3IQSUA10wy4tpfHC-dJIW-xYmHoSzoqRRc4,1077
 alita_sdk/configurations/jira.py,sha256=OxvkJ7a_ig7DQGqpL8rSZod_lJzN25KFaJDRnjt2xPs,1380
 alita_sdk/configurations/pgvector.py,sha256=P-Q07ocIg4CXN_7hUBDM6r9gN62XS1N2jyP79tM9Tig,500
 alita_sdk/configurations/postman.py,sha256=IdLjfFDZqGuvIAnVDlQWeFm4VOC8viUMpKgWruYgDso,1121
 alita_sdk/configurations/qtest.py,sha256=LHM6RXxs_iSwSUdBjNXXVvqiiehT9fkBESE-ECDukt0,695
+alita_sdk/configurations/rally.py,sha256=1rwYh7bVV3XXufWRuPbr3Gz6zVPnfbA42bJYvJYsY-o,1515
 alita_sdk/configurations/service_now.py,sha256=EXYrhXImaL5AT7zDJ244sVVf_0Vc-tHHIGdfRBo2XCg,1197
 alita_sdk/configurations/slack.py,sha256=x-Hz_Uo1xXCtlpMfJUY39q9kXm5IB0IR9l4wSLAFnR0,1315
 alita_sdk/configurations/testrail.py,sha256=k0fPmHBIrWAfEKhrDdB9Rdirw-UFHFoXkRePyrsqcWI,725
@@ -213,7 +215,7 @@ alita_sdk/tools/custom_open_api/__init__.py,sha256=9aT5SPNPWcJC6jMZEM-3rUCXVULj_
 alita_sdk/tools/custom_open_api/api_wrapper.py,sha256=sDSFpvEqpSvXHGiBISdQQcUecfO3md-_F8hAi6p2dvg,4340
 alita_sdk/tools/elastic/__init__.py,sha256=iwnSRppRpzvJ1da2K3Glu8Uu41MhBDCYbguboLkEbW0,2818
 alita_sdk/tools/elastic/api_wrapper.py,sha256=pl8CqQxteJAGwyOhMcld-ZgtOTFwwbv42OITQVe8rM0,1948
-alita_sdk/tools/figma/__init__.py,sha256=ndkDiXDHcpJjqRk5k06y6ioNzA4Mk-rOwEk4G7K3Qe8,5424
+alita_sdk/tools/figma/__init__.py,sha256=xaYyPdg9qZkp6y5h0EeOmyXqFznkRbLUSEg326Hwbbo,4763
 alita_sdk/tools/figma/api_wrapper.py,sha256=Rtgt9FvR8VD0oPdYhlgvVyXLVqLTjtiOPTlwNeaV80w,20560
 alita_sdk/tools/github/__init__.py,sha256=BEtPsqP-yuyt05pINlic2fJ41j_tx7OmGGqCN75kgEo,5406
 alita_sdk/tools/github/api_wrapper.py,sha256=uDwYckdnpYRJtb0uZnDkaz2udvdDLVxuCh1tSwspsiU,8411
@@ -222,11 +224,11 @@ alita_sdk/tools/github/graphql_client_wrapper.py,sha256=d3AGjzLGH_hdQV2V8HeAX92d
 alita_sdk/tools/github/schemas.py,sha256=yFsqivfjCPRk9GxFJrL8sTz6nnjFCZ0j5DIfPtGSsvA,13852
 alita_sdk/tools/github/tool.py,sha256=Jnnv5lenV5ds8AAdyo2m8hSzyJ117HZBjzHC6T1ck-M,1037
 alita_sdk/tools/github/tool_prompts.py,sha256=y6ZW_FpUCE87Uop3WuQAZVRnzxO5t7xjBOI5bCqiluw,30194
-alita_sdk/tools/gitlab/__init__.py,sha256=2jYZi3SFY65PmPbN-f_TXSKMP7IrnLUD_5ry4D8tFPs,5056
+alita_sdk/tools/gitlab/__init__.py,sha256=JniVJI_lYC0OSrc2lES-pyZY4zw5zUR3nMA2NoOmNVU,4618
 alita_sdk/tools/gitlab/api_wrapper.py,sha256=KYCRO2pF8EPTLhWuEj64XsHPCYSucsf8S3R_ofJttrA,22301
 alita_sdk/tools/gitlab/tools.py,sha256=vOGTlSaGaFmWn6LS6YFP-FuTqUPun9vnv1VrUcUHAZQ,16500
 alita_sdk/tools/gitlab/utils.py,sha256=Z2XiqIg54ouqqt1to-geFybmkCb1I6bpE91wfnINH1I,2320
-alita_sdk/tools/gitlab_org/__init__.py,sha256=_DJ5y92E6GuNBtCuaEZakGNInxbuFtvLYZMTMCDf3Js,3713
+alita_sdk/tools/gitlab_org/__init__.py,sha256=k8ECZzeGCmgKJ9UaVmPPlFxf-h1E6taC2SnPVrxPpHY,3775
 alita_sdk/tools/gitlab_org/api_wrapper.py,sha256=RYt5gIZffpukKAvynCMgHbXJy8c8ABMbm56ZVPttCbU,30130
 alita_sdk/tools/gmail/__init__.py,sha256=RkVWqVT335tpSUEVZUWqqPYMOYnxjkPmkBKLYdkpKto,887
 alita_sdk/tools/gmail/gmail_wrapper.py,sha256=t0IYM3zb77Ub8o9kv6HugNm_OoG5tN9T730hYmY8F-c,1312
@@ -280,7 +282,7 @@ alita_sdk/tools/pptx/pptx_wrapper.py,sha256=yyCYcTlIY976kJ4VfPo4dyxj4yeii9j9TWP6
 alita_sdk/tools/qtest/__init__.py,sha256=4vXCB9GSKNFeRTimSB7wklAnO-4reZgrw0Nw1_QuRKE,4070
 alita_sdk/tools/qtest/api_wrapper.py,sha256=cWXpmjjel9CYIXXjetJkARLYZXqvHufSghctTHN0ggc,22296
 alita_sdk/tools/qtest/tool.py,sha256=kKzNPS4fUC76WQQttQ6kdVANViHEvKE8Kf174MQiNYU,562
-alita_sdk/tools/rally/__init__.py,sha256=gw5EkDTeVQ77zGz7NSOkDuhycUpJ_DQC7g_bwwVn64k,4105
+alita_sdk/tools/rally/__init__.py,sha256=JvLt_hW_hC1WiCcwBwi1TlOH7QudJpM2z7XXGWYVaqI,3423
 alita_sdk/tools/rally/api_wrapper.py,sha256=mouzU6g0KML4UNapdk0k6Q0pU3MpJuWnNo71n9PSEHM,11752
 alita_sdk/tools/report_portal/__init__.py,sha256=716wEBO8-acrdXIZuuzurAYSZMHjRSVKu_vVFzstz38,3064
 alita_sdk/tools/report_portal/api_wrapper.py,sha256=-xiCeWyAnvDa9qLZ2cUr_KCR3feZ-UzUFw1PabkhrPU,9512
@@ -327,8 +329,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=JAeWf-RXohsxheUpT0iMDClc_izj-
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0AI_j27xVO5Gk5HQMFrqPTd4uvuVTpiZUicBrdfEpKg,2796
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.244.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.244.dist-info/METADATA,sha256=iHCZk1YXDHaduaIGgDIeuuy3xQtcRpBltoNlIPIMLlY,18897
-alita_sdk-0.3.244.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.244.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.244.dist-info/RECORD,,
+alita_sdk-0.3.245.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.245.dist-info/METADATA,sha256=4jO-XaHDi2ABmhFG4j5bJ9vz2yw4fxxWWKWRqgCyWgY,18897
+alita_sdk-0.3.245.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.245.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.245.dist-info/RECORD,,