@cogitator-ai/next 0.2.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/README.md +369 -0
- package/dist/client/index.d.ts +4 -0
- package/dist/client/index.d.ts.map +1 -0
- package/dist/client/index.js +4 -0
- package/dist/client/index.js.map +1 -0
- package/dist/client/retry.d.ts +3 -0
- package/dist/client/retry.d.ts.map +1 -0
- package/dist/client/retry.js +47 -0
- package/dist/client/retry.js.map +1 -0
- package/dist/client/sse-parser.d.ts +3 -0
- package/dist/client/sse-parser.d.ts.map +1 -0
- package/dist/client/sse-parser.js +39 -0
- package/dist/client/sse-parser.js.map +1 -0
- package/dist/client/use-agent.d.ts +3 -0
- package/dist/client/use-agent.d.ts.map +1 -0
- package/dist/client/use-agent.js +53 -0
- package/dist/client/use-agent.js.map +1 -0
- package/dist/client/use-chat-state.d.ts +52 -0
- package/dist/client/use-chat-state.d.ts.map +1 -0
- package/dist/client/use-chat-state.js +68 -0
- package/dist/client/use-chat-state.js.map +1 -0
- package/dist/client/use-chat.d.ts +3 -0
- package/dist/client/use-chat.d.ts.map +1 -0
- package/dist/client/use-chat.js +215 -0
- package/dist/client/use-chat.js.map +1 -0
- package/dist/handlers/agent.d.ts +4 -0
- package/dist/handlers/agent.d.ts.map +1 -0
- package/dist/handlers/agent.js +76 -0
- package/dist/handlers/agent.js.map +1 -0
- package/dist/handlers/chat.d.ts +4 -0
- package/dist/handlers/chat.d.ts.map +1 -0
- package/dist/handlers/chat.js +124 -0
- package/dist/handlers/chat.js.map +1 -0
- package/dist/index.d.ts +7 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +5 -0
- package/dist/index.js.map +1 -0
- package/dist/streaming/encoder.d.ts +4 -0
- package/dist/streaming/encoder.d.ts.map +1 -0
- package/dist/streaming/encoder.js +15 -0
- package/dist/streaming/encoder.js.map +1 -0
- package/dist/streaming/protocol.d.ts +64 -0
- package/dist/streaming/protocol.d.ts.map +1 -0
- package/dist/streaming/protocol.js +31 -0
- package/dist/streaming/protocol.js.map +1 -0
- package/dist/streaming/stream-writer.d.ts +19 -0
- package/dist/streaming/stream-writer.d.ts.map +1 -0
- package/dist/streaming/stream-writer.js +57 -0
- package/dist/streaming/stream-writer.js.map +1 -0
- package/dist/types.d.ts +97 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +56 -0
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
export function chatReducer(state, action) {
|
|
2
|
+
switch (action.type) {
|
|
3
|
+
case 'SET_INPUT':
|
|
4
|
+
return { ...state, input: action.payload };
|
|
5
|
+
case 'SET_THREAD_ID':
|
|
6
|
+
return { ...state, threadId: action.payload };
|
|
7
|
+
case 'START_LOADING':
|
|
8
|
+
return { ...state, isLoading: true, error: null };
|
|
9
|
+
case 'STOP_LOADING':
|
|
10
|
+
return { ...state, isLoading: false };
|
|
11
|
+
case 'SET_ERROR':
|
|
12
|
+
return { ...state, error: action.payload, isLoading: false };
|
|
13
|
+
case 'ADD_USER_MESSAGE':
|
|
14
|
+
return { ...state, messages: [...state.messages, action.payload] };
|
|
15
|
+
case 'APPEND_MESSAGE':
|
|
16
|
+
return { ...state, messages: [...state.messages, action.payload] };
|
|
17
|
+
case 'START_ASSISTANT_MESSAGE':
|
|
18
|
+
return {
|
|
19
|
+
...state,
|
|
20
|
+
currentMessageId: action.payload,
|
|
21
|
+
currentContent: '',
|
|
22
|
+
currentToolCalls: [],
|
|
23
|
+
};
|
|
24
|
+
case 'APPEND_CONTENT':
|
|
25
|
+
return { ...state, currentContent: state.currentContent + action.payload };
|
|
26
|
+
case 'ADD_TOOL_CALL':
|
|
27
|
+
return { ...state, currentToolCalls: [...state.currentToolCalls, action.payload] };
|
|
28
|
+
case 'FINISH_ASSISTANT_MESSAGE': {
|
|
29
|
+
if (!state.currentMessageId)
|
|
30
|
+
return state;
|
|
31
|
+
const assistantMessage = {
|
|
32
|
+
id: state.currentMessageId,
|
|
33
|
+
role: 'assistant',
|
|
34
|
+
content: state.currentContent,
|
|
35
|
+
toolCalls: state.currentToolCalls.length > 0 ? state.currentToolCalls : undefined,
|
|
36
|
+
createdAt: new Date(),
|
|
37
|
+
};
|
|
38
|
+
return {
|
|
39
|
+
...state,
|
|
40
|
+
messages: [...state.messages, assistantMessage],
|
|
41
|
+
currentMessageId: null,
|
|
42
|
+
currentContent: '',
|
|
43
|
+
currentToolCalls: [],
|
|
44
|
+
};
|
|
45
|
+
}
|
|
46
|
+
case 'REMOVE_LAST_MESSAGE':
|
|
47
|
+
return { ...state, messages: state.messages.slice(0, -1) };
|
|
48
|
+
case 'SET_MESSAGES':
|
|
49
|
+
return { ...state, messages: action.payload };
|
|
50
|
+
case 'CLEAR_MESSAGES':
|
|
51
|
+
return { ...state, messages: [] };
|
|
52
|
+
default:
|
|
53
|
+
return state;
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
export function createInitialState(initialMessages, threadId) {
|
|
57
|
+
return {
|
|
58
|
+
messages: initialMessages ?? [],
|
|
59
|
+
input: '',
|
|
60
|
+
isLoading: false,
|
|
61
|
+
error: null,
|
|
62
|
+
threadId,
|
|
63
|
+
currentMessageId: null,
|
|
64
|
+
currentContent: '',
|
|
65
|
+
currentToolCalls: [],
|
|
66
|
+
};
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=use-chat-state.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"use-chat-state.js","sourceRoot":"","sources":["../../src/client/use-chat-state.ts"],"names":[],"mappings":"AA6BA,MAAM,UAAU,WAAW,CAAC,KAAgB,EAAE,MAAkB;IAC9D,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,WAAW;YACd,OAAO,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAE7C,KAAK,eAAe;YAClB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAEhD,KAAK,eAAe;YAClB,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAEpD,KAAK,cAAc;YACjB,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAExC,KAAK,WAAW;YACd,OAAO,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAE/D,KAAK,kBAAkB;YACrB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAErE,KAAK,gBAAgB;YACnB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAErE,KAAK,yBAAyB;YAC5B,OAAO;gBACL,GAAG,KAAK;gBACR,gBAAgB,EAAE,MAAM,CAAC,OAAO;gBAChC,cAAc,EAAE,EAAE;gBAClB,gBAAgB,EAAE,EAAE;aACrB,CAAC;QAEJ,KAAK,gBAAgB;YACnB,OAAO,EAAE,GAAG,KAAK,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;QAE7E,KAAK,eAAe;YAClB,OAAO,EAAE,GAAG,KAAK,EAAE,gBAAgB,EAAE,CAAC,GAAG,KAAK,CAAC,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAErF,KAAK,0BAA0B,CAAC,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC;YAE1C,MAAM,gBAAgB,GAAgB;gBACpC,EAAE,EAAE,KAAK,CAAC,gBAAgB;gBAC1B,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,KAAK,CAAC,cAAc;gBAC7B,SAAS,EAAE,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS;gBACjF,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;YAEF,OAAO;gBACL,GAAG,KAAK;gBACR,QAAQ,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,gBAAgB,CAAC;gBAC/C,gBAAgB,EAAE,IAAI;gBACtB,cAAc,EAAE,EAAE;gBAClB,gBAAgB,EAAE,EAAE;aACrB,CAAC;QACJ,CAAC;QAED,KAAK,qBAAqB;YACxB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAE7D,KAAK,cAAc;YACjB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAEhD,KAAK,gBAAgB;YACnB,OAAO,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAEpC;YACE,OAAO,KAAK,CAAC;IACjB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,eAA+B,EAAE,QAAiB;IACnF,OAAO;QACL,QAAQ,EAAE,eAAe,IAAI,EAAE;QAC/B,KAAK,EAAE,EAAE;QACT,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,IAAI;QACX,QAAQ;QACR,gBAAgB,EAAE,IAAI;QACtB,cAAc,EAAE,EAAE;QAClB,gBAAgB,EAAE,EAAE;KACrB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"use-chat.d.ts","sourceRoot":"","sources":["../../src/client/use-chat.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,cAAc,EACd,aAAa,EAId,MAAM,aAAa,CAAC;AAWrB,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,cAAc,GAAG,aAAa,CAyPvE"}
|
|
@@ -0,0 +1,215 @@
|
|
|
1
|
+
'use client';
|
|
2
|
+
import { useCallback, useReducer, useRef } from 'react';
|
|
3
|
+
import { parseSSEStream } from './sse-parser.js';
|
|
4
|
+
import { chatReducer, createInitialState } from './use-chat-state.js';
|
|
5
|
+
import { withRetry } from './retry.js';
|
|
6
|
+
let idCounter = 0;
|
|
7
|
+
function generateClientId() {
|
|
8
|
+
return `msg_${Date.now().toString(36)}_${(idCounter++).toString(36)}`;
|
|
9
|
+
}
|
|
10
|
+
export function useCogitatorChat(options) {
|
|
11
|
+
const { api, initialMessages, headers, onError, onFinish, onToolCall, onToolResult, retry } = options;
|
|
12
|
+
const [state, dispatch] = useReducer(chatReducer, createInitialState(initialMessages, options.threadId));
|
|
13
|
+
const abortControllerRef = useRef(null);
|
|
14
|
+
const handleStreamEvent = useCallback((event, toolCalls) => {
|
|
15
|
+
switch (event.type) {
|
|
16
|
+
case 'start':
|
|
17
|
+
dispatch({ type: 'START_ASSISTANT_MESSAGE', payload: event.messageId });
|
|
18
|
+
break;
|
|
19
|
+
case 'text-delta':
|
|
20
|
+
dispatch({ type: 'APPEND_CONTENT', payload: event.delta });
|
|
21
|
+
break;
|
|
22
|
+
case 'tool-call-start':
|
|
23
|
+
toolCalls.set(event.id, { name: event.toolName, args: '' });
|
|
24
|
+
break;
|
|
25
|
+
case 'tool-call-delta': {
|
|
26
|
+
const tc = toolCalls.get(event.id);
|
|
27
|
+
if (tc) {
|
|
28
|
+
tc.args += event.argsTextDelta;
|
|
29
|
+
}
|
|
30
|
+
break;
|
|
31
|
+
}
|
|
32
|
+
case 'tool-call-end': {
|
|
33
|
+
const tc = toolCalls.get(event.id);
|
|
34
|
+
if (tc) {
|
|
35
|
+
let args = {};
|
|
36
|
+
try {
|
|
37
|
+
args = JSON.parse(tc.args);
|
|
38
|
+
}
|
|
39
|
+
catch { }
|
|
40
|
+
const toolCall = {
|
|
41
|
+
id: event.id,
|
|
42
|
+
name: tc.name,
|
|
43
|
+
arguments: args,
|
|
44
|
+
};
|
|
45
|
+
dispatch({ type: 'ADD_TOOL_CALL', payload: toolCall });
|
|
46
|
+
onToolCall?.(toolCall);
|
|
47
|
+
}
|
|
48
|
+
break;
|
|
49
|
+
}
|
|
50
|
+
case 'tool-result': {
|
|
51
|
+
const toolResult = {
|
|
52
|
+
id: event.id,
|
|
53
|
+
toolCallId: event.toolCallId,
|
|
54
|
+
result: event.result,
|
|
55
|
+
};
|
|
56
|
+
onToolResult?.(toolResult);
|
|
57
|
+
break;
|
|
58
|
+
}
|
|
59
|
+
case 'error':
|
|
60
|
+
dispatch({ type: 'SET_ERROR', payload: new Error(event.message) });
|
|
61
|
+
onError?.(new Error(event.message));
|
|
62
|
+
break;
|
|
63
|
+
}
|
|
64
|
+
}, [onError, onToolCall, onToolResult]);
|
|
65
|
+
const processStream = useCallback(async (response) => {
|
|
66
|
+
if (!response.body) {
|
|
67
|
+
throw new Error('Response body is null');
|
|
68
|
+
}
|
|
69
|
+
const reader = response.body.getReader();
|
|
70
|
+
const toolCalls = new Map();
|
|
71
|
+
try {
|
|
72
|
+
for await (const event of parseSSEStream(reader)) {
|
|
73
|
+
handleStreamEvent(event, toolCalls);
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
finally {
|
|
77
|
+
reader.releaseLock();
|
|
78
|
+
}
|
|
79
|
+
dispatch({ type: 'FINISH_ASSISTANT_MESSAGE' });
|
|
80
|
+
dispatch({ type: 'STOP_LOADING' });
|
|
81
|
+
if (onFinish && state.currentMessageId) {
|
|
82
|
+
onFinish({
|
|
83
|
+
id: state.currentMessageId,
|
|
84
|
+
role: 'assistant',
|
|
85
|
+
content: state.currentContent,
|
|
86
|
+
toolCalls: state.currentToolCalls.length > 0 ? state.currentToolCalls : undefined,
|
|
87
|
+
});
|
|
88
|
+
}
|
|
89
|
+
}, [
|
|
90
|
+
handleStreamEvent,
|
|
91
|
+
onFinish,
|
|
92
|
+
state.currentContent,
|
|
93
|
+
state.currentMessageId,
|
|
94
|
+
state.currentToolCalls,
|
|
95
|
+
]);
|
|
96
|
+
const send = useCallback(async (inputOverride, metadata) => {
|
|
97
|
+
const messageContent = inputOverride ?? state.input;
|
|
98
|
+
if (!messageContent.trim())
|
|
99
|
+
return;
|
|
100
|
+
abortControllerRef.current?.abort();
|
|
101
|
+
abortControllerRef.current = new AbortController();
|
|
102
|
+
const userMessage = {
|
|
103
|
+
id: generateClientId(),
|
|
104
|
+
role: 'user',
|
|
105
|
+
content: messageContent,
|
|
106
|
+
metadata,
|
|
107
|
+
createdAt: new Date(),
|
|
108
|
+
};
|
|
109
|
+
dispatch({ type: 'ADD_USER_MESSAGE', payload: userMessage });
|
|
110
|
+
dispatch({ type: 'SET_INPUT', payload: '' });
|
|
111
|
+
dispatch({ type: 'START_LOADING' });
|
|
112
|
+
const allMessages = [...state.messages, userMessage];
|
|
113
|
+
const doFetch = async () => {
|
|
114
|
+
const response = await fetch(api, {
|
|
115
|
+
method: 'POST',
|
|
116
|
+
headers: {
|
|
117
|
+
'Content-Type': 'application/json',
|
|
118
|
+
...headers,
|
|
119
|
+
},
|
|
120
|
+
body: JSON.stringify({
|
|
121
|
+
messages: allMessages.map((m) => ({
|
|
122
|
+
id: m.id,
|
|
123
|
+
role: m.role,
|
|
124
|
+
content: m.content,
|
|
125
|
+
metadata: m.metadata,
|
|
126
|
+
})),
|
|
127
|
+
threadId: state.threadId,
|
|
128
|
+
metadata,
|
|
129
|
+
}),
|
|
130
|
+
signal: abortControllerRef.current.signal,
|
|
131
|
+
});
|
|
132
|
+
if (!response.ok) {
|
|
133
|
+
const errorText = await response.text();
|
|
134
|
+
throw new Error(errorText || `HTTP ${response.status}`);
|
|
135
|
+
}
|
|
136
|
+
return response;
|
|
137
|
+
};
|
|
138
|
+
try {
|
|
139
|
+
const response = await withRetry(doFetch, retry);
|
|
140
|
+
await processStream(response);
|
|
141
|
+
}
|
|
142
|
+
catch (err) {
|
|
143
|
+
if (err instanceof Error && err.name === 'AbortError') {
|
|
144
|
+
dispatch({ type: 'STOP_LOADING' });
|
|
145
|
+
return;
|
|
146
|
+
}
|
|
147
|
+
const error = err instanceof Error ? err : new Error('Unknown error');
|
|
148
|
+
dispatch({ type: 'SET_ERROR', payload: error });
|
|
149
|
+
onError?.(error);
|
|
150
|
+
}
|
|
151
|
+
}, [api, headers, onError, processStream, retry, state.input, state.messages, state.threadId]);
|
|
152
|
+
const stop = useCallback(() => {
|
|
153
|
+
abortControllerRef.current?.abort();
|
|
154
|
+
dispatch({ type: 'STOP_LOADING' });
|
|
155
|
+
}, []);
|
|
156
|
+
const reload = useCallback(async () => {
|
|
157
|
+
if (state.messages.length === 0)
|
|
158
|
+
return;
|
|
159
|
+
let lastUserIndex = -1;
|
|
160
|
+
for (let i = state.messages.length - 1; i >= 0; i--) {
|
|
161
|
+
if (state.messages[i].role === 'user') {
|
|
162
|
+
lastUserIndex = i;
|
|
163
|
+
break;
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
if (lastUserIndex === -1)
|
|
167
|
+
return;
|
|
168
|
+
const lastUserMessage = state.messages[lastUserIndex];
|
|
169
|
+
const messagesBeforeReload = state.messages.slice(0, lastUserIndex);
|
|
170
|
+
dispatch({ type: 'SET_MESSAGES', payload: messagesBeforeReload });
|
|
171
|
+
await send(lastUserMessage.content, lastUserMessage.metadata);
|
|
172
|
+
}, [send, state.messages]);
|
|
173
|
+
const setInput = useCallback((value) => {
|
|
174
|
+
dispatch({ type: 'SET_INPUT', payload: value });
|
|
175
|
+
}, []);
|
|
176
|
+
const setThreadId = useCallback((id) => {
|
|
177
|
+
dispatch({ type: 'SET_THREAD_ID', payload: id });
|
|
178
|
+
}, []);
|
|
179
|
+
const appendMessage = useCallback((message) => {
|
|
180
|
+
dispatch({ type: 'APPEND_MESSAGE', payload: message });
|
|
181
|
+
}, []);
|
|
182
|
+
const clearMessages = useCallback(() => {
|
|
183
|
+
dispatch({ type: 'CLEAR_MESSAGES' });
|
|
184
|
+
}, []);
|
|
185
|
+
const setMessages = useCallback((messages) => {
|
|
186
|
+
dispatch({ type: 'SET_MESSAGES', payload: messages });
|
|
187
|
+
}, []);
|
|
188
|
+
const displayMessages = state.currentMessageId
|
|
189
|
+
? [
|
|
190
|
+
...state.messages,
|
|
191
|
+
{
|
|
192
|
+
id: state.currentMessageId,
|
|
193
|
+
role: 'assistant',
|
|
194
|
+
content: state.currentContent,
|
|
195
|
+
toolCalls: state.currentToolCalls.length > 0 ? state.currentToolCalls : undefined,
|
|
196
|
+
},
|
|
197
|
+
]
|
|
198
|
+
: state.messages;
|
|
199
|
+
return {
|
|
200
|
+
messages: displayMessages,
|
|
201
|
+
input: state.input,
|
|
202
|
+
setInput,
|
|
203
|
+
send,
|
|
204
|
+
isLoading: state.isLoading,
|
|
205
|
+
error: state.error,
|
|
206
|
+
stop,
|
|
207
|
+
reload,
|
|
208
|
+
threadId: state.threadId,
|
|
209
|
+
setThreadId,
|
|
210
|
+
appendMessage,
|
|
211
|
+
clearMessages,
|
|
212
|
+
setMessages,
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
//# sourceMappingURL=use-chat.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"use-chat.js","sourceRoot":"","sources":["../../src/client/use-chat.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AASxD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,IAAI,SAAS,GAAG,CAAC,CAAC;AAClB,SAAS,gBAAgB;IACvB,OAAO,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,OAAuB;IACtD,MAAM,EAAE,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,KAAK,EAAE,GACzF,OAAO,CAAC;IAEV,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,UAAU,CAClC,WAAW,EACX,kBAAkB,CAAC,eAAe,EAAE,OAAO,CAAC,QAAQ,CAAC,CACtD,CAAC;IAEF,MAAM,kBAAkB,GAAG,MAAM,CAAyB,IAAI,CAAC,CAAC;IAEhE,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,KAAkB,EAAE,SAAsD,EAAE,EAAE;QAC7E,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,OAAO;gBACV,QAAQ,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;gBACxE,MAAM;YAER,KAAK,YAAY;gBACf,QAAQ,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC3D,MAAM;YAER,KAAK,iBAAiB;gBACpB,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC5D,MAAM;YAER,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACnC,IAAI,EAAE,EAAE,CAAC;oBACP,EAAE,CAAC,IAAI,IAAI,KAAK,CAAC,aAAa,CAAC;gBACjC,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACnC,IAAI,EAAE,EAAE,CAAC;oBACP,IAAI,IAAI,GAA4B,EAAE,CAAC;oBACvC,IAAI,CAAC;wBACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;oBAAC,MAAM,CAAC,CAAA,CAAC;oBAEV,MAAM,QAAQ,GAAa;wBACzB,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,IAAI,EAAE,EAAE,CAAC,IAAI;wBACb,SAAS,EAAE,IAAI;qBAChB,CAAC;oBAEF,QAAQ,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;oBACvD,UAAU,EAAE,CAAC,QAAQ,CAAC,CAAC;gBACzB,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,UAAU,GAAoB;oBAClC,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,MAAM,EAAE,KAAK,CAAC,MAAM;iBACrB,CAAC;gBACF,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC;gBAC3B,MAAM;YACR,CAAC;YAED,KAAK,OAAO;gBACV,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACnE,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;gBACpC,MAAM;QACV,CAAC;IACH,CAAC,EACD,CAAC,OAAO,EAAE,UAAU,EAAE,YAAY,CAAC,CACpC,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAC/B,KAAK,EAAE,QAAkB,EAAE,EAAE;QAC3B,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,IAAI,GAAG,EAA0C,CAAC;QAEpE,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjD,iBAAiB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,QAAQ,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC,CAAC;QAC/C,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QAEnC,IAAI,QAAQ,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YACvC,QAAQ,CAAC;gBACP,EAAE,EAAE,KAAK,CAAC,gBAAgB;gBAC1B,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,KAAK,CAAC,cAAc;gBAC7B,SAAS,EAAE,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS;aAClF,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD;QACE,iBAAiB;QACjB,QAAQ;QACR,KAAK,CAAC,cAAc;QACpB,KAAK,CAAC,gBAAgB;QACtB,KAAK,CAAC,gBAAgB;KACvB,CACF,CAAC;IAEF,MAAM,IAAI,GAAG,WAAW,CACtB,KAAK,EAAE,aAAsB,EAAE,QAAkC,EAAE,EAAE;QACnE,MAAM,cAAc,GAAG,aAAa,IAAI,KAAK,CAAC,KAAK,CAAC;QACpD,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE;YAAE,OAAO;QAEnC,kBAAkB,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QACpC,kBAAkB,CAAC,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEnD,MAAM,WAAW,GAAgB;YAC/B,EAAE,EAAE,gBAAgB,EAAE;YACtB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,cAAc;YACvB,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,QAAQ,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAC7D,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,QAAQ,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;QAEpC,MAAM,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;YACzB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,GAAG,OAAO;iBACX;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAChC,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;qBACrB,CAAC,CAAC;oBACH,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,QAAQ;iBACT,CAAC;gBACF,MAAM,EAAE,kBAAkB,CAAC,OAAQ,CAAC,MAAM;aAC3C,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1D,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACjD,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACtD,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;gBACnC,OAAO;YACT,CAAC;YAED,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACtE,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAChD,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;QACnB,CAAC;IACH,CAAC,EACD,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAC3F,CAAC;IAEF,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,EAAE;QAC5B,kBAAkB,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QACpC,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;IACrC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;QACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAExC,IAAI,aAAa,GAAG,CAAC,CAAC,CAAC;QACvB,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACpD,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtC,aAAa,GAAG,CAAC,CAAC;gBAClB,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,aAAa,KAAK,CAAC,CAAC;YAAE,OAAO;QAEjC,MAAM,eAAe,GAAG,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;QAEpE,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;QAClE,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,QAAQ,CAAC,CAAC;IAChE,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE3B,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,KAAa,EAAE,EAAE;QAC7C,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,EAAU,EAAE,EAAE;QAC7C,QAAQ,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,aAAa,GAAG,WAAW,CAAC,CAAC,OAAoB,EAAE,EAAE;QACzD,QAAQ,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IACzD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;QACrC,QAAQ,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;IACvC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,QAAuB,EAAE,EAAE;QAC1D,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IACxD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,eAAe,GAAG,KAAK,CAAC,gBAAgB;QAC5C,CAAC,CAAC;YACE,GAAG,KAAK,CAAC,QAAQ;YACjB;gBACE,EAAE,EAAE,KAAK,CAAC,gBAAgB;gBAC1B,IAAI,EAAE,WAAoB;gBAC1B,OAAO,EAAE,KAAK,CAAC,cAAc;gBAC7B,SAAS,EAAE,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS;aAClF;SACF;QACH,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC;IAEnB,OAAO;QACL,QAAQ,EAAE,eAAe;QACzB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,QAAQ;QACR,IAAI;QACJ,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,IAAI;QACJ,MAAM;QACN,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW;QACX,aAAa;QACb,aAAa;QACb,WAAW;KACZ,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { Cogitator, Agent } from '@cogitator-ai/core';
|
|
2
|
+
import type { AgentHandlerOptions } from '../types.js';
|
|
3
|
+
export declare function createAgentHandler(cogitator: Cogitator, agent: Agent, options?: AgentHandlerOptions): (req: Request) => Promise<Response>;
|
|
4
|
+
//# sourceMappingURL=agent.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/handlers/agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,mBAAmB,EAA6B,MAAM,aAAa,CAAC;AAYlF,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,SAAS,EACpB,KAAK,EAAE,KAAK,EACZ,OAAO,CAAC,EAAE,mBAAmB,IAEf,KAAK,OAAO,KAAG,OAAO,CAAC,QAAQ,CAAC,CAwE/C"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
function parseDefaultInput(body) {
|
|
2
|
+
const data = body;
|
|
3
|
+
return {
|
|
4
|
+
input: typeof data.input === 'string' ? data.input : '',
|
|
5
|
+
context: data.context,
|
|
6
|
+
threadId: typeof data.threadId === 'string' ? data.threadId : undefined,
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
export function createAgentHandler(cogitator, agent, options) {
|
|
10
|
+
return async (req) => {
|
|
11
|
+
let body;
|
|
12
|
+
try {
|
|
13
|
+
body = await req.json();
|
|
14
|
+
}
|
|
15
|
+
catch {
|
|
16
|
+
return new Response(JSON.stringify({ error: 'Invalid JSON' }), {
|
|
17
|
+
status: 400,
|
|
18
|
+
headers: { 'Content-Type': 'application/json' },
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
let input;
|
|
22
|
+
try {
|
|
23
|
+
input = options?.parseInput ? await options.parseInput(req) : parseDefaultInput(body);
|
|
24
|
+
}
|
|
25
|
+
catch (err) {
|
|
26
|
+
return new Response(JSON.stringify({ error: err instanceof Error ? err.message : 'Parse error' }), { status: 400, headers: { 'Content-Type': 'application/json' } });
|
|
27
|
+
}
|
|
28
|
+
let runContext = {};
|
|
29
|
+
if (options?.beforeRun) {
|
|
30
|
+
try {
|
|
31
|
+
const ctx = await options.beforeRun(req, input);
|
|
32
|
+
if (ctx)
|
|
33
|
+
runContext = ctx;
|
|
34
|
+
}
|
|
35
|
+
catch (err) {
|
|
36
|
+
return new Response(JSON.stringify({ error: err instanceof Error ? err.message : 'Unauthorized' }), { status: 401, headers: { 'Content-Type': 'application/json' } });
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
try {
|
|
40
|
+
const result = await cogitator.run(agent, {
|
|
41
|
+
input: input.input,
|
|
42
|
+
threadId: input.threadId,
|
|
43
|
+
context: input.context,
|
|
44
|
+
...runContext,
|
|
45
|
+
});
|
|
46
|
+
if (options?.afterRun) {
|
|
47
|
+
await options.afterRun(result);
|
|
48
|
+
}
|
|
49
|
+
const response = {
|
|
50
|
+
output: result.output,
|
|
51
|
+
threadId: result.threadId,
|
|
52
|
+
usage: {
|
|
53
|
+
inputTokens: result.usage.inputTokens,
|
|
54
|
+
outputTokens: result.usage.outputTokens,
|
|
55
|
+
totalTokens: result.usage.totalTokens,
|
|
56
|
+
},
|
|
57
|
+
toolCalls: [...result.toolCalls],
|
|
58
|
+
trace: {
|
|
59
|
+
traceId: result.trace.traceId,
|
|
60
|
+
spans: [...result.trace.spans],
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
return new Response(JSON.stringify(response), {
|
|
64
|
+
headers: { 'Content-Type': 'application/json' },
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
catch (err) {
|
|
68
|
+
const message = err instanceof Error ? err.message : 'Internal server error';
|
|
69
|
+
return new Response(JSON.stringify({ error: message }), {
|
|
70
|
+
status: 500,
|
|
71
|
+
headers: { 'Content-Type': 'application/json' },
|
|
72
|
+
});
|
|
73
|
+
}
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=agent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/handlers/agent.ts"],"names":[],"mappings":"AAGA,SAAS,iBAAiB,CAAC,IAAa;IACtC,MAAM,IAAI,GAAG,IAAgF,CAAC;IAE9F,OAAO;QACL,KAAK,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;QACvD,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACxE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,SAAoB,EACpB,KAAY,EACZ,OAA6B;IAE7B,OAAO,KAAK,EAAE,GAAY,EAAqB,EAAE;QAC/C,IAAI,IAAa,CAAC;QAClB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,EAAE;gBAC7D,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAED,IAAI,KAAiB,CAAC;QACtB,IAAI,CAAC;YACH,KAAK,GAAG,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACxF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,EAC7E,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;QACJ,CAAC;QAED,IAAI,UAAU,GAA4B,EAAE,CAAC;QAC7C,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAChD,IAAI,GAAG;oBAAE,UAAU,GAAG,GAAG,CAAC;YAC5B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,EAC9E,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;gBACxC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,GAAG,UAAU;aACd,CAAC,CAAC;YAEH,IAAI,OAAO,EAAE,QAAQ,EAAE,CAAC;gBACtB,MAAM,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACjC,CAAC;YAED,MAAM,QAAQ,GAAkB;gBAC9B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;oBACrC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;oBACvC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;iBACtC;gBACD,SAAS,EAAE,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC;gBAChC,KAAK,EAAE;oBACL,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC;iBAC/B;aACF,CAAC;YAEF,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;gBAC5C,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAAC;YAC7E,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE;gBACtD,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { Cogitator, Agent } from '@cogitator-ai/core';
|
|
2
|
+
import type { ChatHandlerOptions } from '../types.js';
|
|
3
|
+
export declare function createChatHandler(cogitator: Cogitator, agent: Agent, options?: ChatHandlerOptions): (req: Request) => Promise<Response>;
|
|
4
|
+
//# sourceMappingURL=chat.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../src/handlers/chat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,kBAAkB,EAA0B,MAAM,aAAa,CAAC;AA0C9E,wBAAgB,iBAAiB,CAC/B,SAAS,EAAE,SAAS,EACpB,KAAK,EAAE,KAAK,EACZ,OAAO,CAAC,EAAE,kBAAkB,IAEd,KAAK,OAAO,KAAG,OAAO,CAAC,QAAQ,CAAC,CAkG/C"}
|
|
@@ -0,0 +1,124 @@
|
|
|
1
|
+
import { StreamWriter } from '../streaming/stream-writer.js';
|
|
2
|
+
import { generateId } from '../streaming/encoder.js';
|
|
3
|
+
const SSE_HEADERS = {
|
|
4
|
+
'Content-Type': 'text/event-stream',
|
|
5
|
+
'Cache-Control': 'no-cache',
|
|
6
|
+
Connection: 'keep-alive',
|
|
7
|
+
'x-vercel-ai-ui-message-stream': 'v1',
|
|
8
|
+
};
|
|
9
|
+
function parseDefaultInput(body) {
|
|
10
|
+
const data = body;
|
|
11
|
+
const messages = [];
|
|
12
|
+
if (Array.isArray(data.messages)) {
|
|
13
|
+
for (const msg of data.messages) {
|
|
14
|
+
if (msg && typeof msg === 'object' && 'role' in msg && 'content' in msg) {
|
|
15
|
+
messages.push({
|
|
16
|
+
id: msg.id ?? generateId('msg'),
|
|
17
|
+
role: msg.role,
|
|
18
|
+
content: String(msg.content),
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
return {
|
|
24
|
+
messages,
|
|
25
|
+
threadId: typeof data.threadId === 'string' ? data.threadId : undefined,
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
function getLastUserMessage(messages) {
|
|
29
|
+
for (let i = messages.length - 1; i >= 0; i--) {
|
|
30
|
+
if (messages[i].role === 'user') {
|
|
31
|
+
return messages[i].content;
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
return '';
|
|
35
|
+
}
|
|
36
|
+
export function createChatHandler(cogitator, agent, options) {
|
|
37
|
+
return async (req) => {
|
|
38
|
+
let body;
|
|
39
|
+
try {
|
|
40
|
+
body = await req.json();
|
|
41
|
+
}
|
|
42
|
+
catch {
|
|
43
|
+
return new Response(JSON.stringify({ error: 'Invalid JSON' }), {
|
|
44
|
+
status: 400,
|
|
45
|
+
headers: { 'Content-Type': 'application/json' },
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
let input;
|
|
49
|
+
try {
|
|
50
|
+
input = options?.parseInput ? await options.parseInput(req) : parseDefaultInput(body);
|
|
51
|
+
}
|
|
52
|
+
catch (err) {
|
|
53
|
+
return new Response(JSON.stringify({ error: err instanceof Error ? err.message : 'Parse error' }), { status: 400, headers: { 'Content-Type': 'application/json' } });
|
|
54
|
+
}
|
|
55
|
+
let runContext = {};
|
|
56
|
+
if (options?.beforeRun) {
|
|
57
|
+
try {
|
|
58
|
+
const ctx = await options.beforeRun(req, input);
|
|
59
|
+
if (ctx)
|
|
60
|
+
runContext = ctx;
|
|
61
|
+
}
|
|
62
|
+
catch (err) {
|
|
63
|
+
return new Response(JSON.stringify({ error: err instanceof Error ? err.message : 'Unauthorized' }), { status: 401, headers: { 'Content-Type': 'application/json' } });
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
const { readable, writable } = new TransformStream();
|
|
67
|
+
const sw = new StreamWriter(writable.getWriter());
|
|
68
|
+
const messageId = generateId('msg');
|
|
69
|
+
const runStream = async () => {
|
|
70
|
+
let currentTextId = generateId('txt');
|
|
71
|
+
let textStarted = false;
|
|
72
|
+
try {
|
|
73
|
+
await sw.start(messageId);
|
|
74
|
+
await sw.textStart(currentTextId);
|
|
75
|
+
textStarted = true;
|
|
76
|
+
const userMessage = getLastUserMessage(input.messages);
|
|
77
|
+
const result = await cogitator.run(agent, {
|
|
78
|
+
input: userMessage,
|
|
79
|
+
threadId: input.threadId,
|
|
80
|
+
...runContext,
|
|
81
|
+
onToken: async (token) => {
|
|
82
|
+
await sw.textDelta(currentTextId, token);
|
|
83
|
+
},
|
|
84
|
+
onToolCall: async (tc) => {
|
|
85
|
+
if (textStarted) {
|
|
86
|
+
await sw.textEnd(currentTextId);
|
|
87
|
+
textStarted = false;
|
|
88
|
+
}
|
|
89
|
+
await sw.toolCallStart(tc.id, tc.name);
|
|
90
|
+
await sw.toolCallDelta(tc.id, JSON.stringify(tc.arguments));
|
|
91
|
+
await sw.toolCallEnd(tc.id);
|
|
92
|
+
},
|
|
93
|
+
onToolResult: async (tr) => {
|
|
94
|
+
await sw.toolResult(generateId('tr'), tr.callId, tr.result);
|
|
95
|
+
currentTextId = generateId('txt');
|
|
96
|
+
await sw.textStart(currentTextId);
|
|
97
|
+
textStarted = true;
|
|
98
|
+
},
|
|
99
|
+
});
|
|
100
|
+
if (textStarted) {
|
|
101
|
+
await sw.textEnd(currentTextId);
|
|
102
|
+
}
|
|
103
|
+
await sw.finish(messageId, {
|
|
104
|
+
inputTokens: result.usage.inputTokens,
|
|
105
|
+
outputTokens: result.usage.outputTokens,
|
|
106
|
+
totalTokens: result.usage.totalTokens,
|
|
107
|
+
});
|
|
108
|
+
if (options?.afterRun) {
|
|
109
|
+
await options.afterRun(result);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
catch (err) {
|
|
113
|
+
const message = err instanceof Error ? err.message : 'Unknown error';
|
|
114
|
+
await sw.error(message);
|
|
115
|
+
}
|
|
116
|
+
finally {
|
|
117
|
+
await sw.close();
|
|
118
|
+
}
|
|
119
|
+
};
|
|
120
|
+
runStream();
|
|
121
|
+
return new Response(readable, { headers: SSE_HEADERS });
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
//# sourceMappingURL=chat.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat.js","sourceRoot":"","sources":["../../src/handlers/chat.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAErD,MAAM,WAAW,GAAG;IAClB,cAAc,EAAE,mBAAmB;IACnC,eAAe,EAAE,UAAU;IAC3B,UAAU,EAAE,YAAY;IACxB,+BAA+B,EAAE,IAAI;CAC7B,CAAC;AAEX,SAAS,iBAAiB,CAAC,IAAa;IACtC,MAAM,IAAI,GAAG,IAAmD,CAAC;IACjE,MAAM,QAAQ,GAAkB,EAAE,CAAC;IAEnC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChC,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,IAAI,GAAG,IAAI,SAAS,IAAI,GAAG,EAAE,CAAC;gBACxE,QAAQ,CAAC,IAAI,CAAC;oBACZ,EAAE,EAAG,GAAuB,CAAC,EAAE,IAAI,UAAU,CAAC,KAAK,CAAC;oBACpD,IAAI,EAAG,GAAwB,CAAC,IAAuC;oBACvE,OAAO,EAAE,MAAM,CAAE,GAA4B,CAAC,OAAO,CAAC;iBACvD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ;QACR,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACxE,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAuB;IACjD,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,SAAoB,EACpB,KAAY,EACZ,OAA4B;IAE5B,OAAO,KAAK,EAAE,GAAY,EAAqB,EAAE;QAC/C,IAAI,IAAa,CAAC;QAClB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,EAAE;gBAC7D,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAED,IAAI,KAAgB,CAAC;QACrB,IAAI,CAAC;YACH,KAAK,GAAG,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACxF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,EAC7E,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;QACJ,CAAC;QAED,IAAI,UAAU,GAA4B,EAAE,CAAC;QAC7C,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAChD,IAAI,GAAG;oBAAE,UAAU,GAAG,GAAG,CAAC;YAC5B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,EAC9E,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,eAAe,EAAc,CAAC;QACjE,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE;YAC3B,IAAI,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,WAAW,GAAG,KAAK,CAAC;YAExB,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC1B,MAAM,EAAE,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;gBAClC,WAAW,GAAG,IAAI,CAAC;gBAEnB,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAEvD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;oBACxC,KAAK,EAAE,WAAW;oBAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,GAAG,UAAU;oBACb,OAAO,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;wBAC/B,MAAM,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;oBAC3C,CAAC;oBACD,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;wBACvB,IAAI,WAAW,EAAE,CAAC;4BAChB,MAAM,EAAE,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;4BAChC,WAAW,GAAG,KAAK,CAAC;wBACtB,CAAC;wBACD,MAAM,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;wBACvC,MAAM,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;wBAC5D,MAAM,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBAC9B,CAAC;oBACD,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;wBACzB,MAAM,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;wBAC5D,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;wBAClC,MAAM,EAAE,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;wBAClC,WAAW,GAAG,IAAI,CAAC;oBACrB,CAAC;iBACF,CAAC,CAAC;gBAEH,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,EAAE,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;gBAClC,CAAC;gBAED,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE;oBACzB,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;oBACrC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;oBACvC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;iBACtC,CAAC,CAAC;gBAEH,IAAI,OAAO,EAAE,QAAQ,EAAE,CAAC;oBACtB,MAAM,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;gBACrE,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1B,CAAC;oBAAS,CAAC;gBACT,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;YACnB,CAAC;QACH,CAAC,CAAC;QAEF,SAAS,EAAE,CAAC;QAEZ,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC;AACJ,CAAC"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
export { createChatHandler } from './handlers/chat.js';
|
|
2
|
+
export { createAgentHandler } from './handlers/agent.js';
|
|
3
|
+
export type { ChatMessage, ChatInput, AgentInput, ChatHandlerOptions, AgentHandlerOptions, AgentResponse, } from './types.js';
|
|
4
|
+
export { StreamWriter } from './streaming/stream-writer.js';
|
|
5
|
+
export { encodeSSE, generateId } from './streaming/encoder.js';
|
|
6
|
+
export type { StreamEvent, Usage } from './streaming/protocol.js';
|
|
7
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAEzD,YAAY,EACV,WAAW,EACX,SAAS,EACT,UAAU,EACV,kBAAkB,EAClB,mBAAmB,EACnB,aAAa,GACd,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAC/D,YAAY,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC"}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { createChatHandler } from './handlers/chat.js';
|
|
2
|
+
export { createAgentHandler } from './handlers/agent.js';
|
|
3
|
+
export { StreamWriter } from './streaming/stream-writer.js';
|
|
4
|
+
export { encodeSSE, generateId } from './streaming/encoder.js';
|
|
5
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAWzD,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../src/streaming/encoder.ts"],"names":[],"mappings":"AAEA,wBAAgB,SAAS,CAAC,IAAI,EAAE,OAAO,GAAG,UAAU,CAEnD;AAED,wBAAgB,UAAU,IAAI,UAAU,CAEvC;AAID,wBAAgB,UAAU,CAAC,MAAM,GAAE,MAAa,GAAG,MAAM,CAKxD"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
const encoder = new TextEncoder();
|
|
2
|
+
export function encodeSSE(data) {
|
|
3
|
+
return encoder.encode(`data: ${JSON.stringify(data)}\n\n`);
|
|
4
|
+
}
|
|
5
|
+
export function encodeDone() {
|
|
6
|
+
return encoder.encode('data: [DONE]\n\n');
|
|
7
|
+
}
|
|
8
|
+
let counter = 0;
|
|
9
|
+
export function generateId(prefix = 'id') {
|
|
10
|
+
const timestamp = Date.now().toString(36);
|
|
11
|
+
const random = Math.random().toString(36).substring(2, 8);
|
|
12
|
+
const count = (counter++).toString(36);
|
|
13
|
+
return `${prefix}_${timestamp}${random}${count}`;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=encoder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"encoder.js","sourceRoot":"","sources":["../../src/streaming/encoder.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAElC,MAAM,UAAU,SAAS,CAAC,IAAa;IACrC,OAAO,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC7D,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,OAAO,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAC5C,CAAC;AAED,IAAI,OAAO,GAAG,CAAC,CAAC;AAEhB,MAAM,UAAU,UAAU,CAAC,SAAiB,IAAI;IAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1D,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,OAAO,GAAG,MAAM,IAAI,SAAS,GAAG,MAAM,GAAG,KAAK,EAAE,CAAC;AACnD,CAAC"}
|