alita-sdk 0.3.321b1__py3-none-any.whl → 0.3.321b2__py3-none-any.whl

alita_sdk/runtime/tools/llm.py

@@ -15,7 +15,6 @@ logger = logging.getLogger(__name__)
 
 def create_llm_input_with_messages(
     prompt: Dict[str, str], 
-    messages: List[BaseMessage], 
     params: Dict[str, Any]
 ) -> List[BaseMessage]:
     """
@@ -23,13 +22,12 @@ def create_llm_input_with_messages(
     
     Args:
         prompt: The prompt configuration with template
-        messages: List of chat history messages
         params: Additional parameters for prompt formatting
         
     Returns:
         List of messages to send to LLM
     """
-    logger.info(f"Creating LLM input with messages: {len(messages)} messages, params: {params}")
+    logger.info(f"Creating LLM input with params: {params}")
     
     # Build the input messages
     input_messages = []
@@ -47,9 +45,13 @@ def create_llm_input_with_messages(
             raise ToolException(error_msg)
     
     # Add the chat history messages
+    messages = params.get('messages', [])
     if messages:
         input_messages.extend(messages)
-    
+    else:
+        # conditionally add a default human message if no chat history
+        input_messages.extend([HumanMessage(content="")])
+
     return input_messages
 
 
@@ -124,12 +126,13 @@ class LLMNode(BaseTool):
         # Create parameters for prompt formatting from state
         params = {}
         if isinstance(state, dict):
-            for var in self.input_variables or []:
-                if var != "messages" and var in state:
-                    params[var] = state[var]
+            params = {var: state[var] for var in (self.input_variables or []) if var != "messages" and var in state}
+            # message as a part of chat history added ONLY if "messages" is in input_variables
+            if "messages" in (self.input_variables or []):
+                params["messages"] = messages
         
         # Create LLM input with proper message handling
-        llm_input = create_llm_input_with_messages(self.prompt, messages, params)
+        llm_input = create_llm_input_with_messages(self.prompt, params)
         
         # Get the LLM client, potentially with tools bound
         llm_client = self.client
@@ -268,17 +271,14 @@ class LLMNode(BaseTool):
                     if json_output_vars:
                         try:
                             response = _extract_json(content) or {}
-                            response_data = {key: response.get(key) for key in json_output_vars if key in response}
-                            
-                            # Always add the messages to the response
-                            new_messages = messages + [AIMessage(content=content)]
-                            response_data['messages'] = new_messages
-                            
-                            return response_data
+                            response_data = {key: response.get(key, content) for key in json_output_vars}
                         except (ValueError, json.JSONDecodeError) as e:
-                            # LLM returned non-JSON content, treat as plain text
-                            logger.warning(f"Expected JSON output but got plain text. Output variables specified: {json_output_vars}. Error: {e}")
-                            # Fall through to plain text handling
+                            logger.warning(
+                                f"Expected JSON output but got plain text. Output variables specified: {json_output_vars}. Error: {e}")
+                            response_data = {var: content for var in json_output_vars}
+                        new_messages = messages + [AIMessage(content=content)]
+                        response_data['messages'] = new_messages
+                        return response_data
                     
                     # Simple text response (either no output variables or JSON parsing failed)
                     new_messages = messages + [AIMessage(content=content)]

{alita_sdk-0.3.321b1.dist-info → alita_sdk-0.3.321b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.321b1
+Version: 0.3.321b2
 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.321b1.dist-info → alita_sdk-0.3.321b2.dist-info}/RECORD

@@ -112,7 +112,7 @@ alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCE
 alita_sdk/runtime/tools/function.py,sha256=ZFpd7TGwIawze2e7BHlKwP0NHwNw42wwrmmnXyJQJhk,2600
 alita_sdk/runtime/tools/graph.py,sha256=MbnZYqdmvZY7SGDp43lOVVIjUt5ARHSgj43mdtBjSjQ,3092
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
-alita_sdk/runtime/tools/llm.py,sha256=NsrsP-SblyxDdzgMCn9_OBUL0sUGDVS5yqer49V7ciE,15069
+alita_sdk/runtime/tools/llm.py,sha256=SwRzC9n_rBsljnbhS8FPMgKc9b_DTSleCK40MqWnGzc,15107
 alita_sdk/runtime/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/runtime/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
 alita_sdk/runtime/tools/mcp_server_tool.py,sha256=trGraI8-AwdbNmTKMjfmlBxgTDMTE4-21heCVtd_lz0,4156
@@ -349,8 +349,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.321b1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.321b1.dist-info/METADATA,sha256=Y83S1tr4PUyPhIW_KngAQJPj2aADp4psfWPYOQ_f8U0,18899
-alita_sdk-0.3.321b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.321b1.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.321b1.dist-info/RECORD,,
+alita_sdk-0.3.321b2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.321b2.dist-info/METADATA,sha256=DibZvPqa7lDNcR3Guti57bcNqcAE4dddj5xoArHUKuo,18899
+alita_sdk-0.3.321b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.321b2.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.321b2.dist-info/RECORD,,