c50699d5e6
<!-- CURSOR_SUMMARY --> > [!NOTE] > **Medium Risk** > Touches multiple toolkits’ runtime entrypoints and context/error/auth plumbing, so breakage risk is mainly around invocation/packaging and tool execution wiring rather than business logic. > > **Overview** > Migrates the BrightData, ClickHouse, LinkedIn, Math, MongoDB, Postgres, and Zendesk OSS toolkits from `arcade-tdk` to `arcade-mcp-server` APIs by updating tool decorators, `Context` types, auth classes, and exception imports. > > Adds per-toolkit `__main__.py` files that construct an `MCPApp`, register module tools, and run via configurable transport/host/port; corresponding `pyproject.toml` updates bump versions, drop `arcade-tdk`/`arcade-serve` deps, and add `project.scripts` console entrypoints. > > Updates tests and eval suites to use `arcade_mcp_server.Context` (mocked) and switches eval `ToolCatalog` imports to `arcade_core`. > > <sup>Written by [Cursor Bugbot](https://cursor.com/dashboard?tab=bugbot) for commit 9b3e31acb4b35e1d72efd47e2d279c5b19e3ecb0. This will update automatically on new commits. Configure [here](https://cursor.com/dashboard?tab=bugbot).</sup> <!-- /CURSOR_SUMMARY -->
118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
from typing import Any, TypedDict
|
|
|
|
import httpx
|
|
from arcade_mcp_server import Context
|
|
|
|
|
|
class WhoAmIResponse(TypedDict, total=False):
|
|
user_id: int
|
|
name: str
|
|
email: str
|
|
role: str
|
|
active: bool
|
|
verified: bool
|
|
locale: str
|
|
time_zone: str
|
|
organization_id: int
|
|
organization_name: str
|
|
organization_domains: list[str]
|
|
zendesk_access: bool
|
|
|
|
|
|
async def build_who_am_i_response(context: Context) -> WhoAmIResponse:
|
|
"""Build comprehensive who am I response for Zendesk."""
|
|
user_info = await _get_current_user(context)
|
|
organization_info = await _get_organization_info(context, user_info.get("organization_id"))
|
|
|
|
response_data = {}
|
|
response_data.update(_extract_user_info(user_info))
|
|
response_data.update(_extract_organization_info(organization_info))
|
|
response_data["zendesk_access"] = True
|
|
|
|
return response_data # type: ignore[return-value]
|
|
|
|
|
|
async def _get_current_user(context: Context) -> dict[str, Any]:
|
|
"""Get current user information from Zendesk API."""
|
|
subdomain = context.get_secret("ZENDESK_SUBDOMAIN")
|
|
base_url = f"https://{subdomain}.zendesk.com"
|
|
|
|
headers = {
|
|
"Authorization": f"Bearer {context.get_auth_token_or_empty()}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
response = await client.get(f"{base_url}/api/v2/users/me", headers=headers)
|
|
response.raise_for_status()
|
|
return response.json().get("user", {}) # type: ignore[no-any-return]
|
|
|
|
|
|
async def _get_organization_info(context: Context, organization_id: int | None) -> dict[str, Any]:
|
|
"""Get organization information from Zendesk API."""
|
|
if not organization_id:
|
|
return {}
|
|
|
|
subdomain = context.get_secret("ZENDESK_SUBDOMAIN")
|
|
base_url = f"https://{subdomain}.zendesk.com"
|
|
|
|
headers = {
|
|
"Authorization": f"Bearer {context.get_auth_token_or_empty()}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
response = await client.get(
|
|
f"{base_url}/api/v2/organizations/{organization_id}", headers=headers
|
|
)
|
|
response.raise_for_status()
|
|
return response.json().get("organization", {}) # type: ignore[no-any-return]
|
|
|
|
|
|
def _extract_user_info(user_info: dict[str, Any]) -> dict[str, Any]:
|
|
"""Extract user information from Zendesk user response."""
|
|
extracted = {}
|
|
|
|
if user_info.get("id"):
|
|
extracted["user_id"] = user_info["id"]
|
|
|
|
if user_info.get("name"):
|
|
extracted["name"] = user_info["name"]
|
|
|
|
if user_info.get("email"):
|
|
extracted["email"] = user_info["email"]
|
|
|
|
if user_info.get("role"):
|
|
extracted["role"] = user_info["role"]
|
|
|
|
if "active" in user_info:
|
|
extracted["active"] = user_info["active"]
|
|
|
|
if "verified" in user_info:
|
|
extracted["verified"] = user_info["verified"]
|
|
|
|
if user_info.get("locale"):
|
|
extracted["locale"] = user_info["locale"]
|
|
|
|
if user_info.get("time_zone"):
|
|
extracted["time_zone"] = user_info["time_zone"]
|
|
|
|
if user_info.get("organization_id"):
|
|
extracted["organization_id"] = user_info["organization_id"]
|
|
|
|
return extracted
|
|
|
|
|
|
def _extract_organization_info(organization_info: dict[str, Any]) -> dict[str, Any]:
|
|
"""Extract organization information from Zendesk organization response."""
|
|
extracted = {}
|
|
|
|
if organization_info.get("name"):
|
|
extracted["organization_name"] = organization_info["name"]
|
|
|
|
if organization_info.get("domain_names"):
|
|
domains = organization_info["domain_names"]
|
|
if domains:
|
|
extracted["organization_domains"] = domains
|
|
|
|
return extracted
|