alita-sdk 0.3.297__tar.gz → 0.3.299__tar.gz

{alita_sdk-0.3.297 → alita_sdk-0.3.299}/PKG-INFO

@@ -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 → alita_sdk-0.3.299}/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-0.3.297 → alita_sdk-0.3.299}/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-0.3.297 → alita_sdk-0.3.299}/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 → alita_sdk-0.3.299}/alita_sdk.egg-info/PKG-INFO

@@ -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 → alita_sdk-0.3.299}/pyproject.toml

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