@easbot/ollama-sdk 0.2.46 → 0.2.48

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 (5) hide show
  1. package/dist/index.cjs +4 -4
  2. package/dist/index.d.cts +62 -17
  3. package/dist/index.d.ts +62 -17
  4. package/dist/index.mjs +4 -4
  5. package/package.json +3 -3
package/dist/index.d.cts CHANGED
@@ -1,5 +1,4 @@
1
- import { Provider, LanguageModel, EmbeddingModel, RerankingModel } from 'ai';
2
- import { LanguageModelV3, LanguageModelV3CallOptions, EmbeddingModelV3, EmbeddingModelV3CallOptions, EmbeddingModelV3Result } from '@ai-sdk/provider';
1
+ import { LanguageModelV3, LanguageModelV3CallOptions, EmbeddingModelV3, EmbeddingModelV3CallOptions, EmbeddingModelV3Result, RerankingModelV3 } from '@ai-sdk/provider';
3
2
 
4
3
  type OllamaModelId = string;
5
4
  interface OllamaConfig {
@@ -20,21 +19,6 @@ interface OllamaProviderOptions {
20
19
  seed?: number;
21
20
  }
22
21
 
23
- interface RerankingModelOptions {
24
- totalScore?: number;
25
- useEnhancedPrompt?: boolean;
26
- promptLanguage?: 'zh' | 'en';
27
- }
28
-
29
- interface OllamaProvider extends Provider {
30
- (modelId: OllamaModelId): LanguageModel;
31
- languageModel(modelId: OllamaModelId): LanguageModel;
32
- chat(modelId: OllamaModelId): LanguageModel;
33
- embeddingModel(modelId: OllamaModelId): EmbeddingModel;
34
- rerankingModel(modelId: OllamaModelId, options?: RerankingModelOptions): RerankingModel;
35
- }
36
- declare function createOllama(config?: OllamaConfig): OllamaProvider;
37
-
38
22
  declare class OllamaLanguageModel implements LanguageModelV3 {
39
23
  readonly specificationVersion: "v3";
40
24
  readonly provider: "ollama";
@@ -71,6 +55,67 @@ declare class OllamaEmbeddingModel implements EmbeddingModelV3 {
71
55
  private estimateTokens;
72
56
  }
73
57
 
58
+ interface RerankingModelOptions {
59
+ totalScore?: number;
60
+ useEnhancedPrompt?: boolean;
61
+ promptLanguage?: 'zh' | 'en';
62
+ }
63
+ declare class OllamaRerankingModel implements RerankingModelV3 {
64
+ readonly specificationVersion: "v3";
65
+ readonly provider = "ollama";
66
+ readonly modelId: OllamaModelId;
67
+ private languageModel;
68
+ private options;
69
+ private log;
70
+ constructor(modelId: OllamaModelId, config: OllamaConfig, options?: RerankingModelOptions);
71
+ doRerank(options: {
72
+ query: string;
73
+ documents: {
74
+ type: 'text';
75
+ values: string[];
76
+ } | {
77
+ type: 'object';
78
+ values: Array<{
79
+ content: string;
80
+ id?: string;
81
+ }>;
82
+ };
83
+ topN?: number;
84
+ abortSignal?: AbortSignal;
85
+ providerOptions?: any;
86
+ headers?: Record<string, string>;
87
+ }): Promise<{
88
+ ranking: Array<{
89
+ index: number;
90
+ relevanceScore: number;
91
+ }>;
92
+ providerMetadata?: any;
93
+ warnings?: Array<{
94
+ type: 'other';
95
+ message: string;
96
+ }>;
97
+ response?: {
98
+ id?: string;
99
+ timestamp?: Date;
100
+ modelId?: string;
101
+ headers?: Record<string, string>;
102
+ body?: unknown;
103
+ };
104
+ }>;
105
+ private buildRerankPrompt;
106
+ private parseRerankResult;
107
+ }
108
+
109
+ interface OllamaProvider {
110
+ readonly specificationVersion: 'v3';
111
+ (modelId: OllamaModelId): OllamaLanguageModel;
112
+ languageModel(modelId: OllamaModelId): OllamaLanguageModel;
113
+ chat(modelId: OllamaModelId): OllamaLanguageModel;
114
+ embeddingModel(modelId: OllamaModelId): OllamaEmbeddingModel;
115
+ rerankingModel(modelId: OllamaModelId, options?: RerankingModelOptions): OllamaRerankingModel;
116
+ }
117
+ declare function createOllama(config?: OllamaConfig): OllamaProvider;
118
+
74
119
  declare const VERSION = "0.1.0";
75
120
 
76
121
  declare class OllamaError extends Error {
package/dist/index.d.ts CHANGED
@@ -1,5 +1,4 @@
1
- import { Provider, LanguageModel, EmbeddingModel, RerankingModel } from 'ai';
2
- import { LanguageModelV3, LanguageModelV3CallOptions, EmbeddingModelV3, EmbeddingModelV3CallOptions, EmbeddingModelV3Result } from '@ai-sdk/provider';
1
+ import { LanguageModelV3, LanguageModelV3CallOptions, EmbeddingModelV3, EmbeddingModelV3CallOptions, EmbeddingModelV3Result, RerankingModelV3 } from '@ai-sdk/provider';
3
2
 
4
3
  type OllamaModelId = string;
5
4
  interface OllamaConfig {
@@ -20,21 +19,6 @@ interface OllamaProviderOptions {
20
19
  seed?: number;
21
20
  }
22
21
 
23
- interface RerankingModelOptions {
24
- totalScore?: number;
25
- useEnhancedPrompt?: boolean;
26
- promptLanguage?: 'zh' | 'en';
27
- }
28
-
29
- interface OllamaProvider extends Provider {
30
- (modelId: OllamaModelId): LanguageModel;
31
- languageModel(modelId: OllamaModelId): LanguageModel;
32
- chat(modelId: OllamaModelId): LanguageModel;
33
- embeddingModel(modelId: OllamaModelId): EmbeddingModel;
34
- rerankingModel(modelId: OllamaModelId, options?: RerankingModelOptions): RerankingModel;
35
- }
36
- declare function createOllama(config?: OllamaConfig): OllamaProvider;
37
-
38
22
  declare class OllamaLanguageModel implements LanguageModelV3 {
39
23
  readonly specificationVersion: "v3";
40
24
  readonly provider: "ollama";
@@ -71,6 +55,67 @@ declare class OllamaEmbeddingModel implements EmbeddingModelV3 {
71
55
  private estimateTokens;
72
56
  }
73
57
 
58
+ interface RerankingModelOptions {
59
+ totalScore?: number;
60
+ useEnhancedPrompt?: boolean;
61
+ promptLanguage?: 'zh' | 'en';
62
+ }
63
+ declare class OllamaRerankingModel implements RerankingModelV3 {
64
+ readonly specificationVersion: "v3";
65
+ readonly provider = "ollama";
66
+ readonly modelId: OllamaModelId;
67
+ private languageModel;
68
+ private options;
69
+ private log;
70
+ constructor(modelId: OllamaModelId, config: OllamaConfig, options?: RerankingModelOptions);
71
+ doRerank(options: {
72
+ query: string;
73
+ documents: {
74
+ type: 'text';
75
+ values: string[];
76
+ } | {
77
+ type: 'object';
78
+ values: Array<{
79
+ content: string;
80
+ id?: string;
81
+ }>;
82
+ };
83
+ topN?: number;
84
+ abortSignal?: AbortSignal;
85
+ providerOptions?: any;
86
+ headers?: Record<string, string>;
87
+ }): Promise<{
88
+ ranking: Array<{
89
+ index: number;
90
+ relevanceScore: number;
91
+ }>;
92
+ providerMetadata?: any;
93
+ warnings?: Array<{
94
+ type: 'other';
95
+ message: string;
96
+ }>;
97
+ response?: {
98
+ id?: string;
99
+ timestamp?: Date;
100
+ modelId?: string;
101
+ headers?: Record<string, string>;
102
+ body?: unknown;
103
+ };
104
+ }>;
105
+ private buildRerankPrompt;
106
+ private parseRerankResult;
107
+ }
108
+
109
+ interface OllamaProvider {
110
+ readonly specificationVersion: 'v3';
111
+ (modelId: OllamaModelId): OllamaLanguageModel;
112
+ languageModel(modelId: OllamaModelId): OllamaLanguageModel;
113
+ chat(modelId: OllamaModelId): OllamaLanguageModel;
114
+ embeddingModel(modelId: OllamaModelId): OllamaEmbeddingModel;
115
+ rerankingModel(modelId: OllamaModelId, options?: RerankingModelOptions): OllamaRerankingModel;
116
+ }
117
+ declare function createOllama(config?: OllamaConfig): OllamaProvider;
118
+
74
119
  declare const VERSION = "0.1.0";
75
120
 
76
121
  declare class OllamaError extends Error {