alita-sdk 0.3.116__py3-none-any.whl → 0.3.118__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/langraph_agent.py

@@ -373,7 +373,7 @@ def create_graph(
                     node['decision'].get('description', ""),
                     decisional_inputs=node['decision'].get('decisional_inputs', ['messages']),
                     default_output=node['decision'].get('default_output', 'END')))
-            elif node.get('condition'):
+            elif node.get('condition') and node_type != 'router':
                 logger.info(f'Adding condition: {node["condition"]}')
                 condition_input = node['condition'].get('condition_input', ['messages'])
                 condition_definition = node['condition'].get('condition_definition', '')

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.116.dist-info → alita_sdk-0.3.118.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.116
+Version: 0.3.118
 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.116.dist-info → alita_sdk-0.3.118.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
@@ -19,7 +19,7 @@ alita_sdk/langchain/assistant.py,sha256=V0MpZG9IQRlKMr1SfAabVbUsIp-gNPO9A1SWxuFi
 alita_sdk/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/langchain/langraph_agent.py,sha256=2s5KY6520reNZpX4wMJbOmGBRRPOch4Y2mwFEdtgWR4,20600
+alita_sdk/langchain/langraph_agent.py,sha256=vxsbjOE0UWZIuyXNGEMR1QJ8t5yHS7Kzjt8a8nLb-jw,20626
 alita_sdk/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/langchain/utils.py,sha256=Npferkn10dvdksnKzLJLBI5bNGQyVWTBwqp3vQtUqmY,6631
@@ -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.116.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.118.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.116.dist-info/METADATA,sha256=IGlTICE0EqRPB57y7OJeI99xj9hHcObbQehXGRN-ArU,7075
-alita_sdk-0.3.116.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-alita_sdk-0.3.116.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.116.dist-info/RECORD,,
+alita_sdk-0.3.118.dist-info/METADATA,sha256=EpmbeqXcaaS8ftTkjf1BBPjWBb3AEmjv5sYE20G5uYw,7075
+alita_sdk-0.3.118.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+alita_sdk-0.3.118.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.118.dist-info/RECORD,,