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

alita_sdk/runtime/utils/streamlit.py

@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 
 ai_icon = b'<plain_txt_msg:img>iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAAAXNSR0IArs4c6QAAAMZlWElmTU0AKgAAAAgABgESAAMAAAABAAEAAAEaAAUAAAABAAAAVgEbAAUAAAABAAAAXgEoAAMAAAABAAIAAAExAAIAAAAVAAAAZodpAAQAAAABAAAAfAAAAAAAAABIAAAAAQAAAEgAAAABUGl4ZWxtYXRvciBQcm8gMy41LjEAAAAEkAQAAgAAABQAAACyoAEAAwAAAAEAAQAAoAIABAAAAAEAAAAgoAMABAAAAAEAAAAgAAAAADIwMjQ6MDQ6MDMgMTk6NDA6NDQATjJeeQAAAAlwSFlzAAALEwAACxMBAJqcGAAAA7BpVFh0WE1MOmNvbS5hZG9iZS54bXAAAAAAADx4OnhtcG1ldGEgeG1sbnM6eD0iYWRvYmU6bnM6bWV0YS8iIHg6eG1wdGs9IlhNUCBDb3JlIDYuMC4wIj4KICAgPHJkZjpSREYeG1sbnM6cmRmPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5LzAyLzIyLXJkZi1zeW50YXgtbnMjIj4KICAgICAgPHJkZjpEZXNjcmlwdGlvbiByZGY6YWJvdPC9IgogICAgICAgICAgICAgeG1sbnM6dGlmZj0iaHR0cDovL25zLmFkb2JlLmNvbS90aWZmLzEuMC8iCiAgICAgICAgICAgIHhtbG5zOmV4aWY9X2h0dHA6Ly9ucy5hZG9iZS5jb20vZXhpZi8xLjAvIgogICAgICAgICAgICB4bWxuczp4bXA9Imh0dHA6Ly9ucy5hZG9iZS5jb20veGFwLzEuMC8iPgogICAgICAgICA8dGlmZjpZUmVzb2x1dGlvbj43MjAwMDAvMTAwMDA8L3RpZmY6WVJlc29sdXRpb24+CiAgICAgICAgIDx0aWZmOlhSZXNvbHV0aW9uPjcyMDAwMC8xMDAwMDwvdGlmZjpYUmVzb2x1dGlvbj4KICAgICAgICAgPHRpZmY6UmVzb2x1dGlvblVuaXQ+MjwvdGlmZjpSZXNvbHV0aW9uVW5pdD4KICAgICAgICAgPHRpZmY6T3JpZW50YXRpb24+MTwvdGlmZjpPcmllbnRhdGlvbj4KICAgICAgICAgPGV4aWY6UGl4ZWxZRGltZW5zaW9uPjMyPC9leGlmOlBpeGVsWURpbWVuc2lvbj4KICAgICAgICAgPGV4aWY6UGl4ZWxYRGltZW5zaW9uPjMyPC9leGlmOlBpeGVsWERpbWVuc2lvbj4KICAgICAgICAgPHhtcDpNZXRhZGF0YURhdGU+MjAyNC0wNC0wM1QxOTo0Mjo1OSswMzowMDwveG1wOk1ldGFkYXRhRGF0ZT4KICAgICAgICAgPHhtcDpDcmVhdGVEYXRlPjIwMjQtMDQtMDNUMTk6NDI6MjMrMDM6MDA8L3htcDpDcmVhdGVEYXRlPgogICAgICAgICA8eG1wOkNyZWF0b3JUb29sPlBpeGVsbWF0b3IgUHJvIDMuNS4xPC94bXA6Q3JlYXRvclRvb2w+CiAgICAgIDwvcmRmOkRlc2NyaXB0aW9uPgogICA8L3JkZjpSREY+CjwveDp4bXBtZXRhPgrdpg3+AAAH30lEQVRYCe1Xe4yU1RU/936P+eab187s7izIo4LIo1IR12AJpRIphNIlrVbQtqSVNgVaGw0+Uki02SiYJlaSltoE0kSJFRUo6m4LBWxdwdZQqSWuPOxSyurCzj5nZ+ab+d739txvd8YBFtz/25vcvWfuveec3/mdc+/9FuB/vZExEkD+ef9rCcfkadM358xOTmvMOYWpJddMUiL5UTWccbh35kKp91hSSXa4EX/wth0rTLTNP8v+ZwEgh7+7e9IErX6pRpVl2BeqkpoeVhJ/L7PPCc+5xbMc+KG8Zxx485P2tscOPVa8FohrAjj6w5amcXr6kRBR50mU6tcyVL3GEYHP/UzRM4+cGux44p5dazqq16vlqwEgr31v18Y5DTdt4QSutqfazlVlBNJ3sueD1XftvP/QaJuuMN66trUuGYs3p8LJBy5RYByo6YNkuCAPOSAVPSAuAwGPhSTw4wp4CRWYLgPTpIoqYR7I/R1d+WzXo417Hnm1sjAifLoTJ7avbdXrk/FHY5HEel8iqo+rOAJzfaC9JsgZE+iABdz2wUeefXQuOvOR85IHkHcATC+Yc3QEpVBgxIOY0R0PcZi37LrG0y+dffvf1SBo9Y9EnC4Jx2seclUpaisELOyu6QJ0FcFH4xbuLkVlKMVG7yY6tZEpJkBmSmATBj51QOYGaAqdPKt24s59izdNr/ZZYWDl1t2p20OTWmVVrRdR+5QAR8pJnwU+argqBTf0aXdC4DkKsRwVPE+TJDckkco6Ru4jEO74EJUyELGzgU88stGYFK+7Ib94/8H8QaQMQA5WAOjEL8x9PH+RTpSGrGCKIM1y0QEvjGFrtFyJ3PPc/3iu9XfMQEfJMXoY4XKNmpzsEzZbVcO3U0oT4nCK4iLIQLx0AfCqAM7RBsXU2TWLZo5LLoIuOCAcBQDWth6fHkqk7roALmglG6iHBYc5dUUxabgrOO7csUxjr2saz/2t6+CH2w5sywsDojVDM629b9K4VKLui+F43RZV1WYKACm/BzTbQOcSEATAGQWrmGwIy+7y1sbtb6/4x7pSAICE9Tu5QtOFeAiyWMnRPhMIRl3dsvn+3+7auvmhNmgLqKteQwAMXoGLOLdv2rRpf3xi9Qt/1rXoglmFcxi9hwCQDQRQyNWBw0KUcWcJeMmf4/4Snb9ha1ivr78FVDks8txfp0MxLFUKrRiVWR8Yv1v79OjOq4EI+ezZs3avM/DjBOtoj5JsQL9IgcsU6B2aABxrS1cj03Ikf73YT2es+moDV+QpXCKESxSMRBiG6vUKgILmn7roZp+EUSIXBkZrc/Jv9M8oHe9WKNYT5p0Bhd78BPCIDOgHQVApGapZKHRlSZaTmJp6hsiIxMFDFoYaYhCVsQ48n1klc2/vJxfPj+boanNTY/KSeMj5EvrGewRZNcdD1kpjEeKdggAIdsTweaFPfdsNeczTh5EhOnEEVQmKtXEwNDKQhdKJHTvW4WUwttbykxVTEjH6lCxzHdMOA9Z4uFC4EY91OXrBAN4x3KkVFikjhICMB0ZQIzouipEhchaJSjRWO2VF8/YxPUQdT35z7oLra/fpGplk4/HpLM6Cjtyt4RIFBMNBD2yjD5kEVU6JpjigKLZwXNkgZNwIspLSI8lNc+Z+Zcf6X+2fi4DF6bqiNWMgJ5/+2sJ0KvScHia3ZJzPQXv+y3C+NBvzrlSCCgIUQWKXJCU4xlRPpg2ihbLVi4EsmMBOVDmtRePfmXzjrceaXz7d8oPHX1wG0KiUUSxCmPf+YukD4+rqWrr5nPmHBr8P7+WboN+dgG+ChM7wGgm6CBDlkUAjWuy8sCGffuP3PdO/8fWPqRYJioPg6yKKRRRK0FlZpooWTzbNbFzc9Owrd7hWNnuO2s7QdWpp6gnQ6i0jAcQGkPDQS1h9w/p4++FDFchYkESkleELinfSOaPzrwGAtg1rcjcsz3zIwhGbUhKqMIGOAxkBVeZGZDw7SiSRmiHhc1x0GFg4ShS7xPC2wzdA0Cz20ioZT1jABAKxidOtRWLtAoAoBG4N9h/hhOTKhShGUQ/CUDAnaBuRq+cr8mXrlVoq64zYK9uwqf+OyWCwDABemj/7XdvIvRc4FI7LzssjGqjMEXzjCM+4zH0/m8u0ZHO9B/BB+ogBG8QcswpbQqdKb3iegkdZwSL+H9a9uSoowvJryIqdnU+pM/UFVFJqgs1l6nEUyCnWk2e753zLeLl4oXPf+/u2tLe1Db8LK1c2R+fefMeiWCh5rwahe6hEtOEUjIAo28B6wK/nY31m/1uCecFA9bEia052bYzUpTdLLqcS5lbkOBhRZnnjjNnf+yO/u/vdbQ8ux3K7sjVv2J3So+ObUlrNNpXL8bJu2Q51fD8z+HHT+j13/6msXQ0Avn30g6TekN6pR1MrJPyYQCABAG4UB/71l/23vb5pzfmy4rXGZ3761uraWP321ad6dSCd3Wc2PfzqfeIVrLTgNir/2rXw5qw3lP+ZY5WODBcM3uXg9eWGer81VufCVrHQ97rjmjt8yl1RkFgzRtbJ/vJy52LvJQDExAvzpp9w+noeLGUH9/qc2aVS4fnBXOEdsTbW1vybVUbfYNevHd/9CL+U8jkr+8yQZ2weTf+SFFRvWHn4eCJC1aWcSKd23nnTyeq1Mcrk2Y2H73aKViGUMY8+vGeV+Fft/+0KBv4LJG7QdeOMt6wAAAAASUVORK5CYII=<!plain_txt_msg>'
 user_icon = b'<plain_txt_msg:img>iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAAAXNSR0IArs4c6QAAAMZlWElmTU0AKgAAAAgABgESAAMAAAABAAEAAAEaAAUAAAABAAAAVgEbAAUAAAABAAAAXgEoAAMAAAABAAIAAAExAAIAAAAVAAAAZodpAAQAAAABAAAAfAAAAAAAAABIAAAAAQAAAEgAAAABUGl4ZWxtYXRvciBQcm8gMy41LjEAAAAEkAQAAgAAABQAAACyoAEAAwAAAAEAAQAAoAIABAAAAAEAAAAgoAMABAAAAAEAAAAgAAAAADIwMjQ6MDQ6MDMgMTk6NDI6MjMAfz7nbAAAAAlwSFlzAAALEwAACxMBAJqcGAAAA7BpVFh0WE1MOmNvbS5hZG9iZS54bXAAAAAAADx4OnhtcG1ldGEgeG1zbnM6eD0iYWRvYmU6bnM6bWV0YS8iIHg6eG1wdGs9IlhNUCBDb3JlIDYuMC4wIj4KICAgPHJkZjpSREYgeG1zbnM6cmRmPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5LzAyLzIyLXJkZi1zeW50YXgtbnMjIj4KICAgICAgPHJkZjpEZXNjcmlwdGlvbiByZGY6YWJvdXQ9IiIKICAgICAgICAgICAgeG1zbnM6dGlmZj0iaHR0cDovL25zLmFkb2JlLmNvbS90aWZmLzEuMC8iCiAgICAgICAgICAgIHhtbG5zOmV4aWY9Imh0dHA6Ly9ucy5hZG9iZS5jb20vZXhpZi8xLjAvIgogICAgICAgICAgICB4bWxuczp4bXA9Imh0dHA6Ly9ucy5hZG9iZS5jb20veGFwLzEuMC8iPgogICAgICAgICA8dGlmZjpZUmVzb2x1dGlvbj43MjAwMDAvMTAwMDA8L3RpZmY6WVJlc29sdXRpb24+CiAgICAgICAgIDx0aWZmOlhSZXNvbHV0aW9uPjcyMDAwMC8xMDAwMDwvdGlmZjpYUmVzb2x1dGlvbj4KICAgICAgICAgPHRpZmY6UmVzb2x1dGlvblVuaXQ+MjwvdGlmZjpSZXNvbHV0aW9uVW5pdD4KICAgICAgICAgPHRpZmY6T3JpZW50YXRpb24+MTwvdGlmZjpPcmllbnRhdGlvbj4KICAgICAgICAgPGV4aWY6UGl4ZWxZRGltZW5zaW9uPjMyPC9leGlmOlBpeGVsWURpbWVuc2lvbj4KICAgICAgICAgPGV4aWY6UGl4ZWxYRGltZW5zaW9uPjMyPC9leGlmOlBpeGVsWERpbWVuc2lvbj4KICAgICAgICAgPHhtcDpNZXRhZGF0YURhdGU+MjAyNC0wNC0wM1QxOTo0Mjo1OSswMzowMDwveG1wOk1ldGFkYXRhRGF0ZT4KICAgICAgICAgPHhtcDpDcmVhdGVEYXRlPjIwMjQtMDQtMDNUMTk6NDI6MjMrMDM6MDA8L3htcDpDcmVhdGVEYXRlPgogICAgICAgICA8eG1wOkNyZWF0b3JUb29sPlBpeGVsbWF0b3IgUHJvIDMuNS4xPC94bXA6Q3JlYXRvclRvb2w+CiAgICAgIDwvcmRmOkRlc2NyaXB0aW9uPgogICA8L3JkZjpSREY+CjwveDp4bXBtZXRhPgorTryeAAAFAUlEQVRYCc1XbWgcRRie2Zndy6XGpD+ijbZWxT8G7QfKnWlremk0JpriR7jUVsUgiGK1EhQKWnGFKiqUSLTJBbRJG6r2IuJHkNgkJrZJSxIQLWiJloI/mhgLirnE3O19jM/s7SZXLnd7VwRduHtnZ96PZ96d9312CfmPL5pvfN0f1NisaI6ovCHi4uVhzgqjLjaD+7GYqrV2B7Z+k4/PvAAc2Ny4KqapxyIqKzU0TiIaww9SXRobKjuhMF7b07JpIRcgSi5KUqfdc2SbQuhxDEvlvSAkIihpJVTsxW2fnJMX5ioXGDmv6yIn3zxplv2/bcvRlQnD6KeEmU4Zide83ts4QBDdsnzbpw/xVTP0C9zX4bfqlDE+ArnJWs8ockJJotGX4CEZPCZqXz6+qz8luOl8WK+Kfdzuu5dQ+i2ygAzRCoAqyBjZWnAEMOQb4tjnC1IffodePPnI19mcMpf7fqzHZGoSpLglm65ccwTwc+jiCiGfbPK4Bp0c9rx1+1/Y/XRSTzzgpO8IIMoWFs8JjZMLTg7NdUHsCli0zWTnCECNu2O2sWDkWnucVVLitjK2aJtJ3xHAdFHpPPJvnnb87cjkyJ737+0vpkSUJe/FZ/Z8JukIQB+uiglBDkgHOAm+N6uO3pPJmZyPRQo+hzBTr5BQczZdueYIwHTgUt9AXSXMOqC077Wa4N0mnBTvsg88/MzwV8jVVnNakFGUZjhFZdlhzq34vYquaoOxPkNjPNl6eQTtuCOs8QuGplRhrtZszbItu9i0p7Bita5TgM5+5QxAumnxdfsiCg9KLrB5wOQE1eKEJDcMF5esruvSb3DcvfSZFwBpINmQSjbUeIOhqeURTXFj978jM2MRrrR+FKjOiw1zOwMyMq6DvuAVBX/En0S7vRPYr0KDUoWgskm5USFrBWU7Hnx+aEtSO7f/nDLQ6essmA+TQJRzP1JfaFKwScPJ1GP3l1Kyxs+FNaV5cL+31wmGI4CA54P1CaKAYGgxnj0xgyE45ACAjOKFZM5w8etxGLfjXeC6VHDQOX3y1duyMmJWAO2eQ/Xo61/KXchMI8Ck4ebbFvonf9OJnnbCZSmWzGqPGkx5H5liSTBsaiKybg3JUBEZAbR5O+9A7Z82gxMRw3PeuXvsiU+cUmqvV+8b/x6Hc72slqjGzpaELq6TlG2v23LZQ9jqbb0SwQcspagQyoZ8gku7wf2eDUShPbLQEoTePLOy7F07aKpMAxD0BxkXRR9CaYVUROrrn51o+jHVKNdx2ZlfdgoqvjP9KPTptR0/WRyx5CENQGgqVIjlSkslvHusSb4HXtbV09MYxxlqksaSzWhCe0yOU680AEXXFP0NhROWUsFBb1dNqkE+Yz+ySYU4bNsIxei2x7ZMA9AI1DEa2gWFeakEB70B76Fy2yAfOXXrTUF0qY3SRkmIwK9PlVtvSkte0gDIpT1je2Zxdu6y1FTk8Yd2b1fDkpnzqHrfxBmU2EOW5tmr/5x+bjmrjGUolds8Xffh6ZndLKGgD3A2GXXoA8UhtQms2YEeoMgSRO+4vD5go7U7IQBYndD6ElLZIGh3BIQ0F1bZjYaL1aPu15jNx2JHdMhTI69s3Gz7Wk5mzYBtILlgTnKByv1ouYWLn2WXfJItfZ5h5+cM9V/iAhuElDrYUGXxx6MuXgdeuAU7LsNPRSZmAeo8xuNRlXZ/+k7laKrd/3r8D5cLzopAT7EBAAAAAElFTkSuQmCC<!plain_txt_msg>'
