@orq-ai/node 3.8.9 → 3.9.0-rc.1

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

@@ -1,35 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { knowledgeUpdateChunk } from "../../funcs/knowledgeUpdateChunk.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.UpdateChunkRequest$inboundSchema,
-};
-
-export const tool$knowledgeUpdateChunk: ToolDefinition<typeof args> = {
-  name: "knowledge-update-chunk",
-  description: `Update a chunk`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await knowledgeUpdateChunk(
-      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/knowledgeUpdateDatasource.ts

@@ -1,35 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { knowledgeUpdateDatasource } from "../../funcs/knowledgeUpdateDatasource.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.UpdateDatasourceRequest$inboundSchema,
-};
-
-export const tool$knowledgeUpdateDatasource: ToolDefinition<typeof args> = {
-  name: "knowledge-update-datasource",
-  description: `Update a datasource`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await knowledgeUpdateDatasource(
-      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/modelsList.ts

@@ -1,30 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { modelsList } from "../../funcs/modelsList.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-export const tool$modelsList: ToolDefinition = {
-  name: "models-list",
-  description: `List models
-
-Lists the currently available models, and provides basic information about each one such as the owner and availability.`,
-  tool: async (client, ctx) => {
-    const [result, apiCall] = await modelsList(
-      client,
-      { 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/promptsCreate.ts

@@ -1,35 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsCreate } from "../../funcs/promptsCreate.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.CreatePromptRequestBody$inboundSchema.optional(),
-};
-
-export const tool$promptsCreate: ToolDefinition<typeof args> = {
-  name: "prompts-create",
-  description: `Create a prompt`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsCreate(
-      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/promptsDelete.ts

@@ -1,33 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsDelete } from "../../funcs/promptsDelete.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.DeletePromptRequest$inboundSchema,
-};
-
-export const tool$promptsDelete: ToolDefinition<typeof args> = {
-  name: "prompts-delete",
-  description: `Delete a prompt`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsDelete(
-      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/promptsGetVersion.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsGetVersion } from "../../funcs/promptsGetVersion.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.GetPromptVersionRequest$inboundSchema,
-};
-
-export const tool$promptsGetVersion: ToolDefinition<typeof args> = {
-  name: "prompts-get-version",
-  description: `Retrieve a prompt version
-
-Retrieves a specific version of a prompt by its ID and version ID.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsGetVersion(
-      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/promptsList.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsList } from "../../funcs/promptsList.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.GetAllPromptsRequest$inboundSchema.optional(),
-};
-
-export const tool$promptsList: ToolDefinition<typeof args> = {
-  name: "prompts-list",
-  description: `List all prompts
-
-Returns a list of your prompts. The prompts are returned sorted by creation date, with the most recent prompts appearing first`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsList(
-      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/promptsListVersions.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsListVersions } from "../../funcs/promptsListVersions.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListPromptVersionsRequest$inboundSchema,
-};
-
-export const tool$promptsListVersions: ToolDefinition<typeof args> = {
-  name: "prompts-list-versions",
-  description: `List all prompt versions
-
-Returns a list of your prompt versions. The prompt versions are returned sorted by creation date, with the most recent prompt versions appearing first`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsListVersions(
-      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/promptsRetrieve.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsRetrieve } from "../../funcs/promptsRetrieve.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.GetOnePromptRequest$inboundSchema,
-};
-
-export const tool$promptsRetrieve: ToolDefinition<typeof args> = {
-  name: "prompts-retrieve",
-  description: `Retrieve a prompt
-
-Retrieves a prompt object`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsRetrieve(
-      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/promptsUpdate.ts

@@ -1,35 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { promptsUpdate } from "../../funcs/promptsUpdate.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.UpdatePromptRequest$inboundSchema,
-};
-
-export const tool$promptsUpdate: ToolDefinition<typeof args> = {
-  name: "prompts-update",
-  description: `Update a prompt`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await promptsUpdate(
-      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/remoteconfigsRetrieve.ts

@@ -1,36 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { remoteconfigsRetrieve } from "../../funcs/remoteconfigsRetrieve.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.RemoteConfigsGetConfigRequestBody$inboundSchema
-    .optional(),
-};
-
-export const tool$remoteconfigsRetrieve: ToolDefinition<typeof args> = {
-  name: "remoteconfigs-retrieve",
-  description: `Retrieve a remote config`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await remoteconfigsRetrieve(
-      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.ts

@@ -1,129 +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 { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { objectOutputType, ZodRawShape, ZodTypeAny } from "zod";
-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 ToolDefinition<Args extends undefined | ZodRawShape = undefined> =
-  Args extends ZodRawShape ? {
-      name: string;
-      description: string;
-      scopes?: MCPScope[];
-      args: Args;
-      tool: (
-        client: OrqCore,
-        args: objectOutputType<Args, ZodTypeAny>,
-        extra: RequestHandlerExtra,
-      ) => CallToolResult | Promise<CallToolResult>;
-    }
-    : {
-      name: string;
-      description: string;
-      scopes?: MCPScope[];
-      args?: undefined;
-      tool: (
-        client: OrqCore,
-        extra: RequestHandlerExtra,
-      ) => CallToolResult | Promise<CallToolResult>;
-    };
-
-// Optional function to assist with formatting tool results
-export async function formatResult(
-  value: unknown,
-  init: { response?: Response | undefined },
-): Promise<CallToolResult> {
-  if (typeof value === "undefined") {
-    return { content: [] };
-  }
-
-  const { response } = init;
-  const contentType = response?.headers.get("content-type") ?? "";
-  let content: CallToolResult["content"] = [];
-
-  if (contentType.search(/\bjson\b/g)) {
-    content = [{ type: "text", text: JSON.stringify(value) }];
-  } else if (
-    contentType.startsWith("text/event-stream")
-    && isAsyncIterable(value)
-  ) {
-    content = await consumeSSE(value);
-  } else if (contentType.startsWith("text/") && typeof value === "string") {
-    content = [{ type: "text", text: value }];
-  } else if (isBinaryData(value) && contentType.startsWith("image/")) {
-    const data = await valueToBase64(value);
-    content = data == null
-      ? []
-      : [{ type: "image", data, mimeType: contentType }];
-  } else {
-    return {
-      content: [{
-        type: "text",
-        text: `Unsupported content type: "${contentType}"`,
-      }],
-      isError: true,
-    };
-  }
-
-  return { content };
-}
-
-async function consumeSSE(
-  value: AsyncIterable<unknown>,
-): Promise<CallToolResult["content"]> {
-  const content: CallToolResult["content"] = [];
-
-  for await (const chunk of value) {
-    if (typeof chunk === "string") {
-      content.push({ type: "text", text: chunk });
-    } else {
-      content.push({ type: "text", text: JSON.stringify(chunk) });
-    }
-  }
-
-  return content;
-}
-
-export function createRegisterTool(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: OrqCore,
-  allowedScopes: Set<MCPScope>,
-  allowedTools?: Set<string>,
-): <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>) => void {
-  return <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>): void => {
-    if (allowedTools && !allowedTools.has(tool.name)) {
-      return;
-    }
-
-    const scopes = tool.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    if (tool.args) {
-      server.tool(tool.name, tool.description, tool.args, async (args, ctx) => {
-        return tool.tool(sdk, args, ctx);
-      });
-    } else {
-      server.tool(tool.name, tool.description, async (ctx) => {
-        return tool.tool(sdk, ctx);
-      });
-    }
-
-    logger.debug("Registered tool", { name: tool.name });
-  };
-}