alita-sdk 0.3.340__tar.gz → 0.3.342__tar.gz

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.340
+Version: 0.3.342
 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.340 → alita_sdk-0.3.342}/alita_sdk/runtime/tools/vectorstore_base.py

@@ -136,7 +136,7 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
     vectorstore_type: Optional[str]  = None
     vectorstore_params: Optional[dict]  = None
     max_docs_per_add: int = 100
-    dataset: str = None
+    dataset: Optional[str] = None
     vectorstore: Any = None
     pg_helper: Any = None
     embeddings: Any = None

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -277,7 +277,7 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
                          test_steps_format: str = 'json'):
         """Creates a new test case in specified suite in Azure DevOps."""
         work_item_wrapper = AzureDevOpsApiWrapper(organization_url=self.organization_url,
-                                                  token=self.token.get_secret_value(), project=self.project)
+                                                  token=self.token.get_secret_value(), project=self.project, llm=self.llm)
         if test_steps_format == 'json':
             steps_xml = self.get_test_steps_xml(json.loads(test_steps))
         elif test_steps_format == 'xml':

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/alita_sdk/tools/figma/api_wrapper.py

@@ -17,6 +17,38 @@ from ..utils.available_tools_decorator import extend_with_parent_available_tools
 from ..utils.content_parser import load_content_from_bytes
 
 GLOBAL_LIMIT = 10000
+GLOBAL_RETAIN = ['id', 'name', 'type', 'document', 'children']
+GLOBAL_REMOVE = []
+GLOBAL_DEPTH_START = 4
+GLOBAL_DEPTH_END = 6
+EXTRA_PARAMS = (
+    Optional[Dict[str, Union[str, int, None]]],
+    Field(
+        description=(
+            "Additional parameters for customizing response processing:\n"
+            "- `limit`: Maximum size of the output in characters.\n"
+            "- `regexp`: Regex pattern to filter or clean the output.\n"
+            "- `fields_retain`: List of field names to always keep in the output, on levels starting from `depth_start`.\n"
+            "- `fields_remove`: List of field names to exclude from the output, unless also present in `fields_retain`.\n"
+            "- `depth_start`: The depth in the object hierarchy at which field filtering begins (fields are retained or removed).\n"
+            "- `depth_end`: The depth at which all fields are ignored and recursion stops.\n"
+            "Use these parameters to control the granularity and size of the returned data, especially for large or deeply nested objects."
+        ),
+        default={
+            "limit": GLOBAL_LIMIT, "regexp": None,
+            "fields_retain": GLOBAL_RETAIN, "fields_remove": GLOBAL_REMOVE,
+            "depth_start": GLOBAL_DEPTH_START, "depth_end": GLOBAL_DEPTH_END,
+        },
+        examples=[
+            {
+                "limit": "1000",
+                "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
+                "fields_retain": GLOBAL_RETAIN, "fields_remove": GLOBAL_REMOVE,
+                "depth_start": GLOBAL_DEPTH_START, "depth_end": GLOBAL_DEPTH_END,
+            }
+        ],
+    ),
+)
 
 
 class ArgsSchema(Enum):
@@ -36,19 +68,7 @@ class ArgsSchema(Enum):
                 examples=["8:6,1:7"],
             ),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     File = create_model(
         "FileNodes",
@@ -67,19 +87,7 @@ class ArgsSchema(Enum):
             Optional[str],
             Field(description="Sets version of file", default=None),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     FileKey = create_model(
         "FileKey",
@@ -90,19 +98,7 @@ class ArgsSchema(Enum):
                 examples=["Fp24FuzPwH0L74ODSrCnQo"],
             ),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     FileComment = create_model(
         "FileComment",
@@ -124,19 +120,7 @@ class ArgsSchema(Enum):
                 default=None,
             ),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     FileImages = create_model(
         "FileImages",
@@ -171,19 +155,7 @@ class ArgsSchema(Enum):
             Optional[str],
             Field(description="A specific version ID to use", default=None),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     TeamProjects = create_model(
         "TeamProjects",
@@ -194,19 +166,7 @@ class ArgsSchema(Enum):
                 examples=["1101853299713989222"],
             ),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
     ProjectFiles = create_model(
         "ProjectFiles",
@@ -217,19 +177,7 @@ class ArgsSchema(Enum):
                 examples=["55391681"],
             ),
         ),
-        extra_params=(
-            Optional[Dict[str, Union[str, int, None]]],
-            Field(
-                description="Additional parameters including limit and regex pattern to be removed from response",
-                default={"limit": GLOBAL_LIMIT, "regexp": None},
-                examples=[
-                    {
-                        "limit": "1000",
-                        "regexp": r'("strokes"|"fills")\s*:\s*("[^"]*"|[^\s,}\[]+)\s*(?=,|\}|\n)',
-                    }
-                ],
-            ),
-        ),
+        extra_params=EXTRA_PARAMS,
     )
 
 