-agent_types = ["pipeline", "react", "xml", "openai"]
+agent_types = ["pipeline", "react", "xml", "openai", "predict"]
 
 def img_to_txt(filename):
     msg = b"<plain_txt_msg:img>"
@@ -1034,6 +1034,7 @@ def run_streamlit(st, ai_icon=None, user_icon=None):
             st.session_state.messages.append({"role": "user", "content": prompt})
             with st.chat_message("assistant", avatar=ai_icon):
                 st_cb = AlitaStreamlitCallback(st)
+                logger.info(st.session_state.messages)
                 response = st.session_state.agent_executor.invoke(
                     {"input": prompt, "chat_history": st.session_state.messages[:-1]},
                     { 'callbacks': [st_cb], 'configurable': {"thread_id": st.session_state.thread_id}}

alita_sdk/tools/__init__.py

@@ -79,6 +79,7 @@ _safe_import_tool('pptx', 'pptx', 'get_tools', 'PPTXToolkit')
 _safe_import_tool('postman', 'postman', 'get_tools', 'PostmanToolkit')
 _safe_import_tool('memory', 'memory', 'get_tools', 'MemoryToolkit')
 _safe_import_tool('zephyr_squad', 'zephyr_squad', 'get_tools', 'ZephyrSquadToolkit')
+_safe_import_tool('slack', 'slack', 'get_tools', 'SlackToolkit')
 
 # Log import summary
 available_count = len(AVAILABLE_TOOLS)

alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -10,7 +10,9 @@ from langchain_core.tools import ToolException
 from msrest.authentication import BasicAuthentication
 from pydantic import create_model, PrivateAttr, model_validator, SecretStr
 from pydantic.fields import FieldInfo as Field
+import xml.etree.ElementTree as ET
 
+from ..work_item import AzureDevOpsApiWrapper
 from ...elitea_base import BaseToolApiWrapper
 
 logger = logging.getLogger(__name__)
@@ -101,6 +103,16 @@ TestCaseAddModel = create_model(
     suite_id=(int, Field(description="ID of the test suite to which test cases are to be added"))
 )
 
+TestCaseCreateModel = create_model(
+    "TestCaseCreateModel",
+    project=(str, Field(description="Project ID or project name")),
+    plan_id=(int, Field(description="ID of the test plan to which test cases are to be added")),
+    suite_id=(int, Field(description="ID of the test suite to which test cases are to be added")),
+    title=(str, Field(description="Test case title")),
+    description=(str, Field(description="Test case description")),
+    test_steps=(str, Field(description="""Json array with test steps. Example: [{"action": "Some action", "expectedResult": "Some expectation"},...]""")),
+)
+
 TestCaseGetModel = create_model(
     "TestCaseGetModel",
     project=(str, Field(description="Project ID or project name")),
@@ -202,17 +214,51 @@ class TestPlanApiWrapper(BaseToolApiWrapper):
             logger.error(f"Error getting test suite(s): {e}")
             return ToolException(f"Error getting test suite(s): {e}")
 
-    def add_test_case(self, suite_test_case_create_update_parameters: str, project: str, plan_id: int, suite_id: int):
+    def add_test_case(self, suite_test_case_create_update_parameters, project: str, plan_id: int, suite_id: int):
         """Add a test case to a suite in Azure DevOps."""
         try:
-            params = json.loads(suite_test_case_create_update_parameters)
-            suite_test_case_create_update_params_obj = [SuiteTestCaseCreateUpdateParameters(**param) for param in params]
+            if isinstance(suite_test_case_create_update_parameters, str):
+                suite_test_case_create_update_parameters = json.loads(suite_test_case_create_update_parameters)
+            suite_test_case_create_update_params_obj = [SuiteTestCaseCreateUpdateParameters(**param) for param in suite_test_case_create_update_parameters]
             test_cases = self._client.add_test_cases_to_suite(suite_test_case_create_update_params_obj, project, plan_id, suite_id)
             return [test_case.as_dict() for test_case in test_cases]
         except Exception as e:
             logger.error(f"Error adding test case: {e}")
             return ToolException(f"Error adding test case: {e}")
 
+    def create_test_case(self, project: str, plan_id: int, suite_id: int, title: str, description: str, test_steps: str):
+        """Creates a new test case in specified suite in Azure DevOps."""
+        work_item_wrapper = AzureDevOpsApiWrapper(organization_url=self.organization_url, token=self.token.get_secret_value(), project=project)
+        work_item_json = self.build_ado_test_case(title, description, json.loads(test_steps))
+        created_work_item_id = work_item_wrapper.create_work_item(work_item_json=json.dumps(work_item_json), wi_type="Test Case")['id']
+        return self.add_test_case([{"work_item":{"id":created_work_item_id}}], project, plan_id, suite_id)
+
+    def build_ado_test_case(self, title, description, steps):
+        """
+        :param title: test title
+        :param description: test description
+        :param steps: steps [(action, expected result), ...]
+        :return: JSON with ADO fields
+        """
+        steps_elem = ET.Element("steps")
+
+        for idx, step in enumerate(steps, start=1):
+            step_elem = ET.SubElement(steps_elem, "step", id=str(idx), type="Action")
+            action_elem = ET.SubElement(step_elem, "parameterizedString", isformatted="true")
+            action_elem.text = step["action"]
+            expected_elem = ET.SubElement(step_elem, "parameterizedString", isformatted="true")
+            expected_elem.text = step["expectedResult"]
+
+        steps_xml = ET.tostring(steps_elem, encoding="unicode")
+
+        return {
+            "fields": {
+                "System.Title": title,
+                "System.Description": description,
+                "Microsoft.VSTS.TCM.Steps": steps_xml
+            }
+        }
+
     def get_test_case(self, project: str, plan_id: int, suite_id: int, test_case_id: str):
         """Get a test case from a suite in Azure DevOps."""
         try:
@@ -280,6 +326,12 @@ class TestPlanApiWrapper(BaseToolApiWrapper):
                 "args_schema": TestCaseAddModel,
                 "ref": self.add_test_case,
             },
+            {
+                "name": "create_test_case",
+                "description": self.create_test_case.__doc__,
+                "args_schema": TestCaseCreateModel,
+                "ref": self.create_test_case,
+            },
             {
                 "name": "get_test_case",
                 "description": self.get_test_case.__doc__,

alita_sdk/tools/ado/work_item/ado_wrapper.py

@@ -150,14 +150,15 @@ class AzureDevOpsApiWrapper(BaseToolApiWrapper):
 
         return parsed_items
 
-    def _transform_work_item(self, work_item_json: str):
+    def _transform_work_item(self, work_item_json):
         try:
             # Convert the input JSON to a Python dictionary
-            params = json.loads(work_item_json)
+            if isinstance(work_item_json, str):
+                work_item_json = json.loads(work_item_json)
         except (json.JSONDecodeError, ValueError) as e:
             raise ToolException(f"Issues during attempt to parse work_item_json: {e}")
 
-        if 'fields' not in params:
+        if 'fields' not in work_item_json:
             raise ToolException("The 'fields' property is missing from the work_item_json.")
 
             # Transform the dictionary into a list of JsonPatchOperation objects
@@ -167,11 +168,11 @@ class AzureDevOpsApiWrapper(BaseToolApiWrapper):
                 "path": f"/fields/{field}",
                 "value": value
             }
-            for field, value in params["fields"].items()
+            for field, value in work_item_json["fields"].items()
         ]
         return patch_document
 
