alita-sdk 0.3.326__py3-none-any.whl → 0.3.328__py3-none-any.whl

alita_sdk/runtime/langchain/langraph_agent.py

@@ -248,19 +248,36 @@ class StateModifierNode(Runnable):
 
         # Collect input variables from state
         input_data = {}
+        
         for var in self.input_variables:
             if var in state:
                 input_data[var] = state.get(var)
-
+        type_of_output = type(state.get(self.output_variables[0])) if self.output_variables else None
         # Render the template using Jinja
         import json
+        import base64
         from jinja2 import Environment
 
         def from_json(value):
-            return json.loads(value)
+            """Convert JSON string to Python object"""
+            try:
+                return json.loads(value)
+            except (json.JSONDecodeError, TypeError) as e:
+                logger.warning(f"Failed to parse JSON value: {e}")
+                return value
+        
+        def base64_to_string(value):
+            """Convert base64 encoded string to regular string"""
+            try:
+                return base64.b64decode(value).decode('utf-8')
+            except Exception as e:
+                logger.warning(f"Failed to decode base64 value: {e}")
+                return value
+        
 
         env = Environment()
         env.filters['from_json'] = from_json
+        env.filters['base64ToString'] = base64_to_string
         
         template = env.from_string(self.template)
         rendered_message = template.render(**input_data)
@@ -269,7 +286,35 @@ class StateModifierNode(Runnable):
         if len(self.output_variables) > 0:
             # Use the first output variable to store the rendered content
             output_var = self.output_variables[0]
-            result[output_var] = rendered_message
+            
+            # Convert rendered_message to the appropriate type
+            if type_of_output is not None:
+                try:
+                    if type_of_output == dict:
+                        result[output_var] = json.loads(rendered_message) if isinstance(rendered_message, str) else dict(rendered_message)
+                    elif type_of_output == list:
+                        result[output_var] = json.loads(rendered_message) if isinstance(rendered_message, str) else list(rendered_message)
+                    elif type_of_output == int:
+                        result[output_var] = int(rendered_message)
+                    elif type_of_output == float:
+                        result[output_var] = float(rendered_message)
+                    elif type_of_output == str:
+                        result[output_var] = str(rendered_message)
+                    elif type_of_output == bool:
+                        if isinstance(rendered_message, str):
+                            result[output_var] = rendered_message.lower() in ('true', '1', 'yes', 'on')
+                        else:
+                            result[output_var] = bool(rendered_message)
+                    elif type_of_output == type(None):
+                        result[output_var] = None
+                    else:
+                        # Fallback to string if type is not recognized
+                        result[output_var] = str(rendered_message)
+                except (ValueError, TypeError, json.JSONDecodeError) as e:
+                    logger.warning(f"Failed to convert rendered_message to {type_of_output.__name__}: {e}. Using string fallback.")
+                    result[output_var] = str(rendered_message)
+            else:
+                result[output_var] = rendered_message
 
         # Clean up specified variables (make them empty, not delete)
         

alita_sdk/runtime/langchain/utils.py

@@ -121,6 +121,8 @@ def parse_type(type_str):
     """Parse a type string into an actual Python type."""
     try:
         # Evaluate the type string using builtins and imported modules
+        if type_str == 'number':
+            type_str = 'int'
         return eval(type_str, {**vars(builtins), **globals()})
     except Exception as e:
         print(f"Error parsing type: {e}")
@@ -138,6 +140,7 @@ def create_state(data: Optional[dict] = None):
             state_dict[key] = Annotated[list[AnyMessage], add_messages]
         elif value in ['str', 'int', 'float', 'bool', 'list', 'dict', 'number']:
             state_dict[key] = parse_type(value)
+    logger.debug(f"Created state: {state_dict}")
     return TypedDict('State', state_dict)
 
 def create_typed_dict_from_yaml(data):

alita_sdk/tools/figma/api_wrapper.py