@@ -238,6 +186,10 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
     oauth2: Optional[SecretStr] = Field(default=None)
     global_limit: Optional[int] = Field(default=GLOBAL_LIMIT)
     global_regexp: Optional[str] = Field(default=None)
+    global_fields_retain: Optional[List[str]] = GLOBAL_RETAIN
+    global_fields_remove: Optional[List[str]] = GLOBAL_REMOVE
+    global_depth_start: Optional[int] = GLOBAL_DEPTH_START
+    global_depth_end: Optional[int] = GLOBAL_DEPTH_END
     _client: Optional[FigmaPy] = PrivateAttr()
 
     def _base_loader(
@@ -441,7 +393,7 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
                 default=None)),
             'node_types_include': (Optional[List[str]], Field(
                 description="List type of nodes to include in index: i.e. ['FRAME', 'COMPONENT', 'RECTANGLE', 'COMPONENT_SET', 'INSTANCE', 'VECTOR', ...].",
-                default=['frame'])),
+                default=None)),
             'node_types_exclude': (Optional[List[str]], Field(
                 description="List type of nodes to exclude from index. It is applied only if node_types_include is not provided: i.e. ['FRAME', 'COMPONENT', 'RECTANGLE', 'COMPONENT_SET', 'INSTANCE', 'VECTOR', ...]",
                 default=None))
@@ -546,6 +498,53 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
                 }
             return obj
 
+        def process_fields(obj, fields_retain=None, fields_remove=None, depth_start=1, depth_end=2, depth=1):
+            """
+            Reduces a nested dictionary or list by retaining or removing specified fields at certain depths.
+
+            - At each level, starting from `depth_start`, only fields in `fields_retain` are kept; fields in `fields_remove` are excluded unless also retained.
+            - Recursion stops at `depth_end`, ignoring all fields at or beyond this depth.
+            - Tracks which fields were retained and removed during processing.
+            - Returns a JSON string of the reduced object, plus lists of retained and removed fields.
+            """
+            fields_retain = set(fields_retain or [])
+            fields_remove = set(fields_remove or []) - fields_retain # fields in remove have lower priority than in retain
+
+            retained = set()
+            removed = set()
+
+            def _process(o, d):
+                if depth_end is not None and d >= depth_end:
+                    return None  # Ignore keys at or beyond cut_depth
+                if isinstance(o, dict):
+                    result = {}
+                    for k, v in o.items():
+                        if k in fields_remove:
+                            removed.add(k)
+                            continue
+                        if d >= depth_start:
+                            if k in fields_retain:
+                                retained.add(k)
+                                result[k] = _process(v, d + 1)  # process recursively
+                            else:
+                                # else: skip keys not in retain/default/to_process
+                                removed.add(k) # remember skipped keys
+                        else:
+                            # retained.add(k) # remember retained keys
+                            result[k] = _process(v, d + 1)
+                    return result
+                elif isinstance(o, list):
+                    return [_process(item, d + 1) for item in o]
+                else:
+                    return o
+
+            new_obj = _process(obj, depth)
+            return {
+                "result": json.dumps(new_obj),
+                "retained_fields": list(retained),
+                "removed_fields": list(removed)
+            }
+
         def fix_trailing_commas(json_string):
             json_string = re.sub(r",\s*,+", ",", json_string)
             json_string = re.sub(r",\s*([\]}])", r"\1", json_string)
@@ -555,10 +554,12 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
         @functools.wraps(func)
         def wrapper(self, *args, **kwargs):
             extra_params = kwargs.pop("extra_params", {})
-
             limit = extra_params.get("limit", self.global_limit)
             regexp = extra_params.get("regexp", self.global_regexp)
-
+            fields_retain = extra_params.get("fields_retain", self.global_fields_retain)
+            fields_remove = extra_params.get("fields_remove", self.global_fields_remove)
+            depth_start = extra_params.get("depth_start", self.global_depth_start)
+            depth_end = extra_params.get("depth_end", self.global_depth_end)
             try:
                 limit = int(limit)
                 result = func(self, *args, **kwargs)
@@ -568,13 +569,26 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
                     return ToolException(
                         "Response result is empty. Check your input parameters or credentials"
                     )
-
                 if isinstance(result, (dict, list)):
