Skip to content

Add get_vistas tool for retrieving DevRev vista information #30

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 2 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
52 changes: 51 additions & 1 deletion src/devrev_mcp/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
import asyncio
import os
import requests
import json
Copy link
Contributor

Choose a reason for hiding this comment

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

These imports are not used, remove them

import httpx

from mcp.server.models import InitializationOptions
import mcp.types as types
Expand All @@ -30,6 +32,20 @@ async def handle_list_tools() -> list[types.Tool]:
description="Fetch the current DevRev user details. When the user specifies 'me' in the query, this tool should be called to get the user details.",
inputSchema={"type": "object", "properties": {}},
),
types.Tool(
name="get_vistas",
Copy link
Contributor

Choose a reason for hiding this comment

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

name of the tool should be "get_vista"

description="Retrieve all available vistas (filtered views) from DevRev",
Copy link
Contributor

Choose a reason for hiding this comment

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

Fix this description, the tool fetches information about a vista in DevRev using its ID.

inputSchema={
"type": "object",
"properties": {
"id": {
"type": "string",
"description": "The DevRev ID of the vista"
}
},
"required": ["id"]
},
),
types.Tool(
name="search",
description="Search DevRev using the provided query",
Expand All @@ -39,7 +55,7 @@ async def handle_list_tools() -> list[types.Tool]:
"query": {"type": "string"},
"namespace": {
"type": "string",
"enum": ["article", "issue", "ticket", "part", "dev_user", "account", "rev_org"],
"enum": ["article", "issue", "ticket", "part", "dev_user", "account", "rev_org", "vista"],
"description": "The namespace to search in. Use this to specify the type of object to search for."
},
},
Expand Down Expand Up @@ -439,6 +455,39 @@ async def handle_call_tool(
text=f"Current DevRev user details: {response.json()}"
)
]

elif name == "get_vistas":
if not arguments:
raise ValueError("Missing arguments")

id = arguments.get("id")
if not id:
raise ValueError("Missing id ")

response = make_devrev_request(
"vistas.get",
{
"id": id
}
)

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

return [
types.TextContent(
type="text",
text=f"Vista details for '{id}':\n{response.json()}"
)
]


elif name == "search":
if not arguments:
raise ValueError("Missing arguments")
Expand Down Expand Up @@ -547,6 +596,7 @@ async def handle_call_tool(
text=f"Object created successfully: {response.json()}"
)
]

elif name == "update_work":
if not arguments:
raise ValueError("Missing arguments")
Expand Down
Loading