@orq-ai/node 3.2.0-rc.9 → 3.2.1

package/packages/orq-rc/src/mcp-server/server.ts

@@ -0,0 +1,123 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { OrqCore } from "../core.js";
+import { SDKOptions } from "../lib/config.js";
+import type { ConsoleLogger } from "./console-logger.js";
+import { createRegisterPrompt } from "./prompts.js";
+import {
+  createRegisterResource,
+  createRegisterResourceTemplate,
+} from "./resources.js";
+import { MCPScope, mcpScopes } from "./scopes.js";
+import { createRegisterTool } from "./tools.js";
+import { tool$contactsCreate } from "./tools/contactsCreate.js";
+import { tool$datasetsClear } from "./tools/datasetsClear.js";
+import { tool$datasetsCreate } from "./tools/datasetsCreate.js";
+import { tool$datasetsCreateDatapoint } from "./tools/datasetsCreateDatapoint.js";
+import { tool$datasetsCreateDatapoints } from "./tools/datasetsCreateDatapoints.js";
+import { tool$datasetsDelete } from "./tools/datasetsDelete.js";
+import { tool$datasetsDeleteDatapoint } from "./tools/datasetsDeleteDatapoint.js";
+import { tool$datasetsList } from "./tools/datasetsList.js";
+import { tool$datasetsListDatapoints } from "./tools/datasetsListDatapoints.js";
+import { tool$datasetsRetrieve } from "./tools/datasetsRetrieve.js";
+import { tool$datasetsRetrieveDatapoint } from "./tools/datasetsRetrieveDatapoint.js";
+import { tool$datasetsUpdate } from "./tools/datasetsUpdate.js";
+import { tool$datasetsUpdateDatapoint } from "./tools/datasetsUpdateDatapoint.js";
+import { tool$deploymentsGetConfig } from "./tools/deploymentsGetConfig.js";
+import { tool$deploymentsInvoke } from "./tools/deploymentsInvoke.js";
+import { tool$deploymentsList } from "./tools/deploymentsList.js";
+import { tool$deploymentsMetricsCreate } from "./tools/deploymentsMetricsCreate.js";
+import { tool$feedbackCreate } from "./tools/feedbackCreate.js";
+import { tool$filesCreate } from "./tools/filesCreate.js";
+import { tool$filesDelete } from "./tools/filesDelete.js";
+import { tool$filesGet } from "./tools/filesGet.js";
+import { tool$filesList } from "./tools/filesList.js";
+import { tool$promptsCreate } from "./tools/promptsCreate.js";
+import { tool$promptsDelete } from "./tools/promptsDelete.js";
+import { tool$promptsGetVersion } from "./tools/promptsGetVersion.js";
+import { tool$promptsList } from "./tools/promptsList.js";
+import { tool$promptsListVersions } from "./tools/promptsListVersions.js";
+import { tool$promptsRetrieve } from "./tools/promptsRetrieve.js";
+import { tool$promptsUpdate } from "./tools/promptsUpdate.js";
+import { tool$remoteconfigsRetrieve } from "./tools/remoteconfigsRetrieve.js";
+
+export function createMCPServer(deps: {
+  logger: ConsoleLogger;
+  allowedTools?: string[] | undefined;
+  scopes?: MCPScope[] | undefined;
+  serverURL?: string | undefined;
+  apiKey?: SDKOptions["apiKey"] | undefined;
+  contactId?: SDKOptions["contactId"] | undefined;
+  environment?: SDKOptions["environment"] | undefined;
+  serverIdx?: SDKOptions["serverIdx"] | undefined;
+}) {
+  const server = new McpServer({
+    name: "Orq",
+    version: "3.2.0-rc.62",
+  });
+
+  const client = new OrqCore({
+    apiKey: deps.apiKey,
+    contactId: deps.contactId,
+    environment: deps.environment,
+    serverURL: deps.serverURL,
+    serverIdx: deps.serverIdx,
+  });
+
+  const scopes = new Set(deps.scopes ?? mcpScopes);
+
+  const allowedTools = deps.allowedTools && new Set(deps.allowedTools);
+  const tool = createRegisterTool(
+    deps.logger,
+    server,
+    client,
+    scopes,
+    allowedTools,
+  );
+  const resource = createRegisterResource(deps.logger, server, client, scopes);
+  const resourceTemplate = createRegisterResourceTemplate(
+    deps.logger,
+    server,
+    client,
+    scopes,
+  );
+  const prompt = createRegisterPrompt(deps.logger, server, client, scopes);
+  const register = { tool, resource, resourceTemplate, prompt };
+  void register; // suppress unused warnings
+
+  tool(tool$contactsCreate);
+  tool(tool$feedbackCreate);
+  tool(tool$deploymentsList);
+  tool(tool$deploymentsGetConfig);
+  tool(tool$deploymentsInvoke);
+  tool(tool$filesCreate);
+  tool(tool$filesList);
+  tool(tool$filesGet);
+  tool(tool$filesDelete);
+  tool(tool$promptsList);
+  tool(tool$promptsRetrieve);
+  tool(tool$promptsUpdate);
+  tool(tool$promptsDelete);
+  tool(tool$promptsCreate);
+  tool(tool$promptsListVersions);
+  tool(tool$promptsGetVersion);
+  tool(tool$remoteconfigsRetrieve);
+  tool(tool$datasetsList);
+  tool(tool$datasetsCreate);
+  tool(tool$datasetsRetrieve);
+  tool(tool$datasetsUpdate);
+  tool(tool$datasetsDelete);
+  tool(tool$datasetsListDatapoints);
+  tool(tool$datasetsCreateDatapoint);
+  tool(tool$datasetsRetrieveDatapoint);
+  tool(tool$datasetsUpdateDatapoint);
+  tool(tool$datasetsDeleteDatapoint);
+  tool(tool$datasetsCreateDatapoints);
+  tool(tool$datasetsClear);
+  tool(tool$deploymentsMetricsCreate);
+
+  return server;
+}