-                    processed_result = simplified_dict(result)
-                    result = json.dumps(processed_result)
+                    raw_result = result
+                    processed_result = simplified_dict(raw_result)
+                    raw_str_result = json.dumps(processed_result)
+                    str_result = raw_str_result
+                    if regexp:
+                        regexp = re.compile(regexp)
+                        str_result = re.sub(regexp, "", raw_str_result)
+                        str_result = fix_trailing_commas(str_result)
+                    if len(str_result) > limit:
+                        reduced = process_fields(raw_result, fields_retain=fields_retain, fields_remove=fields_remove, depth_start=depth_start, depth_end=depth_end)
+                        note = (f"Size of the output exceeds limit {limit}. Data reducing has been applied. "
+                                f"Starting from the depth_start = {depth_start} the following object fields were removed: {reduced['removed_fields']}. "
+                                f"The following fields were retained: {reduced['retained_fields']}. "
+                                f"Starting from depth_end = {depth_end} all fields were ignored. "
+                                f"You can adjust fields_retain, fields_remove, depth_start, depth_end, limit and regexp parameters to get more precise output")
+                        return f"## NOTE:\n{note}.\n## Result: {reduced['result']}"[:limit]
+                    return str_result
                 else:
                     result = json.dumps(result)
-
                 if regexp:
                     regexp = re.compile(regexp)
                     result = re.sub(regexp, "", result)

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/alita_sdk/tools/jira/api_wrapper.py

@@ -1047,6 +1047,22 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
         """
         if isinstance(field_data, dict) and 'filename' in field_data and 'content' in field_data:
             return f"!{field_data['filename']}|alt={field_data['filename']}!"
+        if 'content' in field_data and isinstance(field_data['content'], list):
+            result = []
+            for content_item in field_data['content']:
+                if 'content' in content_item and isinstance(content_item['content'], list) and len(content_item['content']) > 0:
+                    if content_item.get('type') == 'mediaSingle':
+                        media = content_item['content'][0]
+                        attrs = media.get('attrs', {})
+                        if attrs.get('type') == 'file':
+                            alt = attrs.get('alt', '')
+                            image_str = f'!{alt}|alt="{alt}"!'
+                            result.append(image_str)
+                    elif content_item.get('type') == 'paragraph':
+                        result.append(content_item['content'][0].get('text', ''))
+                    else:
+                        result.append(self._extract_image_data(content_item))
+            return '\n'.join(result)
         return str(field_data)
 
     def get_field_with_image_descriptions(self, jira_issue_key: str, field_name: str, prompt: Optional[str] = None,
@@ -1223,18 +1239,22 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
             comment_created = comment.get('created', 'Unknown date')
 
             # Process the comment body by replacing image references with descriptions
-            processed_body = re.sub(image_pattern,
-                                    lambda match: self.process_image_match(match, comment_body, attachment_resolver, context_radius, prompt),
-                                    comment_body)
-
-            # Add the processed comment to our results
-            processed_comments.append({
-                "author": comment_author,
-                "created": comment_created,
-                "id": comment.get('id'),
-                "original_content": comment_body,
-                "processed_content": processed_body
-            })
+            try:
+                processed_body = re.sub(image_pattern,
+                                        lambda match: self.process_image_match(match, comment_body, attachment_resolver, context_radius, prompt),
+                                        comment_body)
+
+                # Add the processed comment to our results
+                processed_comments.append({
+                    "author": comment_author,
+                    "created": comment_created,
+                    "id": comment.get('id'),
+                    "original_content": comment_body,
+                    "processed_content": processed_body
+                })
+            except Exception as e:
+                logger.error(f"Error processing image references in comment: {str(e)}")
+                # TODO process comment in api 3 format
         return processed_comments
 
     def get_comments_with_image_descriptions(self, jira_issue_key: str, prompt: Optional[str] = None, context_radius: int = 500):
@@ -1419,7 +1439,7 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
         if validate_query is not None:
             params["validateQuery"] = validate_query
 
-        url = self._client.resource_url("search")
+        url = self._client.resource_url("search/jql" if self.api_version == '3' else "search")
 
         while True:
             params["startAt"] = int(start)
@@ -1437,6 +1457,8 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
                 break
             if not response["issues"]:
                 break
+            if len(issues) < limit:
+                break
             start += len(issues)
 
     def _process_issue_for_indexing(self, issue: dict, fields_to_index=None) -> Document:

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/alita_sdk/tools/postman/__init__.py

@@ -55,8 +55,7 @@ class PostmanToolkit(BaseToolkit):
             name,
             postman_configuration=(Optional[PostmanConfiguration], Field(description="Postman Configuration",
                                                                          json_schema_extra={'configuration_types': ['postman']})),
-            collection_id=(str, Field(description="Default collection ID", json_schema_extra={
-                           'toolkit_name': True, 'max_toolkit_length': PostmanToolkit.toolkit_max_length})),
+            collection_id=(str, Field(description="Default collection ID")),
             environment_config=(dict, Field(
                 description="JSON configuration for request execution (auth headers, project IDs, base URLs, etc.)",
                 default={})),

{alita_sdk-0.3.340 → alita_sdk-0.3.342}/alita_sdk.egg-info/PKG-INFO

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

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