-    def create_work_item(self, work_item_json: str, wi_type="Task"):
+    def create_work_item(self, work_item_json, wi_type="Task"):
         """Create a work item in Azure DevOps."""
         try:
             patch_document = self._transform_work_item(work_item_json)
@@ -185,7 +186,10 @@ class AzureDevOpsApiWrapper(BaseToolApiWrapper):
                 project=self.project,
                 type=wi_type
             )
-            return f"Work item {work_item.id} created successfully. View it at {work_item.url}."
+            return {
+                "id": work_item.id,
+                "message": f"Work item {work_item.id} created successfully. View it at {work_item.url}."
+            }
         except Exception as e:
             if "unknown value" in str(e):
                 logger.error(f"Unable to create work item due to incorrect assignee: {e}")

alita_sdk/tools/carrier/api_wrapper.py

@@ -70,6 +70,9 @@ class CarrierAPIWrapper(BaseModel):
     def get_integrations(self, name: str):
         return self._client.get_integrations(name)
 
+    def get_available_locations(self):
+        return self._client.get_available_locations()
+
     def run_test(self, test_id: str, json_body):
         return self._client.run_test(test_id, json_body)
 

alita_sdk/tools/carrier/backend_tests_tool.py

@@ -79,8 +79,8 @@ class RunTestByIDTool(BaseTool):
     description: str = "Execute test plan from the Carrier platform."
     args_schema: Type[BaseModel] = create_model(
         "RunTestByIdInput",
-        test_id=(str, Field(default=None, description="Test id to execute")),
-        name=(str, Field(default=None, description="Test name to execute")),
+        test_id=(int, Field(default=None, description="Test id to execute. Use test_id if user provide id in int format")),
+        name=(str, Field(default=None, description="Test name to execute. Use name if user provide name in str format")),
         test_parameters=(list, Field(
             default=None,
             description=(
@@ -89,9 +89,26 @@ class RunTestByIDTool(BaseTool):
                 "contain parameter names and their values."
             )
         )),
+        location=(str, Field(
+            default=None,
+            description=(
+                "Location to execute the test. Choose from public_regions, project_regions, "
+                "or cloud_regions. For cloud_regions, additional parameters may be required."
+            )
+        )),
+        cloud_settings=(dict, Field(
+            default={},
+            description=(
+                "Additional parameters for cloud_regions. Provide as a dictionary, "
+                "e.g., {'region_name': 'us-west-1', 'instance_type': 't2.large'}. "
+                "Don't provide this parameter as string! It should be a dictionary!"
+                "If no changes are needed, respond with 'use default'."
+                "Ensure these settings are passed as a valid dictionary not string"
+            )
+        )),
     )
 
