@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.
Files changed (54) hide show
  1. package/README.md +369 -0
  2. package/dist/client/index.d.ts +4 -0
  3. package/dist/client/index.d.ts.map +1 -0
  4. package/dist/client/index.js +4 -0
  5. package/dist/client/index.js.map +1 -0
  6. package/dist/client/retry.d.ts +3 -0
  7. package/dist/client/retry.d.ts.map +1 -0
  8. package/dist/client/retry.js +47 -0
  9. package/dist/client/retry.js.map +1 -0
  10. package/dist/client/sse-parser.d.ts +3 -0
  11. package/dist/client/sse-parser.d.ts.map +1 -0
  12. package/dist/client/sse-parser.js +39 -0
  13. package/dist/client/sse-parser.js.map +1 -0
  14. package/dist/client/use-agent.d.ts +3 -0
  15. package/dist/client/use-agent.d.ts.map +1 -0
  16. package/dist/client/use-agent.js +53 -0
  17. package/dist/client/use-agent.js.map +1 -0
  18. package/dist/client/use-chat-state.d.ts +52 -0
  19. package/dist/client/use-chat-state.d.ts.map +1 -0
  20. package/dist/client/use-chat-state.js +68 -0
  21. package/dist/client/use-chat-state.js.map +1 -0
  22. package/dist/client/use-chat.d.ts +3 -0
  23. package/dist/client/use-chat.d.ts.map +1 -0
  24. package/dist/client/use-chat.js +215 -0
  25. package/dist/client/use-chat.js.map +1 -0
  26. package/dist/handlers/agent.d.ts +4 -0
  27. package/dist/handlers/agent.d.ts.map +1 -0
  28. package/dist/handlers/agent.js +76 -0
  29. package/dist/handlers/agent.js.map +1 -0
  30. package/dist/handlers/chat.d.ts +4 -0
  31. package/dist/handlers/chat.d.ts.map +1 -0
  32. package/dist/handlers/chat.js +124 -0
  33. package/dist/handlers/chat.js.map +1 -0
  34. package/dist/index.d.ts +7 -0
  35. package/dist/index.d.ts.map +1 -0
  36. package/dist/index.js +5 -0
  37. package/dist/index.js.map +1 -0
  38. package/dist/streaming/encoder.d.ts +4 -0
  39. package/dist/streaming/encoder.d.ts.map +1 -0
  40. package/dist/streaming/encoder.js +15 -0
  41. package/dist/streaming/encoder.js.map +1 -0
  42. package/dist/streaming/protocol.d.ts +64 -0
  43. package/dist/streaming/protocol.d.ts.map +1 -0
  44. package/dist/streaming/protocol.js +31 -0
  45. package/dist/streaming/protocol.js.map +1 -0
  46. package/dist/streaming/stream-writer.d.ts +19 -0
  47. package/dist/streaming/stream-writer.d.ts.map +1 -0
  48. package/dist/streaming/stream-writer.js +57 -0
  49. package/dist/streaming/stream-writer.js.map +1 -0
  50. package/dist/types.d.ts +97 -0
  51. package/dist/types.d.ts.map +1 -0
  52. package/dist/types.js +2 -0
  53. package/dist/types.js.map +1 -0
  54. 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,3 @@
1
+ import type { UseChatOptions, UseChatReturn } from '../types.js';
2
+ export declare function useCogitatorChat(options: UseChatOptions): UseChatReturn;
3
+ //# sourceMappingURL=use-chat.d.ts.map
@@ -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"}
@@ -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,4 @@
1
+ export declare function encodeSSE(data: unknown): Uint8Array;
2
+ export declare function encodeDone(): Uint8Array;
3
+ export declare function generateId(prefix?: string): string;
4
+ //# sourceMappingURL=encoder.d.ts.map
@@ -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"}