alita-sdk 0.3.131__py3-none-any.whl → 0.3.133__py3-none-any.whl

alita_sdk/clients/client.py

@@ -8,6 +8,7 @@ from langchain_core.messages import (
     AIMessage, HumanMessage,
     SystemMessage, BaseMessage,
 )
+from langchain_core.tools import ToolException
 
 from ..langchain.assistant import Assistant as LangChainAssistant
 # from ..llamaindex.assistant import Assistant as LLamaAssistant
@@ -174,7 +175,12 @@ class AlitaClient:
             tools = []
         if chat_history is None:
             chat_history = []
-        data = self.get_app_version_details(application_id, application_version_id)
+        try:
+            data = self.get_app_version_details(application_id, application_version_id)
+        except ApiDetailsRequestError as e:
+            error_msg = f"Failed to fetch application version details for {application_id}/{application_version_id}\nDetails: {e}"
+            logger.error(error_msg)
+            raise ToolException(error_msg)
 
         if application_variables:
             for var in data.get('variables', {}):

alita_sdk/community/analysis/ado_analyse/api_wrapper.py

@@ -17,7 +17,7 @@ from elitea_analyse.ado.main import (
 
 from alita_tools.elitea_base import BaseToolApiWrapper
 
-from src.alita_sdk.utils.save_dataframe import save_dataframe_to_artifact
+from ....utils.save_dataframe import save_dataframe_to_artifact
 from ....tools.artifact import ArtifactWrapper
 from ....utils.logging import with_streamlit_logs
 

alita_sdk/community/analysis/github_analyse/api_wrapper.py

@@ -12,7 +12,7 @@ from elitea_analyse.github.main_github import (
 )
 
 from alita_tools.elitea_base import BaseToolApiWrapper
-from src.alita_sdk.utils.save_dataframe import save_dataframe_to_artifact
+from ....utils.save_dataframe import save_dataframe_to_artifact
 from ....tools.artifact import ArtifactWrapper
 from ....utils.logging import with_streamlit_logs
 

alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py

@@ -13,7 +13,7 @@ from elitea_analyse.git.git_search import GitLabV4Search
 
 
 from alita_tools.elitea_base import BaseToolApiWrapper
-from src.alita_sdk.utils.save_dataframe import save_dataframe_to_artifact
+from ....utils.save_dataframe import save_dataframe_to_artifact
 from ....tools.artifact import ArtifactWrapper
 
 

alita_sdk/langchain/assistant.py

@@ -16,7 +16,7 @@ from .constants import REACT_ADDON, REACT_VARS, XML_ADDON
 from .chat_message_template import Jinja2TemplatedChatMessagesTemplate
 from ..tools.echo import EchoTool
 from ..toolkits.tools import get_tools
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, ToolException
 
 logger = logging.getLogger(__name__)
 
@@ -53,6 +53,11 @@ class Assistant:
             target_name
         )
         self.client = target_cls(**model_params)
+        # validate agents compatibility: non-pipeline agents cannot have pipelines as toolkits
+        if app_type != "pipeline" and any(tool['agent_type'] == 'pipeline' for tool in data['tools']):
+            raise ToolException("Non-pipeline agents cannot have pipelines as a toolkits. "
+                                "Review toolkits configuration or use pipeline as master agent.")
+
         self.tools = get_tools(data['tools'], alita_client=alita, llm=self.client)
         if app_type == "pipeline":
             self.prompt = data['instructions']

alita_sdk/tools/llm.py

@@ -3,7 +3,7 @@ from traceback import format_exc
 from typing import Any, Optional, Dict, List
 
 from langchain_core.messages import HumanMessage, BaseMessage
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, ToolException
 
 from ..langchain.utils import _extract_json, create_pydantic_model, create_params
 
@@ -13,7 +13,12 @@ logger = logging.getLogger(__name__)
 def create_llm_input(prompt: Dict[str, str], params: Dict[str, Any], kwargs: Dict[str, Any]) -> list[BaseMessage]:
     logger.info(f"Creating LLM input with prompt: {prompt}, params: {params}, kwargs: {kwargs}")
     if prompt.get('type') == 'fstring' and params:
