alita-sdk 0.3.201__tar.gz → 0.3.203__tar.gz

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.201
+Version: 0.3.203
 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.201 → alita_sdk-0.3.203}/alita_sdk/tools/carrier/api_wrapper.py

@@ -61,6 +61,9 @@ class CarrierAPIWrapper(BaseModel):
     def get_reports_list(self) -> List[Dict[str, Any]]:
         return self._client.get_reports_list()
 
+    def add_tag_to_report(self, report_id, tag_name):
+        return self._client.add_tag_to_report(report_id, tag_name)
+
     def get_tests_list(self) -> List[Dict[str, Any]]:
         return self._client.get_tests_list()
 

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/carrier/backend_reports_tool.py

@@ -23,10 +23,11 @@ class GetReportsTool(BaseTool):
     description: str = "Get list of reports from the Carrier platform."
     args_schema: Type[BaseModel] = create_model(
         "GetReportsInput",
+        name=(str, Field(default="", description="Optional parameter. Report name to filter reports")),
         tag_name=(str, Field(default="", description="Optional parameter. Tag name to filter reports")),
     )
 
-    def _run(self, tag_name=""):
+    def _run(self, name="", tag_name=""):
         try:
             reports = self.api_wrapper.get_reports_list()
 
@@ -38,6 +39,10 @@ class GetReportsTool(BaseTool):
 
             trimmed_reports = []
             for report in reports:
+                # Filter reports by name
+                if name and not name == report["name"]:
+                    continue
+
                 # Filter by tag title
                 tags = report.get("tags", [])
                 if tag_name and not any(tag.get("title") == tag_name for tag in tags):
@@ -111,6 +116,29 @@ class GetReportByIDTool(BaseTool):
             raise ToolException(stacktrace)
 
 
+class AddTagToReportTool(BaseTool):
+    api_wrapper: CarrierAPIWrapper = Field(..., description="Carrier API Wrapper instance")
+    name: str = "add_tag_to_report"
+    description: str = "Add tag to backend report"
+    args_schema: Type[BaseModel] = create_model(
+        "AddTagToReportInput",
+        report_id=(str, Field(description="Report id to update")),
+        tag_name=(str, Field(description="Tag name to add to report")),
+    )
+
+    def _run(self, report_id: str, tag_name: str):
+        try:
+            res = self.api_wrapper.add_tag_to_report(report_id, tag_name)
+            if "Tags was updated" in res.text:
+                return f"Added tag {tag_name} to report id {report_id}"
+            else:
+                return f"Failed to add new tag to report id {report_id}"
+        except Exception:
+            stacktrace = traceback.format_exc()
+            logger.error(f"Error downloading reports: {stacktrace}")
+            raise ToolException(stacktrace)
+
+
 class CreateExcelReportTool(BaseTool):
     api_wrapper: CarrierAPIWrapper = Field(..., description="Carrier API Wrapper instance")
     name: str = "create_excel_report"

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/carrier/backend_tests_tool.py

