-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
index.ts
72 lines (65 loc) · 1.57 KB
/
index.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import {
defaultProvider,
DefaultProviderInit,
} from "@aws-sdk/credential-provider-node";
import type { BaseChatModelParams } from "@langchain/core/language_models/chat_models";
import { BaseBedrockInput } from "../../utils/bedrock/index.js";
import { BedrockChat as BaseBedrockChat } from "./web.js";
export interface BedrockChatFields
extends Partial<BaseBedrockInput>,
BaseChatModelParams,
Partial<DefaultProviderInit> {}
/**
* @example
* ```typescript
* const model = new BedrockChat({
* model: "anthropic.claude-v2",
* region: "us-east-1",
* });
* const res = await model.invoke([{ content: "Tell me a joke" }]);
* console.log(res);
* ```
*/
export class BedrockChat extends BaseBedrockChat {
static lc_name() {
return "BedrockChat";
}
constructor(fields?: BedrockChatFields) {
const {
profile,
filepath,
configFilepath,
ignoreCache,
mfaCodeProvider,
roleAssumer,
roleArn,
webIdentityTokenFile,
roleAssumerWithWebIdentity,
...rest
} = fields ?? {};
super({
...rest,
credentials:
rest?.credentials ??
defaultProvider({
profile,
filepath,
configFilepath,
ignoreCache,
mfaCodeProvider,
roleAssumer,
roleArn,
webIdentityTokenFile,
roleAssumerWithWebIdentity,
}),
});
}
}
export {
convertMessagesToPromptAnthropic,
convertMessagesToPrompt,
} from "./web.js";
/**
* @deprecated Use `BedrockChat` instead.
*/
export const ChatBedrock = BedrockChat;