@openkeyai/sdk 0.1.0 → 0.2.0

package/dist/index.js

@@ -117,6 +117,41 @@ var SecureKeyConsumedError = class extends HubSdkError {
     this.name = "SecureKeyConsumedError";
   }
 };
+var ProviderNotConfiguredError = class extends HubSdkError {
+  constructor(provider) {
+    super(
+      "provider_not_configured",
+      `Provider "${provider}" is not registered with the hub proxy. Available providers are listed at /api/proxy (coming) or in the hub's src/lib/proxy/providers.ts.`,
+      404
+    );
+    this.name = "ProviderNotConfiguredError";
+  }
+};
+var ProviderError = class extends HubSdkError {
+  /** The provider slug we tried to call. */
+  provider;
+  /** The path on the provider that returned non-2xx. */
+  path;
+  /**
+   * The upstream response body. Parsed as JSON when the upstream's
+   * content-type was JSON; raw text otherwise.
+   */
+  body;
+  /** The upstream response status code (relayed unchanged). */
+  upstreamStatus;
+  constructor(provider, path, upstreamStatus, body) {
+    super(
+      "provider_error",
+      `${provider} ${path} returned ${upstreamStatus}.`,
+      upstreamStatus
+    );
+    this.name = "ProviderError";
+    this.provider = provider;
+    this.path = path;
+    this.upstreamStatus = upstreamStatus;
+    this.body = body;
+  }
+};
 function errorFromResponse(status, body, context) {
   const code = body?.error ?? "internal";
   switch (code) {
@@ -134,6 +169,8 @@ function errorFromResponse(status, body, context) {
       return new NotSubscribedError();
     case "provider_not_granted":
       return new ProviderNotGrantedError(context.provider ?? "unknown");
+    case "provider_not_configured":
+      return new ProviderNotConfiguredError(context.provider ?? "unknown");
     case "rate_limited":
       return new RateLimitedError(context.retryAfter ?? 60);
     case "key_not_found":
@@ -143,6 +180,25 @@ function errorFromResponse(status, body, context) {
       return new InternalError(status);
   }
 }
+function isHubErrorBody(body) {
+  if (typeof body !== "object" || body === null) return false;
+  const errorField = body.error;
+  if (typeof errorField !== "string") return false;
+  const knownCodes = [
+    "missing_token",
+    "bad_token",
+    "missing_scope",
+    "subscription_inactive",
+    "tool_not_found",
+    "not_subscribed",
+    "provider_not_granted",
+    "rate_limited",
+    "key_not_found",
+    "internal",
+    "provider_not_configured"
+  ];
+  return knownCodes.includes(errorField);
+}
 var resolvers = /* @__PURE__ */ new Map();
 function getJwksResolver(hubUrl) {
   const cached = resolvers.get(hubUrl);
@@ -197,6 +253,7 @@ async function verify(jwt, opts = {}) {
   const jti = payload.jti;
   const iat = payload.iat;
   const exp = payload.exp;
+  const rawMode = payload.mode;
   if (typeof sub !== "string" || sub.length === 0) {
     throw new BadTokenError("sub claim missing or invalid.");
   }
@@ -215,12 +272,19 @@ async function verify(jwt, opts = {}) {
   if (typeof iat !== "number" || typeof exp !== "number") {
     throw new BadTokenError("iat / exp claims missing or invalid.");
   }
+  const mode = typeof rawMode === "string" && (rawMode === "production" || rawMode === "owner_test") ? rawMode : rawMode === void 0 ? "production" : null;
+  if (mode === null) {
+    throw new BadTokenError(
+      `mode claim must be 'production' or 'owner_test' (got ${JSON.stringify(rawMode)}).`
+    );
+  }
   return {
     iss: "https://openkeyai.com",
     sub,
     aud,
     scopes,
     subscription_active: subscriptionActive,
+    mode,
     iat,
     exp,
     jti
@@ -355,6 +419,298 @@ async function get(jwt, provider, opts = {}) {
   });
 }
 
+// src/proxy.ts
+var PROXY_PATH = "/api/proxy";
+function ensureBearer(token) {
+  if (!token || typeof token !== "string") {
+    throw new BadTokenError("No JWT provided.");
+  }
+}
+function buildUrl(opts) {
+  if (!opts.provider) {
+    throw new BadTokenError("ProxyCallOptions.provider is required.");
+  }
+  if (!opts.path || !opts.path.startsWith("/")) {
+    throw new BadTokenError(
+      "ProxyCallOptions.path must start with '/' (e.g. '/v1/chat/completions')."
+    );
+  }
+  const hubUrl = opts.hubUrl ?? "https://openkeyai.com";
+  return new URL(
+    `${PROXY_PATH}/${encodeURIComponent(opts.provider)}${opts.path}`,
+    hubUrl
+  );
+}
+function buildRequestInit(token, opts) {
+  const headers = new Headers();
+  if (opts.headers) {
+    for (const [k, v] of Object.entries(opts.headers)) {
+      headers.set(k, v);
+    }
+  }
+  headers.set("authorization", `Bearer ${token}`);
+  let body;
+  if (opts.method === "GET" || opts.method === "DELETE" || opts.body == null) {
+    body = void 0;
+  } else if (typeof opts.body === "string" || opts.body instanceof Blob || opts.body instanceof FormData || opts.body instanceof ArrayBuffer || opts.body instanceof ReadableStream) {
+    body = opts.body;
+  } else {
+    body = JSON.stringify(opts.body);
+    if (!headers.has("content-type")) {
+      headers.set("content-type", "application/json");
+    }
+  }
+  if (!headers.has("accept")) {
+    headers.set("accept", "application/json");
+  }
+  return {
+    method: opts.method,
+    headers,
+    body,
+    signal: opts.signal,
+    // @ts-expect-error duplex is part of fetch's RequestInit on Workers + recent Node
+    duplex: body instanceof ReadableStream ? "half" : void 0
+  };
+}
+async function rawFetch(token, opts) {
+  ensureBearer(token);
+  const url = buildUrl(opts);
+  try {
+    return await fetch(url.toString(), buildRequestInit(token, opts));
+  } catch (err) {
+    if (err instanceof DOMException && err.name === "AbortError") {
+      throw err;
+    }
+    throw new NetworkError(err);
+  }
+}
+async function parseBodyByContentType(response) {
+  const ct = response.headers.get("content-type") ?? "";
+  if (ct.includes("application/json")) {
+    try {
+      return await response.json();
+    } catch {
+      return null;
+    }
+  }
+  try {
+    return await response.text();
+  } catch {
+    return null;
+  }
+}
+async function handleResponse(response, opts, expectJson) {
+  if (response.ok) {
+    if (!expectJson) return response;
+    return await response.json();
+  }
+  const body = await parseBodyByContentType(response);
+  if (isHubErrorBody(body)) {
+    const retryAfter = parseInt(response.headers.get("retry-after") ?? "60", 10);
+    throw errorFromResponse(
+      response.status,
+      { error: body.error },
+      {
+        provider: opts.provider,
+        scopeNeeded: "keys.read",
+        retryAfter: Number.isFinite(retryAfter) ? retryAfter : 60
+      }
+    );
+  }
+  throw new ProviderError(opts.provider, opts.path, response.status, body);
+}
+async function call(token, opts) {
+  const response = await rawFetch(token, opts);
+  return handleResponse(response, opts, true);
+}
+async function callRaw(token, opts) {
+  const response = await rawFetch(token, opts);
+  return handleResponse(response, opts, false);
+}
+async function callStream(token, opts) {
+  const response = await rawFetch(token, opts);
+  if (!response.ok) {
+    const body = await parseBodyByContentType(response);
+    if (isHubErrorBody(body)) {
+      const retryAfter = parseInt(
+        response.headers.get("retry-after") ?? "60",
+        10
+      );
+      throw errorFromResponse(
+        response.status,
+        { error: body.error },
+        {
+          provider: opts.provider,
+          scopeNeeded: "keys.read",
+          retryAfter: Number.isFinite(retryAfter) ? retryAfter : 60
+        }
+      );
+    }
+    throw new ProviderError(opts.provider, opts.path, response.status, body);
+  }
+  if (response.body == null) {
+    throw new BadTokenError("Upstream returned 2xx with no body to stream.");
+  }
+  return response.body;
+}
+
+// src/providers/openai.ts
+var openai = {
+  images: {
+    generate(token, params, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "openai",
+        method: "POST",
+        path: "/v1/images/generations",
+        body: params
+      });
+    }
+  },
+  chat: {
+    completions: {
+      create(token, params, opts = {}) {
+        return call(token, {
+          ...opts,
+          provider: "openai",
+          method: "POST",
+          path: "/v1/chat/completions",
+          body: params
+        });
+      },
+      stream(token, params, opts = {}) {
+        return callStream(token, {
+          ...opts,
+          provider: "openai",
+          method: "POST",
+          path: "/v1/chat/completions",
+          body: { ...params, stream: true }
+        });
+      }
+    }
+  },
+  embeddings: {
+    create(token, params, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "openai",
+        method: "POST",
+        path: "/v1/embeddings",
+        body: params
+      });
+    }
+  },
+  audio: {
+    transcriptions: {
+      create(token, params, opts = {}) {
+        const fd = new FormData();
+        fd.append("file", params.file, params.filename ?? "audio.webm");
+        fd.append("model", params.model);
+        if (params.language) fd.append("language", params.language);
+        if (params.prompt) fd.append("prompt", params.prompt);
+        if (params.response_format)
+          fd.append("response_format", params.response_format);
+        if (params.temperature !== void 0)
+          fd.append("temperature", String(params.temperature));
+        return call(token, {
+          ...opts,
+          provider: "openai",
+          method: "POST",
+          path: "/v1/audio/transcriptions",
+          body: fd
+        });
+      }
+    },
+    speech: {
+      create(token, params, opts = {}) {
+        return callRaw(token, {
+          ...opts,
+          provider: "openai",
+          method: "POST",
+          path: "/v1/audio/speech",
+          body: params
+        });
+      }
+    }
+  }
+};
+
+// src/providers/anthropic.ts
+var anthropic = {
+  messages: {
+    create(token, params, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "anthropic",
+        method: "POST",
+        path: "/v1/messages",
+        body: params
+      });
+    },
+    stream(token, params, opts = {}) {
+      return callStream(token, {
+        ...opts,
+        provider: "anthropic",
+        method: "POST",
+        path: "/v1/messages",
+        body: { ...params, stream: true }
+      });
+    }
+  }
+};
+
+// src/providers/replicate.ts
+var replicate = {
+  predictions: {
+    create(token, params, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "replicate",
+        method: "POST",
+        path: "/v1/predictions",
+        body: params
+      });
+    },
+    get(token, predictionId, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "replicate",
+        method: "GET",
+        path: `/v1/predictions/${encodeURIComponent(predictionId)}`
+      });
+    },
+    cancel(token, predictionId, opts = {}) {
+      return call(token, {
+        ...opts,
+        provider: "replicate",
+        method: "POST",
+        path: `/v1/predictions/${encodeURIComponent(predictionId)}/cancel`
+      });
+    },
+    /**
+     * Convenience: create + poll until the prediction reaches a terminal
+     * status. Polls every `pollIntervalMs` (default 1000ms) with the
+     * provided AbortSignal honoured.
+     *
+     * For long-running predictions consider using webhooks via
+     * `create({ webhook, webhook_events_filter })` instead so you're not
+     * holding open a long fetch.
+     */
+    async run(token, params, opts = {}) {
+      const interval = opts.pollIntervalMs ?? 1e3;
+      let prediction = await this.create(token, params, opts);
+      while (prediction.status === "starting" || prediction.status === "processing") {
+        if (opts.signal?.aborted) {
+          throw opts.signal.reason ?? new DOMException("Aborted", "AbortError");
+        }
+        await new Promise((resolve) => setTimeout(resolve, interval));
+        prediction = await this.get(token, prediction.id, opts);
+      }
+      return prediction;
+    }
+  }
+};
+
 // src/index.ts
 var session = {
   verify
@@ -362,8 +718,13 @@ var session = {
 var keys = {
   get
 };
-var SDK_VERSION = "0.1.0";
+var proxy = {
+  call,
+  callRaw,
+  callStream
+};
+var SDK_VERSION = "0.2.0";
 
-export { BadTokenError, HubSdkError, InternalError, KeyNotFoundError, MissingScopeError, MissingTokenError, NetworkError, NotSubscribedError, ProviderNotGrantedError, RateLimitedError, SDK_VERSION, SecureKey, SecureKeyConsumedError, SubscriptionInactiveError, ToolNotFoundError, keys, session };
+export { BadTokenError, HubSdkError, InternalError, KeyNotFoundError, MissingScopeError, MissingTokenError, NetworkError, NotSubscribedError, ProviderError, ProviderNotConfiguredError, ProviderNotGrantedError, RateLimitedError, SDK_VERSION, SecureKey, SecureKeyConsumedError, SubscriptionInactiveError, ToolNotFoundError, anthropic, keys, openai, proxy, replicate, session };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map