package/packages/orq-rc/src/mcp-server/shared.ts

@@ -0,0 +1,75 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { bytesToBase64 } from "../lib/base64.js";
+
+type BinaryData =
+  | Uint8Array
+  | ArrayBuffer
+  | Blob
+  | ReadableStream
+  | Response
+  | string;
+
+export async function consumeStream(
+  stream: ReadableStream<Uint8Array>,
+): Promise<Uint8Array> {
+  const reader = stream.getReader();
+  const chunks: Uint8Array[] = [];
+
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (value != null) chunks.push(value);
+      if (done) break;
+    }
+  } finally {
+    reader.releaseLock();
+  }
+
+  return new Uint8Array(await new Blob(chunks).arrayBuffer());
+}
+
+export function isAsyncIterable(
+  value: unknown,
+): value is AsyncIterable<string> {
+  return (
+    typeof value === "object" && value != null && Symbol.asyncIterator in value
+  );
+}
+
+export function isBinaryData(value: unknown): value is BinaryData {
+  return (
+    value instanceof Uint8Array
+    || value instanceof ArrayBuffer
+    || value instanceof Blob
+    || value instanceof ReadableStream
+    || value instanceof Response
+    || typeof value === "string"
+  );
+}
+
+const base64Schema = z.string().base64();
+
+export async function valueToBase64(
+  value: BinaryData | null | undefined,
+): Promise<string | null> {
+  if (value == null) {
+    return null;
+  } else if (value instanceof Uint8Array) {
+    return bytesToBase64(value);
+  } else if (value instanceof ArrayBuffer) {
+    return bytesToBase64(new Uint8Array(value));
+  } else if (value instanceof Response || value instanceof Blob) {
+    return bytesToBase64(new Uint8Array(await value.arrayBuffer()));
+  } else if (value instanceof ReadableStream) {
+    return bytesToBase64(await consumeStream(value));
+  } else if (typeof value === "string") {
+    return base64Schema.parse(value);
+  } else {
+    value satisfies never;
+    throw new Error(`Unsupported image value type: ${typeof value}`);
+  }
+}

