alita-sdk 0.3.406__py3-none-any.whl → 0.3.407__py3-none-any.whl

alita_sdk/tools/sharepoint/authorization_helper.py

@@ -1,8 +1,10 @@
 from datetime import datetime, timezone
-from urllib.parse import unquote, urlparse
+from urllib.parse import unquote, urlparse, quote
 
 import jwt
 import requests
+from botocore.response import get_response
+
 
 class SharepointAuthorizationHelper:
 
@@ -57,6 +59,13 @@ class SharepointAuthorizationHelper:
         except jwt.InvalidTokenError:
             return False
 
+    def _validate_response(self, response, required_field, error_prefix=None):
+        if response.status_code != 200:
+            raise RuntimeError(f"{error_prefix or 'Request'} failed: {response.status_code} {response.text}")
+        json_data = response.json()
+        if required_field not in json_data:
+            raise KeyError(f"'{required_field}' missing in response")
+        return json_data[required_field]
 
     def generate_token_and_site_id(self, site_url: str) -> tuple[str, str]:
         try:
@@ -65,16 +74,10 @@ class SharepointAuthorizationHelper:
             site_path = parsed.path.strip('/')
             if not domain or not site_path:
                 raise ValueError(f"site_url missing domain or site path: {site_url}")
-            #
             app_name = domain.split('.')[0]
             openid_config_url = f"https://login.microsoftonline.com/{app_name}.onmicrosoft.com/v2.0/.well-known/openid-configuration"
             response = requests.get(openid_config_url)
-            if response.status_code != 200:
-                raise RuntimeError(f"Failed to get OpenID config: {response.status_code} {response.text}")
-            token_url = response.json().get("token_endpoint")
-            if not token_url:
-                raise KeyError("'token_endpoint' missing in OpenID config response")
-            #
+            token_url = self._validate_response(response, required_field="token_endpoint", error_prefix="OpenID config")
             token_data = {
                 "grant_type": "client_credentials",
                 "client_id": self.client_id,
@@ -82,21 +85,11 @@ class SharepointAuthorizationHelper:
                 "scope": "https://graph.microsoft.com/.default"
             }
             token_response = requests.post(token_url, data=token_data)
-            if token_response.status_code != 200:
-                raise RuntimeError(f"Failed to get access token: {token_response.status_code} {token_response.text}")
-            access_token = token_response.json().get("access_token")
-            if not access_token:
-                raise KeyError("'access_token' missing in token response")
-            #
+            access_token = self._validate_response(token_response, required_field="access_token", error_prefix="Token request")
             graph_site_url = f"https://graph.microsoft.com/v1.0/sites/{domain}:/{site_path}"
             headers = {"Authorization": f"Bearer {access_token}"}
             site_response = requests.get(graph_site_url, headers=headers)
-            if site_response.status_code != 200:
-                raise RuntimeError(f"Failed to get site info: {site_response.status_code} {site_response.text}")
-            site_id = site_response.json().get("id")
-            if not site_id:
-                raise KeyError("'id' missing in site response")
-            #
+            site_id = self._validate_response(site_response, required_field="id", error_prefix="Site info")
             return access_token, site_id
         except Exception as e:
             raise RuntimeError(f"Error while obtaining access_token and site_id: {e}")
@@ -111,50 +104,58 @@ class SharepointAuthorizationHelper:
             headers = {"Authorization": f"Bearer {access_token}"}
             drives_url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives"
             drives_response = requests.get(drives_url, headers=headers)
