alita-sdk 0.3.179__py3-none-any.whl → 0.3.180__py3-none-any.whl

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.dist-info → alita_sdk-0.3.180.dist-info}/METADATA

@@ -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.dist-info → alita_sdk-0.3.180.dist-info}/RECORD

@@ -17,7 +17,7 @@ alita_sdk/runtime/clients/client.py,sha256=6ezOJ92CSw6b2PVs4uFMQKQdp40uT1awoFEqW
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/langchain/assistant.py,sha256=VE6Gq7O1ErJV0eTHgM-7TOxdATd62aFimUKprz92o-I,11526
+alita_sdk/runtime/langchain/assistant.py,sha256=eY36htHBvB7n0sW4cr79HZ-FsTh-rQIQga66th_Tpe0,11759
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
@@ -295,8 +295,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=rq4jOb3lRW2GXvAguk4H1KinO5f-zpygzhBJf-E1Ucw,2773
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=iOMxyE7vOc_LwFB_nBMiSFXkNtvbptA4i-BrTlo7M0A,5854
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=IYUJoMFOMA70knLhLtAnuGoy3OK80RuqeQZ710oyIxE,3631
-alita_sdk-0.3.179.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.179.dist-info/METADATA,sha256=GkBo37E1OM9ej52nimuYYnsfys22QnRSuuR4rDutZgg,18753
-alita_sdk-0.3.179.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.179.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.179.dist-info/RECORD,,
+alita_sdk-0.3.180.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.180.dist-info/METADATA,sha256=1_Zfqgg2jYnpzujyCog3Z8_bNx07CjpO1ZjHGqlWzLE,18753
+alita_sdk-0.3.180.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.180.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.180.dist-info/RECORD,,