Replace per-operation MCP tools with three meta-tools (search, describe, call) (#857)

Synced from sferarc/pgbeam@e56ff3e

Author: SferaDev

src/generated/index.ts

@@ -109,6 +109,7 @@ export type { Region, RegionProviderEnumKey, RegionStatusEnumKey } from "./types
 export type { RejectApprovalRequestData, RejectApprovalRequestPathApprovalId, RejectApprovalRequestPathProjectId, RejectApprovalRequestRequestConfig, RejectApprovalRequestResponse, RejectApprovalRequestResponses, RejectApprovalRequestStatus200, RejectApprovalRequestStatus400, RejectApprovalRequestStatus401, RejectApprovalRequestStatus403, RejectApprovalRequestStatus404, RejectApprovalRequestStatus409 } from "./types/RejectApprovalRequest";
 export type { Replica } from "./types/Replica";
 export type { RevokeAgentCredentialPathAgentId, RevokeAgentCredentialPathProjectId, RevokeAgentCredentialRequestConfig, RevokeAgentCredentialResponse, RevokeAgentCredentialResponses, RevokeAgentCredentialStatus204, RevokeAgentCredentialStatus400, RevokeAgentCredentialStatus401, RevokeAgentCredentialStatus403, RevokeAgentCredentialStatus404 } from "./types/RevokeAgentCredential";
+export type { RotateAgentCredentialPathAgentId, RotateAgentCredentialPathProjectId, RotateAgentCredentialRequestConfig, RotateAgentCredentialResponse, RotateAgentCredentialResponses, RotateAgentCredentialStatus200, RotateAgentCredentialStatus400, RotateAgentCredentialStatus401, RotateAgentCredentialStatus403, RotateAgentCredentialStatus404, RotateAgentCredentialStatus409 } from "./types/RotateAgentCredential";
 export type { RowFilter } from "./types/RowFilter";
 export type { SSLModeKey } from "./types/SSLMode";
 export type { StatementRules } from "./types/StatementRules";

src/generated/operations-map.gen.ts

@@ -9,6 +9,7 @@ import type { CreateAgentCredentialRequestConfig, CreateAgentCredentialStatus201
 import type { GetAgentCredentialRequestConfig, GetAgentCredentialStatus200 } from "./types/GetAgentCredential";
 import type { UpdateAgentCredentialStatusRequestConfig, UpdateAgentCredentialStatusStatus200 } from "./types/UpdateAgentCredentialStatus";
 import type { RevokeAgentCredentialRequestConfig } from "./types/RevokeAgentCredential";
+import type { RotateAgentCredentialRequestConfig, RotateAgentCredentialStatus200 } from "./types/RotateAgentCredential";
 import type { ListAnomalyAlertsRequestConfig, ListAnomalyAlertsStatus200 } from "./types/ListAnomalyAlerts";
 import type { UpdateAnomalyAlertRequestConfig, UpdateAnomalyAlertStatus200 } from "./types/UpdateAnomalyAlert";
 import type { ListApprovalRequestsRequestConfig, ListApprovalRequestsStatus200 } from "./types/ListApprovalRequests";
@@ -74,6 +75,7 @@ export const operationsByTag = {
     getAgentCredential: { method: "GET", path: "/v1/projects/{project_id}/agents/{agent_id}" },
     updateAgentCredentialStatus: { method: "PATCH", path: "/v1/projects/{project_id}/agents/{agent_id}" },
     revokeAgentCredential: { method: "DELETE", path: "/v1/projects/{project_id}/agents/{agent_id}" },
+    rotateAgentCredential: { method: "POST", path: "/v1/projects/{project_id}/agents/{agent_id}/rotate" },
     listAuditLogs: { method: "GET", path: "/v1/projects/{project_id}/audit-logs" },
   },
   anomalies: {
@@ -154,6 +156,7 @@ export const operationsByPath = {
   "GET /v1/projects/{project_id}/agents/{agent_id}": { method: "GET", path: "/v1/projects/{project_id}/agents/{agent_id}", operationId: "getAgentCredential" },
   "PATCH /v1/projects/{project_id}/agents/{agent_id}": { method: "PATCH", path: "/v1/projects/{project_id}/agents/{agent_id}", operationId: "updateAgentCredentialStatus" },
   "DELETE /v1/projects/{project_id}/agents/{agent_id}": { method: "DELETE", path: "/v1/projects/{project_id}/agents/{agent_id}", operationId: "revokeAgentCredential" },
+  "POST /v1/projects/{project_id}/agents/{agent_id}/rotate": { method: "POST", path: "/v1/projects/{project_id}/agents/{agent_id}/rotate", operationId: "rotateAgentCredential" },
   "GET /v1/projects/{project_id}/anomalies": { method: "GET", path: "/v1/projects/{project_id}/anomalies", operationId: "listAnomalyAlerts" },
   "PATCH /v1/projects/{project_id}/anomalies/{anomaly_id}": { method: "PATCH", path: "/v1/projects/{project_id}/anomalies/{anomaly_id}", operationId: "updateAnomalyAlert" },
   "GET /v1/projects/{project_id}/approvals": { method: "GET", path: "/v1/projects/{project_id}/approvals", operationId: "listApprovalRequests" },
@@ -213,6 +216,7 @@ type CreateAgentCredentialParams = { pathParams: NonNullable<CreateAgentCredenti
 type GetAgentCredentialParams = { pathParams: NonNullable<GetAgentCredentialRequestConfig["pathParams"]> };
 type UpdateAgentCredentialStatusParams = { pathParams: NonNullable<UpdateAgentCredentialStatusRequestConfig["pathParams"]>; body: NonNullable<UpdateAgentCredentialStatusRequestConfig["data"]> };
 type RevokeAgentCredentialParams = { pathParams: NonNullable<RevokeAgentCredentialRequestConfig["pathParams"]> };
+type RotateAgentCredentialParams = { pathParams: NonNullable<RotateAgentCredentialRequestConfig["pathParams"]> };
 type ListAnomalyAlertsParams = { pathParams: NonNullable<ListAnomalyAlertsRequestConfig["pathParams"]>; queryParams?: NonNullable<ListAnomalyAlertsRequestConfig["queryParams"]> };
 type UpdateAnomalyAlertParams = { pathParams: NonNullable<UpdateAnomalyAlertRequestConfig["pathParams"]>; body: NonNullable<UpdateAnomalyAlertRequestConfig["data"]> };
 type ListApprovalRequestsParams = { pathParams: NonNullable<ListApprovalRequestsRequestConfig["pathParams"]>; queryParams?: NonNullable<ListApprovalRequestsRequestConfig["queryParams"]> };
@@ -275,6 +279,7 @@ export interface ApiOperations {
     getAgentCredential(params: GetAgentCredentialParams): Promise<GetAgentCredentialStatus200>;
     updateAgentCredentialStatus(params: UpdateAgentCredentialStatusParams): Promise<UpdateAgentCredentialStatusStatus200>;
     revokeAgentCredential(params: RevokeAgentCredentialParams): Promise<void>;
+    rotateAgentCredential(params: RotateAgentCredentialParams): Promise<RotateAgentCredentialStatus200>;
     listAuditLogs(params: ListAuditLogsParams): Promise<ListAuditLogsStatus200>;
   };
   anomalies: {
@@ -355,6 +360,7 @@ export interface RequestMap {
   "GET /v1/projects/{project_id}/agents/{agent_id}": { params: GetAgentCredentialParams; response: GetAgentCredentialStatus200 };
   "PATCH /v1/projects/{project_id}/agents/{agent_id}": { params: UpdateAgentCredentialStatusParams; response: UpdateAgentCredentialStatusStatus200 };
   "DELETE /v1/projects/{project_id}/agents/{agent_id}": { params: RevokeAgentCredentialParams; response: void };
+  "POST /v1/projects/{project_id}/agents/{agent_id}/rotate": { params: RotateAgentCredentialParams; response: RotateAgentCredentialStatus200 };
   "GET /v1/projects/{project_id}/anomalies": { params: ListAnomalyAlertsParams; response: ListAnomalyAlertsStatus200 };
   "PATCH /v1/projects/{project_id}/anomalies/{anomaly_id}": { params: UpdateAnomalyAlertParams; response: UpdateAnomalyAlertStatus200 };
   "GET /v1/projects/{project_id}/approvals": { params: ListApprovalRequestsParams; response: ListApprovalRequestsStatus200 };

src/generated/types/AgentCredentialSecrets.ts

@@ -22,8 +22,8 @@ export type AgentCredentialSecrets = {
     */
     connection_string: string;
     /**
-     * @description Hosted MCP endpoint URL for this credential.
-     * @example https://api.pgbeam.com/mcp
+     * @description Hosted agent-database MCP endpoint. Paste into an MCP client (Claude Code, Cursor, …) with the mcp_token as a bearer token. The endpoint is token-scoped, so the same URL serves every credential.
+     * @example https://mcp.pgbeam.app/mcp
      * @type string
     */
     mcp_url: string;

src/generated/types/OnboardingProgress.ts

@@ -43,6 +43,11 @@ export type OnboardingProgress = {
      * @type boolean
     */
     first_query_run: boolean;
+    /**
+     * @description Whether an agent credential has been issued.
+     * @type boolean
+    */
+    agent_credential_created: boolean;
     /**
      * @description Whether the user has dismissed the onboarding checklist.
      * @type boolean

src/generated/types/RotateAgentCredential.ts

@@ -0,0 +1,94 @@
+/**
+* Generated by Kubb (https://kubb.dev/).
+* Do not edit manually.
+*/
+
+import type { AgentCredentialSecrets } from "./AgentCredentialSecrets";
+import type { Error } from "./Error";
+
+/**
+ * @description Unique project identifier (prefixed, e.g. prj_xxx).
+ * @pattern ^[a-zA-Z0-9_.-]+$
+ * @type string
+*/
+export type RotateAgentCredentialPathProjectId = string;
+
+/**
+ * @description Unique agent credential identifier (prefixed, e.g. agt_xxx).
+ * @pattern ^[a-zA-Z0-9_.-]+$
+ * @type string
+*/
+export type RotateAgentCredentialPathAgentId = string;
+
+/**
+ * @description One-time secrets returned only at creation. The connection string and MCP token are not retrievable again.
+ * @type object
+*/
+export type RotateAgentCredentialStatus200 = AgentCredentialSecrets;
+
+/**
+ * @description Standard error response envelope for PgBeam API requests.
+ * @type object
+*/
+export type RotateAgentCredentialStatus400 = Error;
+
+/**
+ * @description Standard error response envelope for PgBeam API requests.
+ * @type object
+*/
+export type RotateAgentCredentialStatus401 = Error;
+
+/**
+ * @description Standard error response envelope for PgBeam API requests.
+ * @type object
+*/
+export type RotateAgentCredentialStatus403 = Error;
+
+/**
+ * @description Standard error response envelope for PgBeam API requests.
+ * @type object
+*/
+export type RotateAgentCredentialStatus404 = Error;
+
+/**
+ * @description Standard error response envelope for PgBeam API requests.
+ * @type object
+*/
+export type RotateAgentCredentialStatus409 = Error;
+
+/**
+ * @type object
+*/
+export type RotateAgentCredentialRequestConfig = {
+    data?: never;
+    /**
+     * @type object
+    */
+    pathParams: {
+        project_id: RotateAgentCredentialPathProjectId;
+        agent_id: RotateAgentCredentialPathAgentId;
+    };
+    queryParams?: never;
+    headerParams?: never;
+    /**
+     * @type string
+    */
+    url: `/v1/projects/${string}/agents/${string}/rotate`;
+};
+
+/**
+ * @type object
+*/
+export type RotateAgentCredentialResponses = {
+    "200": RotateAgentCredentialStatus200;
+    "400": RotateAgentCredentialStatus400;
+    "401": RotateAgentCredentialStatus401;
+    "403": RotateAgentCredentialStatus403;
+    "404": RotateAgentCredentialStatus404;
+    "409": RotateAgentCredentialStatus409;
+};
+
+/**
+ * @description Union of all possible responses
+*/
+export type RotateAgentCredentialResponse = (RotateAgentCredentialStatus200 | RotateAgentCredentialStatus400 | RotateAgentCredentialStatus401 | RotateAgentCredentialStatus403 | RotateAgentCredentialStatus404 | RotateAgentCredentialStatus409);

src/generated/types/UpdateOnboardingRequest.ts

@@ -8,7 +8,8 @@ export const updateOnboardingRequestStepEnum = {
     database_added: "database_added",
     connection_tested: "connection_tested",
     connection_string_copied: "connection_string_copied",
-    first_query_run: "first_query_run"
+    first_query_run: "first_query_run",
+    agent_credential_created: "agent_credential_created"
 } as const;
 
 export type UpdateOnboardingRequestStepEnumKey = (typeof updateOnboardingRequestStepEnum)[keyof typeof updateOnboardingRequestStepEnum];

src/generated/types/index.ts

@@ -109,6 +109,7 @@ export type { Region, RegionProviderEnumKey, RegionStatusEnumKey } from "./Regio
 export type { RejectApprovalRequestData, RejectApprovalRequestPathApprovalId, RejectApprovalRequestPathProjectId, RejectApprovalRequestRequestConfig, RejectApprovalRequestResponse, RejectApprovalRequestResponses, RejectApprovalRequestStatus200, RejectApprovalRequestStatus400, RejectApprovalRequestStatus401, RejectApprovalRequestStatus403, RejectApprovalRequestStatus404, RejectApprovalRequestStatus409 } from "./RejectApprovalRequest";
 export type { Replica } from "./Replica";
 export type { RevokeAgentCredentialPathAgentId, RevokeAgentCredentialPathProjectId, RevokeAgentCredentialRequestConfig, RevokeAgentCredentialResponse, RevokeAgentCredentialResponses, RevokeAgentCredentialStatus204, RevokeAgentCredentialStatus400, RevokeAgentCredentialStatus401, RevokeAgentCredentialStatus403, RevokeAgentCredentialStatus404 } from "./RevokeAgentCredential";
+export type { RotateAgentCredentialPathAgentId, RotateAgentCredentialPathProjectId, RotateAgentCredentialRequestConfig, RotateAgentCredentialResponse, RotateAgentCredentialResponses, RotateAgentCredentialStatus200, RotateAgentCredentialStatus400, RotateAgentCredentialStatus401, RotateAgentCredentialStatus403, RotateAgentCredentialStatus404, RotateAgentCredentialStatus409 } from "./RotateAgentCredential";
 export type { RowFilter } from "./RowFilter";
 export type { SSLModeKey } from "./SSLMode";
 export type { StatementRules } from "./StatementRules";