-        return [HumanMessage(content=prompt['value'].format(**params))]
+        try:
+            return [HumanMessage(content=prompt['value'].format(**params))]
+        except KeyError as e:
+            error_msg = f"KeyError in input formatting - make sure you have added all required state variables as input to the node: {e}"
+            logger.error(error_msg)
+            raise ToolException(error_msg)
     else:
         return kwargs.get("messages") + [HumanMessage(prompt['value'])]
 

alita_sdk/utils/save_dataframe.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, Optional
 from langchain_core.tools import ToolException
 import pandas as pd
 
-from src.alita_sdk.tools.artifact import ArtifactWrapper
+from ..tools.artifact import ArtifactWrapper
 
 logger = logging.getLogger(__name__)
 

{alita_sdk-0.3.131.dist-info → alita_sdk-0.3.133.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.131
+Version: 0.3.133
 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>
 Project-URL: Homepage, https://projectalita.ai
@@ -70,7 +70,7 @@ Requires-Dist: rlpycairo==0.3.0
 Requires-Dist: cairocffi==1.7.1
 Requires-Dist: docx2txt==0.8
 Requires-Dist: mammoth==1.9.0
-Requires-Dist: elitea-analyse==0.1.1
+Requires-Dist: elitea-analyse==0.1.2
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.25.0
 Dynamic: license-file
 

{alita_sdk-0.3.131.dist-info → alita_sdk-0.3.133.dist-info}/RECORD

@@ -4,24 +4,24 @@ alita_sdk/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 alita_sdk/agents/llamaAgentParser.py,sha256=N_Nw6WJ8xrNX3fr_JFwjUIg_Urai9lLU3poMgZz0Cyk,1701
 alita_sdk/clients/__init__.py,sha256=5O_7WsZmvg5z5uZf_Jkx-f8j5C6yKIuSwQR6AQ-TM84,31
 alita_sdk/clients/artifact.py,sha256=W6oLlthtsUHfUWjhihaDdhLZdRVqj1D2j7oHHELxJfs,2639
-alita_sdk/clients/client.py,sha256=egrBkmuNh79WTaCLO2sBiTHA1YBT6R4pif-nWZbnSNw,18475
+alita_sdk/clients/client.py,sha256=_-2S7wNZYkr6JgOtJJkUQbFWT_XFYdOt1iAi9BivNEE,18793
 alita_sdk/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/community/utils.py,sha256=lvuCJaNqVPHOORJV6kIPcXJcdprVW_TJvERtYAEgpjM,249
 alita_sdk/community/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/community/analysis/ado_analyse/__init__.py,sha256=JU-WsNdLp6GyhgftwW0COE37bAH3mciOFYYw58FawwQ,4056
-alita_sdk/community/analysis/ado_analyse/api_wrapper.py,sha256=qLkpm4XPmhuDC0nJwXAaDOcuI3XURVjYLufpKZ9NJnY,8649
+alita_sdk/community/analysis/ado_analyse/api_wrapper.py,sha256=efuFvIVlB2sLaViBW0mqCXFozeYPM4jlC6l7-FaR91A,8639
 alita_sdk/community/analysis/github_analyse/__init__.py,sha256=QHVuYoTw6r89smShN3Iurhz_XLicPk1G8s2HeOMEk1s,3550
-alita_sdk/community/analysis/github_analyse/api_wrapper.py,sha256=pDZlpufTWSCtD0m-TI9aIgctFe62uJ7diX3LTsPdWjY,6836
+alita_sdk/community/analysis/github_analyse/api_wrapper.py,sha256=Otx1kbDqn0O1phz3qNJ5-0soO9_uU4zZ_YnB0tN6l9w,6826
 alita_sdk/community/analysis/gitlab_analyse/__init__.py,sha256=J_HwgZGfiQPxEqFD0IOulgettTI_fvpuqYdR_ciKRFw,4131
-alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py,sha256=GDP_cW8Faa695357M9YXPsQ12WBP33JZQy5CGcCAS1A,6343
+alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py,sha256=FZ2StzLH0MtO26guL51wV-ar457ygJtuPEvhqg-IXGM,6333
 alita_sdk/community/analysis/jira_analyse/__init__.py,sha256=Rm-HKEi_HIxrgHdq9mZ-XzxMKLXm8-81eJwJT2lar-c,5945
 alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=naEgBSdwNonNleUtHCb1UOkWiYdM64ZJ9dfsyszmeX8,9668
 alita_sdk/community/browseruse/__init__.py,sha256=uAxPZEX7ihpt8HtcGDFrzTNv9WcklT1wG1ItTwUO8y4,3601
 alita_sdk/community/browseruse/api_wrapper.py,sha256=Y05NKWfTROPmBxe8ZFIELSGBX5v3RTNP30OTO2Tj8uI,10838
 alita_sdk/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/langchain/assistant.py,sha256=J_xhwbNl934BgDKSpAMC9a1u6v03DZQcTYaamCztEPk,6272
+alita_sdk/langchain/assistant.py,sha256=Wq4_XtlFKKkrkelvTa4qzkI3BBTHy5zwRmk3v1i22lQ,6676
 alita_sdk/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
@@ -85,7 +85,7 @@ alita_sdk/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI
 alita_sdk/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
 alita_sdk/tools/function.py,sha256=wzPS5Y8ScmsXspmn73MQnkCcAghsllAg7BLh61ZNuVw,2543
 alita_sdk/tools/indexer_tool.py,sha256=P9S_omk5TZkizb6UXyxMO87Pzj4UCaye0CuXBgCnhTU,4258
-alita_sdk/tools/llm.py,sha256=JA0OnSU13CLdkS5NFv6iRk8P7k-B47L-oPjk8xrzk48,3223
+alita_sdk/tools/llm.py,sha256=UUd4U4CuCNImHxVJAF5daYbew-7X9H8k21bUWy2PDvg,3502
 alita_sdk/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
 alita_sdk/tools/mcp_server_tool.py,sha256=xcH9AiqfR2TYrwJ3Ixw-_A7XDodtJCnwmq1SsikXpYk,1930
@@ -98,16 +98,16 @@ alita_sdk/utils/AlitaCallback.py,sha256=cvpDhR4QLVCNQci6CO6TEUrUVDZU9_CRSwzcHGm3
 alita_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/utils/evaluate.py,sha256=iM1P8gzBLHTuSCe85_Ng_h30m52hFuGuhNXJ7kB1tgI,1872
 alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,3124
-alita_sdk/utils/save_dataframe.py,sha256=6aHFr28Ssw2jmsnrCTVfZIE2MCpVkG6GV58gzMN-Eao,1477
+alita_sdk/utils/save_dataframe.py,sha256=tNwnaVCvN4_B0oi3F4Y3Z13elbIRtwKmdKHsMcj658g,1465
 alita_sdk/utils/streamlit.py,sha256=zp8owZwHI3HZplhcExJf6R3-APtWx-z6s5jznT2hY_k,29124
 alita_sdk/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk-0.3.131.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.133.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_ado_analysis.py,sha256=wsxB4B2Ycxoiykthh6YbPQ9hqsDbPFie8D9ZK1i_6kg,3311
 tests/test_github_analysis.py,sha256=ulR4CEGmiMRPydJuX7aQcglzvhC7kFOAtZRLLBB9F_M,3148
 tests/test_gitlab_analysis.py,sha256=J7Y2mNi5Sj8-rH2PMRmVbT3uwZ17YeR9pcs0MDIyNW4,3352
 tests/test_jira_analysis.py,sha256=6F3Elikt02L28N6sS_AKDy9lgqgD81_hr979NcdZeg4,3359
-alita_sdk-0.3.131.dist-info/METADATA,sha256=u6rbcQEd9_oXtzErr6KKmk6ZUxEq9XoUmH6Y5Utfjzk,7076
-alita_sdk-0.3.131.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.131.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.131.dist-info/RECORD,,
+alita_sdk-0.3.133.dist-info/METADATA,sha256=DSI3loS__2aAfwa-Mdbvo5hYY42e3WY0WFxHmQ3FvJ8,7076
+alita_sdk-0.3.133.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.133.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.133.dist-info/RECORD,,