@metorial/sdk 1.2.0 → 2.0.1

package/dist/index.js

@@ -1,439 +1,189 @@
-import { createMetorialCoreSDK } from '@metorial/core';
-import { MetorialMcpSession } from '@metorial/mcp-session';
-import { metorialOpenAI } from '@metorial/openai';
-import { metorialAnthropic } from '@metorial/anthropic';
-import { metorialDeepseek } from '@metorial/deepseek';
-import { metorialGoogle } from '@metorial/google';
-import { metorialMistral } from '@metorial/mistral';
-import { metorialXai } from '@metorial/xai';
-import { metorialTogetherAi } from '@metorial/togetherai';
+import { createMetorialPulsarCoreSDK, createMetorialMagnetarCoreSDK } from '@metorial/core';
+import { MetorialMcpSession, MetorialMagnetarMcpSession } from '@metorial/mcp-session';
 
 // src/metorial.ts
-var runWithOpenAI = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...openaiOptions
-  } = config;
-  return withProviderSession(
-    metorialOpenAI.chatCompletions,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => requestedTools.includes(t.name)) : session.tools;
-      let messages = [
-        { role: "user", content: message }
-      ];
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.chat.completions.create({
-          model,
-          messages,
-          tools,
-          ...openaiOptions
-        });
-        let choice = response.choices[0];
-        if (!choice.message.tool_calls) {
-          return {
-            text: choice.message.content || "",
-            steps: step + 1
-          };
-        }
-        let toolResponses = await session.callTools(choice.message.tool_calls);
-        messages.push(
-          { role: "assistant", tool_calls: choice.message.tool_calls },
-          ...toolResponses
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-var runWithAnthropic = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    max_tokens = 4096,
-    ...anthropicOptions
-  } = config;
-  return withProviderSession(
-    metorialAnthropic,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => requestedTools.includes(t.name)) : session.tools;
-      let messages = [
-        { role: "user", content: message }
-      ];
-      let uniqueTools = Array.from(new Map(tools.map((t) => [t.name, t])).values());
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.messages.create({
-          model,
-          max_tokens,
-          messages,
-          tools: uniqueTools,
-          ...anthropicOptions
-        });
-        if (response.stop_reason !== "tool_use") {
-          let textContent = response.content.find(
-            (block) => block.type === "text"
-          );
-          return {
-            text: (textContent == null ? void 0 : textContent.text) || "",
-            steps: step + 1
-          };
-        }
-        let toolUseBlocks = response.content.filter(
-          (block) => block.type === "tool_use"
-        );
-        let toolResponse = await session.callTools(toolUseBlocks);
-        let toolResponseMessage = {
-          role: "user",
-          content: toolResponse.content.map((block) => ({
-            type: "tool_result",
-            tool_use_id: block.tool_use_id,
-            content: block.content
-          }))
-        };
-        messages.push(
-          { role: "assistant", content: response.content },
-          toolResponseMessage
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-var runWithDeepSeek = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...deepseekOptions
-  } = config;
-  return withProviderSession(
-    metorialDeepseek,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => {
-        var _a;
-        return requestedTools.includes(((_a = t.function) == null ? void 0 : _a.name) || t.name);
-      }) : session.tools;
-      let uniqueTools = Array.from(
-        new Map(tools.map((t) => {
-          var _a;
-          return [((_a = t.function) == null ? void 0 : _a.name) || t.name, t];
-        })).values()
-      );
-      let messages = [
-        { role: "user", content: message }
-      ];
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.chat.completions.create({
-          model,
-          messages,
-          tools: uniqueTools,
-          ...deepseekOptions
-        });
-        let choice = response.choices[0];
-        if (!choice.message.tool_calls || choice.message.tool_calls.length === 0) {
-          return {
-            text: choice.message.content || "",
-            steps: step + 1
-          };
-        }
-        let toolResponses = await session.callTools(choice.message.tool_calls);
-        messages.push(
-          { role: "assistant", tool_calls: choice.message.tool_calls },
-          ...toolResponses
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-var runWithGoogle = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...googleOptions
-  } = config;
-  return withProviderSession(
-    metorialGoogle,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
-      let tools = requestedTools ? session.tools.filter((t) => requestedTools.includes(t.name)) : session.tools;
-      let response = await client.models.generateContent({
-        model,
-        contents: [
-          {
-            role: "user",
-            parts: [{ text: message }]
-          }
-        ],
-        config: {
-          tools,
-          ...googleOptions
-        }
-      });
-      let text = (_e = (_d = (_c = (_b = (_a = response.candidates) == null ? void 0 : _a[0]) == null ? void 0 : _b.content) == null ? void 0 : _c.parts) == null ? void 0 : _d[0]) == null ? void 0 : _e.text;
-      let functionCalls = (_i = (_h = (_g = (_f = response.candidates) == null ? void 0 : _f[0]) == null ? void 0 : _g.content) == null ? void 0 : _h.parts) == null ? void 0 : _i.filter((part) => part.functionCall).map((part) => part.functionCall);
-      if (functionCalls && functionCalls.length > 0) {
-        let toolResponses = await session.callTools(functionCalls);
-        return {
-          text: text || "",
-          toolResponses,
-          steps: 1
-        };
-      }
-      return {
-        text: text || "",
-        steps: 1
-      };
+var Metorial = class {
+  constructor(init) {
+    let apiHost = init.apiHost;
+    let mcpHost = init.mcpHost;
+    if (mcpHost && !apiHost) {
+      apiHost = mcpHost.replace("://connect.", "://api.");
     }
-  );
-};
-var extractTextFromContent = (content) => {
-  if (typeof content === "string") {
-    return content;
+    this.pulsarSdk = createMetorialPulsarCoreSDK({ ...init, apiHost });
+    this.magnetarSdk = createMetorialMagnetarCoreSDK({ ...init, apiHost, mcpHost });
   }
-  if (Array.isArray(content)) {
-    return content.filter((chunk) => chunk && typeof chunk === "object").map((chunk) => chunk.text || chunk.content || JSON.stringify(chunk)).join("");
+  // ── Magnetar (default) ───────────────────────────────────────────
+  get providers() {
+    return this.magnetarSdk.providers;
   }
-  return "";
-};
-var runWithMistral = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...mistralOptions
-  } = config;
-  return withProviderSession(
-    metorialMistral,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => {
-        var _a;
-        return requestedTools.includes(((_a = t.function) == null ? void 0 : _a.name) || t.name);
-      }) : session.tools;
-      let fixedTools = tools.map((tool) => {
-        var _a;
-        if ((_a = tool.function) == null ? void 0 : _a.parameters) {
-          let fixedParams = { ...tool.function.parameters };
-          fixedParams.additionalProperties = false;
-          if (fixedParams.properties) {
-            Object.values(fixedParams.properties).forEach((prop) => {
-              if (prop && typeof prop === "object" && prop.type === "object") {
-                prop.additionalProperties = false;
-              }
-            });
-          }
-          return {
-            ...tool,
-            function: {
-              ...tool.function,
-              parameters: fixedParams
-            }
-          };
-        }
-        return tool;
-      });
-      let messages = [
-        { role: "user", content: message }
-      ];
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.chat.complete({
-          model,
-          messages,
-          tools: fixedTools,
-          ...mistralOptions
-        });
-        let choice = response.choices[0];
-        let toolCalls = choice.message.toolCalls;
-        if (!toolCalls || toolCalls.length === 0) {
-          return {
-            text: extractTextFromContent(choice.message.content),
-            steps: step + 1
-          };
-        }
-        let toolResponses = await session.callTools(toolCalls);
-        messages.push(
-          { role: "assistant", toolCalls },
-          ...toolResponses
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-var runWithXAI = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...xaiOptions
-  } = config;
-  return withProviderSession(
-    metorialXai,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => {
-        var _a;
-        return requestedTools.includes(((_a = t.function) == null ? void 0 : _a.name) || t.name);
-      }) : session.tools;
-      let uniqueTools = Array.from(
-        new Map(tools.map((t) => [t.function.name, t])).values()
-      );
-      let messages = [
-        { role: "user", content: message }
-      ];
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.chat.completions.create({
-          model,
-          messages,
-          tools: uniqueTools,
-          ...xaiOptions
-        });
-        let choice = response.choices[0];
-        if (!choice.message.tool_calls || choice.message.tool_calls.length === 0) {
-          return {
-            text: choice.message.content || "",
-            steps: step + 1
-          };
-        }
-        let toolResponses = await session.callTools(choice.message.tool_calls);
-        messages.push(
-          { role: "assistant", tool_calls: choice.message.tool_calls },
-          ...toolResponses
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-var runWithTogetherAI = async (config) => {
-  let {
-    message,
-    serverDeployments,
-    client,
-    model,
-    maxSteps = 25,
-    tools: requestedTools,
-    withProviderSession,
-    ...togetheraiOptions
-  } = config;
-  return withProviderSession(
-    metorialTogetherAi,
-    {
-      serverDeployments: Array.isArray(serverDeployments) ? serverDeployments : [serverDeployments]
-    },
-    async (session) => {
-      let tools = requestedTools ? session.tools.filter((t) => {
-        var _a;
-        return requestedTools.includes(((_a = t.function) == null ? void 0 : _a.name) || t.name);
-      }) : session.tools;
-      let messages = [
-        { role: "user", content: message }
-      ];
-      for (let step = 0; step < maxSteps; step++) {
-        let response = await client.chat.completions.create({
-          model,
-          messages,
-          tools,
-          ...togetheraiOptions
-        });
-        let choice = response.choices[0];
-        if (!choice.message.tool_calls || choice.message.tool_calls.length === 0) {
-          return {
-            text: choice.message.content || "",
-            steps: step + 1
-          };
-        }
-        let toolResponses = await session.callTools(choice.message.tool_calls);
-        messages.push(
-          { role: "assistant", tool_calls: choice.message.tool_calls },
-          ...toolResponses
-        );
-      }
-      throw new Error(`Max steps (${maxSteps}) reached without final response`);
-    }
-  );
-};
-
-// src/metorial.ts
-var Metorial = class {
-  constructor(init) {
-    this.sdk = createMetorialCoreSDK(init);
+  get providerDeployments() {
+    let deployments = this.magnetarSdk.providerDeployments;
+    Object.assign(deployments.setupSessions, {
+      waitForCompletion: this.waitForSetupSession.bind(this)
+    });
+    return deployments;
+  }
+  get sessions() {
+    return this.magnetarSdk.sessions;
+  }
+  get sessionTemplates() {
+    return this.magnetarSdk.sessionTemplates;
+  }
+  get providerRuns() {
+    return this.magnetarSdk.providerRuns;
   }
   get instance() {
-    return this.sdk.instance;
+    return this.magnetarSdk.instance;
   }
-  get secrets() {
-    return this.sdk.secrets;
+  get publishers() {
+    return this.magnetarSdk.publishers;
+  }
+  get providerCategories() {
+    return this.magnetarSdk.providerCategories;
+  }
+  get providerCollections() {
+    return this.magnetarSdk.providerCollections;
+  }
+  get providerGroups() {
+    return this.magnetarSdk.providerGroups;
   }
+  get providerListings() {
+    return this.magnetarSdk.providerListings;
+  }
+  get providerSetupSessions() {
+    return this.magnetarSdk.providerSetupSessions;
+  }
+  get toolCalls() {
+    return this.magnetarSdk.toolCalls;
+  }
+  get customProviders() {
+    return this.magnetarSdk.customProviders;
+  }
+  // ── Backward compat (Pulsar endpoints at top level) ──────────────
   get servers() {
-    return this.sdk.servers;
+    return this.pulsarSdk.servers;
   }
-  get sessions() {
-    return this.sdk.sessions;
+  get secrets() {
+    return this.pulsarSdk.secrets;
   }
   get oauth() {
     return {
-      ...this.sdk.oauth,
+      ...this.pulsarSdk.oauth,
       waitForCompletion: this.waitForOAuthCompletion.bind(this)
     };
   }
   get _config() {
-    return this.sdk._config;
+    return this.pulsarSdk._config;
+  }
+  // ── v1 namespace (legacy Pulsar) ─────────────────────────────────
+  get v1() {
+    let pulsarSdk = this.pulsarSdk;
+    let self = this;
+    return {
+      get instance() {
+        return pulsarSdk.instance;
+      },
+      get secrets() {
+        return pulsarSdk.secrets;
+      },
+      get servers() {
+        return pulsarSdk.servers;
+      },
+      get sessions() {
+        return pulsarSdk.sessions;
+      },
+      get oauth() {
+        return self.oauth;
+      },
+      get _config() {
+        return pulsarSdk._config;
+      },
+      mcp: {
+        createSession: (init) => new MetorialMcpSession(pulsarSdk, init),
+        withSession: self._pulsarWithSession.bind(self),
+        withProviderSession: self._pulsarWithProviderSession.bind(self),
+        createConnection: self._pulsarCreateMcpConnection.bind(self)
+      },
+      withSession: self._pulsarWithSession.bind(self),
+      withProviderSession: self._pulsarWithProviderSession.bind(self),
+      createMcpConnection: self._pulsarCreateMcpConnection.bind(self)
+    };
   }
+  // ── MCP (Magnetar default) ───────────────────────────────────────
   get mcp() {
     return {
-      createSession: (init) => new MetorialMcpSession(this.sdk, init),
+      createSession: (init) => new MetorialMagnetarMcpSession(this.magnetarSdk, init),
       withSession: this.withSession.bind(this),
       withProviderSession: this.withProviderSession.bind(this),
       createConnection: this.createMcpConnection.bind(this)
     };
   }
+  session(init) {
+    return new MetorialMagnetarMcpSession(this.magnetarSdk, init);
+  }
   async createMcpConnection(init) {
-    let session = new MetorialMcpSession(this.sdk, {
+    let session = new MetorialMagnetarMcpSession(this.magnetarSdk, init);
+    return await session.getClient();
+  }
+  async withSession(init, action) {
+    let session = new MetorialMagnetarMcpSession(this.magnetarSdk, init);
+    try {
+      return await action(session);
+    } finally {
+      await session.close();
+    }
+  }
+  async withProviderSession(provider, init, action) {
+    if (init.streaming) {
+      return this.withMagnetarStreamingSession(provider, init, action);
+    }
+    return this.withSession(init, async (session) => {
+      let providerData = await provider(session);
+      return action({
+        ...providerData,
+        session,
+        getSession: session.getSession.bind(session),
+        getCapabilities: session.getCapabilities.bind(session),
+        getToolManager: session.getToolManager.bind(session),
+        closeSession: async () => {
+        }
+      });
+    });
+  }
+  async withMagnetarStreamingSession(provider, init, action) {
+    let session = new MetorialMagnetarMcpSession(this.magnetarSdk, init);
+    let sessionClosed = false;
+    const closeSession = async () => {
+      if (!sessionClosed) {
+        sessionClosed = true;
+        await session.close();
+      }
+    };
+    try {
+      let providerData = await provider(session);
+      let result = await action({
+        ...providerData,
+        session,
+        getSession: session.getSession.bind(session),
+        getCapabilities: session.getCapabilities.bind(session),
+        getToolManager: session.getToolManager.bind(session),
+        closeSession
+      });
+      let timeout = setTimeout(async () => {
+        if (!sessionClosed) {
+          await closeSession();
+        }
+      }, 6e4);
+      timeout.unref();
+      return result;
+    } catch (error) {
+      await closeSession();
+      throw error;
+    }
+  }
+  // ── Pulsar MCP helpers (used by v1) ──────────────────────────────
+  async _pulsarCreateMcpConnection(init) {
+    let session = new MetorialMcpSession(this.pulsarSdk, {
       serverDeployments: [init]
     });
     let deployments = await session.getServerDeployments();
@@ -441,19 +191,19 @@ var Metorial = class {
       deploymentId: deployments[0].id
     });
   }
-  async withSession(init, action) {
-    let session = new MetorialMcpSession(this.sdk, init);
+  async _pulsarWithSession(init, action) {
+    let session = new MetorialMcpSession(this.pulsarSdk, init);
     try {
       return await action(session);
     } finally {
       await session.close();
     }
   }
-  async withProviderSession(provider, init, action) {
+  async _pulsarWithProviderSession(provider, init, action) {
     if (init.streaming) {
       return this.withStreamingSession(provider, init, action);
     }
-    return this.withSession(init, async (session) => {
+    return this._pulsarWithSession(init, async (session) => {
       let providerData = await provider(session);
       return action({
         ...providerData,
@@ -467,12 +217,11 @@ var Metorial = class {
     });
   }
   async withStreamingSession(provider, init, action) {
-    let session = new MetorialMcpSession(this.sdk, init);
+    let session = new MetorialMcpSession(this.pulsarSdk, init);
     let sessionClosed = false;
     const closeSession = async () => {
       if (!sessionClosed) {
         sessionClosed = true;
-        console.log("[Metorial] Closing streaming session");
         await session.close();
       }
     };
@@ -488,42 +237,51 @@ var Metorial = class {
         getToolManager: session.getToolManager.bind(session),
         closeSession
       });
-      setTimeout(async () => {
+      let timeout = setTimeout(async () => {
         if (!sessionClosed) {
-          console.log("[Metorial] Streaming timeout reached - auto-closing session");
           await closeSession();
         }
       }, 6e4);
+      timeout.unref();
       return result;
     } catch (error) {
       await closeSession();
       throw error;
     }
   }
-  inferProvider(model) {
-    let modelLower = model.toLowerCase();
-    if (modelLower.startsWith("claude-")) {
-      return "anthropic";
-    }
-    if (modelLower.startsWith("gpt-") || modelLower.startsWith("o1-")) {
-      return "openai";
-    }
-    if (modelLower.includes("deepseek")) {
-      return "deepseek";
-    }
-    if (modelLower.startsWith("gemini-") || modelLower.includes("google")) {
-      return "google";
-    }
-    if (modelLower.startsWith("mistral-") || modelLower.includes("mistral")) {
-      return "mistral";
-    }
-    if (modelLower.startsWith("x-") || modelLower === "grok-beta") {
-      return "xai";
+  async waitForSetupSession(sessions, options) {
+    var _a, _b;
+    let sessionList = Array.isArray(sessions) ? sessions : [sessions];
+    let pollInterval = Math.max((_a = options == null ? void 0 : options.pollInterval) != null ? _a : 5e3, 2e3);
+    let timeout = (_b = options == null ? void 0 : options.timeout) != null ? _b : 6e5;
+    let startTime = Date.now();
+    if (sessionList.length === 0) {
+      return [];
     }
-    if (modelLower.includes("together") || modelLower.includes("llama") || modelLower.includes("mixtral") || modelLower.includes("qwen") || modelLower.includes("/")) {
-      return "togetherai";
+    while (true) {
+      if (Date.now() - startTime > timeout) {
+        throw new Error(`Setup session timed out after ${timeout / 1e3} seconds`);
+      }
+      try {
+        let statuses = await Promise.all(
+          sessionList.map((s) => this.magnetarSdk.providerDeployments.setupSessions.get(s.id))
+        );
+        let failed = statuses.filter((s) => s.status === "failed");
+        if (failed.length > 0) {
+          throw new Error(`${failed.length} setup session(s) failed`);
+        }
+        let allCompleted = statuses.every((s) => s.status === "completed");
+        if (allCompleted) {
+          return statuses;
+        }
+        await new Promise((resolve) => setTimeout(resolve, pollInterval));
+      } catch (error) {
+        if (error instanceof Error && (error.message.includes("setup session") && (error.message.includes("failed") || error.message.includes("timed out")))) {
+          throw error;
+        }
+        await new Promise((resolve) => setTimeout(resolve, pollInterval));
+      }
     }
-    throw new Error(`Unable to infer provider from model "${model}".`);
   }
   async waitForOAuthCompletion(sessions, options) {
     var _a, _b;
@@ -560,55 +318,6 @@ var Metorial = class {
       }
     }
   }
-  async run(config) {
-    let provider = this.inferProvider(config.model);
-    switch (provider) {
-      case "openai":
-        return runWithOpenAI({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "anthropic":
-        return runWithAnthropic({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "deepseek":
-        return runWithDeepSeek({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "google":
-        return runWithGoogle({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "mistral":
-        return runWithMistral({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "xai":
-        return runWithXAI({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      case "togetherai":
-        return runWithTogetherAI({
-          ...config,
-          client: config.client,
-          withProviderSession: this.withProviderSession.bind(this)
-        });
-      default:
-        throw new Error(`Unsupported provider: ${provider}`);
-    }
-  }
 };
 
 export { Metorial };