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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.179
+Version: 0.3.181
 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,>=0.4.8; extra == "runtime"
+Requires-Dist: langgraph>=0.5; 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.179 → alita_sdk-0.3.181}/alita_sdk/runtime/clients/client.py

@@ -218,8 +218,6 @@ 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-0.3.181/alita_sdk/runtime/langchain/agents/react_agent.py

@@ -0,0 +1,157 @@
+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

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

@@ -12,6 +12,7 @@ from langgraph.store.base import BaseStore
 # 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
@@ -75,13 +76,31 @@ 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 == "react":
-                messages.append(HumanMessage(REACT_ADDON))
-            elif app_type == "xml":
+            if app_type == "xml":
                 messages.append(HumanMessage(XML_ADDON))
             elif app_type in ['openai', 'dial']:
                 messages.append(HumanMessage("{{input}}"))
@@ -93,7 +112,7 @@ class Assistant:
                     variables[variable['name']] = variable['value']
                 else:
                     input_variables.append(variable['name'])
-            if app_type in ["react", "xml"]:
+            if app_type in ["xml"]:
                 input_variables = list(set(input_variables + REACT_VARS))
 
             if chat_history and isinstance(chat_history, list):
@@ -186,75 +205,16 @@ 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.
-        """
-        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(
+        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=simple_tools,
-            checkpointer=memory,
-            state_modifier=self._create_auto_continue_state_modifier()
+            tools=self.tools,
+            memory=self.memory,
+            debug=True
         )
-        
-        return agent
-    
-    def _create_auto_continue_state_modifier(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.
-        """
-        MAX_CONTINUATIONS = 3  # Maximum number of auto-continuations allowed
-        
-        def auto_continue_modifier(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 AI message 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 logic
-                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]
-                    }
-            
-            return state
-        
-        return auto_continue_modifier
+

{alita_sdk-0.3.179 → alita_sdk-0.3.181}/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.graph import END, START
+from langgraph.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-0.3.179 → alita_sdk-0.3.181}/alita_sdk/runtime/llms/alita.py

@@ -12,31 +12,24 @@
 # 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, Mapping, Type
+from typing import Any, List, Optional, AsyncIterator, Dict, Iterator
 from tiktoken import get_encoding, encoding_for_model
 from langchain_core.callbacks import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-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.language_models import BaseChatModel
+from langchain_core.messages import AIMessageChunk, BaseMessage
 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, field_validator, ValidationInfo
+from pydantic import Field, model_validator
 
 logger = logging.getLogger(__name__)
 

{alita_sdk-0.3.179 → alita_sdk-0.3.181}/alita_sdk/runtime/utils/streamlit.py

@@ -1038,9 +1038,14 @@ 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}}
                 )
-                st.write(response["output"])
+                try:
+                    output =response["output"]
+                except KeyError:
+                    logger.info(response)
+                    output = response['messages'][0].content
+                st.write(output)
                 st.session_state.thread_id = response.get("thread_id", None)
-                st.session_state.messages.append({"role": "assistant", "content": response["output"]})
+                st.session_state.messages.append({"role": "assistant", "content": 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.179 → alita_sdk-0.3.181}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.179
+Version: 0.3.181
 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,>=0.4.8; extra == "runtime"
+Requires-Dist: langgraph>=0.5; 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.179 → alita_sdk-0.3.181}/alita_sdk.egg-info/SOURCES.txt

@@ -35,6 +35,7 @@ alita_sdk/runtime/langchain/mixedAgentRenderes.py
 alita_sdk/runtime/langchain/store_manager.py
 alita_sdk/runtime/langchain/utils.py
 alita_sdk/runtime/langchain/agents/__init__.py
+alita_sdk/runtime/langchain/agents/react_agent.py
 alita_sdk/runtime/langchain/agents/xml_chat.py
 alita_sdk/runtime/langchain/document_loaders/AlitaBDDScenariosLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaCSVLoader.py

{alita_sdk-0.3.179 → alita_sdk-0.3.181}/alita_sdk.egg-info/requires.txt

@@ -37,7 +37,7 @@ langchain-openai~=0.3.0
 langgraph-checkpoint-sqlite~=2.0.0
 langgraph-checkpoint-postgres~=2.0.1
 langsmith>=0.3.45
-langgraph<0.5,>=0.4.8
+langgraph>=0.5
 langchain_chroma~=0.2.2
 langchain-unstructured~=0.1.6
 langchain-postgres~=0.0.13

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.179"
+version = "0.3.181"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -32,7 +32,7 @@ name = "Artem Dubrovskiy"
 email = "ad13box@gmail.com"
 
 [project.optional-dependencies]
