@orq-ai/node 3.6.2 → 3.7.0-rc.2

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

@@ -1,117 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
-import {
-  objectOutputType,
-  ZodOptional,
-  ZodType,
-  ZodTypeAny,
-  ZodTypeDef,
-} from "zod";
-import { OrqCore } from "../core.js";
-import { ConsoleLogger } from "./console-logger.js";
-import { MCPScope } from "./scopes.js";
-
-// '@modelcontextprotocol/sdk' currently does not export this type
-export type PromptArgsRawShape = {
-  [k: string]:
-    | ZodType<string, ZodTypeDef, string>
-    | ZodOptional<ZodType<string, ZodTypeDef, string>>;
-};
-
-export type PromptDefinition<
-  Args extends undefined | PromptArgsRawShape = undefined,
-> = Args extends PromptArgsRawShape ? {
-    name: string;
-    description?: string;
-    scopes?: MCPScope[];
-    args: Args;
-    prompt: (
-      client: OrqCore,
-      args: objectOutputType<Args, ZodTypeAny>,
-      extra: RequestHandlerExtra,
-    ) => GetPromptResult | Promise<GetPromptResult>;
-  }
-  : {
-    name: string;
-    description?: string;
-    scopes?: MCPScope[];
-    args?: undefined;
-    prompt: (
-      client: OrqCore,
-      extra: RequestHandlerExtra,
-    ) => GetPromptResult | Promise<GetPromptResult>;
-  };
-
-// Optional function to assist with formatting prompt results
-export async function formatResult(value: string): Promise<GetPromptResult> {
-  return {
-    messages: [
-      {
-        role: "user",
-        content: {
-          type: "text",
-          text: value,
-        },
-      },
-    ],
-  };
-}
-
-export function createRegisterPrompt(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: OrqCore,
-  allowedScopes: Set<MCPScope>,
-): <A extends PromptArgsRawShape | undefined>(
-  prompt: PromptDefinition<A>,
-) => void {
-  return <A extends PromptArgsRawShape | undefined>(
-    prompt: PromptDefinition<A>,
-  ): void => {
-    const scopes = prompt.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    if (prompt.args) {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      } else {
-        server.prompt(
-          prompt.name,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      }
-    } else {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          async (ctx) => prompt.prompt(sdk, ctx),
-        );
-      } else {
-        server.prompt(prompt.name, async (ctx) => prompt.prompt(sdk, ctx));
-      }
-    }
-
-    logger.debug("Registered prompt", { name: prompt.name });
-  };
-}

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

@@ -1,172 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import {
-  McpServer,
-  ResourceMetadata,
-  ResourceTemplate,
-} from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import { Variables } from "@modelcontextprotocol/sdk/shared/uriTemplate.js";
-import { ReadResourceResult } from "@modelcontextprotocol/sdk/types.js";
-import { OrqCore } from "../core.js";
-import { ConsoleLogger } from "./console-logger.js";
-import { MCPScope } from "./scopes.js";
-import { isAsyncIterable, isBinaryData, valueToBase64 } from "./shared.js";
-
-export type ReadResourceCallback = (
-  client: OrqCore,
-  uri: URL,
-  extra: RequestHandlerExtra,
-) => ReadResourceResult | Promise<ReadResourceResult>;
-
-export type ResourceDefinition = {
-  name: string;
-  description?: string;
-  metadata?: ResourceMetadata;
-  scopes?: MCPScope[];
-  resource: string;
-  read: ReadResourceCallback;
-};
-
-export type ReadResourceTemplateCallback = (
-  client: OrqCore,
-  uri: URL,
-  vars: Variables,
-  extra: RequestHandlerExtra,
-) => ReadResourceResult | Promise<ReadResourceResult>;
-
-export type ResourceTemplateDefinition = {
-  name: string;
-  description: string;
-  metadata?: ResourceMetadata;
-  scopes?: MCPScope[];
-  resource: ResourceTemplate;
-  read: ReadResourceTemplateCallback;
-};
-
-// Optional function to assist with formatting resource results
-export async function formatResult(
-  value: unknown,
-  uri: URL,
-  init: { mimeType?: string | undefined; response?: Response | undefined },
-): Promise<ReadResourceResult> {
-  if (typeof value === "undefined") {
-    return { contents: [] };
-  }
-
-  let contents: ReadResourceResult["contents"] = [];
-
-  const mimeType = init.mimeType ?? init.response?.headers.get("content-type")
-    ?? "";
-
-  if (mimeType.search(/\bjson\b/g) !== -1) {
-    contents = [{ uri: uri.toString(), mimeType, text: JSON.stringify(value) }];
-  } else if (
-    mimeType.startsWith("text/event-stream")
-    && isAsyncIterable(value)
-  ) {
-    contents = [
-      {
-        uri: uri.toString(),
-        mimeType: "application/json",
-        text: await stringifySSEToJSON(value),
-      },
-    ];
-  } else if (
-    (mimeType.startsWith("text/") || mimeType.startsWith("application/"))
-    && typeof value === "string"
-  ) {
-    contents = [{ uri: uri.toString(), mimeType, text: value }];
-  } else if (isBinaryData(value)) {
-    const blob = await valueToBase64(value);
-    contents = blob == null ? [] : [{ uri: uri.toString(), blob, mimeType }];
-  } else {
-    throw new Error(`Unsupported content type: "${mimeType}"`);
-  }
-
-  return { contents };
-}
-
-async function stringifySSEToJSON(
-  value: AsyncIterable<unknown>,
-): Promise<string> {
-  const payloads = [];
-
-  for await (const chunk of value) {
-    payloads.push(chunk);
-  }
-
-  return JSON.stringify(payloads);
-}
-
-export function createRegisterResource(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: OrqCore,
-  allowedScopes: Set<MCPScope>,
-): (resource: ResourceDefinition) => void {
-  return (resource: ResourceDefinition): void => {
-    const scopes = resource.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    const metadata: ResourceMetadata = {
-      ...resource.metadata,
-      description: resource.description,
-    };
-
-    server.resource(
-      resource.name,
-      resource.resource,
-      metadata,
-      async (uri, ctx) => resource.read(sdk, uri, ctx),
-    );
-
-    logger.debug("Registered resource", { name: resource.name });
-  };
-}
-
-export function createRegisterResourceTemplate(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: OrqCore,
-  allowedScopes: Set<MCPScope>,
-): (resource: ResourceTemplateDefinition) => void {
-  return (resource: ResourceTemplateDefinition): void => {
-    const scopes = resource.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    const metadata: ResourceMetadata = {
-      ...resource.metadata,
-      description: resource.description,
-    };
-
-    server.resource(
-      resource.name,
-      resource.resource,
-      metadata,
-      async (uri, vars, ctx) => resource.read(sdk, uri, vars, ctx),
-    );
-
-    logger.debug("Registered resource template", { name: resource.name });
-  };
-}

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

