alita-sdk 0.3.188__tar.gz → 0.3.190__tar.gz

{alita_sdk-0.3.188 → alita_sdk-0.3.190}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.188
+Version: 0.3.190
 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 → alita_sdk-0.3.190}/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-0.3.188 → alita_sdk-0.3.190}/alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -96,13 +96,53 @@ 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:
+    [
+        {'{'}
+            plan_id: str
+            suite_id: str
+            title: str
+            description: str
+            test_steps: str
+            test_steps_format: str
+        {'}'}
+        ...
+    ]
+    Where:
+    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 +257,38 @@ 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(
+            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_xml: 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 +297,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 +406,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 → alita_sdk-0.3.190}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.188
+Version: 0.3.190
 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 → alita_sdk-0.3.190}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.188"
+version = "0.3.190"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"