@@ -7,7 +7,6 @@ from pydantic.fields import Field
 from pydantic import create_model, BaseModel
 from .api_wrapper import CarrierAPIWrapper
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -79,8 +78,10 @@ class RunTestByIDTool(BaseTool):
     description: str = "Execute test plan from the Carrier platform."
     args_schema: Type[BaseModel] = create_model(
         "RunTestByIdInput",
-        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_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=(
@@ -200,7 +201,6 @@ class RunTestByIDTool(BaseTool):
                 # 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
@@ -307,7 +307,8 @@ class CreateBackendTestInput(BaseModel):
     test_type: str = Field(..., description="Test type")
     env_type: str = Field(..., description="Env type")
     entrypoint: str = Field(..., description="Entrypoint for the test (JMeter script path or Gatling simulation path)")
-    custom_cmd: str = Field(..., description="Custom command line to execute the test (e.g., -l /tmp/reports/jmeter.jtl -e -o /tmp/reports/html_report)")
+    custom_cmd: str = Field(...,
+                            description="Custom command line to execute the test (e.g., -l /tmp/reports/jmeter.jtl -e -o /tmp/reports/html_report)")
     runner: str = Field(..., description="Test runner (Gatling or JMeter)")
     source: Optional[Dict[str, Optional[str]]] = Field(
         None,
@@ -474,8 +475,11 @@ class CreateBackendTestTool(BaseTool):
                 return {
                     "message": "Do you want to configure email integration?",
                     "instructions": (
+                        "If the user indicates no integrations are needed make sure to pass email_integration"
+                        " as empty dict to _run method and invoke it ones again with email_integration={}."
                         "If yes, select an integration from the available options below and provide email recipients.\n"
-                        "If no, respond with 'no'."
+                        "If no, respond with 'no'. "
+
                     ),
                     "available_integrations": [
                         {
@@ -491,6 +495,28 @@ class CreateBackendTestTool(BaseTool):
                     },
                 }
 
+            # Ensure email_integrations is an empty dict if the user indicates no integrations are needed
+            if isinstance(email_integration, str) and email_integration.lower() == "no":
+                email_integration = {}
+            elif (
+                    len(integrations_list) > 0
+                    and isinstance(email_integration, dict)
+                    and "integration_id" in email_integration
+                    and "recipients" in email_integration
+            ):
+                email_integration = {
+                    "reporters": {
+                        "reporter_email": {
+                            "id": email_integration["integration_id"],
+                            "is_local": True,
+                            "project_id": integrations_list[0]["project_id"],
+                            "recipients": email_integration["recipients"],
+                        }
+                    }
+                }
+            else:
+                email_integration = {}
+
             # Prepare the final data dictionary
             data = {
                 "common_params": {
@@ -512,16 +538,7 @@ class CreateBackendTestTool(BaseTool):
                     "location": "default",  # TODO update location
                 },
                 "test_parameters": test_parameters,
-                "integrations": {
-                    "reporters": {
-                        "reporter_email": {
-                            "id": email_integration["integration_id"],
-                            "is_local": True,
-                            "project_id": integrations_list[0]["project_id"],  # Example project_id
-                            "recipients": email_integration["recipients"],
-                        }
-                    }
-                },
+                "integrations": email_integration,
                 "scheduling": [],
                 "run_test": False,
             }
@@ -539,17 +556,3 @@ class CreateBackendTestTool(BaseTool):
             stacktrace = traceback.format_exc()
             logger.error(f"Error while creating test: {stacktrace}")
             raise ToolException(stacktrace)
-
-# data = {"common_params":{"name":"toolkit_demo","test_type":"toolkit_demo","env_type":"toolkit_demo",
-#                                      "entrypoint":"tests/BasicEcommerceWithTransaction.jmx","runner":"v5.6.3",
-#                                      "source":{"name":"git_https","repo":"https://git.epam.com/epm-perf/boilerplate.git",
-#                                                "branch":"jmeter","username":"mykhailo_hunko@epam.com",
-#                                                "password":"{{secret.mykhailo_gitlab}}"},
-#                                      "env_vars":{"cpu_quota":2,"memory_quota":6,"cloud_settings":{},
-#                                                  "custom_cmd":"-l /tmp/reports/jmeter.jtl -e -o /tmp/reports/html_report"},
-#                                      "parallel_runners":1,"cc_env_vars":{},"customization":{},"location":"default"},
-#                     "test_parameters":[{"name":"VUSERS","default":"5","type":"string","description":"","action":""},
-#                                        {"name":"DURATION","default":"60","type":"string","description":"","action":""}],
-#                     "integrations":{"reporters":{"reporter_email":{"id":1,"is_local":True,"project_id":36,
-#                                                                    "recipients":["mykhailo_hunko@epam.com"]}}},
-#                     "scheduling":[],"run_test":True}

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/carrier/carrier_sdk.py

@@ -77,6 +77,17 @@ class CarrierClient(BaseModel):
         endpoint = f"api/v1/backend_performance/reports/{self.credentials.project_id}"
         return self.request('get', endpoint).get("rows", [])
 
+    def add_tag_to_report(self, report_id, tag_name):
+        endpoint = f"api/v1/backend_performance/tags/{self.credentials.project_id}/{report_id}"
+        full_url = f"{self.credentials.url.rstrip('/')}/{endpoint.lstrip('/')}"
+        headers = {
+            'Authorization': f'bearer {self.credentials.token}',
+            'content-type': 'application/json'
+        }
+        data = {"tags": [{"title": tag_name, "hex": "#5933c6"}]}
+        res = requests.post(full_url, headers=headers, json=data)
+        return res
+
     def get_tests_list(self) -> List[Dict[str, Any]]:
         endpoint = f"api/v1/backend_performance/tests/{self.credentials.project_id}"
         return self.request('get', endpoint).get("rows", [])
@@ -90,9 +101,6 @@ class CarrierClient(BaseModel):
         form_data = {"data": dumps(data)}
         # Send the POST request
         res = requests.post(full_url, headers=headers, data=form_data)
-        print("************************* response")
-        print(res.text)
-        print("**********************************")
         return res
 
     def run_test(self, test_id: str, json_body):
@@ -150,7 +158,8 @@ class CarrierClient(BaseModel):
         for file_name in file_list:
             if file_name.startswith(report_archive_prefix) and "excel_report" not in file_name:
                 report_files_list.append(file_name)
-        test_log_file_path, errors_log_file_path = self.download_and_merge_reports(report_files_list, lg_type, bucket_name, extract_to)
+        test_log_file_path, errors_log_file_path = self.download_and_merge_reports(report_files_list, lg_type,
+                                                                                   bucket_name, extract_to)
 
         return report_info, test_log_file_path, errors_log_file_path
 
@@ -312,4 +321,4 @@ class CarrierClient(BaseModel):
             }
         }
 
-        return self.request('put', endpoint, json=cancel_body)
+        return self.request('put', endpoint, json=cancel_body)

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/carrier/tools.py

@@ -1,6 +1,6 @@
 # import all available tools
 from .tickets_tool import FetchTicketsTool, CreateTicketTool
-from .backend_reports_tool import GetReportsTool, GetReportByIDTool, CreateExcelReportTool
+from .backend_reports_tool import GetReportsTool, GetReportByIDTool, CreateExcelReportTool, AddTagToReportTool
 from .backend_tests_tool import GetTestsTool, GetTestByIDTool, RunTestByIDTool, CreateBackendTestTool
 from .ui_reports_tool import GetUIReportsTool, GetUIReportByIDTool, GetUITestsTool
 from .run_ui_test_tool import RunUITestTool
@@ -14,6 +14,7 @@ __all__ = [
     {"name": "create_ticket", "tool": CreateTicketTool},
     {"name": "get_reports", "tool": GetReportsTool},
     {"name": "get_report_by_id", "tool": GetReportByIDTool},
+    {"name": "add_tag_to_report", "tool": AddTagToReportTool},
     {"name": "create_excel_report", "tool": CreateExcelReportTool},
     {"name": "get_tests", "tool": GetTestsTool},
     {"name": "get_test_by_id", "tool": GetTestByIDTool},

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/postman/api_wrapper.py

@@ -91,12 +91,16 @@ PostmanUpdateCollectionDescription = create_model(
 
 PostmanUpdateCollectionVariables = create_model(
     "PostmanUpdateCollectionVariables",
-    variables=(List[Dict], Field(description="Updated collection variables"))
+    variables=(Optional[List[Dict[str, Any]]], Field(default=None,
+        description="List of collection variables objects. "
+                    "Example: [{'key': 'project_id', 'type': 'string', 'value': '15', 'enabled': true}]"))
 )
 
 PostmanUpdateCollectionAuth = create_model(
     "PostmanUpdateCollectionAuth",
-    auth=(Dict, Field(description="Updated authentication settings"))
+    auth=(Optional[Dict[str, Any]], Field(default=None,
+         description="Updated authentication settings. Example: {'type': 'bearer',token '': 'your_token'}"
+     ))
 )
 
 PostmanDeleteCollection = create_model(
@@ -190,19 +194,21 @@ PostmanUpdateRequestDescription = create_model(
 PostmanUpdateRequestHeaders = create_model(
     "PostmanUpdateRequestHeaders",
     request_path=(str, Field(description="Path to the request (folder/requestName)")),
-    headers=(List[Dict], Field(description="Request headers"))
+    headers=(Optional[List[Dict[str, Any]]], Field(default=None, description="Request headers."))
 )
 
 PostmanUpdateRequestBody = create_model(
     "PostmanUpdateRequestBody",
     request_path=(str, Field(description="Path to the request (folder/requestName)")),
-    body=(Dict, Field(description="Request body"))
+    body=(Optional[Dict[str, Any]], Field(default=None, description="Request body."))
 )
 
 PostmanUpdateRequestAuth = create_model(
     "PostmanUpdateRequestAuth",
     request_path=(str, Field(description="Path to the request (folder/requestName)")),
-    auth=(Dict, Field(description="Request authentication"))
+    auth=(Optional[Dict[str, Any]], Field(default=None,
+         description="Updated authentication settings. Example: {'type': 'bearer',token '': 'your_token'}"
+     ))
 )
 
 PostmanUpdateRequestTests = create_model(
@@ -692,7 +698,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
     def get_collections(self, **kwargs) -> str:
         """Get all Postman collections accessible to the user."""
         try:
-            response = self._make_request('GET', '/collections')
+            response = self._make_request('GET', f'/collections?workspace={self.workspace_id}')
             return json.dumps(response, indent=2)
         except Exception as e:
             stacktrace = format_exc()
@@ -1214,7 +1220,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to update collection {self.collection_id} description: {str(e)}")
 
-    def update_collection_variables(self, variables: List[Dict], **kwargs) -> str:
+    def update_collection_variables(self, variables: List[Dict[str, Any]], **kwargs) -> str:
         """Update collection variables."""
         try:
             # Get current collection
@@ -1234,7 +1240,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to update collection {self.collection_id} variables: {str(e)}")
 
-    def update_collection_auth(self, auth: Dict, **kwargs) -> str:
+    def update_collection_auth(self, auth: Dict[str, Any], **kwargs) -> str:
         """Update collection authentication settings."""
         try:
             # Get current collection
@@ -1283,7 +1289,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             self.analyzer.remove_item_ids(collection_data.get("item", []))
 
             response = self._make_request(
-                'POST', '/collections', json={"collection": collection_data})
+                'POST', f'/collections?workspace={self.workspace_id}', json={"collection": collection_data})
             return json.dumps(response, indent=2)
         except Exception as e:
             stacktrace = format_exc()
@@ -1638,7 +1644,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to update request '{request_path}' description: {str(e)}")
 
-    def update_request_headers(self, request_path: str, headers: List[Dict], **kwargs) -> str:
+    def update_request_headers(self, request_path: str, headers: Dict[str, Any], **kwargs) -> str:
         """Update request headers."""
         try:
             # Get request item and ID
@@ -1659,7 +1665,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to update request '{request_path}' headers: {str(e)}")
 
-    def update_request_body(self, request_path: str, body: Dict, **kwargs) -> str:
+    def update_request_body(self, request_path: str, body: Dict[str, Any], **kwargs) -> str:
         """Update request body."""
         try:
             # Get request item and ID
@@ -1680,7 +1686,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to update request '{request_path}' body: {str(e)}")
 
-    def update_request_auth(self, request_path: str, auth: Dict, **kwargs) -> str:
+    def update_request_auth(self, request_path: str, auth: Dict[str, Any], **kwargs) -> str:
         """Update request authentication."""
         try:
             # Get request item and ID

{alita_sdk-0.3.201 → alita_sdk-0.3.203}/alita_sdk/tools/slack/__init__.py

@@ -25,6 +25,7 @@ def get_tools(tool):
 
 class SlackToolkit(BaseToolkit):
     tools: List[BaseTool] = []
+    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
@@ -52,7 +53,7 @@ class SlackToolkit(BaseToolkit):
                                                                              'configuration_title': True})),
              slack_token=(SecretStr, Field(description="Slack Token like XOXB-*****-*****-*****-*****",
                                            json_schema_extra={'secret': True, 'configuration': True})),
-             channel_id=(str, Field(description="Channel ID", json_schema_extra={'configuration': True})),
+             channel_id=(Optional[str], Field(default=None, description="Channel ID", json_schema_extra={'configuration': True})),
              selected_tools=(List[Literal[tuple(selected_tools)]],
                              Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
              __config__={'json_schema_extra': {'metadata': {"label": "Slack", "icon_url": "slack-icon.svg"}}}
@@ -74,7 +75,7 @@ class SlackToolkit(BaseToolkit):
             tools.append(BaseAction(                
                 api_wrapper=slack_api_wrapper,
                 name=prefix + tool["name"],
-                description=tool["description"],
+                description=f"Slack Tool: {tool['description']}",
                 args_schema=tool["args_schema"],
             ))
         return cls(tools=tools)