alita-sdk 0.3.118__tar.gz → 0.3.120__tar.gz

{alita_sdk-0.3.118 → alita_sdk-0.3.120}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.118
+Version: 0.3.120
 Summary: SDK for building langchain agents using resouces 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>
 Project-URL: Homepage, https://projectalita.ai

{alita_sdk-0.3.118 → alita_sdk-0.3.120}/pyproject.toml

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

{alita_sdk-0.3.118 → alita_sdk-0.3.120}/src/alita_sdk/langchain/document_loaders/AlitaConfluenceLoader.py

@@ -1,6 +1,7 @@
 from io import BytesIO
-from typing import Optional
+from typing import Optional, List
 
+import requests
 from PIL import Image
 from langchain_community.document_loaders import ConfluenceLoader
 from langchain_community.document_loaders.confluence import ContentFormat
@@ -9,7 +10,7 @@ from pdf2image import convert_from_bytes
 from reportlab.graphics import renderPM
 from svglib.svglib import svg2rlg
 
-from ..tools.utils import image_to_byte_array, bytes_to_base64
+from src.alita_sdk.langchain.tools.utils import image_to_byte_array, bytes_to_base64
 
 Image.MAX_IMAGE_PIXELS = 300_000_000
 
@@ -73,6 +74,61 @@ class AlitaConfluenceLoader(ConfluenceLoader):
         kwargs['content_format'] = mapping.get(content_formant, ContentFormat.VIEW)
         return super().load(**kwargs)
 
+    def process_attachment(
+        self,
+        page_id: str,
+        ocr_languages: Optional[str] = None,
+    ) -> List[str]:
+        """
+        Process attachments from a Confluence page and extract text from them.
+        Note: if the attachment is corrupted, it will be skipped and an error will be logged.
+        """
+
+        try:
+            from PIL import Image  # noqa: F401
+        except ImportError:
+            raise ImportError(
+                "`Pillow` package not found, " "please run `pip install Pillow`"
+            )
+
+        attachments = self.confluence.get_attachments_from_content(page_id)["results"]
+        texts = []
+        for attachment in attachments:
+            media_type = attachment["metadata"]["mediaType"]
+            absolute_url = self.base_url + attachment["_links"]["download"]
+            title = attachment["title"]
+            try:
+                if media_type == "application/pdf":
+                    text = title + self.process_pdf(absolute_url, ocr_languages)
+                elif (
+                    media_type == "image/png"
+                    or media_type == "image/jpg"
+                    or media_type == "image/jpeg"
+                ):
+                    text = title + self.process_image(absolute_url, ocr_languages)
+                elif (
+                    media_type == "application/vnd.openxmlformats-officedocument"
+                    ".wordprocessingml.document"
+                ):
+                    text = title + self.process_doc(absolute_url)
+                elif media_type == "application/vnd.ms-excel":
+                    text = title + self.process_xls(absolute_url)
+                # elif media_type == "image/svg+xml":
+                #     text = title + self.process_svg(absolute_url, ocr_languages)
+                else:
+                    continue
+                texts.append(text)
+            except requests.HTTPError as e:
+                if e.response.status_code == 404:
+                    print(f"Attachment not found at {absolute_url}")  # noqa: T201
+                    continue
+                else:
+                    raise
+            except Exception as e:
+                print(f"Error processing attachment {absolute_url}: {e}")
+                continue
+        return texts
+
     def process_pdf(
             self,
             link: str,

{alita_sdk-0.3.118 → alita_sdk-0.3.120}/src/alita_sdk/langchain/interfaces/loaders.py

@@ -19,7 +19,7 @@ from ..document_loaders.AlitaCSVLoader import AlitaCSVLoader
 from ..document_loaders.AlitaExcelLoader import AlitaExcelLoader
 from ..document_loaders.AlitaDirectoryLoader import AlitaDirectoryLoader
 from ..document_loaders.AlitaGitRepoLoader import AlitaGitRepoLoader
-from ..document_loaders.AlitaConfluenceLoader import AlitaConfluenceLoader
+from alita_tools.confluence.loader import AlitaConfluenceLoader
 from ..document_loaders.AlitaBDDScenariosLoader import BDDScenariosLoader
 from ..document_loaders.AlitaJiraLoader import AlitaJiraLoader
 

{alita_sdk-0.3.118 → alita_sdk-0.3.120}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.118
+Version: 0.3.120
 Summary: SDK for building langchain agents using resouces 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>
 Project-URL: Homepage, https://projectalita.ai