alita-sdk 0.3.297__py3-none-any.whl → 0.3.299__py3-none-any.whl

alita_sdk/runtime/clients/client.py

@@ -81,20 +81,18 @@ class AlitaClient:
     def mcp_tool_call(self, params: dict[str, Any]):
         if user_id := self._get_real_user_id():
             url = f"{self.mcp_tools_call}/{user_id}"
-
-            if 'args' in params.get('params', {}).get('arguments', {}):
-                args_obj = params['params']['arguments']['args']
-                if hasattr(args_obj, "__dict__"):
-                    params['params']['arguments'] = vars(args_obj)
-                else:
-                    params['params']['arguments'] = args_obj
-
+            #
+            # This loop iterates over each key-value pair in the arguments dictionary,
+            # and if a value is a Pydantic object, it replaces it with its dictionary representation using .dict().
+            for arg_name, arg_value in params.get('params', {}).get('arguments', {}).items():
+                if hasattr(arg_value, "dict") and callable(arg_value.dict):
+                    params['params']['arguments'][arg_name] = arg_value.dict()
+            #
             response = requests.post(url, headers=self.headers, json=params, verify=False)
             try:
-                data = response.json()
+                return response.json()
             except (ValueError, TypeError):
-                data = response.text
-            return data
+                return response.text
         else:
             return f"Error: Could not determine user ID for MCP tool call"
 

alita_sdk/runtime/langchain/document_loaders/AlitaJSONLoader.py

@@ -68,6 +68,9 @@ class AlitaJSONLoader(BaseLoader):
         else:
             data_dict = content_json
         chunks = RecursiveJsonSplitter(max_chunk_size=self.max_tokens).split_json(json_data=data_dict)
+        chunk_id = 1
         for chunk in chunks:
-            metadata = {"source": str(self.file_path) if hasattr(self, 'file_path') else self.file_name}
+            metadata = {"source": str(self.file_path) if hasattr(self, 'file_path') else self.file_name,
+                        "chunk_id": chunk_id}
+            chunk_id+=1
             yield Document(page_content=json.dumps(chunk), metadata=metadata)

alita_sdk/tools/utils/content_parser.py

@@ -238,7 +238,6 @@ def process_content_by_type(document: Document, content, extension_source: str,
                 loader_kwargs.pop(LoaderProperties.PROMPT_DEFAULT.value)
                 loader_kwargs[LoaderProperties.PROMPT.value] = image_processing_prompt
             loader = loader_cls(file_path=temp_file_path, **loader_kwargs)
-            counter = 1
             try:
                 chunks = loader.load()
             except Exception as e:
@@ -250,14 +249,9 @@ def process_content_by_type(document: Document, content, extension_source: str,
                 )
                 return
             for chunk in chunks:
-                if 'chunk_id' not in chunk.metadata:
-                    chunk.metadata['chunk_id'] = counter
-                document_metadata = document.metadata.copy()
-                document_metadata['id'] = f"{document.metadata['id']}_{chunk.metadata['chunk_id']}"
-                counter+=1
                 yield Document(
                     page_content=sanitize_for_postgres(chunk.page_content),
-                    metadata={**document_metadata, **chunk.metadata}
+                    metadata={**document.metadata, **chunk.metadata}
                 )
     finally:
         if temp_file_path and os.path.exists(temp_file_path):

{alita_sdk-0.3.297.dist-info → alita_sdk-0.3.299.dist-info}/METADATA

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

@@ -36,7 +36,7 @@ alita_sdk/configurations/zephyr_essential.py,sha256=tUIrh-PRNvdrLBj6rJXqlF-h6oaM
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=TPvROw1qu4IyUEGuf7x40IKRpb5eFZpYGN3-8LfQE0M,3461
-alita_sdk/runtime/clients/client.py,sha256=fRx-vUJiHEIO6t3svS1VQrLOczNzTcrZav7r-EUDWCc,43123
+alita_sdk/runtime/clients/client.py,sha256=q9dRNrwDQJmb3dlFa6iWp-rB-XsgNLJh7gtrFhp9V1w,43240
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -59,7 +59,7 @@ alita_sdk/runtime/langchain/document_loaders/AlitaDocxMammothLoader.py,sha256=9h
 alita_sdk/runtime/langchain/document_loaders/AlitaExcelLoader.py,sha256=YBFYikrOEITfIavU0Xu7BQSNvPCFKzcmbJ_VDeQ6KdI,3078
 alita_sdk/runtime/langchain/document_loaders/AlitaGitRepoLoader.py,sha256=5WXGcyHraSVj3ANHj_U6X4EDikoekrIYtS0Q_QqNIng,2608
 alita_sdk/runtime/langchain/document_loaders/AlitaImageLoader.py,sha256=QwgBJE-BvOasjgT1hYHZc0MP0F_elirUjSzKixoM6fY,6610
-alita_sdk/runtime/langchain/document_loaders/AlitaJSONLoader.py,sha256=1mGZjltnqsSXkp1Jw-lQroyNFiCPpjb9ZbdoqOlqPeU,3354
+alita_sdk/runtime/langchain/document_loaders/AlitaJSONLoader.py,sha256=Nav2cgCQKOHQi_ZgYYn_iFdP_Os56KVlVR5nHGXecBc,3445
 alita_sdk/runtime/langchain/document_loaders/AlitaJiraLoader.py,sha256=M2q3YThkps0yAZOjfoLcyE7qycVTYKcXEGtpmp0N6C8,10950
 alita_sdk/runtime/langchain/document_loaders/AlitaMarkdownLoader.py,sha256=RGHDfleYTn7AAc3H-yFZrjm06L0Ux14ZtEJpFlVBNCA,2474
 alita_sdk/runtime/langchain/document_loaders/AlitaPDFLoader.py,sha256=toXdQbT9TuBCdB4t62t2cPalBY_2RZy2lqKSMU7YVhw,3386
@@ -327,7 +327,7 @@ alita_sdk/tools/testrail/__init__.py,sha256=0kETjWKLU7R6mugBWsjwEUsh10pipbAeNSGJ
 alita_sdk/tools/testrail/api_wrapper.py,sha256=xKQbjwL602J55KZiAdMcMtsuzK2jky0DUcrrdsazj0A,32981
 alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
 alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
-alita_sdk/tools/utils/content_parser.py,sha256=duLnLVdjAfswpIQL763J1JU1dgggCshwKAZBMTjCGzE,14044
+alita_sdk/tools/utils/content_parser.py,sha256=8cq-6dHYp-jEYU1Yt3P6rVedVgVaOfgnNIyQ2qU4Yrk,13722
 alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=ypBEAkFRGHv5edW0N9rdo1yKurNGQ4pRVEWtrN_7SeA,17656
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/xray/__init__.py,sha256=AsHDvRgyD-6vvGyAyQDPWHbOD2WoMJ5Llt029bLuu6U,4277
@@ -349,8 +349,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=2M6yFoOmfH5BwxMwTLQO76yD0Ydfv
 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.297.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.297.dist-info/METADATA,sha256=Kxb0lVoqjdrMfuKm2VhjpYr9FGpDnAPkw2C-yOLeTbk,18897
-alita_sdk-0.3.297.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.297.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.297.dist-info/RECORD,,
+alita_sdk-0.3.299.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.299.dist-info/METADATA,sha256=NGZZBnw7yBPOMT0jmpCiCbOE59ln1xZ3YVulQ--CCLo,18897
+alita_sdk-0.3.299.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.299.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.299.dist-info/RECORD,,