-    def _run(self, test_id=None, name=None, test_parameters=None):
+    def _run(self, test_id=None, name=None, test_parameters=None, location=None, cloud_settings=None):
         try:
             if not test_id and not name:
                 return {"message": "Please provide test id or test name to start"}
@@ -115,9 +132,18 @@ class RunTestByIDTool(BaseTool):
             # If no test_parameters are provided, return the default ones for confirmation
             if test_parameters is None:
                 return {
-                    "message": "Please confirm or override the following test parameters to proceed with the test execution.",
+                    "message": "The test requires confirmation or customization of the following parameters before execution.",
                     "default_test_parameters": default_test_parameters,
-                    "instruction": "To override parameters, provide a list of dictionaries for 'test_parameters', e.g., [{'vUsers': '5', 'duration': '120'}].",
+                    "instruction": (
+                        "If the user has already indicated that default parameters should be used, "
+                        "pass 'default_test_parameters' as 'test_parameters' and invoke the '_run' method again without prompting the user.\n"
+                        "If the user wants to proceed with default parameters, respond with 'use default'.\n"
+                        "In this case, the agent should pass 'default_test_parameters' as 'test_parameters' to the tool.\n"
+                        "If the user provides specific overrides, parse them into a list of dictionaries in the following format:\n"
+                        "[{'vUsers': '5', 'duration': '120'}].\n"
+                        "Each dictionary should contain the parameter name and its desired value.\n"
+                        "Ensure that you correctly parse and validate the user's input before invoking the '_run' method."
+                    ),
                 }
 
             # Normalize test_parameters if provided in an incorrect format