-            if drives_response.status_code != 200:
-                raise RuntimeError(f"Failed to get drives: {drives_response.status_code} {drives_response.text}")
-            drives_json = drives_response.json()
-            if "value" not in drives_json or not drives_json["value"]:
-                raise KeyError("'value' missing or empty in drives response")
-            drive_id = drives_json["value"][0].get("id")
-            drive_path = unquote(urlparse(drives_json["value"][0].get("webUrl")).path)
-            if not drive_id:
-                raise KeyError("'id' missing in drive object")
-            #
-            # Build the correct endpoint for folder or root
-            if folder_name:
-                # Validate folder_name for safe URL usage
-                if any(c in folder_name for c in ['..', '//', '\\']):
-                    raise ValueError(f"Unsafe folder_name: {folder_name}")
-                url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives/{drive_id}/root:/{folder_name}:/children?$top={limit_files}"
-            else:
-                url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives/{drive_id}/root/children?$top={limit_files}"
-            response = requests.get(url, headers=headers)
-            if response.status_code != 200:
-                raise RuntimeError(f"Failed to get files list: {response.status_code} {response.text}")
-            files_json = response.json()
-            if "value" not in files_json:
-                raise KeyError("'value' missing in files response")
-            #
+            drives = self._validate_response(drives_response, required_field="value", error_prefix="Drives request")
             result = []
-            for file in files_json["value"]:
-                file_name = file.get('name', '')
-                parts = [drive_path.strip('/'), folder_name.strip('/') if folder_name else '', file_name.strip('/')]
-                #
-                temp_props = {
-                    'Name': file_name,
-                    'Path': '/'.join([part for part in parts if part]),
-                    'Created': file.get('createdDateTime'),
-                    'Modified': file.get('lastModifiedDateTime'),
-                    'Link': file.get('webUrl'),
-                    'id': file.get('id')
-                }
-                if not all([temp_props['Name'], temp_props['Path'], temp_props['id']]):
-                    raise KeyError(f"Missing required file fields in: {file}")
-                result.append(temp_props)
-            # If API doesn't respect $top, slice in Python
-            if limit_files is not None:
-                result = result[:limit_files]
+            def _recurse_drive(drive_id, drive_path, parent_folder, limit_files):
+                # Escape folder_name for URL safety if present
+                if parent_folder:
+                    safe_folder_name = quote(parent_folder.strip('/'), safe="/")
+                    url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives/{drive_id}/root:/{safe_folder_name}:/children?$top={limit_files}"
+                else:
+                    url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives/{drive_id}/root/children?$top={limit_files}"
+                response = requests.get(url, headers=headers)
+                if response.status_code != 200:
+                    return []
+                files_json = response.json()
+                if "value" not in files_json:
+                    return []
+                files = []
+                for file in files_json["value"]:
+                    file_name = file.get('name', '')
+                    # Build full path reflecting nested folders
+                    if parent_folder:
+                        full_path = '/' + '/'.join([drive_path.strip('/'), parent_folder.strip('/'), file_name.strip('/')])
+                    else:
+                        full_path = '/' + '/'.join([drive_path.strip('/'), file_name.strip('/')])
+                    temp_props = {
+                        'Name': file_name,
+                        'Path': full_path,
+                        'Created': file.get('createdDateTime'),
+                        'Modified': file.get('lastModifiedDateTime'),
+                        'Link': file.get('webUrl'),
+                        'id': file.get('id')
+                    }
+                    if not all([temp_props['Name'], temp_props['Path'], temp_props['id']]):
+                        continue  # skip files with missing required fields
+                    if 'folder' in file:
+                        # Recursively extract files from this folder
+                        inner_folder = parent_folder + '/' + file_name if parent_folder else file_name
+                        inner_files = _recurse_drive(drive_id, drive_path, inner_folder, limit_files)
+                        files.extend(inner_files)
+                    else:
+                        files.append(temp_props)
+                    if limit_files is not None and len(result) + len(files) >= limit_files:
+                        return files[:limit_files - len(result)]
+                return files
+            for drive in drives:
+                drive_id = drive.get("id")
+                drive_path = unquote(urlparse(drive.get("webUrl")).path) if drive.get("webUrl") else ""
+                if not drive_id:
+                    continue  # skip drives without id
+                files = _recurse_drive(drive_id, drive_path, folder_name, limit_files)
+                result.extend(files)
+                if limit_files is not None and len(result) >= limit_files:
+                    return result[:limit_files]
             return result
         except Exception as e:
             raise RuntimeError(f"Error in get_files_list: {e}")
@@ -165,21 +166,22 @@ class SharepointAuthorizationHelper:
             headers = {"Authorization": f"Bearer {access_token}"}
             drives_url = f"https://graph.microsoft.com/v1.0/sites/{site_id}/drives"
             drives_response = requests.get(drives_url, headers=headers)
