alita-sdk 0.3.181__py3-none-any.whl → 0.3.182__py3-none-any.whl

alita_sdk/runtime/clients/client.py

@@ -218,6 +218,8 @@ class AlitaClient:
             return LangChainAssistant(self, data, client,
                                       chat_history, app_type,
                                       tools=tools, memory=memory, store=store).runnable()
+        elif runtime == 'llama':
+            raise NotImplementedError("LLama runtime is not supported")
 
     def datasource(self, datasource_id: int) -> AlitaDataSource:
         url = f"{self.datasources}/{datasource_id}"

alita_sdk/runtime/langchain/assistant.py

@@ -7,15 +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 .agents.react_agent import get_langgraph_agent_with_auto_continue
 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
@@ -34,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']
@@ -48,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)
@@ -76,31 +69,13 @@ class Assistant:
         self.tools = get_tools(data['tools'], alita_client=alita, llm=self.client, memory_store=self.store)
         if app_type == "pipeline":
             self.prompt = data['instructions']
-        elif app_type == "react":
-            self.tools += tools
-            messages = [SystemMessage(content=data['instructions'])]
-            # messages.append(HumanMessage("{{input}}"))
-            self.prompt = Jinja2TemplatedChatMessagesTemplate(messages=messages)
-            variables = {}
-            input_variables = []
-            for variable in data['variables']:
-                if variable['value'] != "":
-                    variables[variable['name']] = variable['value']
-                else:
-                    input_variables.append(variable['name'])
-            if input_variables:
-                self.prompt.input_variables = input_variables
-            if variables:
-                self.prompt.partial_variables = variables
-            try:
-                logger.info(f"Client was created with client setting: temperature - {self.client._get_model_default_parameters}")
-            except Exception as e:
-                logger.info(f"Client was created with client setting: temperature - {self.client.temperature} : {self.client.max_tokens}")
         else:
             self.tools += tools
             messages = [SystemMessage(content=data['instructions'])]
             messages.append(MessagesPlaceholder("chat_history"))
-            if app_type == "xml":
+            if app_type == "react":
+                messages.append(HumanMessage(REACT_ADDON))
+            elif app_type == "xml":
                 messages.append(HumanMessage(XML_ADDON))
             elif app_type in ['openai', 'dial']:
                 messages.append(HumanMessage("{{input}}"))
@@ -112,7 +87,7 @@ class Assistant:
                     variables[variable['name']] = variable['value']
                 else:
                     input_variables.append(variable['name'])
-            if app_type in ["xml"]:
+            if app_type in ["react", "xml"]:
                 input_variables = list(set(input_variables + REACT_VARS))
 
             if chat_history and isinstance(chat_history, list):