@@ -126,6 +152,55 @@ class RunTestByIDTool(BaseTool):
             # Apply user-provided test parameters
             updated_test_parameters = self._apply_test_parameters(default_test_parameters, test_parameters)
 
+            # Fetch available locations
+            available_locations = self.api_wrapper.get_available_locations()
+            # If location is not provided, prompt the user with available options
+            if not location:
+                return {
+                    "message": "Please select a location to execute the test.",
+                    "available_locations": {
+                        "public_regions": available_locations["public_regions"],
+                        "project_regions": available_locations["project_regions"],
+                        "cloud_regions": [region["name"] for region in available_locations["cloud_regions"]],
+                    },
+                    "instruction": (
+                        "For public_regions and project_regions, provide the region name. "
+                        "For cloud_regions, provide the region name and optionally override cloud_settings."
+                    )
+                }
+
+            # Handle cloud_regions with additional parameters
+            selected_cloud_region = next(
+                (region for region in available_locations["cloud_regions"] if region["name"] == location),
+                None
+            )
+
+            if selected_cloud_region:
+                # Extract available cloud_settings from the selected cloud region
+                available_cloud_settings = selected_cloud_region["cloud_settings"]
+
+                # Add default values for instance_type and ec2_instance_type
+                available_cloud_settings["instance_type"] = "spot"
+                available_cloud_settings["ec2_instance_type"] = "t2.medium"
+
+                # If cloud_settings are not provided, prompt the user with available parameters
+                if not cloud_settings:
+                    return {
+                        "message": f"Please confirm or override the following cloud settings for the selected location: {location}",
+                        "available_cloud_settings": available_cloud_settings,
+                        "instruction": (
+                            "Provide a dictionary to override cloud settings, e.g., "
+                            "{'region_name': 'us-west-1', 'instance_type': 't2.large'}. "
+                            "Don't provide this parameter as string! It should be a dictionary! "
+                            "Ensure these settings are passed to the 'cloud_settings' argument, not 'test_parameters'."
+                            "Ensure these settings are passed as a valid dictionary not string"
+                        )
+                    }
+
+                # Validate and merge user-provided cloud_settings with available parameters
+                cloud_settings = self._merge_cloud_settings(available_cloud_settings, cloud_settings)
+
+
             # Build common_params dictionary
             common_params = {
                 param["name"]: param
@@ -136,7 +211,8 @@ class RunTestByIDTool(BaseTool):
             # Add env_vars, parallel_runners, and location to common_params
             common_params["env_vars"] = test_data.get("env_vars", {})
             common_params["parallel_runners"] = test_data.get("parallel_runners")
-            common_params["location"] = test_data.get("location")
+            common_params["location"] = location
+            common_params["env_vars"]["cloud_settings"] = cloud_settings or {}
 
             # Build the JSON body
             json_body = {
@@ -146,7 +222,7 @@ class RunTestByIDTool(BaseTool):
             }
 
             # Execute the test
-            report_id = self.api_wrapper.run_test(test_id, json_body)
+            report_id = self.api_wrapper.run_test(test_data.get("id"), json_body)
             return f"Test started. Report id: {report_id}. Link to report:" \
                    f"{self.api_wrapper.url.rstrip('/')}/-/performance/backend/results?result_id={report_id}"
 
@@ -207,6 +283,24 @@ class RunTestByIDTool(BaseTool):
             })
         return updated_parameters
 
