alita-sdk 0.3.188__py3-none-any.whl → 0.3.189__py3-none-any.whl

alita_sdk/runtime/tools/llm.py

@@ -35,9 +35,10 @@ def create_llm_input_with_messages(
     input_messages = []
     
     # Add system message from prompt if available
-    if prompt and prompt.get('template'):
+    if prompt:
         try:
-            system_content = prompt['template'].format(**params) if params else prompt['template']
+            # Format the system message using the prompt template or value and params
+            system_content = (prompt['template'] if 'template' in prompt else prompt['value']).format(**params) if params else prompt['template']
             input_messages.append(SystemMessage(content=system_content))
         except KeyError as e:
             error_msg = f"KeyError in prompt formatting: {e}. Available params: {list(params.keys())}"

alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -96,13 +96,55 @@ TestCaseAddModel = create_model(
     suite_id=(int, Field(description="ID of the test suite to which test cases are to be added"))
 )
 
+
+test_steps_description = """Json or XML array string with test steps. 
+    Json example: [{"stepNumber": 1, "action": "Some action", "expectedResult": "Some expectation"},...]
+    XML example: 
+    <Steps>
+    <Step>
+      <StepNumber>1</StepNumber>
+      <Action>Some action</Action>
+      <ExpectedResult>Some expectation</ExpectedResult>
+    </Step>
+    ...
+    </Steps>
+    """
+
+TestCasesCreateModel = create_model(
+    "TestCasesCreateModel",
+    create_test_cases_parameters=(str, Field(description=f"""Json array where each object is separate test case to be created.
+    Input format:
+    [
+        {'{'}
+            project: str
+            plan_id: str
+            suite_id: str
+            title: str
+            description: str
+            test_steps: str
+            test_steps_format: str
+        {'}'}
+        ...
+    ]
+    Where:
+    project - Project ID or project name;
+    plan_id - ID of the test plan to which test cases are to be added;
+    suite_id - ID of the test suite to which test cases are to be added
+    title - Test case title;
+    description - Test case description;
+    test_steps - {test_steps_description}
+    test_steps_format - Format of provided test steps. Possible values: json, xml
+    """))
+)
+
 TestCaseCreateModel = create_model(
     "TestCaseCreateModel",
     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"},...]""")),
+    test_steps=(str, Field(description=test_steps_description)),
+    test_steps_format=(Optional[str], Field(description="Format of provided test steps. Possible values: json, xml", default='json'))
 )
 
 TestCaseGetModel = create_model(
@@ -217,31 +259,39 @@ class TestPlanApiWrapper(BaseToolApiWrapper):
             logger.error(f"Error adding test case: {e}")
             return ToolException(f"Error adding test case: {e}")
 
-    def create_test_case(self, plan_id: int, suite_id: int, title: str, description: str, test_steps: str):
+    def create_test_cases(self, create_test_cases_parameters):
+        """Creates new test cases in specified suite in Azure DevOps."""
+        test_cases = json.loads(create_test_cases_parameters)
+        return [self.create_test_case(
+            project=test_case['project'],
+            plan_id=test_case['plan_id'],
+            suite_id=test_case['suite_id'],
+            title=test_case['title'],
+            description=test_case['description'],
+            test_steps=test_case['test_steps'],
+            test_steps_format=test_case['test_steps_format']) for test_case in test_cases]
+
+
+    def create_test_case(self, plan_id: int, suite_id: int, title: str, description: str, test_steps: str, test_steps_format: str = 'json'):
         """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=self.project)
-        work_item_json = self.build_ado_test_case(title, description, json.loads(test_steps))
+        if test_steps_format == 'json':
+            steps_xml = self.get_test_steps_xml(json.loads(test_steps))
+        elif test_steps_format == 'xml':
+            steps_xml = self.convert_steps_tag_to_ado_steps(test_steps)
+        else:
+            return ToolException("Unknown test steps format: " + test_steps_format)
+        work_item_json = self.build_ado_test_case(title, description, steps_xml)
         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}}], plan_id, suite_id)
 
-    def build_ado_test_case(self, title, description, steps):
+    def build_ado_test_case(self, title, description, steps_xml):
         """
         :param title: test title
         :param description: test description
-        :param steps: steps [(action, expected result), ...]
+        :param steps: steps xml
         :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,
@@ -250,6 +300,42 @@ class TestPlanApiWrapper(BaseToolApiWrapper):
             }
         }
 