-runtime = [ "langchain_core==0.3.49", "langchain<=0.3.22", "langchain_community~=0.3.7", "langchain-openai~=0.3.0", "langgraph-checkpoint-sqlite~=2.0.0", "langgraph-checkpoint-postgres~=2.0.1", "langsmith>=0.3.45", "langgraph>=0.4.8,<0.5", "langchain_chroma~=0.2.2", "langchain-unstructured~=0.1.6", "langchain-postgres~=0.0.13", "keybert==0.8.3", "charset_normalizer==3.3.2", "unstructured[local-inference]==0.16.23", "unstructured_pytesseract==0.3.13", "unstructured_inference==0.8.7", "python-pptx==1.0.2", "pdf2image==1.16.3", "pikepdf==8.7.1", "pypdf==4.3.1", "pdfminer.six==20240706", "opencv-python==4.11.0.86", "python-docx==1.1.2", "openpyxl==3.1.2", "markdown==3.5.1", "beautifulsoup4==4.12.2", "gensim==4.3.3", "chromadb==0.5.20", "pgvector==0.2.5", "scipy==1.13.1", "pytesseract==0.3.13", "reportlab==4.2.5", "svglib==1.5.1", "rlpycairo==0.3.0", "cairocffi==1.7.1", "docx2txt==0.8", "mammoth==1.9.0", "opentelemetry-exporter-otlp-proto-grpc==1.25.0", "opentelemetry_api==1.25.0", "opentelemetry_instrumentation==0.46b0", "grpcio_status==1.63.0rc1", "protobuf==4.25.7", "sentence-transformers==2.7.0",]
+runtime = [ "langchain_core==0.3.49", "langchain<=0.3.22", "langchain_community~=0.3.7", "langchain-openai~=0.3.0", "langgraph-checkpoint-sqlite~=2.0.0", "langgraph-checkpoint-postgres~=2.0.1", "langsmith>=0.3.45", "langgraph>=0.5", "langchain_chroma~=0.2.2", "langchain-unstructured~=0.1.6", "langchain-postgres~=0.0.13", "keybert==0.8.3", "charset_normalizer==3.3.2", "unstructured[local-inference]==0.16.23", "unstructured_pytesseract==0.3.13", "unstructured_inference==0.8.7", "python-pptx==1.0.2", "pdf2image==1.16.3", "pikepdf==8.7.1", "pypdf==4.3.1", "pdfminer.six==20240706", "opencv-python==4.11.0.86", "python-docx==1.1.2", "openpyxl==3.1.2", "markdown==3.5.1", "beautifulsoup4==4.12.2", "gensim==4.3.3", "chromadb==0.5.20", "pgvector==0.2.5", "scipy==1.13.1", "pytesseract==0.3.13", "reportlab==4.2.5", "svglib==1.5.1", "rlpycairo==0.3.0", "cairocffi==1.7.1", "docx2txt==0.8", "mammoth==1.9.0", "opentelemetry-exporter-otlp-proto-grpc==1.25.0", "opentelemetry_api==1.25.0", "opentelemetry_instrumentation==0.46b0", "grpcio_status==1.63.0rc1", "protobuf==4.25.7", "sentence-transformers==2.7.0",]
 tools = [ "dulwich==0.21.6", "paramiko==3.3.1", "pygithub==2.3.0", "python-gitlab==4.5.0", "gitpython==3.1.43", "atlassian-python-api~=3.41", "atlassian_python_api==3.41.16", "jira==3.8.0", "qtest-swagger-client==0.0.3", "testrail-api==1.13.2", "azure-devops==7.1.0b4", "msrest==0.7.1", "python-graphql-client~=0.4.3", "zephyr-python-api==0.1.0", "pyral==1.6.0", "boto3>=1.37.23", "azure-core==1.30.2", "azure-identity==1.16.0", "azure-keyvault-keys==4.9.0", "azure-keyvault-secrets==4.8.0", "azure-mgmt-core==1.4.0", "azure-mgmt-resource==23.0.1", "azure-mgmt-storage==21.1.0", "azure-storage-blob==12.23.1", "azure-search-documents==11.5.2", "PyMySQL==1.1.1", "psycopg2-binary==2.9.10", "Office365-REST-Python-Client==2.5.14", "python-docx==1.1.2", "python-pptx==1.0.2", "pypdf2~=3.0.1", "FigmaPy==2018.1.0", "pandas==2.2.3", "factor_analyzer==0.5.1", "statsmodels==0.14.4", "tabulate==0.9.0", "tree_sitter==0.20.2", "tree-sitter-languages==1.10.2", "astor~=0.8.1", "markdownify~=1.1.0", "requests_openapi==1.0.5", "duckduckgo_search==5.3.0", "playwright>=1.52.0", "google-api-python-client==2.154.0", "wikipedia==1.4.0", "lxml==5.2.2", "beautifulsoup4", "pymupdf==1.24.9", "googlemaps==4.10.0", "yagmail==0.15.293", "pysnc==1.1.10", "shortuuid==1.0.13", "yarl==1.17.1", "langmem==0.0.27", "textract-py3==2.1.1",]
 community = [ "retry-extended==0.2.3", "pyobjtojson==0.3", "elitea-analyse==0.1.2",]
 all = [ "alita-sdk[runtime]", "alita-sdk[tools]", "alita-sdk[community]",]