9c1b2b9275
Zendesk who_am_i tool. --------- Co-authored-by: Francisco Liberal <francisco@arcade.dev>
120 lines
3.6 KiB
Python
120 lines
3.6 KiB
Python
from typing import Any, TypedDict
|
|
|
|
import httpx
|
|
from arcade_tdk import ToolContext
|
|
|
|
|
|
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: ToolContext) -> 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: ToolContext) -> 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: ToolContext, 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
|