-            if drives_response.status_code != 200:
-                raise RuntimeError(f"Failed to get drives: {drives_response.status_code} {drives_response.text}")
-            drives_json = drives_response.json()
-            if "value" not in drives_json or not drives_json["value"]:
-                raise KeyError("'value' missing or empty in drives response")
-            drive_id = drives_json["value"][0].get("id")
-            if not drive_id:
-                raise KeyError("'id' missing in drive object")
+            drives = self._validate_response(drives_response, required_field="value", error_prefix="Drives request")
+            path = path.strip('/')
             #
-            # Build the endpoint for file content
-            url = f"https://graph.microsoft.com/v1.0/drives/{drive_id}/root:/{path}:/content"
-            response = requests.get(url, headers=headers)
-            if response.status_code != 200:
-                raise RuntimeError(f"Failed to get files list: {response.status_code} {response.text}")
-            #
-            return response.content
+            for drive in drives:
+                drive_path = unquote(urlparse(drive.get("webUrl")).path).strip('/')
+                if not drive_path or not path.startswith(drive_path):
+                    continue
+                drive_id = drive.get("id")
+                if not drive_id:
+                    continue
+                path = path.replace(drive_path, '').strip('/')
+                safe_path = quote(path, safe="")
+                url = f"https://graph.microsoft.com/v1.0/drives/{drive_id}/root:/{safe_path}:/content"
+                response = requests.get(url, headers=headers)
+                if response.status_code == 200:
+                    return response.content
+            raise RuntimeError(f"File '{path}' not found in any private or shared documents.")
         except Exception as e:
-            raise RuntimeError(f"Error in get_files_list: {e}")
+            raise RuntimeError(f"Error in get_file_content: {e}")

{alita_sdk-0.3.406.dist-info → alita_sdk-0.3.407.dist-info}/METADATA

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

@@ -318,7 +318,7 @@ alita_sdk/tools/servicenow/api_wrapper.py,sha256=WpH-bBLGFdhehs4g-K-WAkNuaD1CSrw
 alita_sdk/tools/servicenow/servicenow_client.py,sha256=Rdqfu-ll-qbnclMzChLZBsfXRDzgoX_FdeI2WLApWxc,3269
 alita_sdk/tools/sharepoint/__init__.py,sha256=5z2iSmm-0kbHKf70wN6OOgS4Px7tOzwkIpHXz0Vrbj4,4045
 alita_sdk/tools/sharepoint/api_wrapper.py,sha256=d8B0I4C9x8qt1dvmLjsfYnas98T_gXueJAWrNykZP0U,14075
-alita_sdk/tools/sharepoint/authorization_helper.py,sha256=-wNPxkKcg42o5Z3tkmzIr6rrrCMkA-Og2j6GbDjq7yU,9166
+alita_sdk/tools/sharepoint/authorization_helper.py,sha256=QvxWFBjYZfhI1h_KkSrDbRh8D5BlFX8xWDLmlIoO4mo,9569
 alita_sdk/tools/sharepoint/utils.py,sha256=fZ1YzAu5CTjKSZeslowpOPH974902S8vCp1Wu7L44LM,446
 alita_sdk/tools/slack/__init__.py,sha256=YiPAoRc6y6uVpfHl0K1Qi-flcyPlWFIMVcVbhicGWXY,3990
 alita_sdk/tools/slack/api_wrapper.py,sha256=5VrV7iSGno8ZcDzEHdGPNhInhtODGPPvAzoZ9W9iQWE,14009
@@ -353,8 +353,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.406.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.406.dist-info/METADATA,sha256=qKUH3FgZr8aK9zs2JOuh5VCEoRswwX5frgyDYb2Vszo,19071
-alita_sdk-0.3.406.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.406.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.406.dist-info/RECORD,,
+alita_sdk-0.3.407.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.407.dist-info/METADATA,sha256=5zQGxSdLNttnUzPpD95nDrhddTdPZmHyoLAtHQJJ0mo,19071
+alita_sdk-0.3.407.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.407.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.407.dist-info/RECORD,,