+    def _merge_cloud_settings(self, available_cloud_settings, user_cloud_settings):
+        """
+        Merge user-provided cloud settings with available cloud settings.
+        Ensure that user-provided values override the defaults.
+        """
+        if not user_cloud_settings:
+            return available_cloud_settings
+
+        # Validate user-provided keys against available keys
+        invalid_keys = [key for key in user_cloud_settings if key not in available_cloud_settings]
+        if invalid_keys:
+            raise ValueError(
+                f"Invalid keys in cloud settings: {invalid_keys}. Allowed keys: {list(available_cloud_settings.keys())}")
+
+        # Merge the settings
+        merged_settings = {**available_cloud_settings, **user_cloud_settings}
+        return merged_settings
+
 
 class CreateBackendTestInput(BaseModel):
     test_name: str = Field(..., description="Test name")

alita_sdk/tools/carrier/carrier_sdk.py

@@ -103,6 +103,10 @@ class CarrierClient(BaseModel):
         endpoint = f"api/v1/integrations/integrations/{self.credentials.project_id}?name={name}"
         return self.request('get', endpoint)
 
+    def get_available_locations(self):
+        endpoint = f"api/v1/shared/locations/default/{self.credentials.project_id}"
+        return self.request('get', endpoint)
+
     def run_ui_test(self, test_id: str, json_body):
         """Run a UI test with the given test ID and JSON body."""
         endpoint = f"api/v1/ui_performance/test/{self.credentials.project_id}/{test_id}"

