alita-sdk 0.3.324__tar.gz → 0.3.326__tar.gz

{alita_sdk-0.3.324 → alita_sdk-0.3.326}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.324
+Version: 0.3.326
 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.324 → alita_sdk-0.3.326}/alita_sdk/runtime/langchain/langraph_agent.py

@@ -253,8 +253,17 @@ class StateModifierNode(Runnable):
                 input_data[var] = state.get(var)
 
         # Render the template using Jinja
-        from jinja2 import Template
-        rendered_message = Template(self.template).render(**input_data)
+        import json
+        from jinja2 import Environment
+
+        def from_json(value):
+            return json.loads(value)
+
+        env = Environment()
+        env.filters['from_json'] = from_json
+        
+        template = env.from_string(self.template)
+        rendered_message = template.render(**input_data)
         result = {}
         # Store the rendered message in the state or messages
         if len(self.output_variables) > 0:

{alita_sdk-0.3.324 → alita_sdk-0.3.326}/alita_sdk/runtime/tools/function.py

@@ -4,7 +4,7 @@ from json import dumps
 from langchain_core.callbacks import dispatch_custom_event
 from langchain_core.messages import ToolCall
 from langchain_core.runnables import RunnableConfig
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, ToolException
 from typing import Any, Optional, Union, Annotated
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from pydantic import ValidationError
@@ -48,7 +48,13 @@ class FunctionTool(BaseTool):
                 return {"messages": [{"role": "assistant", "content": dumps(tool_result)}]}
             else:
                 if self.output_variables[0] == "messages":
-                    return {"messages": [{"role": "assistant", "content": dumps(tool_result)}]}
+                    return {
+                        "messages": [{
+                            "role": "assistant",
+                            "content": dumps(tool_result) if not isinstance(tool_result, ToolException) else str(
+                                tool_result)
+                        }]
+                    }
                 else:
                     return { self.output_variables[0]: tool_result }
         except ValidationError:

{alita_sdk-0.3.324 → alita_sdk-0.3.326}/alita_sdk/runtime/utils/streamlit.py

@@ -1102,7 +1102,7 @@ def run_streamlit(st, ai_icon=None, user_icon=None):
                 st_cb = AlitaStreamlitCallback(st)
                 logger.info(st.session_state.messages)
                 response = st.session_state.agent_executor.invoke(
-                    {"input": prompt, "chat_history": st.session_state.messages[:-1]},
+                    {"input": [prompt], "chat_history": st.session_state.messages[:-1]},
                     { 'callbacks': [st_cb], 'configurable': {"thread_id": st.session_state.thread_id}}
                 )
                 st.write(response["output"])

{alita_sdk-0.3.324 → alita_sdk-0.3.326}/alita_sdk/tools/confluence/api_wrapper.py

@@ -539,7 +539,9 @@ class ConfluenceAPIWrapper(NonCodeIndexerToolkit):
         :param title: title
         :param type: type of content: page or blogpost. Defaults to page
         """
-        return self.client.get_page_id(space=self.space, title=title, type=type)
+
+        result = self.client.get_page_id(space=self.space, title=title, type=type)
+        return result if result else "Page not found. Check the title or space."
 
     def _strip_base64_images(self, content):
         base64_md_pattern = r'data:image/(png|jpeg|gif);base64,[a-zA-Z0-9+/=]+'
@@ -562,7 +564,7 @@ class ConfluenceAPIWrapper(NonCodeIndexerToolkit):
                 }
                 pages_info.append(page_info)
             start += self.limit
-        return str(pages_info)
+        return str(pages_info) if pages_info else f"Unable to find anything using query {cql}. Check space or query."
 
     def search_pages(self, query: str, skip_images: bool = False):
         """Search pages in Confluence by query text in title or page content."""

{alita_sdk-0.3.324 → alita_sdk-0.3.326}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.324
+Version: 0.3.326
 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.324 → alita_sdk-0.3.326}/pyproject.toml

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