@botonic/plugin-ai-agents 0.43.0-beta.0 → 0.44.0-alpha.0
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/lib/cjs/agent-builder.d.ts +14 -3
- package/lib/cjs/agent-builder.js +43 -13
- package/lib/cjs/agent-builder.js.map +1 -1
- package/lib/cjs/constants.d.ts +3 -0
- package/lib/cjs/constants.js +6 -1
- package/lib/cjs/constants.js.map +1 -1
- package/lib/cjs/guardrails/input.js +4 -6
- package/lib/cjs/guardrails/input.js.map +1 -1
- package/lib/cjs/hubtype-api-client.js +97 -106
- package/lib/cjs/hubtype-api-client.js.map +1 -1
- package/lib/cjs/index.js +63 -62
- package/lib/cjs/index.js.map +1 -1
- package/lib/cjs/openai.js +19 -5
- package/lib/cjs/openai.js.map +1 -1
- package/lib/cjs/runner.js +39 -38
- package/lib/cjs/runner.js.map +1 -1
- package/lib/cjs/tools/retrieve-knowledge.d.ts +8 -1
- package/lib/cjs/tools/retrieve-knowledge.js +5 -5
- package/lib/cjs/tools/retrieve-knowledge.js.map +1 -1
- package/lib/cjs/types.d.ts +2 -3
- package/lib/esm/agent-builder.d.ts +14 -3
- package/lib/esm/agent-builder.js +57 -23
- package/lib/esm/agent-builder.js.map +1 -1
- package/lib/esm/constants.d.ts +3 -0
- package/lib/esm/constants.js +15 -7
- package/lib/esm/constants.js.map +1 -1
- package/lib/esm/guardrails/index.js +4 -1
- package/lib/esm/guardrails/index.js.map +1 -1
- package/lib/esm/guardrails/input.js +11 -9
- package/lib/esm/guardrails/input.js.map +1 -1
- package/lib/esm/hubtype-api-client.js +105 -110
- package/lib/esm/hubtype-api-client.js.map +1 -1
- package/lib/esm/index.js +76 -72
- package/lib/esm/index.js.map +1 -1
- package/lib/esm/openai.js +33 -15
- package/lib/esm/openai.js.map +1 -1
- package/lib/esm/runner.js +47 -42
- package/lib/esm/runner.js.map +1 -1
- package/lib/esm/structured-output/carousel.js +15 -12
- package/lib/esm/structured-output/carousel.js.map +1 -1
- package/lib/esm/structured-output/exit.js +7 -3
- package/lib/esm/structured-output/exit.js.map +1 -1
- package/lib/esm/structured-output/index.js +10 -7
- package/lib/esm/structured-output/index.js.map +1 -1
- package/lib/esm/structured-output/text-with-buttons.js +12 -8
- package/lib/esm/structured-output/text-with-buttons.js.map +1 -1
- package/lib/esm/structured-output/text.js +8 -5
- package/lib/esm/structured-output/text.js.map +1 -1
- package/lib/esm/tools/index.js +6 -2
- package/lib/esm/tools/index.js.map +1 -1
- package/lib/esm/tools/retrieve-knowledge.d.ts +8 -1
- package/lib/esm/tools/retrieve-knowledge.js +15 -12
- package/lib/esm/tools/retrieve-knowledge.js.map +1 -1
- package/lib/esm/types.d.ts +2 -3
- package/lib/esm/types.js +2 -1
- package/package.json +5 -5
- package/src/agent-builder.ts +79 -26
- package/src/constants.ts +7 -0
- package/src/guardrails/input.ts +1 -1
- package/src/hubtype-api-client.ts +21 -14
- package/src/index.ts +10 -10
- package/src/openai.ts +19 -3
- package/src/runner.ts +15 -6
- package/src/tools/retrieve-knowledge.ts +3 -2
- package/src/types.ts +4 -3
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/guardrails/index.ts"],"names":[],"mappings":"AAAA,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/guardrails/index.ts"],"names":[],"mappings":";;;AAAA,kDAAuB"}
|
|
@@ -1,18 +1,20 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createInputGuardrail = createInputGuardrail;
|
|
4
|
+
const agents_1 = require("@openai/agents");
|
|
5
|
+
const zod_1 = require("zod");
|
|
6
|
+
function createInputGuardrail(rules) {
|
|
7
|
+
const outputType = zod_1.z.object(Object.fromEntries(rules.map(rule => [rule.name, zod_1.z.boolean().describe(rule.description)])));
|
|
8
|
+
const agent = new agents_1.Agent({
|
|
7
9
|
name: 'InputGuardrail',
|
|
8
10
|
instructions: 'Check if the user triggers some of the following guardrails.',
|
|
9
11
|
outputType,
|
|
10
12
|
});
|
|
11
13
|
return {
|
|
12
14
|
name: 'InputGuardrail',
|
|
13
|
-
execute: ({ input, context }) =>
|
|
15
|
+
execute: async ({ input, context }) => {
|
|
14
16
|
const lastMessage = input[input.length - 1];
|
|
15
|
-
const result =
|
|
17
|
+
const result = await (0, agents_1.run)(agent, [lastMessage], { context });
|
|
16
18
|
const finalOutput = result.finalOutput;
|
|
17
19
|
if (finalOutput === undefined) {
|
|
18
20
|
throw new Error('Guardrail agent failed to produce output');
|
|
@@ -23,7 +25,7 @@ export function createInputGuardrail(rules) {
|
|
|
23
25
|
outputInfo: triggeredGuardrails,
|
|
24
26
|
tripwireTriggered: triggered,
|
|
25
27
|
};
|
|
26
|
-
}
|
|
28
|
+
},
|
|
27
29
|
};
|
|
28
30
|
}
|
|
29
31
|
//# sourceMappingURL=input.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"input.js","sourceRoot":"","sources":["../../../src/guardrails/input.ts"],"names":[],"mappings":";
|
|
1
|
+
{"version":3,"file":"input.js","sourceRoot":"","sources":["../../../src/guardrails/input.ts"],"names":[],"mappings":";;AAKA,oDAiCC;AAtCD,2CAA4E;AAC5E,6BAAuB;AAIvB,SAAgB,oBAAoB,CAAC,KAAsB;IACzD,MAAM,UAAU,GAAG,OAAC,CAAC,MAAM,CACzB,MAAM,CAAC,WAAW,CAChB,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CACvE,CACF,CAAA;IAED,MAAM,KAAK,GAAG,IAAI,cAAK,CAAC;QACtB,IAAI,EAAE,gBAAgB;QACtB,YAAY,EACV,8DAA8D;QAChE,UAAU;KACX,CAAC,CAAA;IAEF,OAAO;QACL,IAAI,EAAE,gBAAgB;QACtB,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;YACpC,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAoB,CAAA;YAC9D,MAAM,MAAM,GAAG,MAAM,IAAA,YAAG,EAAC,KAAK,EAAE,CAAC,WAAW,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAA;YAC3D,MAAM,WAAW,GAAG,MAAM,CAAC,WAAsC,CAAA;YACjE,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;YAC7D,CAAC;YACD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CAAA;YAC1E,MAAM,mBAAmB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CACzD,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,IAAI,CACjC,CAAA;YACD,OAAO;gBACL,UAAU,EAAE,mBAAmB;gBAC/B,iBAAiB,EAAE,SAAS;aAC7B,CAAA;QACH,CAAC;KACF,CAAA;AACH,CAAC"}
|
|
@@ -1,122 +1,116 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.HubtypeApiClient = void 0;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const axios_1 = tslib_1.__importDefault(require("axios"));
|
|
6
|
+
const constants_1 = require("./constants");
|
|
7
|
+
class HubtypeApiClient {
|
|
5
8
|
constructor(authToken) {
|
|
6
9
|
this.authToken = authToken;
|
|
7
10
|
}
|
|
8
|
-
retrieveSimilarChunks(query, sourceIds) {
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
return chunks;
|
|
26
|
-
});
|
|
11
|
+
async retrieveSimilarChunks(query, sourceIds) {
|
|
12
|
+
const url = `${constants_1.HUBTYPE_API_URL}/external/v1/ai/knowledge_base/similar_chunks/`;
|
|
13
|
+
const headers = {
|
|
14
|
+
'Content-Type': 'application/json',
|
|
15
|
+
Authorization: `Bearer ${this.authToken}`,
|
|
16
|
+
};
|
|
17
|
+
const params = {
|
|
18
|
+
query,
|
|
19
|
+
source_ids: sourceIds,
|
|
20
|
+
num_chunks: 5,
|
|
21
|
+
};
|
|
22
|
+
const response = await axios_1.default.post(url, params, { headers });
|
|
23
|
+
const chunks = response.data.chunks.map((chunk) => ({
|
|
24
|
+
id: chunk.id,
|
|
25
|
+
text: chunk.text,
|
|
26
|
+
}));
|
|
27
|
+
return chunks;
|
|
27
28
|
}
|
|
28
|
-
getLocalMessages(maxMemoryLength) {
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
return filteredMessages.slice(-maxMemoryLength);
|
|
43
|
-
});
|
|
29
|
+
async getLocalMessages(maxMemoryLength) {
|
|
30
|
+
const localBotonicState = typeof globalThis !== 'undefined' && globalThis.localStorage
|
|
31
|
+
? globalThis.localStorage.getItem('botonicState')
|
|
32
|
+
: null;
|
|
33
|
+
const botonicState = JSON.parse(localBotonicState || '{}');
|
|
34
|
+
const messages = botonicState.messages;
|
|
35
|
+
const filteredMessages = messages
|
|
36
|
+
.filter(message => message.data.text)
|
|
37
|
+
.map(message => ({
|
|
38
|
+
role: message.sentBy === 'user' ? 'user' : 'assistant',
|
|
39
|
+
content: message.data.text,
|
|
40
|
+
}))
|
|
41
|
+
.map(message => this.formatMessage(message));
|
|
42
|
+
return filteredMessages.slice(-maxMemoryLength);
|
|
44
43
|
}
|
|
45
|
-
getMessages(request, maxMemoryLength) {
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
}
|
|
68
|
-
});
|
|
44
|
+
async getMessages(request, maxMemoryLength) {
|
|
45
|
+
const url = `${constants_1.HUBTYPE_API_URL}/external/v1/ai/agent/message_history/`;
|
|
46
|
+
const headers = {
|
|
47
|
+
'Content-Type': 'application/json',
|
|
48
|
+
Authorization: `Bearer ${this.authToken}`,
|
|
49
|
+
};
|
|
50
|
+
const params = {
|
|
51
|
+
last_message_id: request.input.message_id,
|
|
52
|
+
num_messages: maxMemoryLength,
|
|
53
|
+
};
|
|
54
|
+
try {
|
|
55
|
+
const response = await axios_1.default.get(url, {
|
|
56
|
+
headers,
|
|
57
|
+
params,
|
|
58
|
+
});
|
|
59
|
+
const messages = response.data.messages;
|
|
60
|
+
return messages.map(message => this.formatMessage(message));
|
|
61
|
+
}
|
|
62
|
+
catch (error) {
|
|
63
|
+
console.error(error);
|
|
64
|
+
throw new Error('Failed to get messages from Hubtype');
|
|
65
|
+
}
|
|
69
66
|
}
|
|
70
|
-
getMessagesV2(request, options = {}) {
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
67
|
+
async getMessagesV2(request, options = {}) {
|
|
68
|
+
const url = `${constants_1.HUBTYPE_API_URL}/external/v2/ai/agent/message_history/`;
|
|
69
|
+
const headers = {
|
|
70
|
+
'Content-Type': 'application/json',
|
|
71
|
+
Authorization: `Bearer ${this.authToken}`,
|
|
72
|
+
};
|
|
73
|
+
const params = {
|
|
74
|
+
last_message_id: request.input.message_id,
|
|
75
|
+
...(options.maxMessages !== undefined && {
|
|
76
|
+
max_messages: options.maxMessages,
|
|
77
|
+
}),
|
|
78
|
+
...(options.includeToolCalls !== undefined && {
|
|
79
|
+
include_tool_calls: options.includeToolCalls,
|
|
80
|
+
}),
|
|
81
|
+
...(options.maxFullToolResults !== undefined && {
|
|
82
|
+
max_full_tool_results: options.maxFullToolResults,
|
|
83
|
+
}),
|
|
84
|
+
...(options.debugMode !== undefined && {
|
|
85
|
+
debug_mode: options.debugMode,
|
|
86
|
+
}),
|
|
87
|
+
};
|
|
88
|
+
try {
|
|
89
|
+
const response = await axios_1.default.get(url, {
|
|
90
|
+
headers,
|
|
91
|
+
params,
|
|
92
|
+
});
|
|
93
|
+
const { messages, conversation_id, truncated } = response.data;
|
|
94
|
+
const formattedMessages = messages
|
|
95
|
+
.map(message => this.formatMessageV2(message))
|
|
96
|
+
.filter((message) => message !== null);
|
|
97
|
+
return {
|
|
98
|
+
messages: formattedMessages,
|
|
99
|
+
conversationId: conversation_id,
|
|
100
|
+
truncated,
|
|
79
101
|
};
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
}
|
|
86
|
-
if (options.maxFullToolResults !== undefined) {
|
|
87
|
-
params.max_full_tool_results = options.maxFullToolResults;
|
|
88
|
-
}
|
|
89
|
-
if (options.debugMode !== undefined) {
|
|
90
|
-
params.debug_mode = options.debugMode;
|
|
91
|
-
}
|
|
92
|
-
try {
|
|
93
|
-
const response = yield axios.get(url, {
|
|
94
|
-
headers,
|
|
95
|
-
params,
|
|
96
|
-
});
|
|
97
|
-
const { messages, conversation_id, truncated } = response.data;
|
|
98
|
-
const formattedMessages = messages
|
|
99
|
-
.map(message => this.formatMessageV2(message))
|
|
100
|
-
.filter((message) => message !== null);
|
|
101
|
-
return {
|
|
102
|
-
messages: formattedMessages,
|
|
103
|
-
conversationId: conversation_id,
|
|
104
|
-
truncated,
|
|
105
|
-
};
|
|
106
|
-
}
|
|
107
|
-
catch (error) {
|
|
108
|
-
console.error(error);
|
|
109
|
-
throw new Error('Failed to get messages from Hubtype V2 API');
|
|
110
|
-
}
|
|
111
|
-
});
|
|
102
|
+
}
|
|
103
|
+
catch (error) {
|
|
104
|
+
console.error(error);
|
|
105
|
+
throw new Error('Failed to get messages from Hubtype V2 API');
|
|
106
|
+
}
|
|
112
107
|
}
|
|
113
108
|
formatMessageV2(message) {
|
|
114
|
-
var _a, _b, _c, _d, _e;
|
|
115
109
|
switch (message.role) {
|
|
116
110
|
case 'user':
|
|
117
111
|
return {
|
|
118
112
|
role: 'user',
|
|
119
|
-
content:
|
|
113
|
+
content: message.content ?? '',
|
|
120
114
|
};
|
|
121
115
|
case 'assistant': {
|
|
122
116
|
const assistantMessage = message;
|
|
@@ -127,7 +121,7 @@ export class HubtypeApiClient {
|
|
|
127
121
|
assistantMessage.tool_calls.length > 0) {
|
|
128
122
|
return {
|
|
129
123
|
role: 'assistant',
|
|
130
|
-
content:
|
|
124
|
+
content: assistantMessage.content ?? '',
|
|
131
125
|
tool_calls: assistantMessage.tool_calls.map(tc => ({
|
|
132
126
|
id: tc.id,
|
|
133
127
|
type: tc.type,
|
|
@@ -144,7 +138,7 @@ export class HubtypeApiClient {
|
|
|
144
138
|
content: [
|
|
145
139
|
{
|
|
146
140
|
type: 'output_text',
|
|
147
|
-
text:
|
|
141
|
+
text: assistantMessage.content ?? '',
|
|
148
142
|
},
|
|
149
143
|
],
|
|
150
144
|
status: 'completed',
|
|
@@ -158,13 +152,13 @@ export class HubtypeApiClient {
|
|
|
158
152
|
return {
|
|
159
153
|
role: 'tool',
|
|
160
154
|
tool_call_id: toolMessage.tool_call_id,
|
|
161
|
-
content:
|
|
155
|
+
content: toolMessage.content ?? '',
|
|
162
156
|
};
|
|
163
157
|
}
|
|
164
158
|
case 'system':
|
|
165
159
|
return {
|
|
166
160
|
role: 'system',
|
|
167
|
-
content:
|
|
161
|
+
content: message.content ?? '',
|
|
168
162
|
};
|
|
169
163
|
default:
|
|
170
164
|
throw new Error(`Invalid message role: ${message.role}`);
|
|
@@ -194,4 +188,5 @@ export class HubtypeApiClient {
|
|
|
194
188
|
}
|
|
195
189
|
}
|
|
196
190
|
}
|
|
191
|
+
exports.HubtypeApiClient = HubtypeApiClient;
|
|
197
192
|
//# sourceMappingURL=hubtype-api-client.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":";;;;AAEA,0DAAyB;AAEzB,2CAA6C;AAgF7C,MAAa,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,KAAa,EACb,SAAmB;QAEnB,MAAM,GAAG,GAAG,GAAG,2BAAe,gDAAgD,CAAA;QAC9E,MAAM,OAAO,GAAG;YACd,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;SAC1C,CAAA;QACD,MAAM,MAAM,GAAG;YACb,KAAK;YACL,UAAU,EAAE,SAAS;YACrB,UAAU,EAAE,CAAC;SACd,CAAA;QACD,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAA;QAC3D,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAY,EAAE,EAAE,CAAC,CAAC;YACzD,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;SACjB,CAAC,CAAC,CAAA;QACH,OAAO,MAAM,CAAA;IACf,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,eAAuB;QAEvB,MAAM,iBAAiB,GACrB,OAAO,UAAU,KAAK,WAAW,IAAI,UAAU,CAAC,YAAY;YAC1D,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC;YACjD,CAAC,CAAC,IAAI,CAAA;QACV,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,IAAI,CAAC,CAAA;QAC1D,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAA;QACtC,MAAM,gBAAgB,GAAG,QAAQ;aAC9B,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;aACpC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACf,IAAI,EAAE,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW;YACtD,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI;SAC3B,CAAC,CAAC;aACF,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;QAC9C,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAA;IACjD,CAAC;IAED,KAAK,CAAC,WAAW,CACf,OAAmB,EACnB,eAAuB;QAEvB,MAAM,GAAG,GAAG,GAAG,2BAAe,wCAAwC,CAAA;QACtE,MAAM,OAAO,GAAG;YACd,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;SAC1C,CAAA;QACD,MAAM,MAAM,GAAG;YACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;YACzC,YAAY,EAAE,eAAe;SAC9B,CAAA;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAAiC,GAAG,EAAE;gBACpE,OAAO;gBACP,MAAM;aACP,CAAC,CAAA;YACF,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;YACvC,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;YACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,OAAmB,EACnB,UAAgC,EAAE;QAElC,MAAM,GAAG,GAAG,GAAG,2BAAe,wCAAwC,CAAA;QACtE,MAAM,OAAO,GAAG;YACd,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;SAC1C,CAAA;QACD,MAAM,MAAM,GAA2B;YACrC,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;YACzC,GAAG,CAAC,OAAO,CAAC,WAAW,KAAK,SAAS,IAAI;gBACvC,YAAY,EAAE,OAAO,CAAC,WAAW;aAClC,CAAC;YACF,GAAG,CAAC,OAAO,CAAC,gBAAgB,KAAK,SAAS,IAAI;gBAC5C,kBAAkB,EAAE,OAAO,CAAC,gBAAgB;aAC7C,CAAC;YACF,GAAG,CAAC,OAAO,CAAC,kBAAkB,KAAK,SAAS,IAAI;gBAC9C,qBAAqB,EAAE,OAAO,CAAC,kBAAkB;aAClD,CAAC;YACF,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI;gBACrC,UAAU,EAAE,OAAO,CAAC,SAAS;aAC9B,CAAC;SACH,CAAA;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAA2B,GAAG,EAAE;gBAC9D,OAAO;gBACP,MAAM;aACP,CAAC,CAAA;YACF,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAA;YAC9D,MAAM,iBAAiB,GAAG,QAAQ;iBAC/B,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;iBAC7C,MAAM,CAAC,CAAC,OAAO,EAAkC,EAAE,CAAC,OAAO,KAAK,IAAI,CAAC,CAAA;YACxE,OAAO;gBACL,QAAQ,EAAE,iBAAiB;gBAC3B,cAAc,EAAE,eAAe;gBAC/B,SAAS;aACV,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;YACpB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAA;QAC/D,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,OAAyB;QAEzB,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,MAAM;gBACT,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;iBAC/B,CAAA;YACH,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,gBAAgB,GAAG,OAAoC,CAAA;gBAC7D,gEAAgE;gBAChE,iEAAiE;gBACjE,gDAAgD;gBAChD,IACE,gBAAgB,CAAC,UAAU;oBAC3B,gBAAgB,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EACtC,CAAC;oBACD,OAAO;wBACL,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,gBAAgB,CAAC,OAAO,IAAI,EAAE;wBACvC,UAAU,EAAE,gBAAgB,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;4BACjD,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,EAAE,CAAC,IAAI;4BACb,QAAQ,EAAE;gCACR,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gCACtB,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS;6BACjC;yBACF,CAAC,CAAC;qBAC8B,CAAA;gBACrC,CAAC;gBACD,+CAA+C;gBAC/C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,aAAa;4BACnB,IAAI,EAAE,gBAAgB,CAAC,OAAO,IAAI,EAAE;yBACrC;qBACF;oBACD,MAAM,EAAE,WAAW;iBACpB,CAAA;YACH,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,+DAA+D;gBAC/D,iEAAiE;gBACjE,gDAAgD;gBAChD,MAAM,WAAW,GAAG,OAA+B,CAAA;gBACnD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,WAAW,CAAC,YAAY;oBACtC,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,EAAE;iBACD,CAAA;YACrC,CAAC;YACD,KAAK,QAAQ;gBACX,OAAO;oBACL,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;iBAC/B,CAAA;YACH;gBACE,MAAM,IAAI,KAAK,CACb,yBAA0B,OAA4B,CAAC,IAAI,EAAE,CAC9D,CAAA;QACL,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,OAAuB;QAC3C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,MAAM;aACb,CAAA;QACH,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACxC,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,aAAa;wBACnB,IAAI,EAAE,OAAO,CAAC,OAAO;qBACtB;iBACF;gBACD,MAAM,EAAE,WAAW;aACpB,CAAA;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;QACzC,CAAC;IACH,CAAC;CACF;AA/MD,4CA+MC"}
|
package/lib/esm/index.js
CHANGED
|
@@ -1,90 +1,93 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const tslib_1 = require("tslib");
|
|
4
|
+
const agents_1 = require("@openai/agents");
|
|
5
|
+
const agent_builder_1 = require("./agent-builder");
|
|
6
|
+
const constants_1 = require("./constants");
|
|
7
|
+
const hubtype_api_client_1 = require("./hubtype-api-client");
|
|
8
|
+
const openai_1 = require("./openai");
|
|
9
|
+
const runner_1 = require("./runner");
|
|
10
|
+
class BotonicPluginAiAgents {
|
|
9
11
|
constructor(options) {
|
|
10
|
-
var _a, _b;
|
|
11
12
|
this.toolDefinitions = [];
|
|
12
|
-
|
|
13
|
-
if (
|
|
13
|
+
(0, openai_1.setUpOpenAI)(options?.maxRetries, options?.timeout);
|
|
14
|
+
if (options?.messageHistoryApiVersion === 'v1' && options?.memory) {
|
|
14
15
|
throw new Error('Cannot use memory when messageHistoryApiVersion is "v1". ' +
|
|
15
16
|
'Either set messageHistoryApiVersion to "v2" or remove memory.');
|
|
16
17
|
}
|
|
17
|
-
this.authToken = options
|
|
18
|
-
this.toolDefinitions =
|
|
19
|
-
this.messageHistoryApiVersion =
|
|
20
|
-
this.memory =
|
|
18
|
+
this.authToken = options?.authToken;
|
|
19
|
+
this.toolDefinitions = options?.customTools || [];
|
|
20
|
+
this.messageHistoryApiVersion = options?.messageHistoryApiVersion ?? 'v2';
|
|
21
|
+
this.memory = options?.memory ?? {};
|
|
21
22
|
}
|
|
22
23
|
pre() {
|
|
23
24
|
return;
|
|
24
25
|
}
|
|
25
|
-
getInference(request, aiAgentArgs) {
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
if (!authToken) {
|
|
31
|
-
throw new Error('Auth token is required');
|
|
32
|
-
}
|
|
33
|
-
const tools = this.buildTools(((_a = aiAgentArgs.activeTools) === null || _a === void 0 ? void 0 : _a.map(tool => tool.name)) || []);
|
|
34
|
-
const agent = new AIAgentBuilder(aiAgentArgs.name, aiAgentArgs.instructions, tools, request.session.user.contact_info || {}, aiAgentArgs.inputGuardrailRules || [], aiAgentArgs.sourceIds || []).build();
|
|
35
|
-
const messages = yield this.getMessages(request, authToken, MAX_MEMORY_LENGTH);
|
|
36
|
-
const context = {
|
|
37
|
-
authToken,
|
|
38
|
-
sourceIds: aiAgentArgs.sourceIds || [],
|
|
39
|
-
knowledgeUsed: {
|
|
40
|
-
query: '',
|
|
41
|
-
sourceIds: [],
|
|
42
|
-
chunksIds: [],
|
|
43
|
-
chunkTexts: [],
|
|
44
|
-
},
|
|
45
|
-
request,
|
|
46
|
-
};
|
|
47
|
-
const runner = new AIAgentRunner(agent);
|
|
48
|
-
return yield runner.run(messages, context);
|
|
26
|
+
async getInference(request, aiAgentArgs) {
|
|
27
|
+
try {
|
|
28
|
+
const authToken = constants_1.isProd ? request.session._access_token : this.authToken;
|
|
29
|
+
if (!authToken) {
|
|
30
|
+
throw new Error('Auth token is required');
|
|
49
31
|
}
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
32
|
+
const tools = this.buildTools(aiAgentArgs.activeTools?.map(tool => tool.name) || []);
|
|
33
|
+
const agent = new agent_builder_1.AIAgentBuilder({
|
|
34
|
+
name: aiAgentArgs.name,
|
|
35
|
+
instructions: aiAgentArgs.instructions,
|
|
36
|
+
tools: tools,
|
|
37
|
+
contactInfo: request.session.user.contact_info || [],
|
|
38
|
+
inputGuardrailRules: aiAgentArgs.inputGuardrailRules || [],
|
|
39
|
+
sourceIds: aiAgentArgs.sourceIds || [],
|
|
40
|
+
campaignContext: request.input.context?.campaign_v2,
|
|
41
|
+
}).build();
|
|
42
|
+
const messages = await this.getMessages(request, authToken, constants_1.MAX_MEMORY_LENGTH);
|
|
43
|
+
const context = {
|
|
44
|
+
authToken,
|
|
45
|
+
sourceIds: aiAgentArgs.sourceIds || [],
|
|
46
|
+
knowledgeUsed: {
|
|
47
|
+
query: '',
|
|
48
|
+
sourceIds: [],
|
|
49
|
+
chunksIds: [],
|
|
50
|
+
chunkTexts: [],
|
|
51
|
+
},
|
|
52
|
+
request,
|
|
53
|
+
};
|
|
54
|
+
const runner = new runner_1.AIAgentRunner(agent);
|
|
55
|
+
return await runner.run(messages, context);
|
|
56
|
+
}
|
|
57
|
+
catch (error) {
|
|
58
|
+
console.error('error plugin returns undefined', error);
|
|
59
|
+
return {
|
|
60
|
+
messages: [],
|
|
61
|
+
toolsExecuted: [],
|
|
62
|
+
memoryLength: 0,
|
|
63
|
+
exit: true,
|
|
64
|
+
error: true,
|
|
65
|
+
inputGuardrailsTriggered: [],
|
|
66
|
+
outputGuardrailsTriggered: [],
|
|
67
|
+
};
|
|
68
|
+
}
|
|
63
69
|
}
|
|
64
|
-
getMessages(request, authToken, memoryLength) {
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
maxFullToolResults: (_c = this.memory.maxFullToolResults) !== null && _c !== void 0 ? _c : 1,
|
|
79
|
-
debugMode: (_d = this.memory.debugMode) !== null && _d !== void 0 ? _d : false,
|
|
80
|
-
});
|
|
81
|
-
return result.messages;
|
|
70
|
+
async getMessages(request, authToken, memoryLength) {
|
|
71
|
+
const hubtypeClient = new hubtype_api_client_1.HubtypeApiClient(authToken);
|
|
72
|
+
if (!constants_1.isProd) {
|
|
73
|
+
return await hubtypeClient.getLocalMessages(memoryLength);
|
|
74
|
+
}
|
|
75
|
+
if (this.messageHistoryApiVersion === 'v1') {
|
|
76
|
+
return await hubtypeClient.getMessages(request, memoryLength);
|
|
77
|
+
}
|
|
78
|
+
// Default to V2
|
|
79
|
+
const result = await hubtypeClient.getMessagesV2(request, {
|
|
80
|
+
maxMessages: this.memory.maxMessages ?? memoryLength,
|
|
81
|
+
includeToolCalls: this.memory.includeToolCalls ?? true,
|
|
82
|
+
maxFullToolResults: this.memory.maxFullToolResults ?? 1,
|
|
83
|
+
debugMode: this.memory.debugMode ?? false,
|
|
82
84
|
});
|
|
85
|
+
return result.messages;
|
|
83
86
|
}
|
|
84
87
|
buildTools(activeToolNames) {
|
|
85
88
|
const availableTools = this.toolDefinitions.filter(tool => activeToolNames.includes(tool.name));
|
|
86
89
|
return availableTools.map(toolDefinition => {
|
|
87
|
-
return tool({
|
|
90
|
+
return (0, agents_1.tool)({
|
|
88
91
|
name: toolDefinition.name,
|
|
89
92
|
description: toolDefinition.description,
|
|
90
93
|
parameters: toolDefinition.schema,
|
|
@@ -93,5 +96,6 @@ export default class BotonicPluginAiAgents {
|
|
|
93
96
|
});
|
|
94
97
|
}
|
|
95
98
|
}
|
|
96
|
-
|
|
99
|
+
exports.default = BotonicPluginAiAgents;
|
|
100
|
+
tslib_1.__exportStar(require("./types"), exports);
|
|
97
101
|
//# sourceMappingURL=index.js.map
|
package/lib/esm/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AACA,2CAAqC;AAErC,mDAAgD;AAChD,2CAAuD;AACvD,6DAAuD;AACvD,qCAAsC;AACtC,qCAAwC;AAYxC,MAAqB,qBAAqB;IASxC,YAAY,OAAoD;QAFzD,oBAAe,GAAuC,EAAE,CAAA;QAG7D,IAAA,oBAAW,EAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;QAElD,IAAI,OAAO,EAAE,wBAAwB,KAAK,IAAI,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CACb,2DAA2D;gBACzD,+DAA+D,CAClE,CAAA;QACH,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAA;QACnC,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,WAAW,IAAI,EAAE,CAAA;QACjD,IAAI,CAAC,wBAAwB,GAAG,OAAO,EAAE,wBAAwB,IAAI,IAAI,CAAA;QACzE,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,EAAE,CAAA;IACrC,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAyC,EACzC,WAAwB;QAExB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,kBAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YACzE,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;YAC3C,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAC3B,WAAW,CAAC,WAAW,EAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CACtD,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,8BAAc,CAAuB;gBACrD,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,KAAK,EAAE,KAAK;gBACZ,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE;gBACpD,mBAAmB,EAAE,WAAW,CAAC,mBAAmB,IAAI,EAAE;gBAC1D,SAAS,EAAE,WAAW,CAAC,SAAS,IAAI,EAAE;gBACtC,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW;aACpD,CAAC,CAAC,KAAK,EAAE,CAAA;YAEV,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CACrC,OAAO,EACP,SAAS,EACT,6BAAiB,CAClB,CAAA;YACD,MAAM,OAAO,GAAkC;gBAC7C,SAAS;gBACT,SAAS,EAAE,WAAW,CAAC,SAAS,IAAI,EAAE;gBACtC,aAAa,EAAE;oBACb,KAAK,EAAE,EAAE;oBACT,SAAS,EAAE,EAAE;oBACb,SAAS,EAAE,EAAE;oBACb,UAAU,EAAE,EAAE;iBACf;gBACD,OAAO;aACR,CAAA;YAED,MAAM,MAAM,GAAG,IAAI,sBAAa,CAAuB,KAAK,CAAC,CAAA;YAC7D,OAAO,MAAM,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;YACtD,OAAO;gBACL,QAAQ,EAAE,EAAE;gBACZ,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,CAAC;gBACf,IAAI,EAAE,IAAI;gBACV,KAAK,EAAE,IAAI;gBACX,wBAAwB,EAAE,EAAE;gBAC5B,yBAAyB,EAAE,EAAE;aAC9B,CAAA;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;QAEpB,MAAM,aAAa,GAAG,IAAI,qCAAgB,CAAC,SAAS,CAAC,CAAA;QAErD,IAAI,CAAC,kBAAM,EAAE,CAAC;YACZ,OAAO,MAAM,aAAa,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAC3D,CAAC;QAED,IAAI,IAAI,CAAC,wBAAwB,KAAK,IAAI,EAAE,CAAC;YAC3C,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;QAC/D,CAAC;QAED,gBAAgB;QAChB,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,aAAa,CAAC,OAAO,EAAE;YACxD,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,YAAY;YACpD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI;YACtD,kBAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,CAAC;YACvD,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK;SAC1C,CAAC,CAAA;QACF,OAAO,MAAM,CAAC,QAAQ,CAAA;IACxB,CAAC;IAEO,UAAU,CAAC,eAAyB;QAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CACxD,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CACpC,CAAA;QACD,OAAO,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE;YACzC,OAAO,IAAA,aAAI,EAA0C;gBACnD,IAAI,EAAE,cAAc,CAAC,IAAI;gBACzB,WAAW,EAAE,cAAc,CAAC,WAAW;gBACvC,UAAU,EAAE,cAAc,CAAC,MAAM;gBACjC,OAAO,EAAE,cAAc,CAAC,IAAI;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AA3HD,wCA2HC;AAED,kDAAuB"}
|
package/lib/esm/openai.js
CHANGED
|
@@ -1,21 +1,39 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.setUpOpenAI = setUpOpenAI;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const agents_1 = require("@openai/agents");
|
|
6
|
+
const openai_1 = tslib_1.__importStar(require("openai"));
|
|
7
|
+
const constants_1 = require("./constants");
|
|
8
|
+
function setUpOpenAI(maxRetries, timeout) {
|
|
9
|
+
if (constants_1.OPENAI_PROVIDER === 'azure') {
|
|
10
|
+
setAzureOpenAIClient(maxRetries, timeout);
|
|
11
|
+
}
|
|
12
|
+
else {
|
|
13
|
+
setOpenAIClient(maxRetries, timeout);
|
|
14
|
+
}
|
|
15
|
+
(0, agents_1.setOpenAIAPI)('chat_completions');
|
|
16
|
+
(0, agents_1.setTracingDisabled)(true);
|
|
17
|
+
}
|
|
18
|
+
function setOpenAIClient(maxRetries, timeout) {
|
|
19
|
+
const client = new openai_1.default({
|
|
20
|
+
apiKey: constants_1.OPENAI_API_KEY,
|
|
21
|
+
timeout: timeout || 16000, // 16 seconds
|
|
22
|
+
maxRetries: maxRetries || 2,
|
|
23
|
+
dangerouslyAllowBrowser: !constants_1.isProd,
|
|
24
|
+
});
|
|
25
|
+
(0, agents_1.setDefaultOpenAIClient)(client);
|
|
8
26
|
}
|
|
9
27
|
function setAzureOpenAIClient(maxRetries, timeout) {
|
|
10
|
-
const client = new AzureOpenAI({
|
|
11
|
-
apiKey: AZURE_OPENAI_API_KEY,
|
|
12
|
-
apiVersion: AZURE_OPENAI_API_VERSION,
|
|
13
|
-
deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
|
|
14
|
-
baseURL: AZURE_OPENAI_API_BASE,
|
|
15
|
-
timeout: timeout ||
|
|
28
|
+
const client = new openai_1.AzureOpenAI({
|
|
29
|
+
apiKey: constants_1.AZURE_OPENAI_API_KEY,
|
|
30
|
+
apiVersion: constants_1.AZURE_OPENAI_API_VERSION,
|
|
31
|
+
deployment: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
|
|
32
|
+
baseURL: constants_1.AZURE_OPENAI_API_BASE,
|
|
33
|
+
timeout: timeout || 16000, // 16 seconds
|
|
16
34
|
maxRetries: maxRetries || 2,
|
|
17
|
-
dangerouslyAllowBrowser: !isProd,
|
|
35
|
+
dangerouslyAllowBrowser: !constants_1.isProd,
|
|
18
36
|
});
|
|
19
|
-
setDefaultOpenAIClient(client);
|
|
37
|
+
(0, agents_1.setDefaultOpenAIClient)(client);
|
|
20
38
|
}
|
|
21
39
|
//# sourceMappingURL=openai.js.map
|
package/lib/esm/openai.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":";;AAiBA,kCAQC;;AAzBD,2CAIuB;AACvB,yDAA4C;AAE5C,2CAQoB;AAEpB,SAAgB,WAAW,CAAC,UAAmB,EAAE,OAAgB;IAC/D,IAAI,2BAAe,KAAK,OAAO,EAAE,CAAC;QAChC,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IAC3C,CAAC;SAAM,CAAC;QACN,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IACtC,CAAC;IACD,IAAA,qBAAY,EAAC,kBAAkB,CAAC,CAAA;IAChC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAA;AAC1B,CAAC;AAED,SAAS,eAAe,CAAC,UAAmB,EAAE,OAAgB;IAC5D,MAAM,MAAM,GAAG,IAAI,gBAAM,CAAC;QACxB,MAAM,EAAE,0BAAc;QACtB,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC;AAED,SAAS,oBAAoB,CAAC,UAAmB,EAAE,OAAgB;IACjE,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC;QAC7B,MAAM,EAAE,gCAAoB;QAC5B,UAAU,EAAE,oCAAwB;QACpC,UAAU,EAAE,4CAAgC;QAC5C,OAAO,EAAE,iCAAqB;QAC9B,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC"}
|