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

alita_sdk/tools/sharepoint/api_wrapper.py

@@ -156,11 +156,28 @@ class SharepointApiWrapper(NonCodeIndexerToolkit):
             self._client.load(file).execute_query()
 
             file_content = file.read()
+            file_name = file.name
             self._client.execute_query()
         except Exception as e:
-            logging.error(f"Failed to load file from SharePoint: {e}. Path: {path}. Please, double check file name and path.")
-            return ToolException("File not found. Please, check file name and path.")
-        return parse_file_content(file_name=file.name,
+            # attempt to get via graph api
+            try:
+                # attempt to get files via graph api
+                from .authorization_helper import SharepointAuthorizationHelper
+                auth_helper = SharepointAuthorizationHelper(
+                    client_id=self.client_id,
+                    client_secret=self.client_secret.get_secret_value(),
+                    tenant="",  # optional for graph api
+                    scope="",  # optional for graph api
+                    token_json="",  # optional for graph api
+                )
+                file_content = auth_helper.get_file_content(self.site_url, path)
+                file_name = path.split('/')[-1]
+            except Exception as graph_e:
+                logging.error(f"Failed to load file from SharePoint via base api: {e}. Path: {path}. Please, double check file name and path.")
+                logging.error(f"Failed to load file from SharePoint via graph api: {graph_e}. Path: {path}. Please, double check file name and path.")
+                return ToolException(f"File not found. Please, check file name and path: {e} and {graph_e}")
+        #
+        return parse_file_content(file_name=file_name,
                                   file_content=file_content,
                                   is_capture_image=is_capture_image,
                                   page_number=page_number,

alita_sdk/tools/sharepoint/authorization_helper.py

@@ -1,5 +1,5 @@
 from datetime import datetime, timezone
-from urllib.parse import urlparse
+from urllib.parse import unquote, urlparse
 
 import jwt
 import requests
@@ -117,6 +117,7 @@ class SharepointAuthorizationHelper:
             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")
             #
@@ -137,9 +138,12 @@ class SharepointAuthorizationHelper:
             #
             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.get('name'),
-                    'Path': file.get('webUrl'),
+                    'Name': file_name,
+                    'Path': '/'.join([part for part in parts if part]),
                     'Created': file.get('createdDateTime'),
                     'Modified': file.get('lastModifiedDateTime'),
                     'Link': file.get('webUrl'),
@@ -154,3 +158,28 @@ class SharepointAuthorizationHelper:
             return result
         except Exception as e:
             raise RuntimeError(f"Error in get_files_list: {e}")
+
+    def get_file_content(self, site_url: str, path: str):
+        try:
+            access_token, site_id = self.generate_token_and_site_id(site_url)
+            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")
+            #
+            # 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
+        except Exception as e:
+            raise RuntimeError(f"Error in get_files_list: {e}")

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

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

@@ -317,8 +317,8 @@ alita_sdk/tools/servicenow/__init__.py,sha256=ziEt2juPrGFyB98ZXbGf25v6gZo4UJTHsz
 alita_sdk/tools/servicenow/api_wrapper.py,sha256=WpH-bBLGFdhehs4g-K-WAkNuaD1CSrwsDpdgB3RG53s,6120
 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=xUQOlJBHha0bmzkz00GVgzUnAsE28saGkRYHqXTj7Ac,13105
-alita_sdk/tools/sharepoint/authorization_helper.py,sha256=WfkSZh28gfB2aOlWk1T4mHjc0PBW5SPwLzVVTPr_dkM,7476
+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/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.405.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.405.dist-info/METADATA,sha256=zNo9z-qhfD29QynTObDSSd-1sgmJKEdTLqdM_NdmCCI,19071
-alita_sdk-0.3.405.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.405.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.405.dist-info/RECORD,,
+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,,