@@ -143,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,
@@ -199,22 +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.
-        """
-
-        # Use the function from react_agent.py to create the agent
-        return get_langgraph_agent_with_auto_continue(
-            prompt=self.prompt,
-            model=self.client,
-            tools=self.tools,
-            memory=self.memory,
-            debug=True
-        )
-

alita_sdk/runtime/langchain/langraph_agent.py

@@ -12,7 +12,7 @@ from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
 from langgraph.channels.ephemeral_value import EphemeralValue
 from langgraph.graph import StateGraph
-from langgraph.graph import END, START
+from langgraph.graph.graph import END, START
 from langgraph.graph.state import CompiledStateGraph
 from langgraph.managed.base import is_managed_value
 from langgraph.prebuilt import InjectedStore

alita_sdk/runtime/llms/alita.py

@@ -12,24 +12,31 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+
+#
+#  This is adoption of https://github.com/langchain-ai/langchain/blob/master/libs/community/langchain_community/chat_models/openai.py
+#
+
 import logging
 import requests
 from time import sleep
 from traceback import format_exc
 
-from typing import Any, List, Optional, AsyncIterator, Dict, Iterator
+from typing import Any, List, Optional, AsyncIterator, Dict, Iterator, Mapping, Type
 from tiktoken import get_encoding, encoding_for_model
 from langchain_core.callbacks import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models import BaseChatModel
-from langchain_core.messages import AIMessageChunk, BaseMessage
+from langchain_core.language_models import BaseChatModel, SimpleChatModel
+from langchain_core.messages import (AIMessageChunk, BaseMessage, HumanMessage, HumanMessageChunk, ChatMessageChunk, 
+                                     FunctionMessageChunk, SystemMessageChunk, ToolMessageChunk, BaseMessageChunk,
+                                     AIMessage)
 from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
 from langchain_core.runnables import run_in_executor
 from langchain_community.chat_models.openai import generate_from_stream, _convert_delta_to_message_chunk
 from ..clients import AlitaClient
-from pydantic import Field, model_validator
+from pydantic import Field, model_validator, field_validator, ValidationInfo
 
 logger = logging.getLogger(__name__)
 

alita_sdk/runtime/utils/streamlit.py

@@ -1038,14 +1038,9 @@ def run_streamlit(st, ai_icon=None, user_icon=None):
                     {"input": prompt, "chat_history": st.session_state.messages[:-1]},
                     { 'callbacks': [st_cb], 'configurable': {"thread_id": st.session_state.thread_id}}
                 )
-                try:
-                    output =response["output"]
-                except KeyError:
-                    logger.info(response)
-                    output = response['messages'][0].content
-                st.write(output)
+                st.write(response["output"])
                 st.session_state.thread_id = response.get("thread_id", None)
-                st.session_state.messages.append({"role": "assistant", "content": output})
+                st.session_state.messages.append({"role": "assistant", "content": response["output"]})
     
     elif st.session_state.llm and st.session_state.show_toolkit_testing and st.session_state.configured_toolkit:
         # Toolkit Testing Main View

{alita_sdk-0.3.181.dist-info → alita_sdk-0.3.182.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.181
+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
@@ -32,7 +32,7 @@ Requires-Dist: langchain-openai~=0.3.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-postgres~=2.0.1; extra == "runtime"
 Requires-Dist: langsmith>=0.3.45; extra == "runtime"
-Requires-Dist: langgraph>=0.5; extra == "runtime"
+Requires-Dist: langgraph<0.5,>=0.4.8; extra == "runtime"
 Requires-Dist: langchain_chroma~=0.2.2; extra == "runtime"
 Requires-Dist: langchain-unstructured~=0.1.6; extra == "runtime"
 Requires-Dist: langchain-postgres~=0.0.13; extra == "runtime"

{alita_sdk-0.3.181.dist-info → alita_sdk-0.3.182.dist-info}/RECORD

@@ -13,21 +13,20 @@ alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=Ui1GBWizIFGFOi98
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=4N2t5x3GibyXLq3Fvrv2o_VA7Z000yNfc-UN4eGsHZg,2679
-alita_sdk/runtime/clients/client.py,sha256=f0-Re9KAjCer5vLqJNOn8qpq_tg7aT6YRbaJKgsZME0,19924
+alita_sdk/runtime/clients/client.py,sha256=6ezOJ92CSw6b2PVs4uFMQKQdp40uT1awoFEqWAfBH_A,20029
 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=QxFIwRdppm6TltoI7xrFMRlNlRjv54Q1lLHhNR8N6BY,9976
+alita_sdk/runtime/langchain/assistant.py,sha256=QJEMiEOrFMJ4GpnK24U2pKFblrvdQpKFdfhZsI2wAUI,7507
 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
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=HvBe8IWpAHcJmmXr3IsgEsqjn31NDmCzeQHdLLNe7tg,40499
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=ac96v-Nr7t1x1NzaOpvV1VuML3LfoNNIDTDl8pb-bSY,40505
 alita_sdk/runtime/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/runtime/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/runtime/langchain/store_manager.py,sha256=w5-0GbPGJAw14g0CCD9BKFMznzk1I-iJ5OGj_HZJZgA,2211
 alita_sdk/runtime/langchain/utils.py,sha256=Npferkn10dvdksnKzLJLBI5bNGQyVWTBwqp3vQtUqmY,6631
 alita_sdk/runtime/langchain/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/langchain/agents/react_agent.py,sha256=y461yEn2J7eP5GYbiPL3qaU3e9R8OmvyJNuFFvbZRCY,5971
 alita_sdk/runtime/langchain/agents/xml_chat.py,sha256=Mx7PK5T97_GrFCwHHZ3JZP42S7MwtUzV0W-_8j6Amt8,6212
 alita_sdk/runtime/langchain/document_loaders/AlitaBDDScenariosLoader.py,sha256=4kFU1ijrM1Jw7cywQv8mUiBHlE6w-uqfzSZP4hUV5P4,3771
 alita_sdk/runtime/langchain/document_loaders/AlitaCSVLoader.py,sha256=TBJuIFqweLDtd0JxgfPqrcY5eED-M617CT_EInp6Lmg,1949
@@ -64,7 +63,7 @@ alita_sdk/runtime/langchain/tools/bdd_parser/bdd_parser.py,sha256=DiEEOqDef2Xo3x
 alita_sdk/runtime/langchain/tools/bdd_parser/feature_types.py,sha256=l3AdjSQnNv1CE1NuHi7wts6h6AsCiK-iPu0PnPf3jf0,399
 alita_sdk/runtime/langchain/tools/bdd_parser/parser.py,sha256=1H1Nd_OH5Wx8A5YV1zUghBxo613yPptZ7fqNo8Eg48M,17289
 alita_sdk/runtime/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/llms/alita.py,sha256=-d0-U8epPeAz5DUMNlCxeWsNIWU2E4xFeTZNuzakNzE,9697
+alita_sdk/runtime/llms/alita.py,sha256=oAALCyrTbQ8gygJO7xzZLjOowpUnTbh8R363argRUVs,10119
 alita_sdk/runtime/llms/preloaded.py,sha256=3AaUbZK3d8fvxAQMjR3ftOoYa0SnkCOL1EvdvDCXIHE,11321
 alita_sdk/runtime/toolkits/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/toolkits/application.py,sha256=akqUuaIL9u7-SsUmS-XgN4qxDEnXFhsK9do4n8inpSo,2432
@@ -97,7 +96,7 @@ alita_sdk/runtime/utils/constants.py,sha256=Xntx1b_uxUzT4clwqHA_U6K8y5bBqf_4lSQw
 alita_sdk/runtime/utils/evaluate.py,sha256=iM1P8gzBLHTuSCe85_Ng_h30m52hFuGuhNXJ7kB1tgI,1872
 alita_sdk/runtime/utils/logging.py,sha256=svPyiW8ztDfhqHFITv5FBCj8UhLxz6hWcqGIY6wpJJE,3331
 alita_sdk/runtime/utils/save_dataframe.py,sha256=i-E1wp-t4wb17Zq3nA3xYwgSILjoXNizaQAA9opWvxY,1576
-alita_sdk/runtime/utils/streamlit.py,sha256=4xMucFDuU_S5ajtHYgaUBw29pbU5OygjuWe0CtTv7hw,85824
+alita_sdk/runtime/utils/streamlit.py,sha256=z4J_bdxkA0zMROkvTB4u379YBRFCkKh-h7PD8RlnZWQ,85644
 alita_sdk/runtime/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
 alita_sdk/tools/__init__.py,sha256=l3KxV-Qtu-04QQ9YYcovbLtEkZ30fGWDZ7o9nuRF16o,9967
 alita_sdk/tools/elitea_base.py,sha256=NQaIxPX6DVIerHCb18jwUR6maZxxk73NZaTsFHkBQWE,21119
@@ -296,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.181.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.181.dist-info/METADATA,sha256=gY8Qcg5Vw_J_tOT5QAwfxp6SKcS1eJHUYsM0Iu1W1j8,18746
-alita_sdk-0.3.181.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.181.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.181.dist-info/RECORD,,
+alita_sdk-0.3.182.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.182.dist-info/METADATA,sha256=WqwMApLsjTHN2hiq_uUi8e6SOE7PdJZZXf50UULLvPk,18753
+alita_sdk-0.3.182.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.182.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.182.dist-info/RECORD,,

alita_sdk/runtime/langchain/agents/react_agent.py

@@ -1,157 +0,0 @@
-import logging
-from typing import Any, Optional, Callable
-from langchain_core.messages import BaseMessage, AIMessage, HumanMessage
-from langchain_core.tools import BaseTool
-from langgraph.prebuilt import create_react_agent
-from langgraph.checkpoint.memory import MemorySaver
-
-logger = logging.getLogger(__name__)
-
-def create_react_agent_with_auto_continue(
-    prompt: Any,
-    model: Any,
-    tools: list[BaseTool],
-    memory: Optional[Any] = None,
-    debug: bool = False
-) -> Any:
-    """
-    Create a LangGraph React 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.
-    
-    Args:
-        prompt: The prompt template to use for the agent
-        model: The language model to use
-        tools: List of tools available to the agent
-        memory: Optional memory store for checkpointing (will create MemorySaver if None)
-        debug: Whether to enable debug mode
-        
-    Returns:
-        A configured React agent with auto-continue capability
-        
-    Note: Requires LangGraph 0.5.x or higher that supports post_model_hook.
-    """
-    # Use simple in-memory checkpointer for auto-continue functionality if not provided
-    if memory is None:
-        memory = MemorySaver()
-    
-    # Set up parameters for the agent
-    kwargs = {
-        "prompt": prompt,
-        "model": model,
-        "tools": tools,
-        "checkpointer": memory,
-        "post_model_hook": _create_auto_continue_hook()  # Auto-continue hook
-    }
-
-    # Create the base React agent with langgraph's prebuilt function
-    base_agent = create_react_agent(**kwargs)
-    
-    return base_agent
-
-def _create_auto_continue_hook() -> Callable:
-    """
-    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
-
-def get_langgraph_agent_with_auto_continue(
-    prompt: Any,
-    model: Any,
-    tools: list[BaseTool],
-    memory: Optional[Any] = None,
-    debug: bool = False
-) -> Any:
-    """
-    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.
-    
-    Args:
-        prompt: The prompt template to use for the agent
-        model: The language model to use
-        tools: List of tools available to the agent
-        memory: Optional memory store for checkpointing (will create MemorySaver if None)
-        debug: Whether to enable debug mode
-        
-    Returns:
-        A configured LangGraphAgentRunnable with auto-continue capability
-        
-    Note: Requires LangGraph 0.5.x or higher that supports post_model_hook.
-    """
-    from ...langchain.langraph_agent import LangGraphAgentRunnable
-    
-    # Use simple in-memory checkpointer for auto-continue functionality if not provided
-    if memory is None:
-        memory = MemorySaver()
-    
-    # Create the base React agent with auto-continue capability
-    base_agent = create_react_agent_with_auto_continue(
-        prompt=prompt,
-        model=model,
-        tools=tools,
-        memory=memory,
-        debug=debug
-    )
-    
-    # Wrap the base agent in our custom LangGraphAgentRunnable to handle input properly
-    # This ensures that our invoke() input handling logic is applied
-    agent = LangGraphAgentRunnable(
-        builder=base_agent.builder,
-        config_type=base_agent.builder.config_schema,
-        nodes=base_agent.nodes,
-        channels=base_agent.channels,
-        input_channels=base_agent.input_channels,
-        stream_mode=base_agent.stream_mode,
-        output_channels=base_agent.output_channels,
-        stream_channels=base_agent.stream_channels,
-        checkpointer=memory,
-        interrupt_before_nodes=base_agent.interrupt_before_nodes,
-        interrupt_after_nodes=base_agent.interrupt_after_nodes,
-        debug=debug,
-        store=base_agent.store, 
-        schema_to_mapper=base_agent.schema_to_mapper
-    )
-    
-    return agent