@@ -307,76 +307,110 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
         else:
             raise ValueError("You must provide at least project_id or file_keys_include.")
 
+    def has_image_representation(self, node):
+        node_type = node.get('type', '').lower()
+        default_images_types = [
+            'image', 'canvas', 'frame', 'vector', 'table', 'slice', 'sticky', 'shape_with_text', 'connector'
+        ]
+        # filter nodes of type which has image representation
+        # or rectangles with image as background
+        if (node_type in default_images_types
+                or (node_type == 'rectangle' and 'fills' in node and any(
+                    fill.get('type') == 'IMAGE' for fill in node['fills'] if isinstance(fill, dict)))):
+            return True
+        return False
+
+    def get_texts_recursive(self, node):
+        texts = []
+        node_type = node.get('type', '').lower()
+        if node_type == 'text':
+            texts.append(node.get('characters', ''))
+        if 'children' in node:
+            for child in node['children']:
+                texts.extend(self.get_texts_recursive(child))
+        return texts
+
     def _process_document(self, document: Document) -> Generator[Document, None, None]:
         file_key = document.metadata.get('id', '')
         self._log_tool_event(f"Loading details (images) for `{file_key}`")
-        #
         figma_pages = self._client.get_file(file_key).document.get('children', [])
         node_ids_include = document.metadata.pop('figma_pages_include', [])
         node_ids_exclude = document.metadata.pop('figma_pages_exclude', [])
-        node_types_include = [t.lower() for t in document.metadata.pop('figma_nodes_include', [])]
-        node_types_exclude = [t.lower() for t in document.metadata.pop('figma_nodes_exclude', [])]
+        node_types_include = [t.strip().lower() for t in document.metadata.pop('figma_nodes_include', [])]
+        node_types_exclude = [t.strip().lower() for t in document.metadata.pop('figma_nodes_exclude', [])]
         self._log_tool_event(f"Included pages: {node_ids_include}. Excluded pages: {node_ids_exclude}.")
         if node_ids_include:
             figma_pages = [node for node in figma_pages if ('id' in node and node['id'].replace(':', '-') in node_ids_include)]
         elif node_ids_exclude:
             figma_pages = [node for node in figma_pages if ('id' in node and node['id'].replace(':', '-') not in node_ids_exclude)]
 
