langchain 0.0.17 → 0.0.19
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/embeddings/openai.d.ts +0 -7
- package/dist/embeddings/openai.js +2 -12
- package/dist/embeddings/openai.js.map +1 -1
- package/dist/llms/index.d.ts +1 -0
- package/dist/llms/index.js +1 -0
- package/dist/llms/index.js.map +1 -1
- package/dist/llms/openai-chat.d.ts +148 -0
- package/dist/llms/openai-chat.js +262 -0
- package/dist/llms/openai-chat.js.map +1 -0
- package/dist/llms/openai.d.ts +3 -8
- package/dist/llms/openai.js +23 -22
- package/dist/llms/openai.js.map +1 -1
- package/dist/llms/tests/openai-chat.int.test.d.ts +1 -0
- package/dist/llms/tests/openai-chat.int.test.js +36 -0
- package/dist/llms/tests/openai-chat.int.test.js.map +1 -0
- package/dist/llms/tests/openai.int.test.js +4 -1
- package/dist/llms/tests/openai.int.test.js.map +1 -1
- package/dist/vectorstores/tests/pinecone.int.test.js +1 -0
- package/dist/vectorstores/tests/pinecone.int.test.js.map +1 -1
- package/dist/vectorstores/tests/pinecone.test.js +1 -0
- package/dist/vectorstores/tests/pinecone.test.js.map +1 -1
- package/package.json +2 -6
|
@@ -1,5 +1,3 @@
|
|
|
1
|
-
import type { Configuration as ConfigurationT, OpenAIApi as OpenAIApiT } from "openai";
|
|
2
|
-
import type fetchAdapterT from "../util/axios-fetch-adapter.js";
|
|
3
1
|
import { Embeddings } from "./base.js";
|
|
4
2
|
interface ModelParams {
|
|
5
3
|
modelName: string;
|
|
@@ -19,10 +17,5 @@ export declare class OpenAIEmbeddings extends Embeddings implements ModelParams
|
|
|
19
17
|
embedDocuments(texts: string[]): Promise<number[][]>;
|
|
20
18
|
embedQuery(text: string): Promise<number[]>;
|
|
21
19
|
private embeddingWithRetry;
|
|
22
|
-
static imports(): Promise<{
|
|
23
|
-
Configuration: typeof ConfigurationT;
|
|
24
|
-
OpenAIApi: typeof OpenAIApiT;
|
|
25
|
-
fetchAdapter: typeof fetchAdapterT;
|
|
26
|
-
}>;
|
|
27
20
|
}
|
|
28
21
|
export {};
|
|
@@ -1,4 +1,6 @@
|
|
|
1
|
+
import { Configuration, OpenAIApi } from "openai";
|
|
1
2
|
import { backOff } from "exponential-backoff";
|
|
3
|
+
import fetchAdapter from "../util/axios-fetch-adapter.js";
|
|
2
4
|
import { chunkArray } from "../util/index.js";
|
|
3
5
|
import { Embeddings } from "./base.js";
|
|
4
6
|
export class OpenAIEmbeddings extends Embeddings {
|
|
@@ -67,7 +69,6 @@ export class OpenAIEmbeddings extends Embeddings {
|
|
|
67
69
|
}
|
|
68
70
|
async embeddingWithRetry(request) {
|
|
69
71
|
if (!this.client) {
|
|
70
|
-
const { Configuration, OpenAIApi, fetchAdapter } = await OpenAIEmbeddings.imports();
|
|
71
72
|
const clientConfig = new Configuration({
|
|
72
73
|
apiKey: this.apiKey,
|
|
73
74
|
baseOptions: { adapter: fetchAdapter },
|
|
@@ -81,16 +82,5 @@ export class OpenAIEmbeddings extends Embeddings {
|
|
|
81
82
|
numOfAttempts: this.maxRetries,
|
|
82
83
|
});
|
|
83
84
|
}
|
|
84
|
-
static async imports() {
|
|
85
|
-
try {
|
|
86
|
-
const { Configuration, OpenAIApi } = await import("openai");
|
|
87
|
-
const { default: fetchAdapter } = await import("../util/axios-fetch-adapter.js");
|
|
88
|
-
return { Configuration, OpenAIApi, fetchAdapter };
|
|
89
|
-
}
|
|
90
|
-
catch (err) {
|
|
91
|
-
console.error(err);
|
|
92
|
-
throw new Error("Please install openai as a dependency with, e.g. `npm install -S openai`");
|
|
93
|
-
}
|
|
94
|
-
}
|
|
95
85
|
}
|
|
96
86
|
//# sourceMappingURL=openai.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/embeddings/openai.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/embeddings/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,SAAS,EAA0B,MAAM,QAAQ,CAAC;AAC1E,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAMvC,MAAM,OAAO,gBAAiB,SAAQ,UAAU;IAW9C,YACE,MAKC;QAED,KAAK,EAAE,CAAC;QAlBV;;;;mBAAY,wBAAwB;WAAC;QAErC;;;;mBAAY,EAAE;WAAC;QAEf;;;;mBAAa,CAAC;WAAC;QAEf;;;;;WAAuB;QAEvB;;;;;WAA0B;QAYxB,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAe;QAClC,MAAM,UAAU,GAAG,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAErD,MAAM,UAAU,GAAG,EAAE,CAAC;QAEtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC5B,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC;gBAC7C,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,KAAK;aACN,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBACxC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACzC;SACF;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC;YAC7C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChC,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,OAA+B;QAC9D,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAChB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,WAAW,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACvC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAC3C;QACD,MAAM,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QACzE,OAAO,OAAO,CAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;SAC/B,CAAC,CAAC;IACL,CAAC;CACF"}
|
package/dist/llms/index.d.ts
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
export { BaseLLM, LLM, SerializedLLM } from "./base.js";
|
|
2
2
|
export { OpenAI, PromptLayerOpenAI } from "./openai.js";
|
|
3
|
+
export { OpenAIChat } from "./openai-chat.js";
|
|
3
4
|
export { Cohere } from "./cohere.js";
|
|
4
5
|
export { HuggingFaceInference } from "./hf.js";
|
|
5
6
|
export { loadLLM } from "./load.js";
|
package/dist/llms/index.js
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
export { BaseLLM, LLM } from "./base.js";
|
|
2
2
|
export { OpenAI, PromptLayerOpenAI } from "./openai.js";
|
|
3
|
+
export { OpenAIChat } from "./openai-chat.js";
|
|
3
4
|
export { Cohere } from "./cohere.js";
|
|
4
5
|
export { HuggingFaceInference } from "./hf.js";
|
|
5
6
|
export { loadLLM } from "./load.js";
|
package/dist/llms/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/llms/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,GAAG,EAAiB,MAAM,WAAW,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/llms/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,GAAG,EAAiB,MAAM,WAAW,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
|
|
@@ -0,0 +1,148 @@
|
|
|
1
|
+
import { ChatCompletionRequestMessage, CreateChatCompletionRequest, ConfigurationParameters } from "openai";
|
|
2
|
+
import { LLM } from "./base.js";
|
|
3
|
+
import { LLMCallbackManager } from "./index.js";
|
|
4
|
+
interface ModelParams {
|
|
5
|
+
/** Sampling temperature to use, between 0 and 2, defaults to 1 */
|
|
6
|
+
temperature: number;
|
|
7
|
+
/** Total probability mass of tokens to consider at each step, between 0 and 1, defaults to 1 */
|
|
8
|
+
topP: number;
|
|
9
|
+
/** Penalizes repeated tokens according to frequency */
|
|
10
|
+
frequencyPenalty: number;
|
|
11
|
+
/** Penalizes repeated tokens */
|
|
12
|
+
presencePenalty: number;
|
|
13
|
+
/** Number of chat completions to generate for each prompt */
|
|
14
|
+
n: number;
|
|
15
|
+
/** Dictionary used to adjust the probability of specific tokens being generated */
|
|
16
|
+
logitBias?: Record<string, number>;
|
|
17
|
+
/** Whether to stream the results or not */
|
|
18
|
+
streaming: boolean;
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Input to OpenAI class.
|
|
22
|
+
* @augments ModelParams
|
|
23
|
+
*/
|
|
24
|
+
interface OpenAIInput extends ModelParams {
|
|
25
|
+
/** Model name to use */
|
|
26
|
+
modelName: string;
|
|
27
|
+
/** ChatGPT messages to pass as a prefix to the prompt */
|
|
28
|
+
prefixMessages?: ChatCompletionRequestMessage[];
|
|
29
|
+
/** Holds any additional parameters that are valid to pass to {@link
|
|
30
|
+
* https://platform.openai.com/docs/api-reference/completions/create |
|
|
31
|
+
* `openai.create`} that are not explicitly specified on this class.
|
|
32
|
+
*/
|
|
33
|
+
modelKwargs?: Kwargs;
|
|
34
|
+
/** Maximum number of retries to make when generating */
|
|
35
|
+
maxRetries: number;
|
|
36
|
+
/** List of stop words to use when generating */
|
|
37
|
+
stop?: string[];
|
|
38
|
+
}
|
|
39
|
+
type Kwargs = Record<string, any>;
|
|
40
|
+
/**
|
|
41
|
+
* Wrapper around OpenAI large language models that use the Chat endpoint.
|
|
42
|
+
*
|
|
43
|
+
* To use you should have the `openai` package installed, with the
|
|
44
|
+
* `OPENAI_API_KEY` environment variable set.
|
|
45
|
+
*
|
|
46
|
+
* @remarks
|
|
47
|
+
* Any parameters that are valid to be passed to {@link
|
|
48
|
+
* https://platform.openai.com/docs/api-reference/chat/create |
|
|
49
|
+
* `openai.createCompletion`} can be passed through {@link modelKwargs}, even
|
|
50
|
+
* if not explicitly available on this class.
|
|
51
|
+
*
|
|
52
|
+
* @augments BaseLLM
|
|
53
|
+
* @augments OpenAIInput
|
|
54
|
+
*/
|
|
55
|
+
export declare class OpenAIChat extends LLM implements OpenAIInput {
|
|
56
|
+
temperature: number;
|
|
57
|
+
topP: number;
|
|
58
|
+
frequencyPenalty: number;
|
|
59
|
+
presencePenalty: number;
|
|
60
|
+
n: number;
|
|
61
|
+
logitBias?: Record<string, number>;
|
|
62
|
+
modelName: string;
|
|
63
|
+
prefixMessages?: ChatCompletionRequestMessage[];
|
|
64
|
+
modelKwargs?: Kwargs;
|
|
65
|
+
maxRetries: number;
|
|
66
|
+
stop?: string[];
|
|
67
|
+
streaming: boolean;
|
|
68
|
+
private batchClient;
|
|
69
|
+
private streamingClient;
|
|
70
|
+
private clientConfig;
|
|
71
|
+
constructor(fields?: Partial<OpenAIInput> & {
|
|
72
|
+
callbackManager?: LLMCallbackManager;
|
|
73
|
+
concurrency?: number;
|
|
74
|
+
cache?: boolean;
|
|
75
|
+
verbose?: boolean;
|
|
76
|
+
openAIApiKey?: string;
|
|
77
|
+
}, configuration?: ConfigurationParameters);
|
|
78
|
+
/**
|
|
79
|
+
* Get the parameters used to invoke the model
|
|
80
|
+
*/
|
|
81
|
+
invocationParams(): Omit<CreateChatCompletionRequest, "messages"> & Kwargs;
|
|
82
|
+
_identifyingParams(): {
|
|
83
|
+
apiKey?: string | Promise<string> | ((name: string) => string) | ((name: string) => Promise<string>) | undefined;
|
|
84
|
+
organization?: string | undefined;
|
|
85
|
+
username?: string | undefined;
|
|
86
|
+
password?: string | undefined;
|
|
87
|
+
accessToken?: string | Promise<string> | ((name?: string | undefined, scopes?: string[] | undefined) => string) | ((name?: string | undefined, scopes?: string[] | undefined) => Promise<string>) | undefined;
|
|
88
|
+
basePath?: string | undefined;
|
|
89
|
+
baseOptions?: any;
|
|
90
|
+
formDataCtor?: (new () => any) | undefined;
|
|
91
|
+
stop?: import("openai").CreateChatCompletionRequestStop | undefined;
|
|
92
|
+
temperature?: number | null | undefined;
|
|
93
|
+
n?: number | null | undefined;
|
|
94
|
+
stream?: boolean | null | undefined;
|
|
95
|
+
model: string;
|
|
96
|
+
top_p?: number | null | undefined;
|
|
97
|
+
presence_penalty?: number | null | undefined;
|
|
98
|
+
frequency_penalty?: number | null | undefined;
|
|
99
|
+
logit_bias?: object | null | undefined;
|
|
100
|
+
user?: string | undefined;
|
|
101
|
+
model_name: string;
|
|
102
|
+
};
|
|
103
|
+
/**
|
|
104
|
+
* Get the identifying parameters for the model
|
|
105
|
+
*/
|
|
106
|
+
identifyingParams(): {
|
|
107
|
+
apiKey?: string | Promise<string> | ((name: string) => string) | ((name: string) => Promise<string>) | undefined;
|
|
108
|
+
organization?: string | undefined;
|
|
109
|
+
username?: string | undefined;
|
|
110
|
+
password?: string | undefined;
|
|
111
|
+
accessToken?: string | Promise<string> | ((name?: string | undefined, scopes?: string[] | undefined) => string) | ((name?: string | undefined, scopes?: string[] | undefined) => Promise<string>) | undefined;
|
|
112
|
+
basePath?: string | undefined;
|
|
113
|
+
baseOptions?: any;
|
|
114
|
+
formDataCtor?: (new () => any) | undefined;
|
|
115
|
+
stop?: import("openai").CreateChatCompletionRequestStop | undefined;
|
|
116
|
+
temperature?: number | null | undefined;
|
|
117
|
+
n?: number | null | undefined;
|
|
118
|
+
stream?: boolean | null | undefined;
|
|
119
|
+
model: string;
|
|
120
|
+
top_p?: number | null | undefined;
|
|
121
|
+
presence_penalty?: number | null | undefined;
|
|
122
|
+
frequency_penalty?: number | null | undefined;
|
|
123
|
+
logit_bias?: object | null | undefined;
|
|
124
|
+
user?: string | undefined;
|
|
125
|
+
model_name: string;
|
|
126
|
+
};
|
|
127
|
+
private formatMessages;
|
|
128
|
+
/**
|
|
129
|
+
* Call out to OpenAI's endpoint with k unique prompts
|
|
130
|
+
*
|
|
131
|
+
* @param prompt - The prompt to pass into the model.
|
|
132
|
+
* @param [stop] - Optional list of stop words to use when generating.
|
|
133
|
+
*
|
|
134
|
+
* @returns The full LLM output.
|
|
135
|
+
*
|
|
136
|
+
* @example
|
|
137
|
+
* ```ts
|
|
138
|
+
* import { OpenAI } from "langchain/llms";
|
|
139
|
+
* const openai = new OpenAI();
|
|
140
|
+
* const response = await openai.generate(["Tell me a joke."]);
|
|
141
|
+
* ```
|
|
142
|
+
*/
|
|
143
|
+
_call(prompt: string, stop?: string[]): Promise<string>;
|
|
144
|
+
/** @ignore */
|
|
145
|
+
completionWithRetry(request: CreateChatCompletionRequest): Promise<import("axios").AxiosResponse<import("openai").CreateChatCompletionResponse, any>>;
|
|
146
|
+
_llmType(): string;
|
|
147
|
+
}
|
|
148
|
+
export {};
|
|
@@ -0,0 +1,262 @@
|
|
|
1
|
+
import { Configuration, OpenAIApi, } from "openai";
|
|
2
|
+
import { createParser } from "eventsource-parser";
|
|
3
|
+
import { backOff } from "exponential-backoff";
|
|
4
|
+
import fetchAdapter from "../util/axios-fetch-adapter.js";
|
|
5
|
+
import { LLM } from "./base.js";
|
|
6
|
+
/**
|
|
7
|
+
* Wrapper around OpenAI large language models that use the Chat endpoint.
|
|
8
|
+
*
|
|
9
|
+
* To use you should have the `openai` package installed, with the
|
|
10
|
+
* `OPENAI_API_KEY` environment variable set.
|
|
11
|
+
*
|
|
12
|
+
* @remarks
|
|
13
|
+
* Any parameters that are valid to be passed to {@link
|
|
14
|
+
* https://platform.openai.com/docs/api-reference/chat/create |
|
|
15
|
+
* `openai.createCompletion`} can be passed through {@link modelKwargs}, even
|
|
16
|
+
* if not explicitly available on this class.
|
|
17
|
+
*
|
|
18
|
+
* @augments BaseLLM
|
|
19
|
+
* @augments OpenAIInput
|
|
20
|
+
*/
|
|
21
|
+
export class OpenAIChat extends LLM {
|
|
22
|
+
constructor(fields, configuration) {
|
|
23
|
+
super(fields?.callbackManager, fields?.verbose, fields?.concurrency, fields?.cache);
|
|
24
|
+
Object.defineProperty(this, "temperature", {
|
|
25
|
+
enumerable: true,
|
|
26
|
+
configurable: true,
|
|
27
|
+
writable: true,
|
|
28
|
+
value: 1
|
|
29
|
+
});
|
|
30
|
+
Object.defineProperty(this, "topP", {
|
|
31
|
+
enumerable: true,
|
|
32
|
+
configurable: true,
|
|
33
|
+
writable: true,
|
|
34
|
+
value: 1
|
|
35
|
+
});
|
|
36
|
+
Object.defineProperty(this, "frequencyPenalty", {
|
|
37
|
+
enumerable: true,
|
|
38
|
+
configurable: true,
|
|
39
|
+
writable: true,
|
|
40
|
+
value: 0
|
|
41
|
+
});
|
|
42
|
+
Object.defineProperty(this, "presencePenalty", {
|
|
43
|
+
enumerable: true,
|
|
44
|
+
configurable: true,
|
|
45
|
+
writable: true,
|
|
46
|
+
value: 0
|
|
47
|
+
});
|
|
48
|
+
Object.defineProperty(this, "n", {
|
|
49
|
+
enumerable: true,
|
|
50
|
+
configurable: true,
|
|
51
|
+
writable: true,
|
|
52
|
+
value: 1
|
|
53
|
+
});
|
|
54
|
+
Object.defineProperty(this, "logitBias", {
|
|
55
|
+
enumerable: true,
|
|
56
|
+
configurable: true,
|
|
57
|
+
writable: true,
|
|
58
|
+
value: void 0
|
|
59
|
+
});
|
|
60
|
+
Object.defineProperty(this, "modelName", {
|
|
61
|
+
enumerable: true,
|
|
62
|
+
configurable: true,
|
|
63
|
+
writable: true,
|
|
64
|
+
value: "gpt-3.5-turbo"
|
|
65
|
+
});
|
|
66
|
+
Object.defineProperty(this, "prefixMessages", {
|
|
67
|
+
enumerable: true,
|
|
68
|
+
configurable: true,
|
|
69
|
+
writable: true,
|
|
70
|
+
value: void 0
|
|
71
|
+
});
|
|
72
|
+
Object.defineProperty(this, "modelKwargs", {
|
|
73
|
+
enumerable: true,
|
|
74
|
+
configurable: true,
|
|
75
|
+
writable: true,
|
|
76
|
+
value: void 0
|
|
77
|
+
});
|
|
78
|
+
Object.defineProperty(this, "maxRetries", {
|
|
79
|
+
enumerable: true,
|
|
80
|
+
configurable: true,
|
|
81
|
+
writable: true,
|
|
82
|
+
value: 6
|
|
83
|
+
});
|
|
84
|
+
Object.defineProperty(this, "stop", {
|
|
85
|
+
enumerable: true,
|
|
86
|
+
configurable: true,
|
|
87
|
+
writable: true,
|
|
88
|
+
value: void 0
|
|
89
|
+
});
|
|
90
|
+
Object.defineProperty(this, "streaming", {
|
|
91
|
+
enumerable: true,
|
|
92
|
+
configurable: true,
|
|
93
|
+
writable: true,
|
|
94
|
+
value: false
|
|
95
|
+
});
|
|
96
|
+
// Used for non-streaming requests
|
|
97
|
+
Object.defineProperty(this, "batchClient", {
|
|
98
|
+
enumerable: true,
|
|
99
|
+
configurable: true,
|
|
100
|
+
writable: true,
|
|
101
|
+
value: void 0
|
|
102
|
+
});
|
|
103
|
+
// Used for streaming requests
|
|
104
|
+
Object.defineProperty(this, "streamingClient", {
|
|
105
|
+
enumerable: true,
|
|
106
|
+
configurable: true,
|
|
107
|
+
writable: true,
|
|
108
|
+
value: void 0
|
|
109
|
+
});
|
|
110
|
+
Object.defineProperty(this, "clientConfig", {
|
|
111
|
+
enumerable: true,
|
|
112
|
+
configurable: true,
|
|
113
|
+
writable: true,
|
|
114
|
+
value: void 0
|
|
115
|
+
});
|
|
116
|
+
const apiKey = fields?.openAIApiKey ?? process.env.OPENAI_API_KEY;
|
|
117
|
+
if (!apiKey) {
|
|
118
|
+
throw new Error("OpenAI API key not found");
|
|
119
|
+
}
|
|
120
|
+
this.modelName = fields?.modelName ?? this.modelName;
|
|
121
|
+
this.prefixMessages = fields?.prefixMessages ?? this.prefixMessages;
|
|
122
|
+
this.modelKwargs = fields?.modelKwargs ?? {};
|
|
123
|
+
this.maxRetries = fields?.maxRetries ?? this.maxRetries;
|
|
124
|
+
this.temperature = fields?.temperature ?? this.temperature;
|
|
125
|
+
this.topP = fields?.topP ?? this.topP;
|
|
126
|
+
this.frequencyPenalty = fields?.frequencyPenalty ?? this.frequencyPenalty;
|
|
127
|
+
this.presencePenalty = fields?.presencePenalty ?? this.presencePenalty;
|
|
128
|
+
this.n = fields?.n ?? this.n;
|
|
129
|
+
this.logitBias = fields?.logitBias;
|
|
130
|
+
this.stop = fields?.stop;
|
|
131
|
+
this.streaming = fields?.streaming ?? false;
|
|
132
|
+
if (this.streaming && this.n > 1) {
|
|
133
|
+
throw new Error("Cannot stream results when n > 1");
|
|
134
|
+
}
|
|
135
|
+
this.clientConfig = {
|
|
136
|
+
apiKey: fields?.openAIApiKey ?? process.env.OPENAI_API_KEY,
|
|
137
|
+
...configuration,
|
|
138
|
+
};
|
|
139
|
+
}
|
|
140
|
+
/**
|
|
141
|
+
* Get the parameters used to invoke the model
|
|
142
|
+
*/
|
|
143
|
+
invocationParams() {
|
|
144
|
+
return {
|
|
145
|
+
model: this.modelName,
|
|
146
|
+
temperature: this.temperature,
|
|
147
|
+
top_p: this.topP,
|
|
148
|
+
frequency_penalty: this.frequencyPenalty,
|
|
149
|
+
presence_penalty: this.presencePenalty,
|
|
150
|
+
n: this.n,
|
|
151
|
+
logit_bias: this.logitBias,
|
|
152
|
+
stop: this.stop,
|
|
153
|
+
stream: this.streaming,
|
|
154
|
+
...this.modelKwargs,
|
|
155
|
+
};
|
|
156
|
+
}
|
|
157
|
+
_identifyingParams() {
|
|
158
|
+
return {
|
|
159
|
+
model_name: this.modelName,
|
|
160
|
+
...this.invocationParams(),
|
|
161
|
+
...this.clientConfig,
|
|
162
|
+
};
|
|
163
|
+
}
|
|
164
|
+
/**
|
|
165
|
+
* Get the identifying parameters for the model
|
|
166
|
+
*/
|
|
167
|
+
identifyingParams() {
|
|
168
|
+
return {
|
|
169
|
+
model_name: this.modelName,
|
|
170
|
+
...this.invocationParams(),
|
|
171
|
+
...this.clientConfig,
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
formatMessages(prompt) {
|
|
175
|
+
const message = {
|
|
176
|
+
role: "user",
|
|
177
|
+
content: prompt,
|
|
178
|
+
};
|
|
179
|
+
return this.prefixMessages ? [...this.prefixMessages, message] : [message];
|
|
180
|
+
}
|
|
181
|
+
/**
|
|
182
|
+
* Call out to OpenAI's endpoint with k unique prompts
|
|
183
|
+
*
|
|
184
|
+
* @param prompt - The prompt to pass into the model.
|
|
185
|
+
* @param [stop] - Optional list of stop words to use when generating.
|
|
186
|
+
*
|
|
187
|
+
* @returns The full LLM output.
|
|
188
|
+
*
|
|
189
|
+
* @example
|
|
190
|
+
* ```ts
|
|
191
|
+
* import { OpenAI } from "langchain/llms";
|
|
192
|
+
* const openai = new OpenAI();
|
|
193
|
+
* const response = await openai.generate(["Tell me a joke."]);
|
|
194
|
+
* ```
|
|
195
|
+
*/
|
|
196
|
+
async _call(prompt, stop) {
|
|
197
|
+
if (this.stop && stop) {
|
|
198
|
+
throw new Error("Stop found in input and default params");
|
|
199
|
+
}
|
|
200
|
+
const params = this.invocationParams();
|
|
201
|
+
params.stop = stop ?? params.stop;
|
|
202
|
+
const { data } = await this.completionWithRetry({
|
|
203
|
+
...params,
|
|
204
|
+
messages: this.formatMessages(prompt),
|
|
205
|
+
});
|
|
206
|
+
let completion = "";
|
|
207
|
+
if (params.stream) {
|
|
208
|
+
completion = await new Promise((resolve, reject) => {
|
|
209
|
+
let innerCompletion = "";
|
|
210
|
+
const parser = createParser((event) => {
|
|
211
|
+
if (event.type === "event") {
|
|
212
|
+
if (event.data === "[DONE]") {
|
|
213
|
+
resolve(innerCompletion);
|
|
214
|
+
}
|
|
215
|
+
else {
|
|
216
|
+
const response = JSON.parse(event.data);
|
|
217
|
+
const part = response.choices[0];
|
|
218
|
+
if (part != null) {
|
|
219
|
+
innerCompletion += part.delta?.content ?? "";
|
|
220
|
+
this.callbackManager.handleNewToken?.(part.delta?.content ?? "", this.verbose);
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
}
|
|
224
|
+
});
|
|
225
|
+
// workaround for incorrect axios types
|
|
226
|
+
const stream = data;
|
|
227
|
+
stream.on("data", (data) => parser.feed(data.toString("utf-8")));
|
|
228
|
+
stream.on("error", (error) => reject(error));
|
|
229
|
+
});
|
|
230
|
+
}
|
|
231
|
+
else {
|
|
232
|
+
completion = data.choices[0].message?.content ?? "";
|
|
233
|
+
}
|
|
234
|
+
return completion;
|
|
235
|
+
}
|
|
236
|
+
/** @ignore */
|
|
237
|
+
async completionWithRetry(request) {
|
|
238
|
+
if (!request.stream && !this.batchClient) {
|
|
239
|
+
const clientConfig = new Configuration({
|
|
240
|
+
...this.clientConfig,
|
|
241
|
+
baseOptions: { adapter: fetchAdapter },
|
|
242
|
+
});
|
|
243
|
+
this.batchClient = new OpenAIApi(clientConfig);
|
|
244
|
+
}
|
|
245
|
+
if (request.stream && !this.streamingClient) {
|
|
246
|
+
const clientConfig = new Configuration(this.clientConfig);
|
|
247
|
+
this.streamingClient = new OpenAIApi(clientConfig);
|
|
248
|
+
}
|
|
249
|
+
const client = !request.stream ? this.batchClient : this.streamingClient;
|
|
250
|
+
const makeCompletionRequest = async () => client.createChatCompletion(request, request.stream ? { responseType: "stream" } : undefined);
|
|
251
|
+
return backOff(makeCompletionRequest, {
|
|
252
|
+
startingDelay: 4,
|
|
253
|
+
maxDelay: 10,
|
|
254
|
+
numOfAttempts: this.maxRetries,
|
|
255
|
+
// TODO(sean) pass custom retry function to check error types.
|
|
256
|
+
});
|
|
257
|
+
}
|
|
258
|
+
_llmType() {
|
|
259
|
+
return "openai";
|
|
260
|
+
}
|
|
261
|
+
}
|
|
262
|
+
//# sourceMappingURL=openai-chat.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-chat.js","sourceRoot":"","sources":["../../src/llms/openai-chat.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAIV,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAqDhC;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,UAAW,SAAQ,GAAG;IAiCjC,YACE,MAMC,EACD,aAAuC;QAEvC,KAAK,CACH,MAAM,EAAE,eAAe,EACvB,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,WAAW,EACnB,MAAM,EAAE,KAAK,CACd,CAAC;QA/CJ;;;;mBAAc,CAAC;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;;WAAmC;QAEnC;;;;mBAAY,eAAe;WAAC;QAE5B;;;;;WAAgD;QAEhD;;;;;WAAqB;QAErB;;;;mBAAa,CAAC;WAAC;QAEf;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB,kCAAkC;QAClC;;;;;WAA+B;QAE/B,8BAA8B;QAC9B;;;;;WAAmC;QAEnC;;;;;WAA8C;QAmB5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QACpE,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC;QAExD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM,EAAE,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1D,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,MAAc;QACnC,MAAM,OAAO,GAAiC;YAC5C,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM;SAChB,CAAC;QACF,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAAe;QACzC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;YAC9C,GAAG,MAAM;YACT,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;SACtC,CAAC,CAAC;QAEH,IAAI,UAAU,GAAG,EAAE,CAAC;QAEpB,IAAI,MAAM,CAAC,MAAM,EAAE;YACjB,UAAU,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACzD,IAAI,eAAe,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE;oBACpC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE;wBAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE;4BAC3B,OAAO,CAAC,eAAe,CAAC,CAAC;yBAC1B;6BAAM;4BACL,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAUrC,CAAC;4BAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;4BACjC,IAAI,IAAI,IAAI,IAAI,EAAE;gCAChB,eAAe,IAAI,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gCAE7C,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,CACnC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,EACzB,IAAI,CAAC,OAAO,CACb,CAAC;6BACH;yBACF;qBACF;gBACH,CAAC,CAAC,CAAC;gBAEH,uCAAuC;gBACvC,MAAM,MAAM,GAAG,IAAkC,CAAC;gBAClD,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CACjC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CACpC,CAAC;gBACF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;SACJ;aAAM;YACL,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;SACrD;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CAAC,OAAoC;QAC5D,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;YACxC,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACvC,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAChD;QACD,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;YAC3C,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SACpD;QACD,MAAM,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACzE,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE,CACvC,MAAM,CAAC,oBAAoB,CACzB,OAAO,EACP,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CACxD,CAAC;QACJ,OAAO,OAAO,CAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;YAC9B,8DAA8D;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
|
package/dist/llms/openai.d.ts
CHANGED
|
@@ -1,5 +1,4 @@
|
|
|
1
|
-
import
|
|
2
|
-
import type fetchAdapterT from "../util/axios-fetch-adapter.js";
|
|
1
|
+
import { CreateCompletionRequest, CreateCompletionResponse, ConfigurationParameters } from "openai";
|
|
3
2
|
import { BaseLLM } from "./base.js";
|
|
4
3
|
import { LLMResult, LLMCallbackManager } from "./index.js";
|
|
5
4
|
interface ModelParams {
|
|
@@ -75,7 +74,8 @@ export declare class OpenAI extends BaseLLM implements OpenAIInput {
|
|
|
75
74
|
maxRetries: number;
|
|
76
75
|
stop?: string[];
|
|
77
76
|
streaming: boolean;
|
|
78
|
-
private
|
|
77
|
+
private batchClient;
|
|
78
|
+
private streamingClient;
|
|
79
79
|
private clientConfig;
|
|
80
80
|
constructor(fields?: Partial<OpenAIInput> & {
|
|
81
81
|
callbackManager?: LLMCallbackManager;
|
|
@@ -164,11 +164,6 @@ export declare class OpenAI extends BaseLLM implements OpenAIInput {
|
|
|
164
164
|
/** @ignore */
|
|
165
165
|
completionWithRetry(request: CreateCompletionRequest): Promise<import("axios").AxiosResponse<CreateCompletionResponse, any>>;
|
|
166
166
|
_llmType(): string;
|
|
167
|
-
static imports(): Promise<{
|
|
168
|
-
Configuration: typeof ConfigurationT;
|
|
169
|
-
OpenAIApi: typeof OpenAIApiT;
|
|
170
|
-
fetchAdapter: typeof fetchAdapterT;
|
|
171
|
-
}>;
|
|
172
167
|
}
|
|
173
168
|
/**
|
|
174
169
|
* PromptLayer wrapper to OpenAI
|
package/dist/llms/openai.js
CHANGED
|
@@ -1,5 +1,7 @@
|
|
|
1
|
+
import { Configuration, OpenAIApi, } from "openai";
|
|
1
2
|
import { createParser } from "eventsource-parser";
|
|
2
3
|
import { backOff } from "exponential-backoff";
|
|
4
|
+
import fetchAdapter from "../util/axios-fetch-adapter.js";
|
|
3
5
|
import { chunkArray } from "../util/index.js";
|
|
4
6
|
import { BaseLLM } from "./base.js";
|
|
5
7
|
/**
|
|
@@ -104,7 +106,15 @@ export class OpenAI extends BaseLLM {
|
|
|
104
106
|
writable: true,
|
|
105
107
|
value: false
|
|
106
108
|
});
|
|
107
|
-
|
|
109
|
+
// Used for non-streaming requests
|
|
110
|
+
Object.defineProperty(this, "batchClient", {
|
|
111
|
+
enumerable: true,
|
|
112
|
+
configurable: true,
|
|
113
|
+
writable: true,
|
|
114
|
+
value: void 0
|
|
115
|
+
});
|
|
116
|
+
// Used for streaming requests
|
|
117
|
+
Object.defineProperty(this, "streamingClient", {
|
|
108
118
|
enumerable: true,
|
|
109
119
|
configurable: true,
|
|
110
120
|
writable: true,
|
|
@@ -262,17 +272,19 @@ export class OpenAI extends BaseLLM {
|
|
|
262
272
|
}
|
|
263
273
|
/** @ignore */
|
|
264
274
|
async completionWithRetry(request) {
|
|
265
|
-
if (!this.
|
|
266
|
-
const
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
275
|
+
if (!request.stream && !this.batchClient) {
|
|
276
|
+
const clientConfig = new Configuration({
|
|
277
|
+
...this.clientConfig,
|
|
278
|
+
baseOptions: { adapter: fetchAdapter },
|
|
279
|
+
});
|
|
280
|
+
this.batchClient = new OpenAIApi(clientConfig);
|
|
281
|
+
}
|
|
282
|
+
if (request.stream && !this.streamingClient) {
|
|
283
|
+
const clientConfig = new Configuration(this.clientConfig);
|
|
284
|
+
this.streamingClient = new OpenAIApi(clientConfig);
|
|
274
285
|
}
|
|
275
|
-
const
|
|
286
|
+
const client = !request.stream ? this.batchClient : this.streamingClient;
|
|
287
|
+
const makeCompletionRequest = async () => client.createCompletion(request, request.stream ? { responseType: "stream" } : undefined);
|
|
276
288
|
return backOff(makeCompletionRequest, {
|
|
277
289
|
startingDelay: 4,
|
|
278
290
|
maxDelay: 10,
|
|
@@ -283,17 +295,6 @@ export class OpenAI extends BaseLLM {
|
|
|
283
295
|
_llmType() {
|
|
284
296
|
return "openai";
|
|
285
297
|
}
|
|
286
|
-
static async imports() {
|
|
287
|
-
try {
|
|
288
|
-
const { Configuration, OpenAIApi } = await import("openai");
|
|
289
|
-
const { default: fetchAdapter } = await import("../util/axios-fetch-adapter.js");
|
|
290
|
-
return { Configuration, OpenAIApi, fetchAdapter };
|
|
291
|
-
}
|
|
292
|
-
catch (err) {
|
|
293
|
-
console.error(err);
|
|
294
|
-
throw new Error("Please install openai as a dependency with, e.g. `npm install -S openai`");
|
|
295
|
-
}
|
|
296
|
-
}
|
|
297
298
|
}
|
|
298
299
|
/**
|
|
299
300
|
* PromptLayer wrapper to OpenAI
|
package/dist/llms/openai.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llms/openai.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llms/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,SAAS,GAKV,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,YAAY,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAoEpC;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,MAAO,SAAQ,OAAO;IAqCjC,YACE,MAMC,EACD,aAAuC;QAEvC,KAAK,CACH,MAAM,EAAE,eAAe,EACvB,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,WAAW,EACnB,MAAM,EAAE,KAAK,CACd,CAAC;QAnDJ;;;;mBAAc,GAAG;WAAC;QAElB;;;;mBAAY,GAAG;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAS,CAAC;WAAC;QAEX;;;;;WAAmC;QAEnC;;;;mBAAY,kBAAkB;WAAC;QAE/B;;;;;WAAqB;QAErB;;;;mBAAY,EAAE;WAAC;QAEf;;;;mBAAa,CAAC;WAAC;QAEf;;;;;WAAgB;QAEhB;;;;mBAAY,KAAK;WAAC;QAElB,kCAAkC;QAClC;;;;;WAA+B;QAE/B,8BAA8B;QAC9B;;;;;WAAmC;QAEnC;;;;;WAA8C;QAmB5C,MAAM,MAAM,GAAG,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;SAC7C;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC;QAExD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;QAE5C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;SAC1D;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,MAAM,EAAE,MAAM,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1D,GAAG,aAAa;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;YAC1B,GAAG,IAAI,CAAC,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;QAChD,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,OAAO,GAA2C,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC9C,GAAG,MAAM;gBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;aACtB,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,MAAM,EAAE;gBACjB,MAAM,MAAM,GAAG,MAAM,IAAI,OAAO,CAC9B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAClB,MAAM,MAAM,GAAyC,EAAE,CAAC;oBACxD,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE;wBACpC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE;4BAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE;gCAC3B,OAAO,CAAC,MAAM,CAAC,CAAC;6BACjB;iCAAM;gCACL,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAGrC,CAAC;gCAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gCACjC,IAAI,IAAI,IAAI,IAAI,EAAE;oCAChB,MAAM,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;oCACtD,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;oCAC1C,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;oCAEhC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,CACnC,IAAI,CAAC,IAAI,IAAI,EAAE,EACf,IAAI,CAAC,OAAO,CACb,CAAC;iCACH;6BACF;yBACF;oBACH,CAAC,CAAC,CAAC;oBAEH,uCAAuC;oBACvC,MAAM,MAAM,GAAG,IAAkC,CAAC;oBAClD,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CACjC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CACpC,CAAC;oBACF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC/C,CAAC,CACF,CAAC;gBAEF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;aACtB;iBAAM;gBACL,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;aAC/B;YAED,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAErB,IAAI,gBAAgB,EAAE;gBACpB,UAAU,CAAC,gBAAgB;oBACzB,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;aACzD;YAED,IAAI,YAAY,EAAE;gBAChB,UAAU,CAAC,YAAY,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC;aACzE;YAED,IAAI,WAAW,EAAE;gBACf,UAAU,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,WAAW,CAAC;aACtE;SACF;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACpE,aAAa,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;YACvB,cAAc,EAAE;gBACd,YAAY,EAAE,MAAM,CAAC,aAAa;gBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B;SACF,CAAC,CAAC,CACJ,CAAC;QACF,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,KAAK,CAAC,mBAAmB,CAAC,OAAgC;QACxD,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;YACxC,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;gBACrC,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACvC,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SAChD;QACD,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;YAC3C,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;SACpD;QACD,MAAM,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACzE,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE,CACvC,MAAM,CAAC,gBAAgB,CACrB,OAAO,EACP,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CACxD,CAAC;QACJ,OAAO,OAAO,CAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;YAC9B,8DAA8D;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,iBAAkB,SAAQ,MAAM;IAK3C,YACE,MAGC;QAED,KAAK,CAAC,MAAM,CAAC,CAAC;QAVhB;;;;;WAA2B;QAE3B;;;;;WAAkB;QAUhB,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,iBAAiB;YACpB,MAAM,EAAE,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QAE/D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;SAChD;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAAgC;QACxD,IAAI,OAAO,CAAC,MAAM,EAAE;YAClB,OAAO,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;SAC3C;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,qDAAqD;QACrD,MAAM,KAAK,CAAC,2CAA2C,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,MAAM,EAAE,kBAAkB;aAC3B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,aAAa,EAAE,0BAA0B;gBACzC,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;gBACzD,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;gBACvB,gBAAgB,EAAE,QAAQ,CAAC,IAAI;gBAC/B,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBACvD,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;gBACnD,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB;aACzC,CAAC;SACH,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { test, expect } from "@jest/globals";
|
|
2
|
+
import { OpenAIChat } from "../openai-chat.js";
|
|
3
|
+
test("Test OpenAI", async () => {
|
|
4
|
+
const model = new OpenAIChat({ modelName: "gpt-3.5-turbo" });
|
|
5
|
+
const res = await model.call("Print hello world");
|
|
6
|
+
console.log({ res });
|
|
7
|
+
});
|
|
8
|
+
test("Test OpenAI with prefix messages", async () => {
|
|
9
|
+
const model = new OpenAIChat({
|
|
10
|
+
prefixMessages: [
|
|
11
|
+
{ role: "user", content: "My name is John" },
|
|
12
|
+
{ role: "assistant", content: "Hi there" },
|
|
13
|
+
],
|
|
14
|
+
});
|
|
15
|
+
const res = await model.call("What is my name");
|
|
16
|
+
console.log({ res });
|
|
17
|
+
});
|
|
18
|
+
test("Test OpenAI in streaming mode", async () => {
|
|
19
|
+
let nrNewTokens = 0;
|
|
20
|
+
let streamedCompletion = "";
|
|
21
|
+
const model = new OpenAIChat({
|
|
22
|
+
modelName: "gpt-3.5-turbo",
|
|
23
|
+
streaming: true,
|
|
24
|
+
callbackManager: {
|
|
25
|
+
handleNewToken(token) {
|
|
26
|
+
nrNewTokens += 1;
|
|
27
|
+
streamedCompletion += token;
|
|
28
|
+
},
|
|
29
|
+
},
|
|
30
|
+
});
|
|
31
|
+
const res = await model.call("Print hello world");
|
|
32
|
+
console.log({ res });
|
|
33
|
+
expect(nrNewTokens > 0).toBe(true);
|
|
34
|
+
expect(res).toBe(streamedCompletion);
|
|
35
|
+
});
|
|
36
|
+
//# sourceMappingURL=openai-chat.int.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-chat.int.test.js","sourceRoot":"","sources":["../../../src/llms/tests/openai-chat.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC,CAAC;IAC7D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC;QAC3B,cAAc,EAAE;YACd,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE;YAC5C,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,EAAE;SAC3C;KACF,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,kBAAkB,GAAG,EAAE,CAAC;IAE5B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC;QAC3B,SAAS,EAAE,eAAe;QAC1B,SAAS,EAAE,IAAI;QACf,eAAe,EAAE;YACf,cAAc,CAAC,KAAK;gBAClB,WAAW,IAAI,CAAC,CAAC;gBACjB,kBAAkB,IAAI,KAAK,CAAC;YAC9B,CAAC;SACF;KACF,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAErB,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC"}
|
|
@@ -7,18 +7,21 @@ test("Test OpenAI", async () => {
|
|
|
7
7
|
});
|
|
8
8
|
test("Test OpenAI in streaming mode", async () => {
|
|
9
9
|
let nrNewTokens = 0;
|
|
10
|
+
let streamedCompletion = "";
|
|
10
11
|
const model = new OpenAI({
|
|
11
12
|
maxTokens: 5,
|
|
12
13
|
modelName: "text-ada-001",
|
|
13
14
|
streaming: true,
|
|
14
15
|
callbackManager: {
|
|
15
|
-
handleNewToken() {
|
|
16
|
+
handleNewToken(token) {
|
|
16
17
|
nrNewTokens += 1;
|
|
18
|
+
streamedCompletion += token;
|
|
17
19
|
},
|
|
18
20
|
},
|
|
19
21
|
});
|
|
20
22
|
const res = await model.call("Print hello world");
|
|
21
23
|
console.log({ res });
|
|
22
24
|
expect(nrNewTokens > 0).toBe(true);
|
|
25
|
+
expect(res).toBe(streamedCompletion);
|
|
23
26
|
});
|
|
24
27
|
//# sourceMappingURL=openai.int.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.int.test.js","sourceRoot":"","sources":["../../../src/llms/tests/openai.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACtE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC;QACvB,SAAS,EAAE,CAAC;QACZ,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,IAAI;QACf,eAAe,EAAE;YACf,cAAc;
|
|
1
|
+
{"version":3,"file":"openai.int.test.js","sourceRoot":"","sources":["../../../src/llms/tests/openai.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;IACtE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,kBAAkB,GAAG,EAAE,CAAC;IAE5B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC;QACvB,SAAS,EAAE,CAAC;QACZ,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,IAAI;QACf,eAAe,EAAE;YACf,cAAc,CAAC,KAAK;gBAClB,WAAW,IAAI,CAAC,CAAC;gBACjB,kBAAkB,IAAI,KAAK,CAAC;YAC9B,CAAC;SACF;KACF,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAErB,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"pinecone.int.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;IAEpC,MAAM,MAAM,CAAC,IAAI,CAAC;QAChB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAqB;QAC9C,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAiB;KACtC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,cAAe,CAAC,CAAC;IAExD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAE1C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAEhC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;QACtB,IAAI,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;IAEpC,MAAM,MAAM,CAAC,IAAI,CAAC;QAChB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAqB;QAC9C,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAiB;KACtC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,cAAe,CAAC,CAAC;IAExD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAE1C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAEhC,mEAAmE;IACnE,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;QACtB,IAAI,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"pinecone.int.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.int.test.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;IAEpC,MAAM,MAAM,CAAC,IAAI,CAAC;QAChB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAqB;QAC9C,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAiB;KACtC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,cAAe,CAAC,CAAC;IAExD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAE1C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAEhC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;QACtB,IAAI,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;IAEpC,MAAM,MAAM,CAAC,IAAI,CAAC;QAChB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAqB;QAC9C,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAiB;KACtC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,cAAe,CAAC,CAAC;IAExD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAE1C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAEhC,mEAAmE;IACnE,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;QACtB,IAAI,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,MAAa,EAAE,UAAU,CAAC,CAAC;IAE3D,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,MAAa,EAAE,UAAU,CAAC,CAAC;IAE3D,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,MAAa,EAAE,UAAU,CAAC,CAAC;IAE3D,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,MAAa,EAAE,UAAU,CAAC,CAAC;IAE3D,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "langchain",
|
|
3
|
-
"version": "0.0.
|
|
3
|
+
"version": "0.0.19",
|
|
4
4
|
"description": "Typescript bindings for langchain",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "./index.js",
|
|
@@ -80,7 +80,6 @@
|
|
|
80
80
|
"husky": "^8.0.3",
|
|
81
81
|
"jest": "^29.4.2",
|
|
82
82
|
"lint-staged": "^13.1.1",
|
|
83
|
-
"openai": "^3.1.0",
|
|
84
83
|
"prettier": "^2.8.3",
|
|
85
84
|
"serpapi": "^1.1.1",
|
|
86
85
|
"srt-parser-2": "^1.2.2",
|
|
@@ -98,7 +97,6 @@
|
|
|
98
97
|
"cohere-ai": "^5.0.2",
|
|
99
98
|
"hnswlib-node": "^1.3.0",
|
|
100
99
|
"huggingface": "^1.4.0",
|
|
101
|
-
"openai": "^3.1.0",
|
|
102
100
|
"serpapi": "^1.1.1",
|
|
103
101
|
"srt-parser-2": "^1.2.2"
|
|
104
102
|
},
|
|
@@ -121,9 +119,6 @@
|
|
|
121
119
|
"hnswlib-node": {
|
|
122
120
|
"optional": true
|
|
123
121
|
},
|
|
124
|
-
"openai": {
|
|
125
|
-
"optional": true
|
|
126
|
-
},
|
|
127
122
|
"serpapi": {
|
|
128
123
|
"optional": true
|
|
129
124
|
},
|
|
@@ -139,6 +134,7 @@
|
|
|
139
134
|
"expr-eval": "^2.0.2",
|
|
140
135
|
"gpt-3-encoder": "^1.1.4",
|
|
141
136
|
"jsonpointer": "^5.0.1",
|
|
137
|
+
"openai": "^3.2.0",
|
|
142
138
|
"p-queue": "^7.3.4",
|
|
143
139
|
"sqlite3": "^5.1.4",
|
|
144
140
|
"unfetch": "^5.0.0",
|