alita-sdk 0.3.117__py3-none-any.whl → 0.3.119__py3-none-any.whl

alita_sdk/community/analysis/jira_analyse/api_wrapper.py

@@ -15,7 +15,7 @@ from elitea_analyse.jira.jira_issues import JiraIssues
 
 from alita_tools.elitea_base import BaseToolApiWrapper
 from ....tools.artifact import ArtifactWrapper
-
+from ....utils.logging import with_streamlit_logs
 
 logger = logging.getLogger(__name__)
 
@@ -59,9 +59,21 @@ class JiraAnalyseWrapper(BaseToolApiWrapper):
         Get projects a user has access to and merge them with issues count.
         after_date: str
             date after which issues are considered
+        project_keys: str
+            one or more projects keys separated with comma
         """
         project_keys = project_keys or self.project_keys
 
+        dispatch_custom_event(
+            name="thinking_step",
+            data={
+                "message": f"I am extracting number of all issues with initial parameters:\
+                    project keys: {project_keys},   after date: {after_date}",
+                "tool_name": "get_number_off_all_issues",
+                "toolkit": "analyse_jira",
+            },
+        )
+
         project_df = jira_projects_overview(
             after_date, project_keys=project_keys, jira=self.jira
         )
@@ -72,22 +84,12 @@ class JiraAnalyseWrapper(BaseToolApiWrapper):
             f"projects_overview_{project_keys}.csv",
             csv_options={"index": False},
         )
-        dispatch_custom_event(
-            name="jira_projects_overview",
-            data={
-                "project_keys": project_keys,
-                "after_date": after_date,
-                "files": [f"projects_overview_{project_keys}.csv"],
-                "project_count": len(project_df),
-                "columns": project_df.columns.tolist(),
-            },
-        )
-
         return {
             "projects": project_df["key"].tolist(),
             "projects_summary": project_df.to_string(),
         }
 
+    @with_streamlit_logs(tool_name="get_jira_issues")
     def get_jira_issues(
         self,
         closed_issues_based_on: int,
@@ -100,7 +102,8 @@ class JiraAnalyseWrapper(BaseToolApiWrapper):
         """
         Extract Jira issues for the specified projects.
         closed_issues_based_on: int
-            define whether issues can be thought as closed based on their status (1) or not empty resolved date (2)
+            define whether issues can be thought as 
+            closed based on their status (1) or not empty resolved date (2)
         resolved_after: str
             resolved after date (i.e. 2023-01-01)
         updated_after: str