@@ -1,7 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-export const mcpScopes = [] as const;
-
-export type MCPScope = (typeof mcpScopes)[number];

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

@@ -1,163 +0,0 @@
-/*
- * 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 } 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$datasetsCreateExperiment } from "./tools/datasetsCreateExperiment.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$deploymentsCreateExperiment } from "./tools/deploymentsCreateExperiment.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$deploymentsStream } from "./tools/deploymentsStream.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$knowledgeCreate } from "./tools/knowledgeCreate.js";
-import { tool$knowledgeCreateChunks } from "./tools/knowledgeCreateChunks.js";
-import { tool$knowledgeCreateDatasource } from "./tools/knowledgeCreateDatasource.js";
-import { tool$knowledgeDelete } from "./tools/knowledgeDelete.js";
-import { tool$knowledgeDeleteChunk } from "./tools/knowledgeDeleteChunk.js";
-import { tool$knowledgeDeleteDatasource } from "./tools/knowledgeDeleteDatasource.js";
-import { tool$knowledgeList } from "./tools/knowledgeList.js";
-import { tool$knowledgeListChunks } from "./tools/knowledgeListChunks.js";
-import { tool$knowledgeListDatasources } from "./tools/knowledgeListDatasources.js";
-import { tool$knowledgeRetrieve } from "./tools/knowledgeRetrieve.js";
-import { tool$knowledgeRetrieveChunk } from "./tools/knowledgeRetrieveChunk.js";
-import { tool$knowledgeRetrieveDatasource } from "./tools/knowledgeRetrieveDatasource.js";
-import { tool$knowledgeSearch } from "./tools/knowledgeSearch.js";
-import { tool$knowledgeUpdate } from "./tools/knowledgeUpdate.js";
-import { tool$knowledgeUpdateChunk } from "./tools/knowledgeUpdateChunk.js";
-import { tool$knowledgeUpdateDatasource } from "./tools/knowledgeUpdateDatasource.js";
-import { tool$modelsList } from "./tools/modelsList.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.7.0-rc.1",
-  });
-
-  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);
-
-  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$deploymentsStream);
-  tool(tool$deploymentsCreateExperiment);
-  tool(tool$filesCreate);
-  tool(tool$filesList);
-  tool(tool$filesGet);
-  tool(tool$filesDelete);
-  tool(tool$promptsList);
-  tool(tool$promptsCreate);
-  tool(tool$promptsRetrieve);
-  tool(tool$promptsUpdate);
-  tool(tool$promptsDelete);
-  tool(tool$promptsListVersions);
-  tool(tool$promptsGetVersion);
-  tool(tool$remoteconfigsRetrieve);
-  tool(tool$modelsList);
-  tool(tool$datasetsList);
-  tool(tool$datasetsCreate);
-  tool(tool$datasetsRetrieve);
-  tool(tool$datasetsUpdate);
-  tool(tool$datasetsDelete);
-  tool(tool$datasetsCreateExperiment);
-  tool(tool$datasetsListDatapoints);
-  tool(tool$datasetsCreateDatapoint);
-  tool(tool$datasetsRetrieveDatapoint);
-  tool(tool$datasetsUpdateDatapoint);
-  tool(tool$datasetsDeleteDatapoint);
-  tool(tool$datasetsCreateDatapoints);
-  tool(tool$datasetsClear);
-  tool(tool$knowledgeList);
-  tool(tool$knowledgeCreate);
-  tool(tool$knowledgeRetrieve);
-  tool(tool$knowledgeUpdate);
-  tool(tool$knowledgeDelete);
-  tool(tool$knowledgeSearch);
-  tool(tool$knowledgeListDatasources);
-  tool(tool$knowledgeCreateDatasource);
-  tool(tool$knowledgeRetrieveDatasource);
-  tool(tool$knowledgeDeleteDatasource);
-  tool(tool$knowledgeUpdateDatasource);
-  tool(tool$knowledgeCreateChunks);
-  tool(tool$knowledgeListChunks);
-  tool(tool$knowledgeUpdateChunk);
-  tool(tool$knowledgeDeleteChunk);
-  tool(tool$knowledgeRetrieveChunk);
-  tool(tool$deploymentsMetricsCreate);
-
-  return server;
-}

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

@@ -1,75 +0,0 @@
-/*
- * 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

@@ -1,37 +0,0 @@
-/*
- * 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

@@ -1,35 +0,0 @@
-/*
- * 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

@@ -1,37 +0,0 @@
-/*
- * 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

@@ -1,37 +0,0 @@
-/*
- * 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);
-  },
-};