@alpic80/rivet-ai-sdk-provider 2.0.0 → 2.0.2

package/README.md

@@ -1,199 +1,41 @@
-# OpenRouter Provider for Vercel AI SDK
+# Rivet Provider for Vercel AI SDK
 
-The [OpenRouter](https://openrouter.ai/) provider for the [Vercel AI SDK](https://sdk.vercel.ai/docs) gives access to over 300 large language models on the OpenRouter chat and completion APIs.
+This is the [Rivet](https://rivet.ironcladapp.com) provider for the [Vercel AI SDK](https://sdk.vercel.ai/docs) giving you access to run your Rivet Graphs as a supported LM.
 
 ## Setup for AI SDK v5
 
 ```bash
 # For pnpm
-pnpm add @openrouter/ai-sdk-provider
+pnpm add @alpic80/rivet-ai-sdk-provider
 
 # For npm
-npm install @openrouter/ai-sdk-provider
+npm install @alpic80/rivet-ai-sdk-provider
 
 # For yarn
-yarn add @openrouter/ai-sdk-provider
-```
-
-## (LEGACY) Setup for AI SDK v4
-
-```bash
-# For pnpm
-pnpm add @openrouter/ai-sdk-provider@ai-sdk-v4
-
-# For npm
-npm install @openrouter/ai-sdk-provider@ai-sdk-v4
-
-# For yarn
-yarn add @openrouter/ai-sdk-provider@ai-sdk-v4
-
+yarn add @alpic80/rivet-ai-sdk-provider
 ```
 
 ## Provider Instance
 
-You can import the default provider instance `openrouter` from `@openrouter/ai-sdk-provider`:
+You can import the default provider instance `rivet` from `@alpic80/rivet-ai-sdk-provider`:
 
 ```ts
-import { openrouter } from '@openrouter/ai-sdk-provider';
+import { rivet } from '@alpic80/rivet-ai-sdk-provider';
 ```
 
 ## Example
 
 ```ts
-import { openrouter } from '@openrouter/ai-sdk-provider';
+import { rivet } from '"@alpic80/rivet-ai-sdk-provider';
 import { generateText } from 'ai';
 
 const { text } = await generateText({
-  model: openrouter('openai/gpt-4o'),
+  model: rivet('myGraph.rivet-project'),
   prompt: 'Write a vegetarian lasagna recipe for 4 people.',
 });
 ```
 
-## Supported models
-
-This list is not a definitive list of models supported by OpenRouter, as it constantly changes as we add new models (and deprecate old ones) to our system. You can find the latest list of models supported by OpenRouter [here](https://openrouter.ai/models).
-
-You can find the latest list of tool-supported models supported by OpenRouter [here](https://openrouter.ai/models?order=newest&supported_parameters=tools). (Note: This list may contain models that are not compatible with the AI SDK.)
-
-## Passing Extra Body to OpenRouter
-
-There are 3 ways to pass extra body to OpenRouter:
-
-1. Via the `providerOptions.openrouter` property:
-
-   ```typescript
-   import { createOpenRouter } from '@openrouter/ai-sdk-provider';
-   import { streamText } from 'ai';
-
-   const openrouter = createOpenRouter({ apiKey: 'your-api-key' });
-   const model = openrouter('anthropic/claude-3.7-sonnet:thinking');
-   await streamText({
-     model,
-     messages: [{ role: 'user', content: 'Hello' }],
-     providerOptions: {
-       openrouter: {
-         reasoning: {
-           max_tokens: 10,
-         },
-       },
-     },
-   });
-   ```
-
-2. Via the `extraBody` property in the model settings:
-
-   ```typescript
-   import { createOpenRouter } from '@openrouter/ai-sdk-provider';
-   import { streamText } from 'ai';
-
-   const openrouter = createOpenRouter({ apiKey: 'your-api-key' });
-   const model = openrouter('anthropic/claude-3.7-sonnet:thinking', {
-     extraBody: {
-       reasoning: {
-         max_tokens: 10,
-       },
-     },
-   });
-   await streamText({
-     model,
-     messages: [{ role: 'user', content: 'Hello' }],
-   });
-   ```
+## Required Configuration
 
-3. Via the `extraBody` property in the model factory.
-
-   ```typescript
-   import { createOpenRouter } from '@openrouter/ai-sdk-provider';
-   import { streamText } from 'ai';
-
-   const openrouter = createOpenRouter({
-     apiKey: 'your-api-key',
-     extraBody: {
-       reasoning: {
-         max_tokens: 10,
-       },
-     },
-   });
-   const model = openrouter('anthropic/claude-3.7-sonnet:thinking');
-   await streamText({
-     model,
-     messages: [{ role: 'user', content: 'Hello' }],
-   });
-   ```
-
-## Anthropic Prompt Caching
-
-You can include Anthropic-specific options directly in your messages when using functions like `streamText`. The OpenRouter provider will automatically convert these messages to the correct format internally.
-
-### Basic Usage
-
-```typescript
-import { createOpenRouter } from '@openrouter/ai-sdk-provider';
-import { streamText } from 'ai';
-
-const openrouter = createOpenRouter({ apiKey: 'your-api-key' });
-const model = openrouter('anthropic/<supported-caching-model>');
-
-await streamText({
-  model,
-  messages: [
-    {
-      role: 'system',
-      content:
-        'You are a podcast summary assistant. You are detail-oriented and critical about the content.',
-    },
-    {
-      role: 'user',
-      content: [
-        {
-          type: 'text',
-          text: 'Given the text body below:',
-        },
-        {
-          type: 'text',
-          text: `<LARGE BODY OF TEXT>`,
-          providerOptions: {
-            openrouter: {
-              cacheControl: { type: 'ephemeral' },
-            },
-          },
-        },
-        {
-          type: 'text',
-          text: 'List the speakers?',
-        },
-      ],
-    },
-  ],
-});
-```
-
-## Use Cases
-
-### Usage Accounting
-
-The provider supports [OpenRouter usage accounting](https://openrouter.ai/docs/use-cases/usage-accounting), which allows you to track token usage details directly in your API responses, without making additional API calls.
-
-```typescript
-// Enable usage accounting
-const model = openrouter('openai/gpt-3.5-turbo', {
-  usage: {
-    include: true,
-  },
-});
-
-// Access usage accounting data
-const result = await generateText({
-  model,
-  prompt: 'Hello, how are you today?',
-});
-
-// Provider-specific usage details (available in providerMetadata)
-if (result.providerMetadata?.openrouter?.usage) {
-  console.log('Cost:', result.providerMetadata.openrouter.usage.cost);
-  console.log(
-    'Total Tokens:',
-    result.providerMetadata.openrouter.usage.totalTokens,
-  );
-}
-```
+In order to use the Rivet AI SDK provider you will need to also install and run the Adion custom Rivet server
+available on npm (@alpic80/rivet-cli) and to configure it with support for PROJECTS_ROOT_DIR (see serve.md for details)

package/dist/index.js

@@ -1109,7 +1109,7 @@ prepare tools input:${printObject(tools)}`);
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.0" : "0.0.0-test";
+var VERSION = true ? "2.0.2" : "0.0.0-test";
 
 // src/post-to-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -1761,7 +1761,7 @@ function createRivetEventSourceResponseHandler(id, model) {
   }));
 }
 function mapRivetEventToOpenAIChunk(eventData, id, model) {
-  var _a15, _b, _c, _d, _e, _f, _g;
+  var _a15, _b, _c, _d, _e, _f, _g, _h;
   const eventType = eventData.type;
   switch (eventType) {
     case "partialOutput":
@@ -1782,6 +1782,24 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
         ],
         usage: void 0
       };
+    case "nodeFinish":
+      return {
+        id,
+        created: Math.floor(Date.now() / 1e3),
+        model,
+        choices: [
+          {
+            delta: {
+              role: "assistant",
+              content: (_b = eventData.delta) != null ? _b : "",
+              tool_calls: void 0
+            },
+            finish_reason: null,
+            index: 0
+          }
+        ],
+        usage: void 0
+      };
     case "done":
       return {
         id,
@@ -1791,14 +1809,14 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
           {
             delta: {
               role: "assistant",
-              content: (_d = (_c = (_b = eventData.graphOutput) == null ? void 0 : _b.response) == null ? void 0 : _c.value) != null ? _d : "",
+              content: (_e = (_d = (_c = eventData.graphOutput) == null ? void 0 : _c.response) == null ? void 0 : _d.value) != null ? _e : "",
               tool_calls: void 0
             },
             finish_reason: "stop",
             index: 0
           }
         ],
-        usage: toOpenAIUsage((_g = (_f = (_e = eventData.graphOutput.usages) == null ? void 0 : _e.value) == null ? void 0 : _f[0]) == null ? void 0 : _g.value)
+        usage: toOpenAIUsage((_h = (_g = (_f = eventData.graphOutput.usages) == null ? void 0 : _f.value) == null ? void 0 : _g[0]) == null ? void 0 : _h.value)
       };
     default:
       return {