@@ -112,6 +115,7 @@ class JiraAnalyseWrapper(BaseToolApiWrapper):
         project_keys: str
             one or more projects keys separated with comma
         """
+
         if not (
             (
                 closed_issues_based_on == 1
@@ -122,7 +126,7 @@ class JiraAnalyseWrapper(BaseToolApiWrapper):
             return (
                 "ERROR: Check input parameters closed_issues_based_on and closed_status"
             )
-        
+
         project_keys = project_keys or self.project_keys
 
         dispatch_custom_event(

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/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/utils/logging.py

@@ -0,0 +1,93 @@
+import logging
+from functools import wraps
+
+from langchain_core.callbacks import dispatch_custom_event
+
+
+class StreamlitCallbackHandler(logging.Handler):
+    """Custom logging handler to send logs to Streamlit."""
+
+    def __init__(self, tool_name: str = "logging"):
+        super().__init__()
+        self.tool_name = tool_name
+
+    def emit(self, record):
+        """Emit a log record."""
+        if record.levelno < logging.INFO:
+            return  # Ignore debug logs
+
+        log_entry = self.format(record)
+        dispatch_custom_event(
+            name="thinking_step",
+            data={
+                "message": log_entry,
+                "tool_name": self.tool_name,
+                "toolkit": "logging",  # ? or pass the toolkit name
+            },
+        )
+
+
+def setup_streamlit_logging(
+    logger_name: str = "", tool_name="logging"
+) -> StreamlitCallbackHandler:
+    """
+    Attach a StreamlitCallbackHandler to the given logger (default: root).
+    Returns the handler so you can remove it later if needed.
+    """
+    logger = logging.getLogger(logger_name)
+    handler = StreamlitCallbackHandler(tool_name)
+
+    # Avoid duplicate handlers
+    if not any(isinstance(h, StreamlitCallbackHandler) for h in logger.handlers):
+        formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+
+    return handler
+
+
+# Decorator version
+def with_streamlit_logs(logger_name: str = "", tool_name="logging"):
+    """
+    Decorator to temporarily attach a StreamlitCallbackHandler to a function's logger.
+
+    Args:
+        logger_name (str): Name of the logger to attach the handler to.
+            Use an empty string "" for the root logger.
+        tool_name (str): Name of the tool to display in Streamlit logs.
+
+    Behavior:
+        - Attaches a StreamlitCallbackHandler before the function runs.
+        - Forwards all INFO and higher log messages—including those
+            from 3rd-party libraries using the specified logger—to Streamlit 
+            via dispatch_custom_event.
+        - Automatically removes the handler after the function completes, 
+            even if an exception occurs.
+
+    Example:
+        @with_streamlit_logs(logger_name="my_logger", tool_name="my_tool")
+        def my_function():
+            logging.info("This is a log message.")
+            # Logs from 3rd-party libraries using "my_logger" will also be sent to Streamlit.
+
+    Returns:
+        The decorated function with Streamlit logging enabled during its execution.
+    """
+
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            logger = logging.getLogger(logger_name)
+            handler = StreamlitCallbackHandler(tool_name=tool_name)
+            handler.setFormatter(
+                logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
+            )
+            logger.addHandler(handler)
+            try:
+                return func(*args, **kwargs)
+            finally:
+                logger.removeHandler(handler)
+
+        return wrapper
+
+    return decorator

{alita_sdk-0.3.117.dist-info → alita_sdk-0.3.119.dist-info}/METADATA

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

@@ -11,7 +11,7 @@ alita_sdk/community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 alita_sdk/community/utils.py,sha256=lvuCJaNqVPHOORJV6kIPcXJcdprVW_TJvERtYAEgpjM,249
 alita_sdk/community/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/community/analysis/jira_analyse/__init__.py,sha256=Rm-HKEi_HIxrgHdq9mZ-XzxMKLXm8-81eJwJT2lar-c,5945
-alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=FPYAOh2SV5ggWg0ywDdkyt6udyR3juQvZM7lOIwvcKU,9327
+alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=JqGSxg_3x0ErzII31UZkY3V7jo9i8Gb5d_pW7lPIOSA,9522
 alita_sdk/community/browseruse/__init__.py,sha256=uAxPZEX7ihpt8HtcGDFrzTNv9WcklT1wG1ItTwUO8y4,3601
 alita_sdk/community/browseruse/api_wrapper.py,sha256=Y05NKWfTROPmBxe8ZFIELSGBX5v3RTNP30OTO2Tj8uI,10838
 alita_sdk/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -27,7 +27,7 @@ alita_sdk/langchain/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 alita_sdk/langchain/agents/xml_chat.py,sha256=Mx7PK5T97_GrFCwHHZ3JZP42S7MwtUzV0W-_8j6Amt8,6212
 alita_sdk/langchain/document_loaders/AlitaBDDScenariosLoader.py,sha256=4kFU1ijrM1Jw7cywQv8mUiBHlE6w-uqfzSZP4hUV5P4,3771
 alita_sdk/langchain/document_loaders/AlitaCSVLoader.py,sha256=TBJuIFqweLDtd0JxgfPqrcY5eED-M617CT_EInp6Lmg,1949
-alita_sdk/langchain/document_loaders/AlitaConfluenceLoader.py,sha256=ehCmV8uaSP9HY9fcBrgEa8j3-ATN5quMb8BDUERly90,5986
+alita_sdk/langchain/document_loaders/AlitaConfluenceLoader.py,sha256=wpZxN9tKSzRACROu4NYQ020wuIVgvuI14_EPQ_CbrU4,8316
 alita_sdk/langchain/document_loaders/AlitaDirectoryLoader.py,sha256=fKezkgvIcLG7S2PVJp1a8sZd6C4XQKNZKAFC87DbQts,7003
 alita_sdk/langchain/document_loaders/AlitaDocxMammothLoader.py,sha256=dBf0JkWMrNBCB4zJ3fjtOxci0OzZuDimMGSPYS7W6Oo,4848
 alita_sdk/langchain/document_loaders/AlitaExcelLoader.py,sha256=fC7L_NzkGZSirYvJhXSbuc37e2FNtNgETQamkKVqcTQ,1457
@@ -42,7 +42,7 @@ alita_sdk/langchain/document_loaders/utils.py,sha256=ifh0UJiweIb2iEb2_THJy2pbAnT
 alita_sdk/langchain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/langchain/interfaces/kwextractor.py,sha256=kSJA9L8g8UArmHu7Bd9dIO0Rrq86JPUb8RYNlnN68FQ,3072
 alita_sdk/langchain/interfaces/llm_processor.py,sha256=oX5feXsQ5D-8q3Mf40Txltv3kkHKb2xBeNPwecy5yDQ,8442
-alita_sdk/langchain/interfaces/loaders.py,sha256=li-O2dubiDNYn-qfVcDsuD4LqP_IZ61cV2vHUZAqeXc,3337
+alita_sdk/langchain/interfaces/loaders.py,sha256=7btXnGW-wODBoF5Jlv9ljPlkMzJU4NwA8E_v_VzsIHg,3326
 alita_sdk/langchain/interfaces/splitters.py,sha256=tW65-Ejj9VYyxXFZNgPts_CKILQ18bWp_1bZ-24FKGc,3630
 alita_sdk/langchain/retrievers/AlitaRetriever.py,sha256=osChtJxUlfpsFESpJSE5mnJAkxTXnzgFZnC6l5mUlbo,6148
 alita_sdk/langchain/retrievers/VectorstoreRetriever.py,sha256=CWZbqle5sjEWxFbmn1-ZuMYUgQLQb_b6sCK1d1KxeWE,2012
@@ -88,12 +88,13 @@ alita_sdk/tools/vectorstore.py,sha256=F-DoHxPa4UVsKB-FEd-wWa59QGQifKMwcSNcZ5WZOK
 alita_sdk/utils/AlitaCallback.py,sha256=cvpDhR4QLVCNQci6CO6TEUrUVDZU9_CRSwzcHGm3SGw,7356
 alita_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/utils/evaluate.py,sha256=iM1P8gzBLHTuSCe85_Ng_h30m52hFuGuhNXJ7kB1tgI,1872
+alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,3124
 alita_sdk/utils/streamlit.py,sha256=zp8owZwHI3HZplhcExJf6R3-APtWx-z6s5jznT2hY_k,29124
 alita_sdk/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk-0.3.117.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.119.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_jira_analysis.py,sha256=I0cErH5R_dHVyutpXrM1QEo7jfBuKWTmDQvJBPjx18I,3281
-alita_sdk-0.3.117.dist-info/METADATA,sha256=UKs2mCU1AvQTt5jMaeRngvrGk8z6WwY4DX6aNBmaEzo,7075
-alita_sdk-0.3.117.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-alita_sdk-0.3.117.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.117.dist-info/RECORD,,
+alita_sdk-0.3.119.dist-info/METADATA,sha256=Icr-9s7-kw-d_4ZHMiPJOkUgvyx_X5mB9xLYNiYWHj4,7075
+alita_sdk-0.3.119.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+alita_sdk-0.3.119.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.119.dist-info/RECORD,,