ollama-ai-provider-v2 2.0.0-beta.0 → 2.0.0

package/README.md

@@ -7,20 +7,6 @@ Use Ollama with the Vercel AI SDK, implementing the official Ollama API. This pr
 [![Node.js](https://img.shields.io/badge/Node.js-18+-green.svg)](https://nodejs.org/)
 [![License: Apache-2.0](https://img.shields.io/badge/License-Apache--2.0-yellow.svg)](https://opensource.org/licenses/Apache-2.0)
 
-## 🎉 AI SDK 6 Beta Support
-
-This provider now supports **AI SDK 6 Beta** features including:
-
-- **🤖 Agent Abstraction** - Build complex agents with `ToolLoopAgent`
-- **🔐 Tool Approval** - Request user confirmation before executing tools  
-- **📊 Structured Output** - Generate typed data alongside tool calling
-- **⚡ Enhanced Performance** - Optimized for the latest AI SDK features
-
-```bash
-# Install AI SDK 6 Beta + Ollama Provider
-npm install ai@beta ollama-ai-provider-v2
-```
-
 ## Why Choose Ollama Provider V2?
 
 - ✅ **Minimal Dependencies** - Lean codebase with just 2 core dependencies
@@ -29,7 +15,7 @@ npm install ai@beta ollama-ai-provider-v2
 - ✅ **Advanced Features** - Tool calling, streaming, thinking mode, embeddings, and completion models
 - ✅ **Type Safety** - Full TypeScript support with comprehensive type definitions
 - ✅ **Zero Configuration** - Works out-of-the-box with sensible defaults
-- ✅ **Actively Maintained** - Regular updates and AI SDK v5+ compatibility
+- ✅ **Actively Maintained** - Regular updates and AI SDK v6+ compatibility
 
 ## Quick Start
 
@@ -206,147 +192,6 @@ ollama serve
 ollama pull llama3.2
 ```
 
-# AI SDK 6 Beta examples
-## Agent Abstraction
-
-AI SDK 6 introduces the `ToolLoopAgent` class for building agents with full control over execution flow.
-
-### Basic Agent
-
-```typescript
-import { ToolLoopAgent } from 'ai';
-import { ollama } from 'ollama-ai-provider-v2';
-
-const weatherAgent = new ToolLoopAgent({
-  model: ollama('llama3.3:70b'),
-  instructions: 'You are a helpful weather assistant.',
-  tools: {
-    weather: weatherTool,
-  },
-});
-
-const result = await weatherAgent.generate({
-  prompt: 'What is the weather in San Francisco?',
-});
-```
-
-### Agent with Call Options
-
-Use call options to pass runtime configuration to agents:
-
-```typescript
-import { ToolLoopAgent } from 'ai';
-import { ollama } from 'ollama-ai-provider-v2';
-import { z } from 'zod';
-
-const supportAgent = new ToolLoopAgent({
-  model: ollama('qwen2.5:32b'),
-  callOptionsSchema: z.object({
-    userId: z.string(),
-    accountType: z.enum(['free', 'pro', 'enterprise']),
-  }),
-  instructions: 'You are a helpful customer support agent.',
-  prepareCall: ({ options, ...settings }) => ({
-    ...settings,
-    instructions: `${settings.instructions}
-
-User context:
-- Account type: ${options.accountType}  
-- User ID: ${options.userId}
-
-Adjust your response based on the user's account level.`,
-  }),
-});
-
-const result = await supportAgent.generate({
-  prompt: 'How do I upgrade my account?',
-  options: {
-    userId: 'user_123',
-    accountType: 'free',
-  },
-});
-```
-
-## Tool Execution Approval
-
-AI SDK 6 allows you to require user approval before executing tools.
-
-### Basic Tool Approval
-
-```typescript
-import { tool } from 'ai';
-import { z } from 'zod';
-
-export const weatherTool = tool({
-  description: 'Get the weather in a location',
-  inputSchema: z.object({
-    city: z.string(),
-  }),
-  needsApproval: true, // Always require approval
-  execute: async ({ city }) => {
-    const weather = await fetchWeather(city);
-    return weather;
-  },
-});
-```
-
-### Dynamic Approval
-
-Make approval decisions based on tool input:
-
-```typescript
-export const paymentTool = tool({
-  description: 'Process a payment',
-  inputSchema: z.object({
-    amount: z.number(),
-    recipient: z.string(),
-  }),
-  needsApproval: async ({ amount }) => amount > 1000, // Only large payments
-  execute: async ({ amount, recipient }) => {
-    return await processPayment(amount, recipient);
-  },
-});
-```
-
-## UI Integration
-
-### Server-side API Route
-
-```typescript
-import { createAgentUIStreamResponse } from 'ai';
-import { weatherAgent } from '@/lib/agents';
-
-export async function POST(request: Request) {
-  const { messages } = await request.json();
-
-  return createAgentUIStreamResponse({
-    agent: weatherAgent,
-    messages,
-  });
-}
-```
-
-### Client-side with Type Safety
-
-```typescript
-import { useChat } from '@ai-sdk/react';
-import { InferAgentUIMessage } from 'ai';
-import { weatherAgent } from '@/lib/agents';
-
-type WeatherAgentUIMessage = InferAgentUIMessage<typeof weatherAgent>;
-
-export function WeatherChat() {
-  const { messages, sendMessage } = useChat<WeatherAgentUIMessage>();
-
-  return (
-    <div>
-      {/* Your chat UI */}
-    </div>
-  );
-}
-```
-
-
 ## Contributing
 
 Contributions are welcome! Here's how to get started:
@@ -363,4 +208,4 @@ Contributions are welcome! Here's how to get started:
 
 Apache-2.0 © [nordwestt](https://github.com/nordwestt/ollama-ai-provider-v2)
 
-See [LICENSE.md](./LICENSE.md) for details.
+See [LICENSE.md](./LICENSE.md) for details.

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { ProviderV2, LanguageModelV2, EmbeddingModelV2 } from '@ai-sdk/provider';
+import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 
@@ -65,36 +65,44 @@ interface OllamaEmbeddingSettings {
   monitor and detect abuse. Learn more.
   */
     user?: string;
+    /**
+  Whether to truncate input text to fit within model's context length.
+     */
+    truncate?: boolean;
+    /**
+  How long to keep the model loaded in memory (e.g., "5m" for 5 minutes).
+     */
+    keepAlive?: string;
 }
 
-interface OllamaProvider extends ProviderV2 {
-    (modelId: OllamaChatModelId): LanguageModelV2;
+interface OllamaProvider extends ProviderV3 {
+    (modelId: OllamaChatModelId): LanguageModelV3;
     /**
   Creates an Ollama model for text generation.
      */
-    languageModel(modelId: OllamaChatModelId): LanguageModelV2;
+    languageModel(modelId: OllamaChatModelId): LanguageModelV3;
     /**
   Creates an Ollama chat model for text generation.
      */
-    chat(modelId: OllamaChatModelId, settings?: OllamaProviderOptions): LanguageModelV2;
+    chat(modelId: OllamaChatModelId, settings?: OllamaProviderOptions): LanguageModelV3;
     /**
   Creates an Ollama completion model for text generation.
      */
-    completion(modelId: OllamaCompletionModelId, settings?: OllamaCompletionSettings): LanguageModelV2;
+    completion(modelId: OllamaCompletionModelId, settings?: OllamaCompletionSettings): LanguageModelV3;
     /**
   Creates a model for text embeddings.
      */
-    embedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    embedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
     /**
   Creates a model for text embeddings.
   
   @deprecated Use `textEmbeddingModel` instead.
      */
-    textEmbedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    textEmbedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
     /**
   Creates a model for text embeddings.
      */
-    textEmbeddingModel(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    textEmbeddingModel(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
 }
 interface OllamaProviderSettings {
     /**

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { ProviderV2, LanguageModelV2, EmbeddingModelV2 } from '@ai-sdk/provider';
+import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 
@@ -65,36 +65,44 @@ interface OllamaEmbeddingSettings {
   monitor and detect abuse. Learn more.
   */
     user?: string;
+    /**
+  Whether to truncate input text to fit within model's context length.
+     */
+    truncate?: boolean;
+    /**
+  How long to keep the model loaded in memory (e.g., "5m" for 5 minutes).
+     */
+    keepAlive?: string;
 }
 
-interface OllamaProvider extends ProviderV2 {
-    (modelId: OllamaChatModelId): LanguageModelV2;
+interface OllamaProvider extends ProviderV3 {
+    (modelId: OllamaChatModelId): LanguageModelV3;
     /**
   Creates an Ollama model for text generation.
      */
-    languageModel(modelId: OllamaChatModelId): LanguageModelV2;
+    languageModel(modelId: OllamaChatModelId): LanguageModelV3;
     /**
   Creates an Ollama chat model for text generation.
      */
-    chat(modelId: OllamaChatModelId, settings?: OllamaProviderOptions): LanguageModelV2;
+    chat(modelId: OllamaChatModelId, settings?: OllamaProviderOptions): LanguageModelV3;
     /**
   Creates an Ollama completion model for text generation.
      */
-    completion(modelId: OllamaCompletionModelId, settings?: OllamaCompletionSettings): LanguageModelV2;
+    completion(modelId: OllamaCompletionModelId, settings?: OllamaCompletionSettings): LanguageModelV3;
     /**
   Creates a model for text embeddings.
      */
-    embedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    embedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
     /**
   Creates a model for text embeddings.
   
   @deprecated Use `textEmbeddingModel` instead.
      */
-    textEmbedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    textEmbedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
     /**
   Creates a model for text embeddings.
      */
-    textEmbeddingModel(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
+    textEmbeddingModel(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV3;
 }
 interface OllamaProviderSettings {
     /**