alita_sdk/tools/slack/__init__.py

@@ -0,0 +1,57 @@
+from typing import List, Optional
+
+from langchain_core.tools import BaseToolkit, BaseTool
+from pydantic import create_model, BaseModel, Field, SecretStr
+from ..base.tool import BaseAction
+
+from .api_wrapper import SlackApiWrapper
+from ..utils import TOOLKIT_SPLITTER, clean_string, get_max_toolkit_length
+
+name = "slack"
+
+def get_tools(tool):
+    return SlackToolkit().get_toolkit(
+        selected_tools=tool['settings'].get('selected_tools', []),
+        slack_token=tool['settings'].get('slack_token'),
+        channel_id=tool['settings'].get('channel_id'),
+        toolkit_name=tool.get('toolkit_name')
+    ).get_tools()
+
+class SlackToolkit(BaseToolkit):
+    tools: List[BaseTool] = []
+
+    @staticmethod
+    def toolkit_config_schema() -> BaseModel:
+         selected_tools = {x['name']: x['args_schema'].schema() for x in SlackApiWrapper.model_construct().get_available_tools()}
+         SlackToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
+         return create_model(
+            name,
+            slack_token=(SecretStr, Field( description="Slack Bot/User OAuth Token like XOXB-*****-*****-*****-*****")),
+            channel_id=(str, Field(title="Channel ID", description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")),
+            selected_tools=(list[str], Field(title="Selected Tools", description="List of tools to enable", default=[])),
+             __config__={'json_schema_extra': {'metadata': {"label": "slack", "icon_url": None, "hidden": True}}}
+        )
+
+    @classmethod
+    def get_toolkit(cls, selected_tools: Optional[List[str]] = None, toolkit_name: Optional[str] = None, **kwargs):
+        if selected_tools is None:
+            selected_tools = []
+        slack_api_wrapper = SlackApiWrapper(**kwargs)
+        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
+        available_tools = slack_api_wrapper.get_available_tools()
+        tools = []
+        for tool in available_tools:
+            if selected_tools and tool["name"] not in selected_tools:
+                continue
+            tools.append(BaseAction(                
+                api_wrapper=slack_api_wrapper,
+                name=prefix + tool["name"],
+                description=tool["description"],
+                args_schema=tool["args_schema"],
+                func=tool["ref"]
+            ))
+        return cls(tools=tools)
+
+    def get_tools(self) -> List[BaseTool]:
+        return self.tools
+

alita_sdk/tools/slack/api_wrapper.py

@@ -0,0 +1,190 @@
+import logging
+from typing import Optional
+from pydantic import BaseModel, Field, SecretStr, create_model, model_validator
+from slack_sdk import WebClient
+from slack_sdk.errors import SlackApiError
+
+from alita_sdk.tools.elitea_base import BaseToolApiWrapper
+
+
+logger = logging.getLogger(__name__)
+
+SendMessageModel = create_model(
+                    "SendMessageModel",                    
+                    message=(str, Field(description="The message text to send."))
+                    )
+
+ReadMessagesModel = create_model(
+                    "ReadMessagesModel",                    
+                    limit=(int, Field(default=10, description="The number of messages to fetch (default is 10)."))
+                    )
+
+CreateChannelModel = create_model(
+                    "CreateChannelModel",                     
+                    channel_name=(str, Field(description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")),
+                    is_private=(bool, Field(default=False, description="Whether to make the channel private (default: False)."))
+                    )
+
+ListUsersModel = create_model(
+                    "ListUsersModel"                 
+                    )
+
+
+class SlackApiWrapper(BaseToolApiWrapper):
+
+    """
+    Slack API wrapper for interacting with Slack channels and messages.
+    """
+    slack_token: Optional[SecretStr] = Field(default=None,description="Slack Bot/User OAuth Token like XOXB-*****-*****-*****-*****")
+    channel_id: Optional[str] = Field(default=None, description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")
+
+    @model_validator(mode="after")
+    @classmethod
+    def validate_toolkit(cls, values):
+        token = values.slack_token.get_secret_value() if values.slack_token else None
+        if not token:
+            logging.error("Slack token is required.")
+            raise ValueError("Slack token is required.")
+        try:
+            cls._client = WebClient(token=token)
+            logging.info("Authenticated with Slack token.")
+        except Exception as e:
+            logging.error(f"Failed to authenticate with Slack: {str(e)}")
+            raise ValueError(f"Failed to authenticate with Slack: {str(e)}")
+        return values
+
+    def _get_client(self):
+        if not self._client:
+            self._client = WebClient(token=self.slack_token.get_secret_value())
+        return self._client
+
+
+    def send_message(self, message: str):
+        """
+        Sends a message to a specified Slack channel, user, or conversation.
+        """
+        
+        try:
+
+            client = self._get_client()
+            response = client.chat_postMessage(channel=self.channel_id, text=message)
+            logger.info(f"Message sent to {self.channel_id}: {message}")
+            return f"Message sent successfully to {self.channel_id}."
+        
+        except SlackApiError as e:
+            logger.error(f"Failed to send message to {self.channel_id}: {e.response['error']}")
+            return f"Received the error :  {e.response['error']}"
+
+    def read_messages(self, limit=10):
+        """
+        Reads the latest messages from a Slack channel or conversation.
+       
+        :param limit: int: The number of messages to fetch (default is 10)
+        :return: list: Returns a list of messages with metadata.
+        """
+        try:
+
+            client = self._get_client()
+            # Fetch conversation history
+            response = client.conversations_history(
+                channel=self.channel_id,
+                limit=limit )
+            
+            # Extract messages from the response
+            messages = self.extract_slack_messages(response.get('messages', []))
+            
+            return messages
+            
+        except SlackApiError as e:
+            # Handle errors from the Slack API
+            logger.error(f"Failed to read message from {self.channel_id}: {e.response['error']}")
+            return f"Received the error :  {e.response['error']}"
+        
+    def create_slack_channel(self,  channel_name: str, is_private=False):
+        """
+        Creates a new Slack channel.
+
+        :param channel_name: str: Desired name for the channel (e.g., "my-new-channel").
+        :param is_private: bool: Whether to make the channel private (default: False).
+        :return: dict: Slack API response or error message.
+        """
+
+        try:
+            client = self._get_client()
+            response = client.conversations_create(
+                name=channel_name,
+                is_private=is_private
+            )
+            channel_id = response["channel"]["id"]
+            print(f"Channel '{channel_name}' created successfully! Channel ID: {channel_id}")
+            return {"success": True, "channel_id": channel_id}
+        except SlackApiError as e:
+            error_message = e.response.get("error", "unknown_error")
+            print(f"Failed to create channel '{channel_name}': {error_message}")
+            return {"success": False, "error": error_message}
+    
+    def list_users(self):
+        """
+        Lists all users in the Slack workspace.
+
+        :return: list: List of users with their IDs and names.
+        """
+        
+        
+        try:
+            client = self._get_client()
+            print(client.auth_test())
+            response = client.users_list()
+            users = response["members"]
+            return [{"id": user["id"], "name": user["name"]} for user in users if not user["is_bot"]]
+        
+        except SlackApiError as e:
+            logger.error(f"Failed to list users: {e.response['error']}")
+            return f"Received the error :  {e.response['error']}"
+    
+    def extract_slack_messages(self, data):
+        extracted_info = []
+
+        for item in data:
+            # Extract 'user' and 'text'
+            user = item.get("user", "Undefined User")
+            message = item.get("text", "No message")
+
+            # Extract 'app name'
+            app_name = item.get("bot_profile", {}).get("name", "No App Name")
+            
+            # Append to result
+            extracted_info.append({"user": user, "message": message, "app_name": app_name})
+        
+        return extracted_info
+    
+    
+    def get_available_tools(self):
+        return [
+            {
+                "name": "send_message",
+                "description": self.send_message.__doc__ or "Send a message to a Slack channel, user, or conversation.",
+                "args_schema": SendMessageModel,
+                "ref": self.send_message
+                
+            },
+            {
+                "name": "read_messages",
+                "description": self.read_messages.__doc__ or "Send a message to a Slack channel, user, or conversation.",
+                "args_schema": ReadMessagesModel,
+                "ref": self.read_messages
+            },
+            {
+                "name": "create_channel",
+                "description": self.create_slack_channel.__doc__ or "Send a message to a Slack channel, user, or conversation.",
+                "args_schema": CreateChannelModel,
+                "ref": self.create_slack_channel             
+            },
+            {
+                "name": "list_users",
+                "description": self.list_users.__doc__ or "List all users in the Slack workspace.",
+                "args_schema": ListUsersModel,
+                "ref": self.list_users
+            }
+            
+        ]

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.182
+Version: 0.3.183
 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
@@ -124,6 +124,7 @@ Requires-Dist: shortuuid==1.0.13; extra == "tools"
 Requires-Dist: yarl==1.17.1; extra == "tools"
 Requires-Dist: langmem==0.0.27; extra == "tools"
 Requires-Dist: textract-py3==2.1.1; extra == "tools"
+Requires-Dist: slack_sdk==3.35.0; extra == "tools"
 Provides-Extra: community
 Requires-Dist: retry-extended==0.2.3; extra == "community"
 Requires-Dist: pyobjtojson==0.3; extra == "community"