listSkills sync action defining the available AI agent skills 🤖

eventually the response of this sync action shall be generated dynamically based on component configuration

Author: soustruh

.github/workflows/push.yml

@@ -67,7 +67,7 @@ jobs:
           TAG="${GITHUB_REF##*/}"
           IS_SEMANTIC_TAG=$(echo "$TAG" | grep -q '^v\?[0-9]\+\.[0-9]\+\.[0-9]\+$' && echo true || echo false)
           echo "is_semantic_tag=$IS_SEMANTIC_TAG" | tee -a $GITHUB_OUTPUT
-          echo "app_image_tag=$TAG" | tee -a $GITHUB_OUTPUT
+          echo "app_image_tag=$TAG-${{ github.run_id }}" | tee -a $GITHUB_OUTPUT
 
       - name: Deploy-Ready check
         id: deploy_ready

src/component.py

@@ -1,4 +1,5 @@
 import csv
+import json
 import logging
 import os
 import shutil
@@ -86,7 +87,7 @@ class Component(ComponentBase):
     def __init__(self):
         super().__init__()
 
-    def run(self, return_data=False):
+    def run(self, return_json=False):
         self.validate_configuration_parameters(REQUIRED_PARAMETERS)
         self.validate_image_parameters(REQUIRED_IMAGE_PARS)
 
@@ -130,7 +131,15 @@ def run(self, return_data=False):
         logging.debug([result for result in results])
         logging.info(f"Downloaded {total_records} records in total")
 
-        if return_data:
+        if return_json:
+            for result in results:
+                if "file" not in result:
+                    result["status"] = "fileError"
+                    continue
+                result["records"] = list(self._csv_result_to_dict(result.pop("file")))
+                result["status"] = "ok"
+            with open("response.json", "w") as f:
+                json.dump(results, f)
             return results
 
         # remove headers and get columns
@@ -180,6 +189,12 @@ def _fix_header_from_csv(results: list[dict]) -> list[str]:
             os.replace(temp_file_path, result_file_path)
         return expected_header
 
+    def _csv_result_to_dict(self, filename: str):
+        with open(filename) as f:
+            reader = csv.DictReader(f)
+            for row in reader:
+                yield row
+
     def set_proxy(self) -> None:
         """Sets proxy if defined"""
         proxy_config = self.configuration.parameters.get(KEY_PROXY, {})
@@ -622,12 +637,105 @@ def load_possible_primary_keys(self) -> list[SelectElement]:
         else:
             raise UserException(f"Invalid {KEY_QUERY_TYPE}")
 
-    # @sync_action("runComponent")
-    # def sync_run_component(self):
-    #     """
-    #     Run the component as a sync action
-    #     """
-    #     return self.run(return_data=True)
+    @sync_action("runSkill")
+    def sync_run_component(self):
+        """
+        Run the component as a skill for an AI agent
+        """
+        return self.run(return_json=True)
+
+    @sync_action("listSkills")
+    def list_skills(self):
+        return [
+            {
+                "name": "Get Salesforce contacts",
+                "description": "Returns list of Salesforce contacts.",
+                "parameters": [
+                    {
+                        "name": "action",
+                        "type": "string",
+                        "required": True,
+                        "enum": ["runSkill"],
+                    },
+                    {
+                        "name": "configData",
+                        "type": "object",
+                        "required": True,
+                        "parameters": [
+                            {
+                                "name": "object",
+                                "type": "string",
+                                "required": True,
+                                "enum": ["contacts"],
+                                "description": "String identifer of the agent skill to run.",
+                            },
+                            {
+                                "name": "username",
+                                "type": "string",
+                                "required": True,
+                                "description": "The Salesforce account username.",
+                            },
+                            {
+                                "name": "#password",
+                                "type": "string",
+                                "required": True,
+                                "description": "The Salesforce account password.",
+                            },
+                            {
+                                "name": "#securityToken",
+                                "type": "string",
+                                "required": True,
+                                "description": "Keboola project security token.",
+                            },
+                        ],
+                    },
+                ],
+            },
+            {
+                "name": "Get Salesforce reports",
+                "description": "Returns list of Salesforce reports.",
+                "parameters": [
+                    {
+                        "name": "action",
+                        "type": "string",
+                        "required": True,
+                        "enum": ["runSkill"],
+                    },
+                    {
+                        "name": "configData",
+                        "type": "object",
+                        "required": True,
+                        "parameters": [
+                            {
+                                "name": "object",
+                                "type": "string",
+                                "required": True,
+                                "enum": ["reports"],
+                                "description": "String identifer of the agent skill to run.",
+                            },
+                            {
+                                "name": "username",
+                                "type": "string",
+                                "required": True,
+                                "description": "The Salesforce account username.",
+                            },
+                            {
+                                "name": "#password",
+                                "type": "string",
+                                "required": True,
+                                "description": "The Salesforce account password.",
+                            },
+                            {
+                                "name": "#securityToken",
+                                "type": "string",
+                                "required": True,
+                                "description": "Keboola project security token.",
+                            },
+                        ],
+                    },
+                ],
+            },
+        ]
 
     def _get_object_name_from_custom_query(self) -> str:
         params = self.configuration.parameters