@easbot/local-model-sdk 0.1.3 → 0.1.11

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. package/README.en.md +7 -7
  2. package/README.md +6 -6
  3. package/dist/index.cjs +118 -28
  4. package/dist/index.d.cts +164 -42
  5. package/dist/index.d.ts +164 -42
  6. package/dist/index.mjs +118 -28
  7. package/package.json +33 -32
package/README.en.md CHANGED
@@ -1,6 +1,6 @@
1
1
  # Local Model SDK
2
2
 
3
- Local model SDK based on Transformers.js + ONNX, supporting Language Model (text generation) and Embedding (text embedding) capabilities, fully compatible with AI SDK specifications.
3
+ Local model SDK based on Transformers.js + ONNX, supporting Language Model (text generation) and Embedding (text embedding) capabilities, fully compatible with AI SDK v3 specifications.
4
4
 
5
5
  ## Features
6
6
 
@@ -9,7 +9,7 @@ Local model SDK based on Transformers.js + ONNX, supporting Language Model (text
9
9
  - ✅ **Streaming Generation**: Real-time streaming text output
10
10
  - ✅ **Batch Processing**: Efficient batch embedding processing
11
11
  - ✅ **Model Caching**: Automatic model file cache management
12
- - ✅ **AI SDK Compatible**: Fully compatible with AI SDK v2 specification
12
+ - ✅ **AI SDK Compatible**: Fully compatible with AI SDK v3 specification
13
13
  - ✅ **TypeScript**: Complete type definition support
14
14
  - ✅ **Offline Operation**: No network required, fully local
15
15
 
@@ -67,7 +67,7 @@ import { embed, embedMany } from 'ai';
67
67
 
68
68
  // Single text embedding
69
69
  const result = await embed({
70
- model: provider.textEmbeddingModel('bge-small-zh-v1.5'),
70
+ model: provider.embeddingModel('bge-small-zh-v1.5'),
71
71
  value: 'Hello world',
72
72
  });
73
73
 
@@ -75,7 +75,7 @@ console.log(result.embedding); // number[]
75
75
 
76
76
  // Batch text embedding
77
77
  const batchResult = await embedMany({
78
- model: provider.textEmbeddingModel('bge-small-zh-v1.5'),
78
+ model: provider.embeddingModel('bge-small-zh-v1.5'),
79
79
  values: ['Text 1', 'Text 2', 'Text 3'],
80
80
  });
81
81
 
@@ -116,13 +116,13 @@ const provider = new LocalModelProvider({
116
116
  #### Methods
117
117
 
118
118
  - `languageModel(modelId: string): LocalLanguageModel` - Create Language Model instance
119
- - `textEmbeddingModel(modelId: string): LocalEmbeddingModel` - Create Embedding Model instance
119
+ - `embeddingModel(modelId: string): LocalEmbeddingModel` - Create Embedding Model instance
120
120
  - `getLoadedModels(): ModelInfo[]` - Get list of loaded models
121
121
  - `dispose(): Promise<void>` - Unload all models and release resources
122
122
 
123
123
  ### LocalLanguageModel
124
124
 
125
- Implements AI SDK's `LanguageModelV2` interface.
125
+ Implements AI SDK's `LanguageModelV3` interface.
126
126
 
127
127
  ```typescript
128
128
  const model = provider.languageModel('qwen2.5-0.5b-instruct');
@@ -147,7 +147,7 @@ const stream = streamText({
147
147
  Implements AI SDK's `EmbeddingModelV2` interface.
148
148
 
149
149
  ```typescript
150
- const model = provider.textEmbeddingModel('bge-small-zh-v1.5');
150
+ const model = provider.embeddingModel('bge-small-zh-v1.5');
151
151
 
152
152
  // Single text
153
153
  const result = await embed({
package/README.md CHANGED
@@ -9,7 +9,7 @@
9
9
  - ✅ **流式生成**：支持实时流式文本输出
10
10
  - ✅ **批量处理**：高效的批量嵌入处理
11
11
  - ✅ **模型缓存**：自动管理模型文件缓存
12
- - ✅ **AI SDK 兼容**：完全兼容 AI SDK v2 规范
12
+ - ✅ **AI SDK 兼容**：完全兼容 AI SDK v3 规范
13
13
  - ✅ **TypeScript**：完整的类型定义支持
14
14
  - ✅ **离线运行**：无需网络连接，完全本地化
15
15
 
@@ -67,7 +67,7 @@ import { embed, embedMany } from 'ai';
67
67
 
68
68
  // 单个文本嵌入
69
69
  const result = await embed({
70
- model: provider.textEmbeddingModel('bge-small-zh-v1.5'),
70
+ model: provider.embeddingModel('bge-small-zh-v1.5'),
71
71
  value: 'Hello world',
72
72
  });
73
73
 
@@ -75,7 +75,7 @@ console.log(result.embedding); // number[]
75
75
 
76
76
  // 批量文本嵌入
77
77
  const batchResult = await embedMany({
78
- model: provider.textEmbeddingModel('bge-small-zh-v1.5'),
78
+ model: provider.embeddingModel('bge-small-zh-v1.5'),
79
79
  values: ['Text 1', 'Text 2', 'Text 3'],
80
80
  });
81
81
 
@@ -116,13 +116,13 @@ const provider = new LocalModelProvider({
116
116
  #### 方法
117
117
 
118
118
  - `languageModel(modelId: string): LocalLanguageModel` - 创建 Language Model 实例
119
- - `textEmbeddingModel(modelId: string): LocalEmbeddingModel` - 创建 Embedding Model 实例
119
+ - `embeddingModel(modelId: string): LocalEmbeddingModel` - 创建 Embedding Model 实例
120
120
  - `getLoadedModels(): ModelInfo[]` - 获取已加载的模型列表
121
121
  - `dispose(): Promise<void>` - 卸载所有模型，释放资源
122
122
 
123
123
  ### LocalLanguageModel
124
124
 
125
- 实现 AI SDK 的 `LanguageModelV2` 接口。
125
+ 实现 AI SDK 的 `LanguageModelV3` 接口。
126
126
 
127
127
  ```typescript
128
128
  const model = provider.languageModel('qwen2.5-0.5b-instruct');
@@ -147,7 +147,7 @@ const stream = streamText({
147
147
  实现 AI SDK 的 `EmbeddingModelV2` 接口。
148
148
 
149
149
  ```typescript
150
- const model = provider.textEmbeddingModel('bge-small-zh-v1.5');
150
+ const model = provider.embeddingModel('bge-small-zh-v1.5');
151
151
 
152
152
  // 单个文本
153
153
  const result = await embed({