@orq-ai/node 3.4.0-rc.37 → 3.5.11

package/{src/mcp-server/tools/toolsDelete.ts → packages/orq-rc/src/mcp-server/tools/promptsDelete.ts}

@@ -2,22 +2,20 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { toolsDelete } from "../../funcs/toolsDelete.js";
+import { promptsDelete } from "../../funcs/promptsDelete.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.DeleteToolRequest$inboundSchema,
+  request: operations.DeletePromptRequest$inboundSchema,
 };
 
-export const tool$toolsDelete: ToolDefinition<typeof args> = {
-  name: "tools-delete",
-  description: `Delete tool
-
-Deletes a tool by key.`,
+export const tool$promptsDelete: ToolDefinition<typeof args> = {
+  name: "prompts-delete",
+  description: `Delete a prompt`,
   args,
   tool: async (client, args, ctx) => {
-    const [result, apiCall] = await toolsDelete(
+    const [result, apiCall] = await promptsDelete(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

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

@@ -0,0 +1,37 @@
+/*
+ * 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/{src/mcp-server/tools/memoryStoresRetrieveMemory.ts → packages/orq-rc/src/mcp-server/tools/promptsList.ts}

@@ -2,22 +2,22 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { memoryStoresRetrieveMemory } from "../../funcs/memoryStoresRetrieveMemory.js";
+import { promptsList } from "../../funcs/promptsList.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.RetrieveMemoryRequest$inboundSchema,
+  request: operations.GetAllPromptsRequest$inboundSchema.optional(),
 };
 
-export const tool$memoryStoresRetrieveMemory: ToolDefinition<typeof args> = {
-  name: "memory-stores-retrieve-memory",
-  description: `Retrieve a specific memory
+export const tool$promptsList: ToolDefinition<typeof args> = {
+  name: "prompts-list",
+  description: `List all prompts
 
-Retrieves details of a specific memory by its ID`,
+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 memoryStoresRetrieveMemory(
+    const [result, apiCall] = await promptsList(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

package/{src/mcp-server/tools/memoryStoresListDocuments.ts → packages/orq-rc/src/mcp-server/tools/promptsListVersions.ts}

@@ -2,22 +2,22 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { memoryStoresListDocuments } from "../../funcs/memoryStoresListDocuments.js";
+import { promptsListVersions } from "../../funcs/promptsListVersions.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.GetAllMemoryDocumentsRequest$inboundSchema,
+  request: operations.ListPromptVersionsRequest$inboundSchema,
 };
 
-export const tool$memoryStoresListDocuments: ToolDefinition<typeof args> = {
-  name: "memory-stores-list-documents",
-  description: `List all documents for a memory
+export const tool$promptsListVersions: ToolDefinition<typeof args> = {
+  name: "prompts-list-versions",
+  description: `List all prompt versions
 
-Retrieves a paginated list of documents associated with a specific memory.`,
+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 memoryStoresListDocuments(
+    const [result, apiCall] = await promptsListVersions(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

package/{src/mcp-server/tools/toolsCreate.ts → packages/orq-rc/src/mcp-server/tools/promptsRetrieve.ts}

@@ -2,22 +2,22 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { toolsCreate } from "../../funcs/toolsCreate.js";
+import { promptsRetrieve } from "../../funcs/promptsRetrieve.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.CreateToolRequestBody$inboundSchema.optional(),
+  request: operations.GetOnePromptRequest$inboundSchema,
 };
 
-export const tool$toolsCreate: ToolDefinition<typeof args> = {
-  name: "tools-create",
-  description: `Create tool
+export const tool$promptsRetrieve: ToolDefinition<typeof args> = {
+  name: "prompts-retrieve",
+  description: `Retrieve a prompt
 
-Creates a new tool in the workspace.`,
+Retrieves a prompt object`,
   args,
   tool: async (client, args, ctx) => {
-    const [result, apiCall] = await toolsCreate(
+    const [result, apiCall] = await promptsRetrieve(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

package/{src/mcp-server/tools/sessionsGet.ts → packages/orq-rc/src/mcp-server/tools/promptsUpdate.ts}

@@ -2,22 +2,20 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { sessionsGet } from "../../funcs/sessionsGet.js";
+import { promptsUpdate } from "../../funcs/promptsUpdate.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.GetSessionRequest$inboundSchema,
+  request: operations.UpdatePromptRequest$inboundSchema,
 };
 
-export const tool$sessionsGet: ToolDefinition<typeof args> = {
-  name: "sessions-get",
-  description: `Get Trace Session
-
-Get a trace session`,
+export const tool$promptsUpdate: ToolDefinition<typeof args> = {
+  name: "prompts-update",
+  description: `Update a prompt`,
   args,
   tool: async (client, args, ctx) => {
-    const [result, apiCall] = await sessionsGet(
+    const [result, apiCall] = await promptsUpdate(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

package/{src/mcp-server/tools/memoryStoresUpdate.ts → packages/orq-rc/src/mcp-server/tools/remoteconfigsRetrieve.ts}

@@ -2,22 +2,21 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { memoryStoresUpdate } from "../../funcs/memoryStoresUpdate.js";
+import { remoteconfigsRetrieve } from "../../funcs/remoteconfigsRetrieve.js";
 import * as operations from "../../models/operations/index.js";
 import { formatResult, ToolDefinition } from "../tools.js";
 
 const args = {
-  request: operations.UpdateMemoryStoreRequest$inboundSchema,
+  request: operations.RemoteConfigsGetConfigRequestBody$inboundSchema
+    .optional(),
 };
 
-export const tool$memoryStoresUpdate: ToolDefinition<typeof args> = {
-  name: "memory-stores-update",
-  description: `Update memory store
-
-Update the memory store configuration`,
+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 memoryStoresUpdate(
+    const [result, apiCall] = await remoteconfigsRetrieve(
       client,
       args.request,
       { fetchOptions: { signal: ctx.signal } },

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

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