@memberjunction/ai-mistral 0.9.17 → 0.9.18
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.
- package/dist/config.d.ts +1 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/generic/mistral.types.d.ts +87 -0
- package/dist/generic/mistral.types.d.ts.map +1 -0
- package/dist/generic/mistral.types.js +3 -0
- package/dist/generic/mistral.types.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/models/mistral.d.ts +4 -4
- package/dist/models/mistral.d.ts.map +1 -0
- package/dist/models/mistral.js +0 -13
- package/dist/models/mistral.js.map +1 -1
- package/dist/models/mistralClient.d.ts +2 -86
- package/dist/models/mistralClient.d.ts.map +1 -0
- package/dist/models/mistralClient.js.map +1 -1
- package/dist/models/mistralEmbedding.d.ts +8 -0
- package/dist/models/mistralEmbedding.d.ts.map +1 -0
- package/dist/models/mistralEmbedding.js +35 -0
- package/dist/models/mistralEmbedding.js.map +1 -0
- package/package.json +3 -3
package/dist/config.d.ts
CHANGED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,aAAa,EAAE,MAAoC,CAAC"}
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { ChatMessage } from "@memberjunction/ai";
|
|
2
|
+
export interface ModelPermission {
|
|
3
|
+
id: string;
|
|
4
|
+
object: 'model_permission';
|
|
5
|
+
created: number;
|
|
6
|
+
allow_create_engine: boolean;
|
|
7
|
+
allow_sampling: boolean;
|
|
8
|
+
allow_logprobs: boolean;
|
|
9
|
+
allow_search_indices: boolean;
|
|
10
|
+
allow_view: boolean;
|
|
11
|
+
allow_fine_tuning: boolean;
|
|
12
|
+
organization: string;
|
|
13
|
+
group: string | null;
|
|
14
|
+
is_blocking: boolean;
|
|
15
|
+
}
|
|
16
|
+
export interface Model {
|
|
17
|
+
id: string;
|
|
18
|
+
object: 'model';
|
|
19
|
+
created: number;
|
|
20
|
+
owned_by: string;
|
|
21
|
+
root: string | null;
|
|
22
|
+
parent: string | null;
|
|
23
|
+
permission: ModelPermission[];
|
|
24
|
+
}
|
|
25
|
+
export interface ListModelsResponse {
|
|
26
|
+
object: 'list';
|
|
27
|
+
data: Model[];
|
|
28
|
+
}
|
|
29
|
+
export interface TokenUsage {
|
|
30
|
+
prompt_tokens: number;
|
|
31
|
+
completion_tokens: number;
|
|
32
|
+
total_tokens: number;
|
|
33
|
+
}
|
|
34
|
+
export type ChatCompletetionRequest = {
|
|
35
|
+
model: string;
|
|
36
|
+
messages: ChatMessage[];
|
|
37
|
+
temperature?: number;
|
|
38
|
+
max_tokens?: number;
|
|
39
|
+
top_p?: number;
|
|
40
|
+
random_seed?: number;
|
|
41
|
+
stream?: boolean;
|
|
42
|
+
safe_prompt?: boolean;
|
|
43
|
+
};
|
|
44
|
+
export interface ChatCompletionResponseChoice {
|
|
45
|
+
index: number;
|
|
46
|
+
message: {
|
|
47
|
+
role: string;
|
|
48
|
+
content: string;
|
|
49
|
+
};
|
|
50
|
+
finish_reason: string;
|
|
51
|
+
}
|
|
52
|
+
export interface ChatCompletionResponseChunkChoice {
|
|
53
|
+
index: number;
|
|
54
|
+
delta: {
|
|
55
|
+
role?: string;
|
|
56
|
+
content?: string;
|
|
57
|
+
};
|
|
58
|
+
finish_reason: string;
|
|
59
|
+
}
|
|
60
|
+
export interface ChatCompletionResponse {
|
|
61
|
+
id: string;
|
|
62
|
+
object: 'chat.completion';
|
|
63
|
+
created: number;
|
|
64
|
+
model: string;
|
|
65
|
+
choices: ChatCompletionResponseChoice[];
|
|
66
|
+
usage: TokenUsage;
|
|
67
|
+
}
|
|
68
|
+
export interface ChatCompletionResponseChunk {
|
|
69
|
+
id: string;
|
|
70
|
+
object: 'chat.completion.chunk';
|
|
71
|
+
created: number;
|
|
72
|
+
model: string;
|
|
73
|
+
choices: ChatCompletionResponseChunkChoice[];
|
|
74
|
+
}
|
|
75
|
+
export interface Embedding {
|
|
76
|
+
id: string;
|
|
77
|
+
object: 'embedding';
|
|
78
|
+
embedding: number[];
|
|
79
|
+
}
|
|
80
|
+
export interface EmbeddingResponse {
|
|
81
|
+
id: string;
|
|
82
|
+
object: 'list';
|
|
83
|
+
data: Embedding[];
|
|
84
|
+
model: string;
|
|
85
|
+
usage: TokenUsage;
|
|
86
|
+
}
|
|
87
|
+
//# sourceMappingURL=mistral.types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistral.types.d.ts","sourceRoot":"","sources":["../../src/generic/mistral.types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,WAAW,eAAe;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,kBAAkB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,CAAC;IAC7B,cAAc,EAAE,OAAO,CAAC;IACxB,cAAc,EAAE,OAAO,CAAC;IACxB,oBAAoB,EAAE,OAAO,CAAC;IAC9B,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,WAAW,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,UAAU,EAAE,eAAe,EAAE,CAAC;CACjC;AAED,MAAM,WAAW,kBAAkB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,KAAK,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,UAAU;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,MAAM,uBAAuB,GAAG;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,WAAW,CAAC,EAAE,OAAO,CAAA;CACxB,CAAA;AAED,MAAM,WAAW,4BAA4B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,iCAAiC;IAC9C,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,sBAAsB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,iBAAiB,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,4BAA4B,EAAE,CAAC;IACxC,KAAK,EAAE,UAAU,CAAC;CACrB;AAED,MAAM,WAAW,2BAA2B;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,uBAAuB,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,iCAAiC,EAAE,CAAC;CAChD;AAED,MAAM,WAAW,SAAS;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,UAAU,CAAC;CACrB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistral.types.js","sourceRoot":"","sources":["../../src/generic/mistral.types.ts"],"names":[],"mappings":""}
|
package/dist/index.d.ts
CHANGED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,2BAA2B,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -15,4 +15,5 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
|
15
15
|
};
|
|
16
16
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
17
|
__exportStar(require("./models/mistral"), exports);
|
|
18
|
+
__exportStar(require("./models/mistralEmbedding"), exports);
|
|
18
19
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,mDAAiC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,mDAAiC;AACjC,4DAA0C"}
|
package/dist/models/mistral.d.ts
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
|
-
import { BaseLLM, ChatParams, ChatResult, ClassifyParams, ClassifyResult,
|
|
2
|
-
import {
|
|
1
|
+
import { BaseLLM, ChatParams, ChatResult, ClassifyParams, ClassifyResult, SummarizeParams, SummarizeResult } from '@memberjunction/ai';
|
|
2
|
+
import { MistralClient } from './mistralClient';
|
|
3
|
+
import { ListModelsResponse } from '../generic/mistral.types';
|
|
3
4
|
export declare class MistralLLM extends BaseLLM {
|
|
4
5
|
static _client: MistralClient;
|
|
5
6
|
constructor(apiKey: string);
|
|
@@ -7,8 +8,6 @@ export declare class MistralLLM extends BaseLLM {
|
|
|
7
8
|
ChatCompletion(params: MistralChatParams): Promise<ChatResult>;
|
|
8
9
|
SummarizeText(params: SummarizeParams): Promise<SummarizeResult>;
|
|
9
10
|
ClassifyText(params: ClassifyParams): Promise<ClassifyResult>;
|
|
10
|
-
createEmbedding(model: string, text: string): Promise<EmbeddingResponse>;
|
|
11
|
-
EmbedText(params: EmbedParams): Promise<EmbedResult>;
|
|
12
11
|
/**
|
|
13
12
|
* Returns a list of available models
|
|
14
13
|
* @returns {Promise<AvailableModelInfo>}
|
|
@@ -19,3 +18,4 @@ export declare class MistralChatParams extends ChatParams {
|
|
|
19
18
|
model: string;
|
|
20
19
|
}
|
|
21
20
|
export declare function LoadMistralLLM(): void;
|
|
21
|
+
//# sourceMappingURL=mistral.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistral.d.ts","sourceRoot":"","sources":["../../src/models/mistral.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAoB,cAAc,EAAE,cAAc,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAEzJ,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAgC,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAE5F,qBACa,UAAW,SAAQ,OAAO;IACnC,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC;gBAClB,MAAM,EAAE,MAAM;IAO1B,IAAW,MAAM,IAAI,aAAa,CAA6B;IAElD,cAAc,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,UAAU,CAAC;IAwC9D,aAAa,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IAIhE,YAAY,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAI1E;;;OAGG;IACU,UAAU,IAAI,OAAO,CAAC,kBAAkB,CAAC;CAIzD;AAED,qBAAa,iBAAkB,SAAQ,UAAU;IAC7C,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,wBAAgB,cAAc,SAE7B"}
|
package/dist/models/mistral.js
CHANGED
|
@@ -61,19 +61,6 @@ let MistralLLM = MistralLLM_1 = class MistralLLM extends ai_1.BaseLLM {
|
|
|
61
61
|
async ClassifyText(params) {
|
|
62
62
|
throw new Error("Method not implemented.");
|
|
63
63
|
}
|
|
64
|
-
async createEmbedding(model, text) {
|
|
65
|
-
const response = await this.client.embeddings(model, text);
|
|
66
|
-
return response;
|
|
67
|
-
}
|
|
68
|
-
async EmbedText(params) {
|
|
69
|
-
const response = await this.client.embeddings(params.model, params.text);
|
|
70
|
-
return {
|
|
71
|
-
object: 'object',
|
|
72
|
-
model: params.model || "mistral-embed", //hard coded for now as theres only one available embedding model
|
|
73
|
-
ModelUsage: new ai_1.ModelUsage(response.usage.prompt_tokens, response.usage.completion_tokens),
|
|
74
|
-
data: response.data[0].embedding
|
|
75
|
-
};
|
|
76
|
-
}
|
|
77
64
|
/**
|
|
78
65
|
* Returns a list of available models
|
|
79
66
|
* @returns {Promise<AvailableModelInfo>}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mistral.js","sourceRoot":"","sources":["../../src/models/mistral.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,
|
|
1
|
+
{"version":3,"file":"mistral.js","sourceRoot":"","sources":["../../src/models/mistral.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAAyJ;AACzJ,mDAAuD;AACvD,mDAAgD;AAIzC,IAAM,UAAU,kBAAhB,MAAM,UAAW,SAAQ,YAAO;IAEnC,YAAY,MAAc;QACtB,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,YAAU,CAAC,OAAO,EAAC,CAAC;YACrB,YAAU,CAAC,OAAO,GAAG,IAAI,6BAAa,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QACvD,CAAC;IACL,CAAC;IAED,IAAW,MAAM,KAAmB,OAAO,YAAU,CAAC,OAAO,CAAC,CAAA,CAAC;IAExD,KAAK,CAAC,cAAc,CAAC,MAAyB;QACjD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACxC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC5B,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAE3B,IAAI,OAAO,GAAuB,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAoC,EAAE,EAAE;YAChG,MAAM,GAAG,GAAqB;gBAC1B,OAAO,EAAE;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;iBAClC;gBACD,aAAa,EAAE,MAAM,CAAC,aAAa;gBACnC,KAAK,EAAE,MAAM,CAAC,KAAK;aACtB,CAAC;YACF,OAAO,GAAG,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,OAAO;YACH,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,IAAI;YAChB,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;YACpD,IAAI,EAAE;gBACF,OAAO,EAAE,OAAO;gBAChB,KAAK,EAAE;oBACH,WAAW,EAAE,YAAY,CAAC,KAAK,CAAC,YAAY;oBAC5C,YAAY,EAAE,YAAY,CAAC,KAAK,CAAC,aAAa;oBAC9C,gBAAgB,EAAE,YAAY,CAAC,KAAK,CAAC,iBAAiB;iBACzD;aACJ;YACD,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,IAAI;SAClB,CAAA;IAEL,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,MAAuB;QAC9C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,MAAsB;QAC5C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,UAAU;QACnB,MAAM,kBAAkB,GAAuB,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9E,OAAO,kBAAkB,CAAC;IAC9B,CAAC;CACJ,CAAA;AAnEY,gCAAU;qBAAV,UAAU;IADtB,IAAA,sBAAa,EAAC,YAAO,EAAE,YAAY,CAAC;GACxB,UAAU,CAmEtB;AAED,MAAa,iBAAkB,SAAQ,eAAU;CAEhD;AAFD,8CAEC;AAED,SAAgB,cAAc;IAC1B,iFAAiF;AACrF,CAAC;AAFD,wCAEC"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { ListModelsResponse, ChatCompletetionRequest, ChatCompletionResponse, EmbeddingResponse } from '../generic/mistral.types';
|
|
2
2
|
/**
|
|
3
3
|
* A simple and lightweight client for the Mistral API
|
|
4
4
|
* @param {*} apiKey can be set as an environment variable MISTRAL_API_KEY,
|
|
@@ -94,88 +94,4 @@ export declare class MistralClient {
|
|
|
94
94
|
*/
|
|
95
95
|
embeddings(model: string, input: string): Promise<EmbeddingResponse>;
|
|
96
96
|
}
|
|
97
|
-
|
|
98
|
-
id: string;
|
|
99
|
-
object: 'model_permission';
|
|
100
|
-
created: number;
|
|
101
|
-
allow_create_engine: boolean;
|
|
102
|
-
allow_sampling: boolean;
|
|
103
|
-
allow_logprobs: boolean;
|
|
104
|
-
allow_search_indices: boolean;
|
|
105
|
-
allow_view: boolean;
|
|
106
|
-
allow_fine_tuning: boolean;
|
|
107
|
-
organization: string;
|
|
108
|
-
group: string | null;
|
|
109
|
-
is_blocking: boolean;
|
|
110
|
-
}
|
|
111
|
-
export interface Model {
|
|
112
|
-
id: string;
|
|
113
|
-
object: 'model';
|
|
114
|
-
created: number;
|
|
115
|
-
owned_by: string;
|
|
116
|
-
root: string | null;
|
|
117
|
-
parent: string | null;
|
|
118
|
-
permission: ModelPermission[];
|
|
119
|
-
}
|
|
120
|
-
export interface ListModelsResponse {
|
|
121
|
-
object: 'list';
|
|
122
|
-
data: Model[];
|
|
123
|
-
}
|
|
124
|
-
export interface TokenUsage {
|
|
125
|
-
prompt_tokens: number;
|
|
126
|
-
completion_tokens: number;
|
|
127
|
-
total_tokens: number;
|
|
128
|
-
}
|
|
129
|
-
export type ChatCompletetionRequest = {
|
|
130
|
-
model: string;
|
|
131
|
-
messages: ChatMessage[];
|
|
132
|
-
temperature?: number;
|
|
133
|
-
max_tokens?: number;
|
|
134
|
-
top_p?: number;
|
|
135
|
-
random_seed?: number;
|
|
136
|
-
stream?: boolean;
|
|
137
|
-
safe_prompt?: boolean;
|
|
138
|
-
};
|
|
139
|
-
export interface ChatCompletionResponseChoice {
|
|
140
|
-
index: number;
|
|
141
|
-
message: {
|
|
142
|
-
role: string;
|
|
143
|
-
content: string;
|
|
144
|
-
};
|
|
145
|
-
finish_reason: string;
|
|
146
|
-
}
|
|
147
|
-
export interface ChatCompletionResponseChunkChoice {
|
|
148
|
-
index: number;
|
|
149
|
-
delta: {
|
|
150
|
-
role?: string;
|
|
151
|
-
content?: string;
|
|
152
|
-
};
|
|
153
|
-
finish_reason: string;
|
|
154
|
-
}
|
|
155
|
-
export interface ChatCompletionResponse {
|
|
156
|
-
id: string;
|
|
157
|
-
object: 'chat.completion';
|
|
158
|
-
created: number;
|
|
159
|
-
model: string;
|
|
160
|
-
choices: ChatCompletionResponseChoice[];
|
|
161
|
-
usage: TokenUsage;
|
|
162
|
-
}
|
|
163
|
-
export interface ChatCompletionResponseChunk {
|
|
164
|
-
id: string;
|
|
165
|
-
object: 'chat.completion.chunk';
|
|
166
|
-
created: number;
|
|
167
|
-
model: string;
|
|
168
|
-
choices: ChatCompletionResponseChunkChoice[];
|
|
169
|
-
}
|
|
170
|
-
export interface Embedding {
|
|
171
|
-
id: string;
|
|
172
|
-
object: 'embedding';
|
|
173
|
-
embedding: number[];
|
|
174
|
-
}
|
|
175
|
-
export interface EmbeddingResponse {
|
|
176
|
-
id: string;
|
|
177
|
-
object: 'list';
|
|
178
|
-
data: Embedding[];
|
|
179
|
-
model: string;
|
|
180
|
-
usage: TokenUsage;
|
|
181
|
-
}
|
|
97
|
+
//# sourceMappingURL=mistralClient.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistralClient.d.ts","sourceRoot":"","sources":["../../src/models/mistralClient.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAMlI;;;;;GAKG;AACH,qBAAa,aAAa;IAEtB,OAAO,CAAC,kBAAkB,CAA6B;IACvD,OAAO,CAAC,QAAQ,CAA4B;IAE5C,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,WAAW,CAAC;gBAEb,MAAM,EAAE;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAC;IAmB1D;;;;;;OAMG;YACW,OAAO;IAiBrB;;;;;;;;;;;OAWG;IACH,0BAA0B;;;;;;;;;MAoBxB;IAGF;;;OAGG;IACU,UAAU,IAAI,OAAO,CAAC,kBAAkB,CAAC;IAKtD;;;;;;;;;;;OAWG;IACU,IAAI,CAAC,MAAM,EAAE,uBAAuB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAQnF;;;;;;;;;;;OAWG;IACH,UAAU;;;;;;;;6CAqCR;IAEF;;;;;;;OAOG;IACU,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAUlF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mistralClient.js","sourceRoot":"","sources":["../../src/models/mistralClient.ts"],"names":[],"mappings":";;;;;;
|
|
1
|
+
{"version":3,"file":"mistralClient.js","sourceRoot":"","sources":["../../src/models/mistralClient.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAC1B,8DAAqC;AAGrC,oDAAoD;AACpD,8BAA8B;AAC9B,6CAA6C;AAE7C;;;;;GAKG;AACH,MAAa,aAAa;IAStB,YAAY,MAA4C;QAPhD,uBAAkB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAC/C,aAAQ,GAAG,wBAAwB,CAAC;QAiD9C;;;;;;;;;;;WAWG;QACH,+BAA0B,GAAG,UAC3B,KAAK,EACL,QAAQ,EACR,WAAW,EACX,SAAS,EACT,IAAI,EACJ,UAAU,EACV,MAAM,EACN,QAAQ;YAER,OAAO;gBACL,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,QAAQ;gBAClB,WAAW,EAAE,WAAW,IAAI,SAAS;gBACrC,UAAU,EAAE,SAAS,IAAI,SAAS;gBAClC,KAAK,EAAE,IAAI,IAAI,SAAS;gBACxB,WAAW,EAAE,UAAU,IAAI,SAAS;gBACpC,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,WAAW,EAAE,QAAQ,IAAI,SAAS;aACnC,CAAC;QACJ,CAAC,CAAC;QAgCF;;;;;;;;;;;WAWG;QACH,eAAU,GAAG,KAAK,SAAS,CAAC,EAAE,EAC5B,KAAK,EACL,QAAQ,EACR,WAAW,EACX,SAAS,EACT,IAAI,EACJ,UAAU,EACV,QAAQ,EAAC;YACT,MAAM,OAAO,GAAG,IAAI,CAAC,0BAA0B,CAC7C,KAAK,EACL,QAAQ,EACR,WAAW,EACX,SAAS,EACT,IAAI,EACJ,UAAU,EACV,IAAI,EACJ,QAAQ,CACT,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAClC,MAAM,EAAE,qBAAqB,EAAE,OAAO,CACvC,CAAC;YAEF,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACnC,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnD,+BAA+B;gBAC/B,MAAM,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3C,4BAA4B;gBAC5B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;oBACnC,mDAAmD;oBACnD,IAAI,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wBAClC,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAChD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;4BAC3B,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;wBAC9B,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QA3JI,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;QAE3D,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;QAErC,IAAA,qBAAU,EAAC,eAAK,EAAE;YAClB,OAAO,EAAE,CAAC;YACV,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE;gBACtB,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACnE,CAAC;YAED,UAAU,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;gBAC9B,OAAO,CAAC,KAAK,CAAC,kBAAkB,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;gBACrD,OAAO,UAAU,GAAG,GAAG,CAAC;YAC5B,CAAC;SACA,CAAC,CAAC;IACP,CAAC;IAEH;;;;;;OAMG;IACK,KAAK,CAAC,OAAO,CAAI,MAAc,EAAE,IAAY,EAAE,OAAa;QAElE,MAAM,QAAQ,GAAG,MAAM,IAAA,eAAK,EAAC;YACzB,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,EAAE;YAC/B,IAAI,EAAE,OAAO,IAAI,EAAE;YACnB,OAAO,EAAE;gBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACzC;YACD,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;SAClD,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACjB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,OAAO,KAAK,CAAC,QAAQ,CAAC;QACxB,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACzB,CAAC;IAqCD;;;OAGG;IACI,KAAK,CAAC,UAAU;QACrB,MAAM,QAAQ,GAAuB,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAC5E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;;;;;OAWG;IACI,KAAK,CAAC,IAAI,CAAC,MAA+B;QAC/C,MAAM,QAAQ,GAA2B,MAAM,IAAI,CAAC,OAAO,CACzD,MAAM,EAAE,qBAAqB,EAAE,MAAM,CACtC,CAAC;QAEF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAqDD;;;;;;;OAOG;IACI,KAAK,CAAC,UAAU,CAAC,KAAa,EAAE,KAAa;QAClD,MAAM,OAAO,GAAG;YACZ,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,KAAK;SACb,CAAC;QACF,MAAM,QAAQ,GAAsB,MAAM,IAAI,CAAC,OAAO,CACpD,MAAM,EAAE,eAAe,EAAE,OAAO,CACjC,CAAC;QACF,OAAO,QAAQ,CAAC;IACpB,CAAC;CACF;AAzLD,sCAyLC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { EmbedParams, EmbedResult, Embeddings } from "@memberjunction/ai";
|
|
2
|
+
import { MistralClient } from './mistralClient';
|
|
3
|
+
export declare class MistralEmbedding extends Embeddings {
|
|
4
|
+
static _client: MistralClient;
|
|
5
|
+
constructor(apiKey: string);
|
|
6
|
+
EmbedText(params: EmbedParams): Promise<EmbedResult>;
|
|
7
|
+
}
|
|
8
|
+
//# sourceMappingURL=mistralEmbedding.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistralEmbedding.d.ts","sourceRoot":"","sources":["../../src/models/mistralEmbedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,UAAU,EAAc,MAAM,oBAAoB,CAAC;AAEtF,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAIhD,qBACa,gBAAiB,SAAQ,UAAU;IAC5C,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC;gBAElB,MAAM,EAAE,MAAM;IAOb,SAAS,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;CASpE"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var MistralEmbedding_1;
|
|
9
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
10
|
+
exports.MistralEmbedding = void 0;
|
|
11
|
+
const ai_1 = require("@memberjunction/ai");
|
|
12
|
+
const global_1 = require("@memberjunction/global");
|
|
13
|
+
const mistralClient_1 = require("./mistralClient");
|
|
14
|
+
let MistralEmbedding = MistralEmbedding_1 = class MistralEmbedding extends ai_1.Embeddings {
|
|
15
|
+
constructor(apiKey) {
|
|
16
|
+
super(apiKey);
|
|
17
|
+
if (!MistralEmbedding_1._client) {
|
|
18
|
+
MistralEmbedding_1._client = new mistralClient_1.MistralClient({ apiKey });
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
async EmbedText(params) {
|
|
22
|
+
const response = await MistralEmbedding_1._client.embeddings(params.model, params.text);
|
|
23
|
+
return {
|
|
24
|
+
object: 'object',
|
|
25
|
+
model: params.model || "mistral-embed", //hard coded for now as theres only one available embedding model
|
|
26
|
+
ModelUsage: new ai_1.ModelUsage(response.usage.prompt_tokens, response.usage.completion_tokens),
|
|
27
|
+
data: response.data[0].embedding
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
};
|
|
31
|
+
exports.MistralEmbedding = MistralEmbedding;
|
|
32
|
+
exports.MistralEmbedding = MistralEmbedding = MistralEmbedding_1 = __decorate([
|
|
33
|
+
(0, global_1.RegisterClass)(ai_1.Embeddings, 'MistralEmbedding')
|
|
34
|
+
], MistralEmbedding);
|
|
35
|
+
//# sourceMappingURL=mistralEmbedding.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mistralEmbedding.js","sourceRoot":"","sources":["../../src/models/mistralEmbedding.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAAsF;AACtF,mDAAuD;AACvD,mDAAgD;AAKzC,IAAM,gBAAgB,wBAAtB,MAAM,gBAAiB,SAAQ,eAAU;IAG5C,YAAY,MAAc;QACtB,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,kBAAgB,CAAC,OAAO,EAAC,CAAC;YAC3B,kBAAgB,CAAC,OAAO,GAAG,IAAI,6BAAa,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,MAAmB;QACtC,MAAM,QAAQ,GAAsB,MAAM,kBAAgB,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACzG,OAAO;YACH,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE,iEAAiE;YACzG,UAAU,EAAE,IAAI,eAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC;YAC1F,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;SACnC,CAAA;IACL,CAAC;CACJ,CAAA;AAnBY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,sBAAa,EAAC,eAAU,EAAE,kBAAkB,CAAC;GACjC,gBAAgB,CAmB5B"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@memberjunction/ai-mistral",
|
|
3
|
-
"version": "0.9.
|
|
3
|
+
"version": "0.9.18",
|
|
4
4
|
"description": "MemberJunction Wrapper for Mistral AI's AI Models",
|
|
5
5
|
"main": "dist/index.js",
|
|
6
6
|
"types": "dist/index.d.ts",
|
|
@@ -19,7 +19,7 @@
|
|
|
19
19
|
"typescript": "^5.3.3"
|
|
20
20
|
},
|
|
21
21
|
"dependencies": {
|
|
22
|
-
"@memberjunction/ai": "^0.9.
|
|
23
|
-
"@memberjunction/global": "^0.9.
|
|
22
|
+
"@memberjunction/ai": "^0.9.160",
|
|
23
|
+
"@memberjunction/global": "^0.9.153"
|
|
24
24
|
}
|
|
25
25
|
}
|