alita-sdk 0.3.417__tar.gz → 0.3.418__tar.gz

{alita_sdk-0.3.417 → alita_sdk-0.3.418}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.417
+Version: 0.3.418
 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.417 → alita_sdk-0.3.418}/alita_sdk/runtime/clients/sandbox_client.py

@@ -143,6 +143,7 @@ class SandboxClient:
         self.configurations_url = f'{self.base_url}{self.api_path}/integrations/integrations/default/{self.project_id}?section=configurations&unsecret=true'
         self.ai_section_url = f'{self.base_url}{self.api_path}/integrations/integrations/default/{self.project_id}?section=ai'
         self.image_generation_url = f'{self.base_url}{self.llm_path}/images/generations'
+        self.auth_user_url = f'{self.base_url}{self.api_path}/auth/user'
         self.configurations: list = configurations or []
         self.model_timeout = kwargs.get('model_timeout', 120)
         self.model_image_generation = kwargs.get('model_image_generation')
@@ -363,3 +364,10 @@ class SandboxClient:
         url = f'{self.artifact_url}/{bucket_name}'
         data = requests.delete(url, headers=self.headers, verify=False, params={'filename': quote(artifact_name)})
         return self._process_requst(data)
+
+    def get_user_data(self) -> Dict[str, Any]:
+        resp = requests.get(self.auth_user_url, headers=self.headers, verify=False)
+        if resp.ok:
+            return resp.json()
+        logger.error(f'Failed to fetch user data: {resp.status_code} - {resp.text}')
+        raise ApiDetailsRequestError(f'Failed to fetch user data with status code {resp.status_code}.')

{alita_sdk-0.3.417 → alita_sdk-0.3.418}/alita_sdk/runtime/toolkits/tools.py

@@ -48,23 +48,24 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
                     selected_tools=tool['settings']['selected_tools'],
                     toolkit_name=tool.get('toolkit_name', '') or tool.get('name', '')
                 ).get_tools())
-            elif tool['type'] == 'application' and tool.get('agent_type', '') != 'pipeline' :
+            elif tool['type'] == 'application':
                 tools.extend(ApplicationToolkit.get_toolkit(
                     alita_client,
                     application_id=int(tool['settings']['application_id']),
                     application_version_id=int(tool['settings']['application_version_id']),
                     selected_tools=[]
                 ).get_tools())
-            elif tool['type'] == 'application' and tool.get('agent_type', '') == 'pipeline':
-                # static get_toolkit returns a list of CompiledStateGraph stubs
-                tools.extend(SubgraphToolkit.get_toolkit(
-                    alita_client,
-                    application_id=int(tool['settings']['application_id']),
-                    application_version_id=int(tool['settings']['application_version_id']),
-                    app_api_key=alita_client.auth_token,
-                    selected_tools=[],
-                    llm=llm
-                ))
+                # backward compatibility for pipeline application type as subgraph node
+                if tool.get('agent_type', '') == 'pipeline':
+                    # static get_toolkit returns a list of CompiledStateGraph stubs
+                    tools.extend(SubgraphToolkit.get_toolkit(
+                        alita_client,
+                        application_id=int(tool['settings']['application_id']),
+                        application_version_id=int(tool['settings']['application_version_id']),
+                        app_api_key=alita_client.auth_token,
+                        selected_tools=[],
+                        llm=llm
+                    ))
             elif tool['type'] == 'memory':
                 tools += MemoryToolkit.get_toolkit(
                     namespace=tool['settings'].get('namespace', str(tool['id'])),

{alita_sdk-0.3.417 → alita_sdk-0.3.418}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.417
+Version: 0.3.418
 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.417 → alita_sdk-0.3.418}/pyproject.toml

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