@okrapdf/ai-sdk 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,72 @@
+import { createOpenAICompatible, type OpenAICompatibleProviderSettings } from "@ai-sdk/openai-compatible";
+export interface OkraProviderSettings {
+    /**
+     * OkraPDF API key. Defaults to `OKRA_API_KEY` env var.
+     */
+    apiKey?: string;
+    /**
+     * Base URL for the OkraPDF API.
+     * @default "https://api.okrapdf.com"
+     */
+    baseURL?: string;
+    /**
+     * Custom headers to include in every request.
+     */
+    headers?: Record<string, string>;
+    /**
+     * Custom fetch implementation.
+     */
+    fetch?: OpenAICompatibleProviderSettings["fetch"];
+}
+export interface OkraProvider {
+    /**
+     * Create an AI SDK language model for a document.
+     *
+     * @example
+     * ```ts
+     * const okra = createOkra({ apiKey: "okra_..." });
+     * const model = okra("doc-abc123");
+     * const { text } = await generateText({ model, prompt: "Summarize this" });
+     * ```
+     */
+    (documentId: string, model?: string): ReturnType<ReturnType<typeof createOpenAICompatible>>;
+    /**
+     * Create a model from an OkraPDF session handle.
+     * Works with any object that has a `modelEndpoint` and `id` property
+     * (i.e. the return value of `okra.sessions.create()` or `okra.sessions.from()`).
+     *
+     * @example
+     * ```ts
+     * const rt = createOkra({ apiKey }); // from @okrapdf/runtime
+     * const session = await rt.sessions.create(file);
+     *
+     * const okra = createOkra({ apiKey }); // from @okrapdf/ai-sdk
+     * const model = okra.fromSession(session);
+     * const { text } = await generateText({ model, prompt: "Summarize" });
+     * ```
+     */
+    fromSession(session: {
+        id: string;
+        modelEndpoint: string;
+    }): ReturnType<ReturnType<typeof createOpenAICompatible>>;
+}
+/**
+ * Create an OkraPDF provider for the Vercel AI SDK.
+ *
+ * Each document in OkraPDF exposes an OpenAI-compatible endpoint.
+ * This provider maps document IDs to AI SDK language models.
+ *
+ * @example
+ * ```ts
+ * import { createOkra } from "@okrapdf/ai-sdk";
+ * import { streamText } from "ai";
+ *
+ * const okra = createOkra({ apiKey: process.env.OKRA_API_KEY });
+ * const result = streamText({
+ *   model: okra("doc-abc123"),
+ *   messages: [{ role: "user", content: "What is the revenue?" }],
+ * });
+ * ```
+ */
+export declare function createOkra(settings?: OkraProviderSettings): OkraProvider;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EACtB,KAAK,gCAAgC,EACtC,MAAM,2BAA2B,CAAC;AAInC,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAEjC;;OAEG;IACH,KAAK,CAAC,EAAE,gCAAgC,CAAC,OAAO,CAAC,CAAC;CACnD;AAED,MAAM,WAAW,YAAY;IAC3B;;;;;;;;;OASG;IACH,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,UAAU,CAC9C,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAC1C,CAAC;IAEF;;;;;;;;;;;;;;OAcG;IACH,WAAW,CAAC,OAAO,EAAE;QACnB,EAAE,EAAE,MAAM,CAAC;QACX,aAAa,EAAE,MAAM,CAAC;KACvB,GAAG,UAAU,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;CAC3D;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,UAAU,CAAC,QAAQ,GAAE,oBAAyB,GAAG,YAAY,CA4B5E"}

package/dist/index.js

@@ -0,0 +1,42 @@
+import { createOpenAICompatible, } from "@ai-sdk/openai-compatible";
+const DEFAULT_BASE_URL = "https://api.okrapdf.com";
+/**
+ * Create an OkraPDF provider for the Vercel AI SDK.
+ *
+ * Each document in OkraPDF exposes an OpenAI-compatible endpoint.
+ * This provider maps document IDs to AI SDK language models.
+ *
+ * @example
+ * ```ts
+ * import { createOkra } from "@okrapdf/ai-sdk";
+ * import { streamText } from "ai";
+ *
+ * const okra = createOkra({ apiKey: process.env.OKRA_API_KEY });
+ * const result = streamText({
+ *   model: okra("doc-abc123"),
+ *   messages: [{ role: "user", content: "What is the revenue?" }],
+ * });
+ * ```
+ */
+export function createOkra(settings = {}) {
+    const apiKey = settings.apiKey ?? globalThis.process?.env?.OKRA_API_KEY;
+    const base = (settings.baseURL ?? DEFAULT_BASE_URL).replace(/\/+$/, "");
+    function forEndpoint(endpoint, model = "default") {
+        const provider = createOpenAICompatible({
+            name: "okrapdf",
+            baseURL: endpoint,
+            apiKey,
+            headers: settings.headers,
+            fetch: settings.fetch,
+        });
+        return provider(model);
+    }
+    const provider = ((documentId, model) => {
+        return forEndpoint(`${base}/v1/documents/${encodeURIComponent(documentId)}`, model);
+    });
+    provider.fromSession = (session) => {
+        return forEndpoint(session.modelEndpoint);
+    };
+    return provider;
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,GAEvB,MAAM,2BAA2B,CAAC;AAEnC,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;AA6DnD;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,UAAU,CAAC,WAAiC,EAAE;IAC5D,MAAM,MAAM,GACV,QAAQ,CAAC,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE,GAAG,EAAE,YAAY,CAAC;IAC3D,MAAM,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,gBAAgB,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAExE,SAAS,WAAW,CAAC,QAAgB,EAAE,QAAgB,SAAS;QAC9D,MAAM,QAAQ,GAAG,sBAAsB,CAAC;YACtC,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,QAAQ;YACjB,MAAM;YACN,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,QAAQ,GAAG,CAAC,CAAC,UAAkB,EAAE,KAAc,EAAE,EAAE;QACvD,OAAO,WAAW,CAChB,GAAG,IAAI,iBAAiB,kBAAkB,CAAC,UAAU,CAAC,EAAE,EACxD,KAAK,CACN,CAAC;IACJ,CAAC,CAAiB,CAAC;IAEnB,QAAQ,CAAC,WAAW,GAAG,CAAC,OAAO,EAAE,EAAE;QACjC,OAAO,WAAW,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC5C,CAAC,CAAC;IAEF,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "@okrapdf/ai-sdk",
+  "version": "0.1.0",
+  "description": "Vercel AI SDK provider for OkraPDF — use any document as an AI model",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  },
+  "keywords": [
+    "okrapdf",
+    "ai-sdk",
+    "vercel",
+    "pdf",
+    "document",
+    "openai-compatible",
+    "language-model"
+  ],
+  "dependencies": {
+    "@ai-sdk/openai-compatible": "^2.0.30"
+  },
+  "peerDependencies": {
+    "ai": "^6.0.0"
+  },
+  "peerDependenciesMeta": {
+    "ai": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@types/node": "^20.14.0",
+    "ai": "^6.0.0",
+    "typescript": "^5.5.0",
+    "vitest": "^2.0.0"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/steventsao/okrapdf",
+    "directory": "packages/ai-sdk"
+  }
+}