Skip to content

feat: Create issue tool #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .devrev/repo.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
deployable: true
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

do we need this? this is not deployable.

1 change: 1 addition & 0 deletions .github/CODEOWNERS
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
* @kpsunil97
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add newline

4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,10 @@ A Model Context Protocol server for DevRev. It is used to search and retrieve in

- `search`: Search for information using the DevRev search API with the provided query and namespace.
- `get_object`: Get all information about a DevRev issue or ticket using its ID.
- `create_issue`: Creates a DevRev issue with a specified title and part. Parameters:
- `title`: The issue title (required)
- `part_name`: Name of the part to associate with the issue (required).
- The issue will be automatically assigned to the currently authenticated user
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can we add the owner too which if not provided can be assigned to the current user?


## Configuration

Expand Down
73 changes: 72 additions & 1 deletion src/devrev_mcp/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from mcp.server import NotificationOptions, Server
from pydantic import AnyUrl
import mcp.server.stdio
from .utils import make_devrev_request
from .utils import make_devrev_request, search_part_by_name, get_current_user_id

server = Server("devrev_mcp")

Expand Down Expand Up @@ -47,6 +47,18 @@ async def handle_list_tools() -> list[types.Tool]:
},
"required": ["id"],
},
),
types.Tool(
name="create_issue",
description="Create a DevRev issue with the specified title. The part_name parameter is used to search for and identify the appropriate part ID, and the issue will be automatically assigned to the currently authenticated user.",
inputSchema={
"type": "object",
"properties": {
"title": {"type": "string"},
"part_name": {"type": "string"},
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add description too, easier of the fields to be predicted by agent. move part to part id and let agent figure out the part id from the name with the search tool.

},
"required": ["title", "part_name"],
},
)
]

Expand Down Expand Up @@ -118,6 +130,65 @@ async def handle_call_tool(
text=f"Object information for '{id}':\n{object_info}"
)
]
elif name == "create_issue":
if not arguments:
raise ValueError("Missing arguments")

title = arguments.get("title")
part_name = arguments.get("part_name")

if not title:
raise ValueError("Missing title parameter")

if not part_name:
raise ValueError("Missing part_name parameter")

# Search for part by name
success, part_id, error_message = search_part_by_name(part_name)
if not success:
return [
types.TextContent(
type="text",
text=error_message
)
]

# Get current user ID
success, user_id, error_message = get_current_user_id()
if not success:
return [
types.TextContent(
type="text",
text=error_message
)
]

response = make_devrev_request(
"works.create",
{
"type": "issue",
"title": title,
"applies_to_part": part_id,
"owned_by": [user_id]
}
)

if response.status_code != 201:
error_text = response.text
return [
types.TextContent(
type="text",
text=f"Create issue failed with status {response.status_code}: {error_text}"
)
]

issue_info = response.json()
return [
types.TextContent(
type="text",
text=f"Issue created successfully with ID: {issue_info['display_id']} Info: {issue_info}"
)
]
else:
raise ValueError(f"Unknown tool: {name}")

Expand Down
55 changes: 54 additions & 1 deletion src/devrev_mcp/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

import os
import requests
from typing import Any, Dict
from typing import Any, Dict, Tuple, Optional

def make_devrev_request(endpoint: str, payload: Dict[str, Any]) -> requests.Response:
"""
Expand Down Expand Up @@ -37,3 +37,56 @@ def make_devrev_request(endpoint: str, payload: Dict[str, Any]) -> requests.Resp
headers=headers,
json=payload
)

def search_part_by_name(part_name: str) -> Tuple[bool, Optional[str], Optional[str]]:
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

instead allow part as namespace in search tool. agent would figure out the relevant part while searching.

"""
Search for a part by name and return its ID if found.

Args:
part_name: The name of the part to search for

Returns:
Tuple containing:
- bool: Success status
- Optional[str]: Part ID if found, None otherwise
- Optional[str]: Error message if there was an error, None otherwise
"""
try:
search_response = make_devrev_request(
"search.hybrid",
{"query": part_name, "namespace": "part"}
)
if search_response.status_code != 200:
return False, None, f"Search for part failed with status {search_response.status_code}: {search_response.text}"

search_results = search_response.json()
if not search_results.get("results") or len(search_results.get("results")) == 0:
return False, None, f"No parts found matching '{part_name}'"

part_id = search_results.get("results")[0].get("part").get("id")
return True, part_id, None
except Exception as e:
return False, None, f"Failed to search for part: {str(e)}"

def get_current_user_id() -> Tuple[bool, Optional[str], Optional[str]]:
"""
Get the ID of the current authenticated user.

Returns:
Tuple containing:
- bool: Success status
- Optional[str]: User ID if successful, None otherwise
- Optional[str]: Error message if there was an error, None otherwise
"""
try:
owned_by = make_devrev_request(
"dev-users.self",
{}
)
if owned_by.status_code != 200:
return False, None, f"Get user failed with status {owned_by.status_code}: {owned_by.text}"

user_id = owned_by.json().get("dev_user").get("id")
return True, user_id, None
except Exception as e:
return False, None, f"Failed to get current user: {str(e)}"