-        # if node_types_include is not provided, default to 'frame'
-        # to avoid downloading too many images and nodes which co=annot be rendered as images
-        if not node_types_include:
-            node_types_include = ['frame']
-
-        node_ids = [
-            child['id']
-            for page in figma_pages
-            if 'children' in page
-            for child in page['children']
-            if 'id' in child
-            and (
-                (node_types_include and child.get('type').lower() in node_types_include)
-                or (node_types_exclude and child.get('type').lower() not in node_types_exclude)
-                or (not node_types_include and not node_types_exclude)
-            )
-        ]
-
-        if not node_ids:
-            yield from ()
-            return
-
-        images = self._client.get_file_images(file_key, node_ids).images or {}
-        total_images = len(images)
-        if total_images == 0:
-            logging.info(f"No images found for file {file_key}.")
-            return
-        progress_step = max(1, total_images // 10)
-        for idx, (node_id, image_url) in enumerate(images.items(), 1):
-            if not image_url:
-                logging.warning(f"Image URL not found for node_id {node_id} in file {file_key}. Skipping.")
-                continue
-            response = requests.get(image_url)
-            if response.status_code == 200:
-                content_type = response.headers.get('Content-Type', '')
-                if 'text/html' not in content_type.lower():
-                    extension = f".{content_type.split('/')[-1]}" if content_type.startswith('image') else '.txt'
-                    page_content = load_content_from_bytes(
-                        file_content=response.content,
-                        extension=extension, llm=self.llm)
+        image_nodes = []
+        text_nodes = {}
+        for page in figma_pages:
+            for node in page.get('children', []):
+                # filter by node_type if specified any include or exclude
+                node_type = node.get('type', '').lower()
+                include = node_types_include and node_type in node_types_include
+                exclude = node_types_exclude and node_type not in node_types_exclude
+                no_filter = not node_types_include and not node_types_exclude
+
+                if include or exclude or no_filter:
+                    node_id = node.get('id')
+                    if node_id:
+                        if self.has_image_representation(node):
+                            image_nodes.append(node['id'])
+                        else:
+                            text_nodes[node['id']] = self.get_texts_recursive(node)
+        # process image nodes
+        if image_nodes:
+            images = self._client.get_file_images(file_key, image_nodes).images or {}
+            total_images = len(images)
+            if total_images == 0:
+                logging.info(f"No images found for file {file_key}.")
+                return
+            progress_step = max(1, total_images // 10)
+            for idx, (node_id, image_url) in enumerate(images.items(), 1):
+                if not image_url:
+                    logging.warning(f"Image URL not found for node_id {node_id} in file {file_key}. Skipping.")
+                    continue
+                response = requests.get(image_url)
+                if response.status_code == 200:
+                    content_type = response.headers.get('Content-Type', '')
+                    if 'text/html' not in content_type.lower():
+                        extension = f".{content_type.split('/')[-1]}" if content_type.startswith('image') else '.txt'
+                        page_content = load_content_from_bytes(
+                            file_content=response.content,
+                            extension=extension, llm=self.llm)
+                        yield Document(
+                            page_content=page_content,
+                            metadata={
+                                'id': node_id,
+                                'updated_on': document.metadata.get('updated_on', ''),
+                                'file_key': file_key,
+                                'node_id': node_id,
+                                'image_url': image_url,
+                                'type': 'image'
+                            }
+                        )
+                if idx % progress_step == 0 or idx == total_images:
+                    percent = int((idx / total_images) * 100)
+                    msg = f"Processed {idx}/{total_images} images ({percent}%) for file {file_key}."
+                    logging.info(msg)
+                    self._log_tool_event(msg)
+        # process text nodes
+        if text_nodes:
+            for node_id, texts in text_nodes.items():
+                if texts:
                     yield Document(
-                        page_content=page_content,
+                        page_content="\n".join(texts),
                         metadata={
                             'id': node_id,
                             'updated_on': document.metadata.get('updated_on', ''),
                             'file_key': file_key,
                             'node_id': node_id,
-                            'image_url': image_url
+                            'type': 'text'
                         }
                     )
-            if idx % progress_step == 0 or idx == total_images:
-                percent = int((idx / total_images) * 100)
-                msg = f"Processed {idx}/{total_images} images ({percent}%) for file {file_key}."
-                logging.info(msg)
-                self._log_tool_event(msg)
 
     def _remove_metadata_keys(self):
         return super()._remove_metadata_keys() + ['figma_pages_include', 'figma_pages_exclude', 'figma_nodes_include', 'figma_nodes_exclude']

alita_sdk/tools/jira/api_wrapper.py

@@ -563,12 +563,14 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
         Use the appropriate issue link type (e.g., "Test", "Relates", "Blocks").
         If we use "Test" linktype, the test is inward issue, the story/other issue is outward issue.."""
 
+        comment = "This test is linked to the story."
+        comment_body = {"content": [{"content": [{"text": comment,"type": "text"}],"type": "paragraph"}],"type": "doc","version": 1} if self.api_version == "3" else comment
         link_data = {
             "type": {"name": f"{linktype}"},
             "inwardIssue": {"key": f"{inward_issue_key}"},
             "outwardIssue": {"key": f"{outward_issue_key}"},
             "comment": {
-                "body": "This test is linked to the story."
+                "body": comment_body
             }
         }
         self._client.create_issue_link(link_data)
@@ -706,6 +708,8 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
     def add_comments(self, issue_key: str, comment: str):
         """ Add a comment to a Jira issue."""
         try:
+            if self.api_version == '3':
+                comment = {"content": [{"content": [{"text": comment,"type": "text"}],"type": "paragraph"}],"type": "doc","version": 1}
             self._client.issue_add_comment(issue_key, comment)
             issue_url = f"{self._client.url}browse/{issue_key}"
             output = f"Done. Comment is added for issue {issue_key}. You can view it at {issue_url}"

alita_sdk/tools/utils/__init__.py

@@ -55,6 +55,8 @@ def parse_type(type_str):
     """Parse a type string into an actual Python type."""
     try:
         # Evaluate the type string using builtins and imported modules
+        if type_str == 'number':
+            type_str = 'int'
         return eval(type_str, {**vars(builtins), **globals()})
     except Exception as e:
         print(f"Error parsing type: {e}")

{alita_sdk-0.3.326.dist-info → alita_sdk-0.3.328.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.326
+Version: 0.3.328
 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
@@ -74,8 +74,7 @@ Requires-Dist: paramiko==3.3.1; extra == "tools"
 Requires-Dist: pygithub==2.3.0; extra == "tools"
 Requires-Dist: python-gitlab==4.5.0; extra == "tools"
 Requires-Dist: gitpython==3.1.43; extra == "tools"
-Requires-Dist: atlassian-python-api~=3.41; extra == "tools"
-Requires-Dist: atlassian_python_api==3.41.16; extra == "tools"
+Requires-Dist: atlassian-python-api~=4.0.7; extra == "tools"
 Requires-Dist: jira==3.8.0; extra == "tools"
 Requires-Dist: qtest-swagger-client==0.0.3; extra == "tools"
 Requires-Dist: testrail-api==1.13.2; extra == "tools"

{alita_sdk-0.3.326.dist-info → alita_sdk-0.3.328.dist-info}/RECORD

@@ -44,11 +44,11 @@ alita_sdk/runtime/langchain/assistant.py,sha256=1Eq8BIefp8suhbC9CssoOXtC-plkemoU
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=74W8noL61DPng8U0eKCn6PZsHHDcBCFgOy7zwc_JEx0,45027
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=z_Bontl600nV7ombsomKXtRCuwCJc-5b5P91wapHYo4,47523
 alita_sdk/runtime/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/runtime/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/runtime/langchain/store_manager.py,sha256=i8Fl11IXJhrBXq1F1ukEVln57B1IBe-tqSUvfUmBV4A,2218
-alita_sdk/runtime/langchain/utils.py,sha256=Npferkn10dvdksnKzLJLBI5bNGQyVWTBwqp3vQtUqmY,6631
+alita_sdk/runtime/langchain/utils.py,sha256=r5X00cgUjx4OwjwxHCnSyO-AbAw2gPVvBx2KfQr87lw,6742
 alita_sdk/runtime/langchain/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/langchain/agents/xml_chat.py,sha256=Mx7PK5T97_GrFCwHHZ3JZP42S7MwtUzV0W-_8j6Amt8,6212
 alita_sdk/runtime/langchain/document_loaders/AlitaBDDScenariosLoader.py,sha256=4kFU1ijrM1Jw7cywQv8mUiBHlE6w-uqfzSZP4hUV5P4,3771
@@ -235,7 +235,7 @@ alita_sdk/tools/custom_open_api/api_wrapper.py,sha256=sDSFpvEqpSvXHGiBISdQQcUecf
 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=W6vIMMkZI2Lmpg6_CRRV3oadaIbVI-qTLmKUh6enqWs,4509
-alita_sdk/tools/figma/api_wrapper.py,sha256=SFuvjhxYgey1qGsO9sakFpY1bK1RSzgAH-uJsAp7FnE,27477
+alita_sdk/tools/figma/api_wrapper.py,sha256=-vsIy0Y4UFJBjKumRAOqmgS7pEXVn0UjaNSNN7pFsLs,29351
 alita_sdk/tools/github/__init__.py,sha256=2rHu0zZyZGnLC5CkHgDIhe14N9yCyaEfrrt7ydH8478,5191
 alita_sdk/tools/github/api_wrapper.py,sha256=uDwYckdnpYRJtb0uZnDkaz2udvdDLVxuCh1tSwspsiU,8411
 alita_sdk/tools/github/github_client.py,sha256=nxnSXsDul2PPbWvYZS8TmAFFmR-5ALyakNoV5LN2D4U,86617
@@ -260,7 +260,7 @@ alita_sdk/tools/google/bigquery/tool.py,sha256=Esf9Hsp8I0e7-5EdkFqQ-bid0cfrg-bfS
 alita_sdk/tools/google_places/__init__.py,sha256=QtmBCI0bHDK79u4hsCSWFcUihu-h4EmPSh9Yll7zz3w,3590
 alita_sdk/tools/google_places/api_wrapper.py,sha256=7nZly6nk4f4Tm7s2MVdnnwlb-1_WHRrDhyjDiqoyPjA,4674
 alita_sdk/tools/jira/__init__.py,sha256=G-9qnOYKFWM_adG0QFexh5-2pj_WaxIxxZanB3ARFqI,6339
-alita_sdk/tools/jira/api_wrapper.py,sha256=kYd6edldpjaEQNKvJjVaPQt8A22K6f3DMZVJk6Tylsc,80916
+alita_sdk/tools/jira/api_wrapper.py,sha256=-juLuxeOCyDKb_-ZS8eTOeUJWEHKcCiBlXyFY2vbL4Q,81296
 alita_sdk/tools/keycloak/__init__.py,sha256=0WB9yXMUUAHQRni1ghDEmd7GYa7aJPsTVlZgMCM9cQ0,3050
 alita_sdk/tools/keycloak/api_wrapper.py,sha256=cOGr0f3S3-c6tRDBWI8wMnetjoNSxiV5rvC_0VHb8uw,3100
 alita_sdk/tools/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -325,7 +325,7 @@ alita_sdk/tools/testio/__init__.py,sha256=NEvQtzsffqAXryaffVk0GpdcxZQ1AMkfeztnxH
 alita_sdk/tools/testio/api_wrapper.py,sha256=BvmL5h634BzG6p7ajnQLmj-uoAw1gjWnd4FHHu1h--Q,21638
 alita_sdk/tools/testrail/__init__.py,sha256=Xg4nVjULL_D8JpIXLYXppnwUfGF4-lguFwKHmP5VwxM,4696
 alita_sdk/tools/testrail/api_wrapper.py,sha256=PKhtf04C6PFDexGCAJm-hjA9Gpu4crx6EXKT5K-b_Pk,32985
-alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
+alita_sdk/tools/utils/__init__.py,sha256=W9rCCUPtHCP5nGAbWp0n5jaNA84572aiRoqKneBnaS4,3330
 alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
 alita_sdk/tools/utils/content_parser.py,sha256=7k5Ddv3Nzp3UoocgslwwSXi1G9ZR7sXzj6593IDeOcM,14063
 alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=ypBEAkFRGHv5edW0N9rdo1yKurNGQ4pRVEWtrN_7SeA,17656
@@ -349,8 +349,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 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.326.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.326.dist-info/METADATA,sha256=gGLnc0mht7c9z_x54ejIbf1umM6I3BtSb-jUWuWW1vc,18897
-alita_sdk-0.3.326.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.326.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.326.dist-info/RECORD,,
+alita_sdk-0.3.328.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.328.dist-info/METADATA,sha256=jRVOPK8heveuCV9P6VcjO4DE5h6kMdu2TUXKU6VOH7Q,18835
+alita_sdk-0.3.328.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.328.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.328.dist-info/RECORD,,