alita-sdk 0.3.422__py3-none-any.whl → 0.3.424__py3-none-any.whl

alita_sdk/runtime/langchain/constants.py

@@ -82,4 +82,5 @@ DEFAULT_MULTIMODAL_PROMPT = """
 """
 
 ELITEA_RS = "elitea_response"
+PRINTER = "printer"
 PRINTER_NODE_RS = "printer_output"

alita_sdk/runtime/langchain/langraph_agent.py

@@ -19,7 +19,7 @@ from langgraph.managed.base import is_managed_value
 from langgraph.prebuilt import InjectedStore
 from langgraph.store.base import BaseStore
 
-from .constants import PRINTER_NODE_RS
+from .constants import PRINTER_NODE_RS, PRINTER
 from .mixedAgentRenderes import convert_message_to_json
 from .utils import create_state, propagate_the_input_mapping, safe_format
 from ..tools.function import FunctionTool
@@ -236,15 +236,17 @@ class StateDefaultNode(Runnable):
 class PrinterNode(Runnable):
     name = "PrinterNode"
 
-    def __init__(self, text_pattern: str, formatting_enabled: bool = True):
-        self.text = text_pattern
-        self.formatting_enabled = formatting_enabled
+    def __init__(self, input_mapping: Optional[dict[str, dict]]):
+        self.input_mapping = input_mapping
 
     def invoke(self, state: BaseStore, config: Optional[RunnableConfig] = None) -> dict:
         logger.info(f"Printer Node - Current state variables: {state}")
         result = {}
-        logger.debug(f"Initial text pattern: {self.text}")
-        formatted_output = safe_format(self.text, state)
+        logger.debug(f"Initial text pattern: {self.input_mapping}")
+        mapping = propagate_the_input_mapping(self.input_mapping, [], state)
+        if mapping.get(PRINTER) is None:
+            raise ToolException(f"PrinterNode requires '{PRINTER}' field in input mapping")
+        formatted_output = mapping[PRINTER]
         logger.debug(f"Formatted output: {formatted_output}")
         result[PRINTER_NODE_RS] = formatted_output
         return result
@@ -647,8 +649,7 @@ def create_graph(
                 ))
             elif node_type == 'printer':
                 lg_builder.add_node(node_id, PrinterNode(
-                    text_pattern=node.get('printer', ''),
-                    formatting_enabled=node.get('formatting_enabled', True)
+                    input_mapping=node.get('input_mapping', {'printer': {'type': 'fixed', 'value': ''}}),
                 ))
 
                 # add interrupts after printer node if specified
@@ -657,7 +658,7 @@ def create_graph(
                 # reset printer output variable to avoid carrying over
                 reset_node_id = f"{node_id}_reset"
                 lg_builder.add_node(reset_node_id, PrinterNode(
-                    text_pattern=''
+                    input_mapping={'printer': {'type': 'fixed', 'value': ''}}
                 ))
                 lg_builder.add_edge(node_id, reset_node_id)
                 lg_builder.add_conditional_edges(reset_node_id, TransitionalEdge(clean_string(node['transition'])))

{alita_sdk-0.3.422.dist-info → alita_sdk-0.3.424.dist-info}/METADATA

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

@@ -43,9 +43,9 @@ alita_sdk/runtime/clients/sandbox_client.py,sha256=kGOGfm3OAFmYeTM4bIuKbhRsOiOhF
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/langchain/assistant.py,sha256=qKoEjbGuUnX-OZDHmSaK3plb1jON9unzEwAjxBT9DY8,16044
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
-alita_sdk/runtime/langchain/constants.py,sha256=EVqL1JZTST_7m0I_5I1XBnDX4WTvxyRVkhlRbUjwtA4,3384
+alita_sdk/runtime/langchain/constants.py,sha256=I3dwexVp_Qq3MueRA2ClLgFDEhk4BkJhgR6m7V0gVPc,3404
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=FvFhphGIAUDKtR5s2m_otJD-CEsFUITaiM8NwF7hCzM,51471
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=sAdDojp7lOC2jyaBPS3MP8hV_EhjsZuyOqUT3abf2Wk,51647
 alita_sdk/runtime/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/runtime/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/runtime/langchain/store_manager.py,sha256=i8Fl11IXJhrBXq1F1ukEVln57B1IBe-tqSUvfUmBV4A,2218
@@ -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.422.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.422.dist-info/METADATA,sha256=DUQJRlH2dVMYnKnsTajjC2bay2fXlhS8_qjQzX5B1xs,19071
-alita_sdk-0.3.422.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.422.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.422.dist-info/RECORD,,
+alita_sdk-0.3.424.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.424.dist-info/METADATA,sha256=tdVnnVJp4PuGYlO1a5x4EV3TFVzntxX9ngHLACfaclM,19071
+alita_sdk-0.3.424.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.424.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.424.dist-info/RECORD,,