Merge pull request #4 from dataiku/feature/create-incident-tool

Merging tool for issue creation

Author: alexbourret

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## [Version 1.1.0](https://github.com/dataiku/dss-plugin-jira/releases/tag/v1.1.0) - Feature release - 2025-04-30
+
+- Add AI agent tool for issue creation
+
 ## [Version 1.0.2](https://github.com/dataiku/dss-plugin-jira/releases/tag/v1.0.2) - Feature release - 2024-12-12
 
 - Add option to access Jira on premise

plugin.json

@@ -1,14 +1,14 @@
 {
     "id": "jira",
-    "version": "1.0.2",
+    "version": "1.1.0",
     "meta": {
         "label": "Jira",
         "description": "Import data from your Jira account",
         "author": "Dataiku",
         "icon": "icon-ticket",
-        "tags": ["Connector", "CRM"],
+        "tags": ["Connector", "CRM", "Generative AI", "Agents", "Tools"],
         "url": "https://www.dataiku.com/product/plugins/jira/",
         "licenseInfo": "Apache Software License",
         "supportLevel": "NOT_SUPPORTED"
     }
-}
+}

python-agent-tools/create-issue/tool.json

@@ -0,0 +1,38 @@
+{
+    "id": "create-jira-issue",
+    "meta": {
+        "icon": "icon-ticket",
+        "label": "Create Jira Issue",
+        "description": "Create a Jira issue based on user input"
+    },
+    
+    "params" : [
+        {
+            "name": "access_type",
+            "label": "Access type. Reviewer: hidden menu to keep for future extension",
+            "visibilityCondition": "false",
+            "type": "SELECT",
+            "defaultValue": "token_access",
+            "selectChoices": [
+                {
+                    "value": "token_access",
+                    "label": "Token"
+                }
+            ]
+        },
+        {
+            "name": "token_access",
+            "label": "Jira connection",
+            "type": "PRESET",
+            "parameterSetId": "basic-auth",
+            "visibilityCondition": "model.access_type == 'token_access'"
+        },
+        {
+            "name": "jira_project_key",
+            "label":"Jira Project Key",
+            "type": "STRING",
+            "mandatory": true,
+            "description": ""
+        }
+    ]
+}

python-agent-tools/create-issue/tool.py

@@ -0,0 +1,73 @@
+from dataiku.llm.agent_tools import BaseAgentTool
+import logging
+from jira_client import JiraClient
+from utils import get_connection_details
+
+
+class JiraCreateIssueTool(BaseAgentTool):
+    def set_config(self, config, plugin_config):
+        # This logger outputs the key in DEBUG mode ...
+        logging.getLogger("jiraapiclient.discovery").setLevel("INFO")
+        logging.info("JiraCreateIssueTool init")
+        self.config = config
+        connection_details = get_connection_details(config)
+        self.client = JiraClient(connection_details)
+        self.client.start_session("issue")
+        self.jira_project_key = config.get("jira_project_key")
+
+    def get_descriptor(self, tool):
+        return {
+            "description": "This tool is a wrapper around atlassian-python-api\'s Jira issue_create API, useful when you need to create a Jira issue. The input to this tool is a dictionary containing the new issue summary and description, e.g. '{'summary':'new issue summary', 'description':'new issue description'}'",            
+            "inputSchema": {
+                "$id": "https://dataiku.com/agents/tools/search/input",
+                "title": "Create Jira issue tool",
+                "type": "object",
+                "properties": {
+                    "summary": {
+                        "type": "string",
+                        "description": "The issue summary"
+                    },
+                    "description": {
+                        "type": "string",
+                        "description": "The issue description"
+                    }
+                },
+                "required": ["summary", "description"]
+            }
+        }
+
+    def create_jira_issue(self, summary: str, description: str, issue_type: str = "Task"):
+        try:
+            new_issue = self.client.create_issue(self.jira_project_key, summary, description, issue_type)
+            return new_issue
+
+        except Exception as exception:
+            return f"Error creating issue: {str(exception)}"
+
+    def invoke(self, input, trace):
+        args = input.get("input", {})
+
+        # Log inputs and config to trace
+        trace.span["name"] = "JIRA_CREATE_ISSUE_TOOL_CALL"
+        for key, value in args.items():
+            trace.inputs[key] = value
+        trace.attributes["config"] = self.config
+
+        summary = args.get("summary")
+        description = args.get("description")
+        jira_instance_url = self.client.get_site_url()
+        created_issue = self.create_jira_issue(summary, description)
+
+        if created_issue and "errors" in created_issue:
+            output_text = "There was a problem while creating the issue ticket: {}".format(
+                created_issue.get("errors", {}).get("description")
+            )
+        else:
+            output_text = f"Issue created: {created_issue.get('key')} available at {jira_instance_url}browse/{created_issue.get('key')}" if isinstance(created_issue, dict) else created_issue
+        
+        # Log outputs to trace
+        trace.outputs["output"] = output_text
+
+        return {
+            "output": output_text
+        }

python-lib/jira_client.py

@@ -148,7 +148,7 @@ def filter_data(self, data, item_value):
         if filtering_key is None:
             return arrayed(data)
         else:
-            return arrayed(extract_data_with_json_path(data,filtering_key))
+            return arrayed(extract_data_with_json_path(data, filtering_key))
 
     def format_data(self, data):
         for key in self.formating:
@@ -214,6 +214,31 @@ def get(self, url, data=None, params=None):
         response = requests.get(url, **args)
         return response
 
+    def post(self, url, data=None, json=None, params=None):
+        headers = self.get_headers()
+        headers['Content-Type'] = 'application/json'
+        headers['Accept'] = 'application/json'
+        headers.pop('X-ExperimentalApi', None)
+        auth = self.get_auth()
+        response = requests.post(url, params=params, auth=auth, data=data, json=json, headers=headers)
+        return response
+
+    def create_issue(self, jira_project_key, summary, description, issue_type):
+        issue_data = {
+            'project': {'key': jira_project_key},
+            'summary': summary,
+            'description': description,
+            'issuetype': {'name': issue_type}
+        }
+        json = {
+            "fields": issue_data,
+        }
+        # This data form does not work on v3
+        url = "/".join([self.get_site_url(), "rest/api/2/issue"])
+        response = self.post(url=url, json=json)
+        json_response = response.json()
+        return json_response
+
     def get_auth(self):
         if self.is_opsgenie_api():
             return None

python-lib/utils.py

@@ -17,4 +17,10 @@ def extract_data_with_json_path(data, json_path):
     keys = json_path.split(".")
     for key in keys:
         data = data.get(key, {})
-    return data
+    return data
+
+
+def get_connection_details(config):
+    access_type = config.get("access_type", "token_access")
+    connection_details = config.get(access_type)
+    return connection_details