alita-sdk 0.3.179__tar.gz → 0.3.180__tar.gz

{alita_sdk-0.3.179 → alita_sdk-0.3.180}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.179
+Version: 0.3.180
 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 <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.179 → alita_sdk-0.3.180}/alita_sdk/runtime/langchain/assistant.py

@@ -186,37 +186,44 @@ class Assistant:
         Create a LangGraph agent with auto-continue capability for when responses get truncated.
         This provides better handling of length-limited responses compared to traditional AgentExecutor.
         Uses simple in-memory checkpointing for auto-continue functionality.
+        
+        Note: Requires LangGraph 0.5.x or higher that supports post_model_hook.
         """
+        
         from langgraph.prebuilt import create_react_agent
         from langgraph.checkpoint.memory import MemorySaver
-    
+        
         # Use simple in-memory checkpointer for auto-continue functionality
-        # This ensures clean, predictable behavior for continuation logic
         memory = MemorySaver()
         
         # Filter tools to only include BaseTool instances
         simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
         
-        # Create the agent with checkpointing enabled
-        agent = create_react_agent(
-            prompt=self.prompt,
-            model=self.client,
-            tools=simple_tools,
-            checkpointer=memory,
-            state_modifier=self._create_auto_continue_state_modifier()
-        )
-        
+        # Set up parameters for the agent
+        kwargs = {
+            "prompt": self.prompt,
+            "model": self.client,
+            "tools": simple_tools,
+            "checkpointer": memory,
+            "post_model_hook": self._create_auto_continue_hook()  # Auto-continue hook
+        }
+
+        # Create the agent with the new hook-based API
+        agent = create_react_agent(**kwargs)
         return agent
     
-    def _create_auto_continue_state_modifier(self):
+    # The state_modifier approach has been removed in favor of post_model_hook which
+    # is the recommended approach for LangGraph 0.5.x and higher
+    
+    def _create_auto_continue_hook(self):
         """
-        Create a state modifier that implements auto-continue logic.
-        This checks if the last AI response was truncated and automatically continues if needed.
-        Limits auto-continue to prevent infinite loops.
+        Create a post-model hook for LangGraph 0.5.x that detects truncated responses
+        and adds continuation prompts.
+        This checks if the last AI message was truncated and automatically continues if needed.
         """
         MAX_CONTINUATIONS = 3  # Maximum number of auto-continuations allowed
         
-        def auto_continue_modifier(state):
+        def post_model_hook(state):
             messages = state.get("messages", [])
             
             # Count how many auto-continue messages we've already sent
@@ -229,8 +236,8 @@ class Assistant:
             # Don't continue if we've reached the limit
             if continuation_count >= MAX_CONTINUATIONS:
                 return state
-            
-            # Check if the last AI message was truncated
+                
+            # Check if the last message is from AI and was truncated
             if messages and isinstance(messages[-1], AIMessage):
                 last_ai_message = messages[-1]
                 
@@ -244,17 +251,13 @@ class Assistant:
                     not last_ai_message.content.rstrip().endswith(('.', '!', '?', ':', ';'))
                 )
                 
-                # Add continuation logic
+                # Add continuation request if truncated
                 if is_truncated:
-                    # Add a human message to continue
-                    continuation_msg = HumanMessage(
-                        content="Continue your previous response from where you left off"
-                    )
-                    return {
-                        **state,
-                        "messages": messages + [continuation_msg]
-                    }
+                    logger.info("Detected truncated response, adding continuation request")
+                    new_messages = messages.copy()
+                    new_messages.append(HumanMessage(content="Continue your previous response from where you left off"))
+                    return {"messages": new_messages}
             
             return state
-        
-        return auto_continue_modifier
+            
+        return post_model_hook

{alita_sdk-0.3.179 → alita_sdk-0.3.180}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.179
+Version: 0.3.180
 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 <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.179 → alita_sdk-0.3.180}/pyproject.toml

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