package/packages/orq-rc/src/mcp-server/tools/contactsCreate.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { contactsCreate } from "../../funcs/contactsCreate.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.CreateContactRequestBody$inboundSchema,
+};
+
+export const tool$contactsCreate: ToolDefinition<typeof args> = {
+  name: "contacts-create",
+  description: `Update user information
+
+Update or add user information to workspace`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await contactsCreate(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsClear.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsClear } from "../../funcs/datasetsClear.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ClearDatasetRequest$inboundSchema,
+};
+
+export const tool$datasetsClear: ToolDefinition<typeof args> = {
+  name: "datasets-clear",
+  description: `Delete all datapoints
+
+Delete all datapoints from a dataset. This action is irreversible.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsClear(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    return formatResult(void 0, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsCreate.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsCreate } from "../../funcs/datasetsCreate.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.CreateDatasetRequestBody$inboundSchema.optional(),
+};
+
+export const tool$datasetsCreate: ToolDefinition<typeof args> = {
+  name: "datasets-create",
+  description: `Create a dataset
+
+Creates a new dataset in the specified project.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsCreate(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsCreateDatapoint.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsCreateDatapoint } from "../../funcs/datasetsCreateDatapoint.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.CreateDatasetItemRequest$inboundSchema,
+};
+
+export const tool$datasetsCreateDatapoint: ToolDefinition<typeof args> = {
+  name: "datasets-create-datapoint",
+  description: `Create a datapoint
+
+Creates a new datapoint in the specified dataset.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsCreateDatapoint(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsCreateDatapoints.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsCreateDatapoints } from "../../funcs/datasetsCreateDatapoints.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.BulkCreateDatapointsRequest$inboundSchema,
+};
+
+export const tool$datasetsCreateDatapoints: ToolDefinition<typeof args> = {
+  name: "datasets-create-datapoints",
+  description: `Create multiple datapoints
+
+Create multiple datapoints at once.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsCreateDatapoints(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsDelete.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsDelete } from "../../funcs/datasetsDelete.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeleteDatasetRequest$inboundSchema,
+};
+
+export const tool$datasetsDelete: ToolDefinition<typeof args> = {
+  name: "datasets-delete",
+  description: `Delete a dataset
+
+Permanently deletes a dataset and all its datapoints. This action is irreversible.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsDelete(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    return formatResult(void 0, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsDeleteDatapoint.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsDeleteDatapoint } from "../../funcs/datasetsDeleteDatapoint.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeleteDatapointRequest$inboundSchema,
+};
+
+export const tool$datasetsDeleteDatapoint: ToolDefinition<typeof args> = {
+  name: "datasets-delete-datapoint",
+  description: `Delete a datapoint
+
+Permanently deletes a specific datapoint from a dataset.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsDeleteDatapoint(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    return formatResult(void 0, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsList.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsList } from "../../funcs/datasetsList.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ListDatasetsRequest$inboundSchema.optional(),
+};
+
+export const tool$datasetsList: ToolDefinition<typeof args> = {
+  name: "datasets-list",
+  description: `List datasets
+
+Retrieves a paginated list of datasets for the current workspace. Results can be paginated using cursor-based pagination.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsList(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsListDatapoints.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsListDatapoints } from "../../funcs/datasetsListDatapoints.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ListDatasetDatapointsRequest$inboundSchema,
+};
+
+export const tool$datasetsListDatapoints: ToolDefinition<typeof args> = {
+  name: "datasets-list-datapoints",
+  description: `List datapoints
+
+Retrieves a paginated list of datapoints from a specific dataset.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsListDatapoints(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsRetrieve.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsRetrieve } from "../../funcs/datasetsRetrieve.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.RetrieveDatasetRequest$inboundSchema,
+};
+
+export const tool$datasetsRetrieve: ToolDefinition<typeof args> = {
+  name: "datasets-retrieve",
+  description: `Retrieve a dataset
+
+Retrieves a specific dataset by its unique identifier`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsRetrieve(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsRetrieveDatapoint.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsRetrieveDatapoint } from "../../funcs/datasetsRetrieveDatapoint.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.RetrieveDatapointRequest$inboundSchema,
+};
+
+export const tool$datasetsRetrieveDatapoint: ToolDefinition<typeof args> = {
+  name: "datasets-retrieve-datapoint",
+  description: `Retrieve a datapoint
+
+Retrieves a datapoint object`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsRetrieveDatapoint(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsUpdate.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsUpdate } from "../../funcs/datasetsUpdate.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.UpdateDatasetRequest$inboundSchema,
+};
+
+export const tool$datasetsUpdate: ToolDefinition<typeof args> = {
+  name: "datasets-update",
+  description: `Update a dataset
+
+Update a dataset`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsUpdate(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/packages/orq-rc/src/mcp-server/tools/datasetsUpdateDatapoint.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsUpdateDatapoint } from "../../funcs/datasetsUpdateDatapoint.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.UpdateDatapointRequest$inboundSchema,
+};
+
+export const tool$datasetsUpdateDatapoint: ToolDefinition<typeof args> = {
+  name: "datasets-update-datapoint",
+  description: `Update a datapoint`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await datasetsUpdateDatapoint(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};