-
Notifications
You must be signed in to change notification settings - Fork 2k
/
function.ts
74 lines (65 loc) 路 1.6 KB
/
function.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
73
74
import {
BaseMessage,
BaseMessageChunk,
type BaseMessageFields,
mergeContent,
_mergeDicts,
type MessageType,
} from "./base.js";
export interface FunctionMessageFieldsWithName extends BaseMessageFields {
name: string;
}
/**
* Represents a function message in a conversation.
*/
export class FunctionMessage extends BaseMessage {
static lc_name() {
return "FunctionMessage";
}
constructor(fields: FunctionMessageFieldsWithName);
constructor(
fields: string | BaseMessageFields,
/** @deprecated */
name: string
);
constructor(
fields: string | FunctionMessageFieldsWithName,
/** @deprecated */
name?: string
) {
if (typeof fields === "string") {
// eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
fields = { content: fields, name: name! };
}
super(fields);
}
_getType(): MessageType {
return "function";
}
}
/**
* Represents a chunk of a function message, which can be concatenated
* with other function message chunks.
*/
export class FunctionMessageChunk extends BaseMessageChunk {
static lc_name() {
return "FunctionMessageChunk";
}
_getType(): MessageType {
return "function";
}
concat(chunk: FunctionMessageChunk) {
return new FunctionMessageChunk({
content: mergeContent(this.content, chunk.content),
additional_kwargs: _mergeDicts(
this.additional_kwargs,
chunk.additional_kwargs
),
response_metadata: _mergeDicts(
this.response_metadata,
chunk.response_metadata
),
name: this.name ?? "",
});
}
}