alita-sdk 0.3.330__tar.gz → 0.3.332__tar.gz

{alita_sdk-0.3.330 → alita_sdk-0.3.332}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.330
+Version: 0.3.332
 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.330 → alita_sdk-0.3.332}/alita_sdk/runtime/langchain/assistant.py

@@ -34,6 +34,7 @@ class Assistant:
         self.app_type = app_type
         self.memory = memory
         self.store = store
+        self.max_iterations = data.get('meta', {}).get('step_limit', 25)
 
         logger.debug("Data for agent creation: %s", data)
         logger.info("App type: %s", app_type)
@@ -145,7 +146,8 @@ class Assistant:
     def _agent_executor(self, agent: Any):
         return AgentExecutor.from_agent_and_tools(agent=agent, tools=self.tools,
                                                   verbose=True, handle_parsing_errors=True,
-                                                  max_execution_time=None, return_intermediate_steps=True)
+                                                  max_execution_time=None, return_intermediate_steps=True,
+                                                  max_iterations=self.max_iterations)
 
     def getAgentExecutor(self):
         # Exclude compiled graph runnables from simple tool agents

{alita_sdk-0.3.330 → alita_sdk-0.3.332}/alita_sdk/runtime/tools/llm.py

@@ -31,26 +31,25 @@ def create_llm_input_with_messages(
     
     # Build the input messages
     input_messages = []
+    messages = params.get('messages', [])
     
     # Add system message from prompt if available
     if prompt:
         try:
             # Format the system message using the prompt template or value and params
             prompt_str = prompt['template'] if 'template' in prompt else prompt['value']
-            system_content = prompt_str.format(**params) if params else prompt_str
-            input_messages.append(SystemMessage(content=system_content))
+            prompt_content = prompt_str.format(**params) if params else prompt_str
+            # if user hasn't specified chat history, add system message as HumanMessage
+            input_messages.append(SystemMessage(content=prompt_content) if messages else HumanMessage(content=prompt_content))
         except KeyError as e:
             error_msg = f"KeyError in prompt formatting: {e}. Available params: {list(params.keys())}"
             logger.error(error_msg)
             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="Reply to this message.")])
 
     return input_messages
 

{alita_sdk-0.3.330 → alita_sdk-0.3.332}/alita_sdk.egg-info/PKG-INFO

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

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