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

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

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

@@ -7,14 +7,10 @@ from langchain.agents import (
     create_json_chat_agent)
 from langgraph.store.base import BaseStore
 
-# Note: Traditional LangChain agents (OpenAI, XML, JSON) do not support 
-# checkpointing natively. Only LangGraph agents support checkpointing.
-# For checkpointing in traditional agents, you need to migrate to LangGraph.
-
 from .agents.xml_chat import create_xml_chat_agent
 from .langraph_agent import create_graph
 from langchain_core.messages import (
-    BaseMessage, SystemMessage, HumanMessage, AIMessage
+    BaseMessage, SystemMessage, HumanMessage
 )
 from langchain_core.prompts import MessagesPlaceholder
 from .constants import REACT_ADDON, REACT_VARS, XML_ADDON
@@ -33,8 +29,7 @@ class Assistant:
                  app_type: str = "openai",
                  tools: Optional[list] = [],
                  memory: Optional[Any] = None,
-                 store: Optional[BaseStore] = None,
-                 use_checkpointing: bool = False):
+                 store: Optional[BaseStore] = None):
 
         self.client = copy(client)
         self.client.max_tokens = data['llm_settings']['max_tokens']
@@ -47,7 +42,6 @@ class Assistant:
         self.app_type = app_type
         self.memory = memory
         self.store = store
-        self.use_checkpointing = use_checkpointing
 
         logger.debug("Data for agent creation: %s", data)
         logger.info("App type: %s", app_type)
@@ -124,17 +118,12 @@ class Assistant:
         if self.app_type == 'pipeline':
             return self.pipeline()
         elif self.app_type == 'openai':
-            # Check if checkpointing is enabled - if so, use LangGraph for auto-continue capability
-            if self.use_checkpointing:
-                return self.getLangGraphAgentWithAutoContinue()
             return self.getOpenAIToolsAgentExecutor()
         elif self.app_type == 'xml':
-            # Check if checkpointing is enabled - if so, use LangGraph for auto-continue capability
-            if self.use_checkpointing:
-                return self.getLangGraphAgentWithAutoContinue()
             return self.getXMLAgentExecutor()
         else:
-            return self.getLangGraphAgentWithAutoContinue()
+            self.tools = [EchoTool()] + self.tools
+            return self.getAgentExecutor()
 
     def _agent_executor(self, agent: Any):
         return AgentExecutor.from_agent_and_tools(agent=agent, tools=self.tools,
@@ -180,84 +169,3 @@ class Assistant:
 
     def predict(self, messages: list[BaseMessage]):
         return self.client.invoke(messages)
-
-    def getLangGraphAgentWithAutoContinue(self):
-        """
-        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
-        memory = MemorySaver()
-        
-        # Filter tools to only include BaseTool instances
-        simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
-        
-        # 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
-    
-    # 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 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 post_model_hook(state):
-            messages = state.get("messages", [])
-            
-            # Count how many auto-continue messages we've already sent
-            continuation_count = sum(
-                1 for msg in messages 
-                if isinstance(msg, HumanMessage) and 
-                "continue your previous response" in msg.content.lower()
-            )
-            
-            # Don't continue if we've reached the limit
-            if continuation_count >= MAX_CONTINUATIONS:
-                return state
-                
-            # Check if the last message is from AI and was truncated
-            if messages and isinstance(messages[-1], AIMessage):
-                last_ai_message = messages[-1]
-                
-                # Check for truncation indicators
-                is_truncated = (
-                    hasattr(last_ai_message, 'response_metadata') and 
-                    last_ai_message.response_metadata.get('finish_reason') == 'length'
-                ) or (
-                    # Fallback: check if message seems to end abruptly
-                    last_ai_message.content and 
-                    not last_ai_message.content.rstrip().endswith(('.', '!', '?', ':', ';'))
-                )
-                
-                # Add continuation request if truncated
-                if is_truncated:
-                    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 post_model_hook

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

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

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