+    def get_test_steps_xml(self, steps: dict):
+        steps_elem = ET.Element("steps")
+        for step in steps:
+            step_number = step.get("stepNumber", 1)
+            action = step.get("action", "")
+            expected_result = step.get("expectedResult", "")
+            steps_elem.append(self.build_step_element(step_number, action, expected_result))
+        return ET.tostring(steps_elem, encoding="unicode")
+
+    def convert_steps_tag_to_ado_steps(self, input_xml: str) -> str:
+        """
+        Converts input XML from format:
+        <Steps><Step><Action>...</Action><ExpectedResult>...</ExpectedResult></Step></Steps>
+        to Azure DevOps test case format:
+        <steps><step id="1" type="Action">...</step>...</steps>
+        """
+        input_root = ET.fromstring(input_xml)
+        steps_elem = ET.Element("steps")
+        for step_node in input_root.findall("Step"):
+            step_number = step_node.findtext("StepNumber", default="1")
+            action = step_node.findtext("Action", default="")
+            expected_result = step_node.findtext("ExpectedResult", default="")
+            steps_elem.append(self.build_step_element(step_number, action, expected_result))
+        return ET.tostring(steps_elem, encoding="unicode")
+
+    def build_step_element(self, step_number: str, action: str, expected_result: str) -> ET.Element:
+        """
+            Creates an individual <step> element for Azure DevOps.
+            """
+        step_elem = ET.Element("step", id=str(step_number), type="Action")
+        action_elem = ET.SubElement(step_elem, "parameterizedString", isformatted="true")
+        action_elem.text = action or ""
+        expected_elem = ET.SubElement(step_elem, "parameterizedString", isformatted="true")
+        expected_elem.text = expected_result or ""
+        return step_elem
+
     def get_test_case(self, plan_id: int, suite_id: int, test_case_id: str):
         """Get a test case from a suite in Azure DevOps."""
         try:
@@ -323,6 +409,12 @@ class TestPlanApiWrapper(BaseToolApiWrapper):
                 "args_schema": TestCaseCreateModel,
                 "ref": self.create_test_case,
             },
+            {
+                "name": "create_test_cases",
+                "description": self.create_test_cases.__doc__,
+                "args_schema": TestCasesCreateModel,
+                "ref": self.create_test_cases,
+            },
             {
                 "name": "get_test_case",
                 "description": self.get_test_case.__doc__,

{alita_sdk-0.3.188.dist-info → alita_sdk-0.3.189.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.188
+Version: 0.3.189
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.188.dist-info → alita_sdk-0.3.189.dist-info}/RECORD

@@ -81,7 +81,7 @@ alita_sdk/runtime/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFp
 alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
 alita_sdk/runtime/tools/function.py,sha256=ZFpd7TGwIawze2e7BHlKwP0NHwNw42wwrmmnXyJQJhk,2600
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
-alita_sdk/runtime/tools/llm.py,sha256=qy3-TYhV-rvT4PzSNMw0vvJTMmgZI9nAmvW4NNhL5QU,14937
+alita_sdk/runtime/tools/llm.py,sha256=OD-oFxg_axZitb_zL-O1hAT5UhBvqiw0XOMy5EPva-o,15043
 alita_sdk/runtime/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/runtime/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
 alita_sdk/runtime/tools/mcp_server_tool.py,sha256=eI8QUt497xblwF4Zhbvi8wCg17yh2yoWjcw_AIzHwGE,2819
@@ -105,7 +105,7 @@ alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,
 alita_sdk/tools/ado/repos/__init__.py,sha256=bzVSEAPwBoH4sY3cNj5_FNXIC3yY8lkofaonNKhwhRk,6286
 alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=_OWKAls7VFfFtEPTwqj_DxE1MSvpC0ivxdTIULEz3Tk,48206
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=K-jhCT7D7bxXbiat3GCP_qDVg_ipcBlnh9r3JhmAubM,4408
-alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=uUaxXPxz2NiwzNEXtR31pPGe77f-p0sj7VL-zZnfTqI,14529
+alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=J7A2etGa3MTcYpdDhejSM0pLSiXF5xSO74sGihynHCw,18257
 alita_sdk/tools/ado/wiki/__init__.py,sha256=5QlISwHbfBl_0he7miv6WbK-Kc2mS-t46d9UFJZqY8A,4308
 alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=l4bc2QoKSUXg9UqNcx0ylv7YL9JPPQd35Ti5MXyEgC4,12690
 alita_sdk/tools/ado/work_item/__init__.py,sha256=vZTTLh-dcefOPCkNOehcmXiPLZrD6n-tg8fKoKWrgJE,4411
@@ -297,8 +297,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.188.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.188.dist-info/METADATA,sha256=i1rCgEanmynBNJEsIjDaLlBJhYVq7pp242X8TZJXHMA,18804
-alita_sdk-0.3.188.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.188.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.188.dist-info/RECORD,,
+alita_sdk-0.3.189.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.189.dist-info/METADATA,sha256=5os2tSZvdoNwZboxU24vXSEqoAHqkZRAendJI1yYJFY,18804
+alita_sdk-0.3.189.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.189.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.189.dist-info/RECORD,,