@convex-dev/agent 0.1.15-alpha.0 → 0.1.15-alpha.1
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/dist/client/index.d.ts +494 -173
- package/dist/client/index.d.ts.map +1 -1
- package/dist/client/index.js +119 -166
- package/dist/client/index.js.map +1 -1
- package/dist/client/listMessages.d.ts +22 -0
- package/dist/client/listMessages.d.ts.map +1 -0
- package/dist/client/listMessages.js +25 -0
- package/dist/client/listMessages.js.map +1 -0
- package/dist/client/search.d.ts +162 -0
- package/dist/client/search.d.ts.map +1 -0
- package/dist/client/search.js +113 -0
- package/dist/client/search.js.map +1 -0
- package/dist/client/streaming.d.ts +17 -3
- package/dist/client/streaming.d.ts.map +1 -1
- package/dist/client/streaming.js +32 -0
- package/dist/client/streaming.js.map +1 -1
- package/dist/client/types.d.ts +5 -1
- package/dist/client/types.d.ts.map +1 -1
- package/dist/component/_generated/api.d.ts +2 -2
- package/dist/component/messages.d.ts +3 -3
- package/dist/component/messages.d.ts.map +1 -1
- package/dist/component/messages.js +12 -7
- package/dist/component/messages.js.map +1 -1
- package/dist/component/vector/index.js +3 -3
- package/dist/component/vector/index.js.map +1 -1
- package/dist/validators.d.ts +41 -52
- package/dist/validators.d.ts.map +1 -1
- package/dist/validators.js +1 -8
- package/dist/validators.js.map +1 -1
- package/package.json +4 -3
- package/src/client/index.ts +245 -268
- package/src/client/listMessages.ts +38 -0
- package/src/client/search.ts +172 -0
- package/src/client/streaming.ts +49 -2
- package/src/client/types.ts +5 -1
- package/src/component/_generated/api.d.ts +2 -2
- package/src/component/messages.ts +13 -7
- package/src/component/vector/index.ts +4 -4
- package/src/validators.ts +2 -10
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* List messages from a thread.
|
|
3
|
+
* @param ctx A ctx object from a query, mutation, or action.
|
|
4
|
+
* @param component The agent component, usually `components.agent`.
|
|
5
|
+
* @param args.threadId The thread to list messages from.
|
|
6
|
+
* @param args.paginationOpts Pagination options (e.g. via usePaginatedQuery).
|
|
7
|
+
* @param args.excludeToolMessages Whether to exclude tool messages.
|
|
8
|
+
* False by default.
|
|
9
|
+
* @param args.statuses What statuses to include. All by default.
|
|
10
|
+
* @returns The MessageDoc's in a format compatible with usePaginatedQuery.
|
|
11
|
+
*/
|
|
12
|
+
export async function listMessages(ctx, component, args) {
|
|
13
|
+
if (args.paginationOpts.numItems === 0) {
|
|
14
|
+
return {
|
|
15
|
+
page: [],
|
|
16
|
+
isDone: true,
|
|
17
|
+
continueCursor: args.paginationOpts.cursor ?? "",
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
return ctx.runQuery(component.messages.listMessagesByThreadId, {
|
|
21
|
+
order: "desc",
|
|
22
|
+
...args,
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=listMessages.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"listMessages.js","sourceRoot":"","sources":["../../src/client/listMessages.ts"],"names":[],"mappings":"AAKA;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,GAAgB,EAChB,SAAyB,EACzB,IAKC;IAED,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QACvC,OAAO;YACL,IAAI,EAAE,EAAE;YACR,MAAM,EAAE,IAAI;YACZ,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,EAAE;SACjD,CAAC;IACJ,CAAC;IACD,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE;QAC7D,KAAK,EAAE,MAAM;QACb,GAAG,IAAI;KACR,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,162 @@
|
|
|
1
|
+
import type { AgentComponent, ContextOptions, RunActionCtx, RunQueryCtx } from "./types.js";
|
|
2
|
+
import type { MessageDoc } from "../component/schema.js";
|
|
3
|
+
import type { CoreMessage } from "ai";
|
|
4
|
+
export type GetEmbedding = (text: string) => Promise<{
|
|
5
|
+
vector: number[];
|
|
6
|
+
vectorModel: string;
|
|
7
|
+
vectorScoreThreshold?: number;
|
|
8
|
+
}>;
|
|
9
|
+
/**
|
|
10
|
+
* Fetch the context messages for a thread.
|
|
11
|
+
* @param ctx Either a query, mutation, or action ctx.
|
|
12
|
+
* If it is not an action context, you can't do text or
|
|
13
|
+
* vector search.
|
|
14
|
+
* @param args The associated thread, user, message
|
|
15
|
+
* @returns
|
|
16
|
+
*/
|
|
17
|
+
export declare function fetchContextMessages(ctx: RunQueryCtx | RunActionCtx, component: AgentComponent, args: {
|
|
18
|
+
userId: string | undefined;
|
|
19
|
+
threadId: string | undefined;
|
|
20
|
+
messages: CoreMessage[];
|
|
21
|
+
/**
|
|
22
|
+
* If provided, it will search for messages up to and including this message.
|
|
23
|
+
* Note: if this is far in the past, text and vector search results may be more
|
|
24
|
+
* limited, as it's post-filtering the results.
|
|
25
|
+
*/
|
|
26
|
+
upToAndIncludingMessageId?: string;
|
|
27
|
+
contextOptions: ContextOptions;
|
|
28
|
+
getEmbedding?: GetEmbedding;
|
|
29
|
+
}): Promise<MessageDoc[]>;
|
|
30
|
+
/**
|
|
31
|
+
* Filter out tool messages that don't have both a tool call and response.
|
|
32
|
+
* @param docs The messages to filter.
|
|
33
|
+
* @returns The filtered messages.
|
|
34
|
+
*/
|
|
35
|
+
export declare function filterOutOrphanedToolMessages(docs: MessageDoc[]): {
|
|
36
|
+
id?: string | undefined;
|
|
37
|
+
userId?: string | undefined;
|
|
38
|
+
embeddingId?: string | undefined;
|
|
39
|
+
fileIds?: string[] | undefined;
|
|
40
|
+
error?: string | undefined;
|
|
41
|
+
agentName?: string | undefined;
|
|
42
|
+
model?: string | undefined;
|
|
43
|
+
provider?: string | undefined;
|
|
44
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
45
|
+
message?: {
|
|
46
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
47
|
+
role: "user";
|
|
48
|
+
content: string | ({
|
|
49
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
50
|
+
type: "text";
|
|
51
|
+
text: string;
|
|
52
|
+
} | {
|
|
53
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
54
|
+
mimeType?: string | undefined;
|
|
55
|
+
type: "image";
|
|
56
|
+
image: string | ArrayBuffer;
|
|
57
|
+
} | {
|
|
58
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
59
|
+
filename?: string | undefined;
|
|
60
|
+
type: "file";
|
|
61
|
+
mimeType: string;
|
|
62
|
+
data: string | ArrayBuffer;
|
|
63
|
+
})[];
|
|
64
|
+
} | {
|
|
65
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
66
|
+
role: "assistant";
|
|
67
|
+
content: string | ({
|
|
68
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
69
|
+
type: "text";
|
|
70
|
+
text: string;
|
|
71
|
+
} | {
|
|
72
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
73
|
+
filename?: string | undefined;
|
|
74
|
+
type: "file";
|
|
75
|
+
mimeType: string;
|
|
76
|
+
data: string | ArrayBuffer;
|
|
77
|
+
} | {
|
|
78
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
79
|
+
signature?: string | undefined;
|
|
80
|
+
type: "reasoning";
|
|
81
|
+
text: string;
|
|
82
|
+
} | {
|
|
83
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
84
|
+
type: "redacted-reasoning";
|
|
85
|
+
data: string;
|
|
86
|
+
} | {
|
|
87
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
88
|
+
type: "tool-call";
|
|
89
|
+
toolCallId: string;
|
|
90
|
+
toolName: string;
|
|
91
|
+
args: any;
|
|
92
|
+
})[];
|
|
93
|
+
} | {
|
|
94
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
95
|
+
role: "tool";
|
|
96
|
+
content: {
|
|
97
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
98
|
+
args?: any;
|
|
99
|
+
experimental_content?: ({
|
|
100
|
+
type: "text";
|
|
101
|
+
text: string;
|
|
102
|
+
} | {
|
|
103
|
+
mimeType?: string | undefined;
|
|
104
|
+
type: "image";
|
|
105
|
+
data: string;
|
|
106
|
+
})[] | undefined;
|
|
107
|
+
isError?: boolean | undefined;
|
|
108
|
+
type: "tool-result";
|
|
109
|
+
toolCallId: string;
|
|
110
|
+
toolName: string;
|
|
111
|
+
result: any;
|
|
112
|
+
}[];
|
|
113
|
+
} | {
|
|
114
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
115
|
+
role: "system";
|
|
116
|
+
content: string;
|
|
117
|
+
} | undefined;
|
|
118
|
+
text?: string | undefined;
|
|
119
|
+
reasoning?: string | undefined;
|
|
120
|
+
usage?: {
|
|
121
|
+
promptTokens: number;
|
|
122
|
+
completionTokens: number;
|
|
123
|
+
totalTokens: number;
|
|
124
|
+
} | undefined;
|
|
125
|
+
providerMetadata?: Record<string, Record<string, any>> | undefined;
|
|
126
|
+
sources?: {
|
|
127
|
+
title?: string | undefined;
|
|
128
|
+
providerOptions?: Record<string, Record<string, any>> | undefined;
|
|
129
|
+
id: string;
|
|
130
|
+
sourceType: "url";
|
|
131
|
+
url: string;
|
|
132
|
+
}[] | undefined;
|
|
133
|
+
reasoningDetails?: ({
|
|
134
|
+
signature?: string | undefined;
|
|
135
|
+
type: "text";
|
|
136
|
+
text: string;
|
|
137
|
+
} | {
|
|
138
|
+
type: "redacted";
|
|
139
|
+
data: string;
|
|
140
|
+
})[] | undefined;
|
|
141
|
+
warnings?: ({
|
|
142
|
+
details?: string | undefined;
|
|
143
|
+
type: "unsupported-setting";
|
|
144
|
+
setting: string;
|
|
145
|
+
} | {
|
|
146
|
+
details?: string | undefined;
|
|
147
|
+
type: "unsupported-tool";
|
|
148
|
+
tool: any;
|
|
149
|
+
} | {
|
|
150
|
+
type: "other";
|
|
151
|
+
message: string;
|
|
152
|
+
})[] | undefined;
|
|
153
|
+
finishReason?: "length" | "error" | "other" | "stop" | "content-filter" | "tool-calls" | "unknown" | undefined;
|
|
154
|
+
_id: string;
|
|
155
|
+
_creationTime: number;
|
|
156
|
+
status: "pending" | "success" | "failed";
|
|
157
|
+
order: number;
|
|
158
|
+
threadId: string;
|
|
159
|
+
stepOrder: number;
|
|
160
|
+
tool: boolean;
|
|
161
|
+
}[];
|
|
162
|
+
//# sourceMappingURL=search.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/client/search.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,cAAc,EACd,YAAY,EACZ,WAAW,EACZ,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AAUtC,MAAM,MAAM,YAAY,GAAG,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC;IACnD,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B,CAAC,CAAC;AAEH;;;;;;;GAOG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,WAAW,GAAG,YAAY,EAC/B,SAAS,EAAE,cAAc,EACzB,IAAI,EAAE;IACJ,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB;;;;OAIG;IACH,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,cAAc,EAAE,cAAc,CAAC;IAC/B,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B,GACA,OAAO,CAAC,UAAU,EAAE,CAAC,CA4FvB;AAED;;;;GAIG;AACH,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,UAAU,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAyB/D"}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
import { assert } from "convex-helpers";
|
|
2
|
+
import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, } from "../shared.js";
|
|
3
|
+
const DEFAULT_VECTOR_SCORE_THRESHOLD = 0.0;
|
|
4
|
+
/**
|
|
5
|
+
* Fetch the context messages for a thread.
|
|
6
|
+
* @param ctx Either a query, mutation, or action ctx.
|
|
7
|
+
* If it is not an action context, you can't do text or
|
|
8
|
+
* vector search.
|
|
9
|
+
* @param args The associated thread, user, message
|
|
10
|
+
* @returns
|
|
11
|
+
*/
|
|
12
|
+
export async function fetchContextMessages(ctx, component, args) {
|
|
13
|
+
assert(args.userId || args.threadId, "Specify userId or threadId");
|
|
14
|
+
const opts = args.contextOptions;
|
|
15
|
+
// Fetch the latest messages from the thread
|
|
16
|
+
let included;
|
|
17
|
+
const contextMessages = [];
|
|
18
|
+
if (args.threadId &&
|
|
19
|
+
(opts.recentMessages !== 0 || args.upToAndIncludingMessageId)) {
|
|
20
|
+
const { page } = await ctx.runQuery(component.messages.listMessagesByThreadId, {
|
|
21
|
+
threadId: args.threadId,
|
|
22
|
+
excludeToolMessages: opts.excludeToolMessages,
|
|
23
|
+
paginationOpts: {
|
|
24
|
+
numItems: opts.recentMessages ?? DEFAULT_RECENT_MESSAGES,
|
|
25
|
+
cursor: null,
|
|
26
|
+
},
|
|
27
|
+
upToAndIncludingMessageId: args.upToAndIncludingMessageId,
|
|
28
|
+
order: "desc",
|
|
29
|
+
statuses: ["success"],
|
|
30
|
+
});
|
|
31
|
+
included = new Set(page.map((m) => m._id));
|
|
32
|
+
contextMessages.push(
|
|
33
|
+
// Reverse since we fetched in descending order
|
|
34
|
+
...page.reverse());
|
|
35
|
+
}
|
|
36
|
+
if (opts.searchOptions?.textSearch || opts.searchOptions?.vectorSearch) {
|
|
37
|
+
const targetMessage = contextMessages.find((m) => m._id === args.upToAndIncludingMessageId)?.message;
|
|
38
|
+
const messagesToSearch = targetMessage ? [targetMessage] : args.messages;
|
|
39
|
+
if (!("runAction" in ctx)) {
|
|
40
|
+
throw new Error("searchUserMessages only works in an action");
|
|
41
|
+
}
|
|
42
|
+
const lastMessage = messagesToSearch.at(-1);
|
|
43
|
+
assert(lastMessage, "No messages to search");
|
|
44
|
+
const text = extractText(lastMessage);
|
|
45
|
+
assert(text, `No text to search in message ${JSON.stringify(lastMessage)}`);
|
|
46
|
+
assert(!args.contextOptions?.searchOptions?.vectorSearch || "runAction" in ctx, "You must do vector search from an action");
|
|
47
|
+
if (opts.searchOptions?.vectorSearch && !args.getEmbedding) {
|
|
48
|
+
throw new Error("You must provide an embedding and embeddingModel to use vector search");
|
|
49
|
+
}
|
|
50
|
+
const embeddingFields = opts.searchOptions?.vectorSearch
|
|
51
|
+
? await args.getEmbedding?.(text)
|
|
52
|
+
: undefined;
|
|
53
|
+
const searchMessages = await ctx.runAction(component.messages.searchMessages, {
|
|
54
|
+
searchAllMessagesForUserId: opts?.searchOtherThreads
|
|
55
|
+
? args.userId ??
|
|
56
|
+
(args.threadId &&
|
|
57
|
+
(await ctx.runQuery(component.threads.getThread, {
|
|
58
|
+
threadId: args.threadId,
|
|
59
|
+
}))?.userId)
|
|
60
|
+
: undefined,
|
|
61
|
+
threadId: args.threadId,
|
|
62
|
+
beforeMessageId: args.upToAndIncludingMessageId,
|
|
63
|
+
limit: opts.searchOptions?.limit ?? 10,
|
|
64
|
+
messageRange: {
|
|
65
|
+
...DEFAULT_MESSAGE_RANGE,
|
|
66
|
+
...opts.searchOptions?.messageRange,
|
|
67
|
+
},
|
|
68
|
+
text,
|
|
69
|
+
vectorScoreThreshold: opts.searchOptions?.vectorScoreThreshold ??
|
|
70
|
+
DEFAULT_VECTOR_SCORE_THRESHOLD,
|
|
71
|
+
...embeddingFields,
|
|
72
|
+
});
|
|
73
|
+
// TODO: track what messages we used for context
|
|
74
|
+
contextMessages.unshift(...searchMessages.filter((m) => !included?.has(m._id)));
|
|
75
|
+
}
|
|
76
|
+
// Ensure we don't include tool messages without a corresponding tool call
|
|
77
|
+
return filterOutOrphanedToolMessages(contextMessages.sort((a, b) =>
|
|
78
|
+
// Sort the raw MessageDocs by order and stepOrder
|
|
79
|
+
a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order));
|
|
80
|
+
}
|
|
81
|
+
/**
|
|
82
|
+
* Filter out tool messages that don't have both a tool call and response.
|
|
83
|
+
* @param docs The messages to filter.
|
|
84
|
+
* @returns The filtered messages.
|
|
85
|
+
*/
|
|
86
|
+
export function filterOutOrphanedToolMessages(docs) {
|
|
87
|
+
const toolCallIds = new Set();
|
|
88
|
+
const result = [];
|
|
89
|
+
for (const doc of docs) {
|
|
90
|
+
if (doc.message?.role === "assistant" &&
|
|
91
|
+
Array.isArray(doc.message.content)) {
|
|
92
|
+
for (const content of doc.message.content) {
|
|
93
|
+
if (content.type === "tool-call") {
|
|
94
|
+
toolCallIds.add(content.toolCallId);
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
result.push(doc);
|
|
98
|
+
}
|
|
99
|
+
else if (doc.message?.role === "tool") {
|
|
100
|
+
if (doc.message.content.every((c) => toolCallIds.has(c.toolCallId))) {
|
|
101
|
+
result.push(doc);
|
|
102
|
+
}
|
|
103
|
+
else {
|
|
104
|
+
console.debug("Filtering out orphaned tool message", doc);
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
else {
|
|
108
|
+
result.push(doc);
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
return result;
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/client/search.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EACL,qBAAqB,EACrB,uBAAuB,EACvB,WAAW,GACZ,MAAM,cAAc,CAAC;AAEtB,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAQ3C;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,GAA+B,EAC/B,SAAyB,EACzB,IAYC;IAED,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,4BAA4B,CAAC,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC;IACjC,4CAA4C;IAC5C,IAAI,QAAiC,CAAC;IACtC,MAAM,eAAe,GAAiB,EAAE,CAAC;IACzC,IACE,IAAI,CAAC,QAAQ;QACb,CAAC,IAAI,CAAC,cAAc,KAAK,CAAC,IAAI,IAAI,CAAC,yBAAyB,CAAC,EAC7D,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,GAAG,CAAC,QAAQ,CACjC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EACzC;YACE,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;YAC7C,cAAc,EAAE;gBACd,QAAQ,EAAE,IAAI,CAAC,cAAc,IAAI,uBAAuB;gBACxD,MAAM,EAAE,IAAI;aACb;YACD,yBAAyB,EAAE,IAAI,CAAC,yBAAyB;YACzD,KAAK,EAAE,MAAM;YACb,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB,CACF,CAAC;QACF,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,eAAe,CAAC,IAAI;QAClB,+CAA+C;QAC/C,GAAG,IAAI,CAAC,OAAO,EAAE,CAClB,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,aAAa,EAAE,UAAU,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,EAAE,CAAC;QACvE,MAAM,aAAa,GAAG,eAAe,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,yBAAyB,CAChD,EAAE,OAAO,CAAC;QACX,MAAM,gBAAgB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;QACzE,IAAI,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,WAAW,GAAG,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;QAC7C,MAAM,CAAC,WAAW,EAAE,uBAAuB,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,EAAE,gCAAgC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC5E,MAAM,CACJ,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,EAAE,YAAY,IAAI,WAAW,IAAI,GAAG,EACvE,0CAA0C,CAC3C,CAAC;QACF,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAC3D,MAAM,IAAI,KAAK,CACb,uEAAuE,CACxE,CAAC;QACJ,CAAC;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY;YACtD,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;YACjC,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,SAAS,CACxC,SAAS,CAAC,QAAQ,CAAC,cAAc,EACjC;YACE,0BAA0B,EAAE,IAAI,EAAE,kBAAkB;gBAClD,CAAC,CAAC,IAAI,CAAC,MAAM;oBACX,CAAC,IAAI,CAAC,QAAQ;wBACZ,CACE,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE;4BAC9C,QAAQ,EAAE,IAAI,CAAC,QAAQ;yBACxB,CAAC,CACH,EAAE,MAAM,CAAC;gBACd,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe,EAAE,IAAI,CAAC,yBAAyB;YAC/C,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;YACtC,YAAY,EAAE;gBACZ,GAAG,qBAAqB;gBACxB,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY;aACpC;YACD,IAAI;YACJ,oBAAoB,EAClB,IAAI,CAAC,aAAa,EAAE,oBAAoB;gBACxC,8BAA8B;YAChC,GAAG,eAAe;SACnB,CACF,CAAC;QACF,gDAAgD;QAChD,eAAe,CAAC,OAAO,CACrB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CACvD,CAAC;IACJ,CAAC;IACD,0EAA0E;IAC1E,OAAO,6BAA6B,CAClC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;IAC5B,kDAAkD;IAClD,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpE,CACF,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,6BAA6B,CAAC,IAAkB;IAC9D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IACE,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,WAAW;YACjC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAClC,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC1C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;YACH,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;aAAM,IAAI,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACxC,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;gBACpE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -1,7 +1,21 @@
|
|
|
1
1
|
import { type ChunkDetector, type StreamTextTransform, type ToolSet } from "ai";
|
|
2
|
-
import type { ProviderOptions, TextStreamPart } from "../validators.js";
|
|
3
|
-
import type {
|
|
4
|
-
import type { RunActionCtx } from "./types.js";
|
|
2
|
+
import type { ProviderOptions, StreamArgs, TextStreamPart } from "../validators.js";
|
|
3
|
+
import type { MessageDoc } from "../component/schema.js";
|
|
4
|
+
import type { AgentComponent, RunActionCtx, RunQueryCtx, SyncStreamsReturnValue } from "./types.js";
|
|
5
|
+
/**
|
|
6
|
+
* A function that handles fetching stream deltas, used with the React hooks
|
|
7
|
+
* `useThreadMessages` or `useStreamingThreadMessages`.
|
|
8
|
+
* @param ctx A ctx object from a query, mutation, or action.
|
|
9
|
+
* @param component The agent component, usually `components.agent`.
|
|
10
|
+
* @param args.threadId The thread to sync streams for.
|
|
11
|
+
* @param args.streamArgs The stream arguments with per-stream cursors.
|
|
12
|
+
* @returns The deltas for each stream from their existing cursor.
|
|
13
|
+
*/
|
|
14
|
+
export declare function syncStreams(ctx: RunQueryCtx, component: AgentComponent, args: {
|
|
15
|
+
threadId: string;
|
|
16
|
+
streamArgs: StreamArgs | undefined;
|
|
17
|
+
includeStatuses?: ("streaming" | "finished" | "aborted")[];
|
|
18
|
+
}): Promise<SyncStreamsReturnValue | undefined>;
|
|
5
19
|
export type StreamingOptions = {
|
|
6
20
|
/**
|
|
7
21
|
* The minimum granularity of deltas to save.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/client/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,aAAa,EAElB,KAAK,mBAAmB,EACxB,KAAK,OAAO,EACb,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EACV,eAAe,
|
|
1
|
+
{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/client/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,aAAa,EAElB,KAAK,mBAAmB,EACxB,KAAK,OAAO,EACb,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EACV,eAAe,EACf,UAAU,EAEV,cAAc,EACf,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EACV,cAAc,EACd,YAAY,EACZ,WAAW,EACX,sBAAsB,EACvB,MAAM,YAAY,CAAC;AAEpB;;;;;;;;GAQG;AACH,wBAAsB,WAAW,CAC/B,GAAG,EAAE,WAAW,EAChB,SAAS,EAAE,cAAc,EACzB,IAAI,EAAE;IACJ,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,UAAU,GAAG,SAAS,CAAC;IAEnC,eAAe,CAAC,EAAE,CAAC,WAAW,GAAG,UAAU,GAAG,SAAS,CAAC,EAAE,CAAC;CAC5D,GACA,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC,CAoB7C;AAGD,MAAM,MAAM,gBAAgB,GAAG;IAC7B;;;;;;OAMG;IACH,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,GAAG,aAAa,CAAC;IACpD;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AACF,eAAO,MAAM,yBAAyB;;;CAIV,CAAC;AAE7B,wBAAgB,eAAe,CAAC,KAAK,SAAS,OAAO,EACnD,OAAO,EAAE,gBAAgB,GAAG,OAAO,GAAG,SAAS,EAC/C,QAAQ,EACJ,mBAAmB,CAAC,KAAK,CAAC,GAC1B,KAAK,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,GACjC,SAAS,yEAgBd;AAED,qBAAa,aAAa;;aAYN,SAAS,EAAE,cAAc;aACzB,GAAG,EAAE,YAAY;aAEjB,QAAQ,EAAE;QACxB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;QAC9B,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;QAC1B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;QAC7B,eAAe,EAAE,eAAe,GAAG,SAAS,CAAC;QAC7C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;QAC1B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;QAC9B,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;KACtC;IAxBI,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,SAAgB,OAAO,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;IAO7C,eAAe,EAAE,eAAe,CAAC;gBAGtB,SAAS,EAAE,cAAc,EACzB,GAAG,EAAE,YAAY,EACjC,OAAO,EAAE,IAAI,GAAG,gBAAgB,EAChB,QAAQ,EAAE;QACxB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;QAC9B,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;QAC1B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;QAC7B,eAAe,EAAE,eAAe,GAAG,SAAS,CAAC;QAC7C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;QAC1B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;QAC9B,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;KACtC;IA0BU,QAAQ,CAAC,KAAK,EAAE,cAAc,EAAE;IAsEhC,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE;CA0B3C"}
|
package/dist/client/streaming.js
CHANGED
|
@@ -1,4 +1,36 @@
|
|
|
1
1
|
import { smoothStream, } from "ai";
|
|
2
|
+
/**
|
|
3
|
+
* A function that handles fetching stream deltas, used with the React hooks
|
|
4
|
+
* `useThreadMessages` or `useStreamingThreadMessages`.
|
|
5
|
+
* @param ctx A ctx object from a query, mutation, or action.
|
|
6
|
+
* @param component The agent component, usually `components.agent`.
|
|
7
|
+
* @param args.threadId The thread to sync streams for.
|
|
8
|
+
* @param args.streamArgs The stream arguments with per-stream cursors.
|
|
9
|
+
* @returns The deltas for each stream from their existing cursor.
|
|
10
|
+
*/
|
|
11
|
+
export async function syncStreams(ctx, component, args) {
|
|
12
|
+
if (!args.streamArgs)
|
|
13
|
+
return undefined;
|
|
14
|
+
if (args.streamArgs.kind === "list") {
|
|
15
|
+
return {
|
|
16
|
+
kind: "list",
|
|
17
|
+
messages: await ctx.runQuery(component.streams.list, {
|
|
18
|
+
threadId: args.threadId,
|
|
19
|
+
startOrder: args.streamArgs.startOrder,
|
|
20
|
+
statuses: args.includeStatuses,
|
|
21
|
+
}),
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
else {
|
|
25
|
+
return {
|
|
26
|
+
kind: "deltas",
|
|
27
|
+
deltas: await ctx.runQuery(component.streams.listDeltas, {
|
|
28
|
+
threadId: args.threadId,
|
|
29
|
+
cursors: args.streamArgs.cursors,
|
|
30
|
+
}),
|
|
31
|
+
};
|
|
32
|
+
}
|
|
33
|
+
}
|
|
2
34
|
export const DEFAULT_STREAMING_OPTIONS = {
|
|
3
35
|
// This chunks by sentences / clauses. Punctuation followed by whitespace.
|
|
4
36
|
chunking: /[\p{P}\s]/u,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../src/client/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,YAAY,GAGb,MAAM,IAAI,CAAC;
|
|
1
|
+
{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../src/client/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,YAAY,GAGb,MAAM,IAAI,CAAC;AAeZ;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,GAAgB,EAChB,SAAyB,EACzB,IAKC;IAED,IAAI,CAAC,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IACvC,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACpC,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE;gBACnD,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU;gBACtC,QAAQ,EAAE,IAAI,CAAC,eAAe;aAC/B,CAAC;SACH,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,UAAU,EAAE;gBACvD,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO;aACjC,CAAC;SACH,CAAC;IACJ,CAAC;AACH,CAAC;AAkBD,MAAM,CAAC,MAAM,yBAAyB,GAAG;IACvC,0EAA0E;IAC1E,QAAQ,EAAE,YAAY;IACtB,UAAU,EAAE,GAAG;CACW,CAAC;AAE7B,MAAM,UAAU,eAAe,CAC7B,OAA+C,EAC/C,QAGa;IAEb,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,MAAM,QAAQ,GACZ,OAAO,OAAO,KAAK,SAAS;QAC1B,CAAC,CAAC,yBAAyB,CAAC,QAAQ;QACpC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;IACvB,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxC,CAAC,CAAC,QAAQ;QACV,CAAC,CAAC,QAAQ;YACR,CAAC,CAAC,CAAC,QAAQ,CAAC;YACZ,CAAC,CAAC,EAAE,CAAC;IACT,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IAC7D,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,OAAO,aAAa;IAYN;IACA;IAEA;IAdX,QAAQ,CAAqB;IACpB,OAAO,CAA6B;IACpD,UAAU,GAAqB,EAAE,CAAC;IAClC,UAAU,CAAS;IACnB,cAAc,CAAS;IACvB,YAAY,GAAW,CAAC,CAAC;IACzB,aAAa,CAA4B;IACzC,OAAO,GAAW,CAAC,CAAC;IACb,eAAe,CAAkB;IAExC,YACkB,SAAyB,EACzB,GAAiB,EACjC,OAAgC,EAChB,QAUf;QAbe,cAAS,GAAT,SAAS,CAAgB;QACzB,QAAG,GAAH,GAAG,CAAc;QAEjB,aAAQ,GAAR,QAAQ,CAUvB;QAED,IAAI,CAAC,OAAO;YACV,OAAO,OAAO,KAAK,SAAS;gBAC1B,CAAC,CAAC,yBAAyB;gBAC3B,CAAC,CAAC;oBACE,GAAG,yBAAyB;oBAC5B,GAAG,OAAO;iBACX,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,KAAK,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,CAAC,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACpD,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,QAAQ,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBACxD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAClB,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE;wBACvD,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,MAAM,EAAE,aAAa;qBACtB,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC/B,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACM,KAAK,CAAC,QAAQ,CAAC,KAAuB;QAC3C,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CACxC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,EAC7B;gBACE,GAAG,IAAI,CAAC,QAAQ;gBAChB,KAAK,EAAE,IAAI,CAAC,UAAU;gBACtB,SAAS,EAAE,IAAI,CAAC,cAAc;aAC/B,CACF,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QAC/B,IACE,CAAC,IAAI,CAAC,aAAa;YACnB,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EACzD,CAAC;YACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACzC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CACxC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,EAC/B,KAAK,CACN,CAAC;YACF,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC/B,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,MAAM,CAAC,CAAC;QACV,CAAC;QACD,uEAAuE;QACvE,IACE,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;YAC1B,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EACzD,CAAC;YACD,yDAAyD;YACzD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACzC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QACjC,CAAC;IACH,CAAC;IAED,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC;QAC3B,MAAM,GAAG,GAAG,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QACD,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK;YACL,GAAG;YACH,KAAK;SACN,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,QAAsB;QACxC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,aAAa,CAAC;YACzB,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QACD,MAAM,WAAW,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACpC,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC;YACpC,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,SAAS,GAAG,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;QAC7D,CAAC;QACD,MAAM,UAAU,GACd,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;QACjB,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,EAAE;YACxD,QAAQ;YACR,UAAU;SACX,CAAC,CAAC;IACL,CAAC;CACF"}
|
package/dist/client/types.d.ts
CHANGED
|
@@ -39,6 +39,10 @@ export type ContextOptions = {
|
|
|
39
39
|
* At least one of textSearch or vectorSearch must be true.
|
|
40
40
|
*/
|
|
41
41
|
vectorSearch?: boolean;
|
|
42
|
+
/**
|
|
43
|
+
* The score threshold for vector search. Default is 0.0.
|
|
44
|
+
*/
|
|
45
|
+
vectorScoreThreshold?: number;
|
|
42
46
|
/**
|
|
43
47
|
* What messages around the search results to include.
|
|
44
48
|
* Default: { before: 2, after: 1 }
|
|
@@ -102,7 +106,7 @@ export type UsageHandler = (ctx: RunActionCtx, args: {
|
|
|
102
106
|
model: string;
|
|
103
107
|
provider: string;
|
|
104
108
|
}) => void | Promise<void>;
|
|
105
|
-
export type RawRequestResponseHandler = (ctx:
|
|
109
|
+
export type RawRequestResponseHandler = (ctx: ActionCtx, args: {
|
|
106
110
|
userId: string | undefined;
|
|
107
111
|
threadId: string | undefined;
|
|
108
112
|
agentName: string | undefined;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/client/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,cAAc,EACd,oBAAoB,EACpB,YAAY,EACZ,kBAAkB,EAClB,SAAS,EACT,4BAA4B,EAC5B,6BAA6B,EAC7B,eAAe,EACf,kBAAkB,EAClB,YAAY,EACZ,kBAAkB,EAClB,UAAU,EACV,gBAAgB,EAChB,iBAAiB,EACjB,UAAU,EACV,OAAO,EACR,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EACV,IAAI,EACJ,MAAM,EACN,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,aAAa,EACb,mBAAmB,EACpB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,KAAK,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EACV,YAAY,EACZ,gBAAgB,EAChB,eAAe,EACf,WAAW,EACX,aAAa,EACb,KAAK,EACN,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAEvD;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,aAAa,CAAC,EAAE;QACd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QACd;;WAEG;QACH,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB;;;WAGG;QACH,YAAY,CAAC,EAAE,OAAO,CAAC;QACvB;;;;;;WAMG;QACH,YAAY,CAAC,EAAE;YAAE,MAAM,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;KAClD,CAAC;IACF;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;;;;;;OAQG;IACH,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,iBAAiB,CAAC;IAClD;;;;;OAKG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B;;;OAGG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9D,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,YAAY,EACjB,IAAI,EAAE;IACJ,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,KAAK,EAAE,KAAK,CAAC;IAEb,gBAAgB,EAAE,gBAAgB,GAAG,SAAS,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB,KACE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,MAAM,yBAAyB,GAAG,CACtC,GAAG,EAAE,
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/client/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,cAAc,EACd,oBAAoB,EACpB,YAAY,EACZ,kBAAkB,EAClB,SAAS,EACT,4BAA4B,EAC5B,6BAA6B,EAC7B,eAAe,EACf,kBAAkB,EAClB,YAAY,EACZ,kBAAkB,EAClB,UAAU,EACV,gBAAgB,EAChB,iBAAiB,EACjB,UAAU,EACV,OAAO,EACR,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EACV,IAAI,EACJ,MAAM,EACN,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,aAAa,EACb,mBAAmB,EACpB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,KAAK,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EACV,YAAY,EACZ,gBAAgB,EAChB,eAAe,EACf,WAAW,EACX,aAAa,EACb,KAAK,EACN,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAEvD;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,aAAa,CAAC,EAAE;QACd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QACd;;WAEG;QACH,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB;;;WAGG;QACH,YAAY,CAAC,EAAE,OAAO,CAAC;QACvB;;WAEG;QACH,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B;;;;;;WAMG;QACH,YAAY,CAAC,EAAE;YAAE,MAAM,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;KAClD,CAAC;IACF;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;;;;;;OAQG;IACH,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,iBAAiB,CAAC;IAClD;;;;;OAKG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B;;;OAGG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9D,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,YAAY,EACjB,IAAI,EAAE;IACJ,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,KAAK,EAAE,KAAK,CAAC;IAEb,gBAAgB,EAAE,gBAAgB,GAAG,SAAS,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB,KACE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,MAAM,yBAAyB,GAAG,CACtC,GAAG,EAAE,SAAS,EACd,IAAI,EAAE;IACJ,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,OAAO,EAAE,4BAA4B,CAAC;IACtC,QAAQ,EAAE,6BAA6B,CAAC;CACzC,KACE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;AAE5C,MAAM,MAAM,QAAQ,CAClB,UAAU,SAAS,OAAO,EAC1B,KAAK,SAAS,OAAO,GAAG,SAAS,GAAG,SAAS,EAC7C,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,IACpB,IAAI,CACN,UAAU,CACR,OAAO,YAAY,CACjB,KAAK,SAAS,SAAS,GAAG,UAAU,GAAG,KAAK,EAC5C,MAAM,EACN,cAAc,CACf,CACF,CAAC,CAAC,CAAC,EACJ,YAAY,GAAG,OAAO,GAAG,OAAO,CACjC,GAAG;IACF;;;;;OAKG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;IACd;;;OAGG;IACH,UAAU,CAAC,EAAE,UAAU,CAAC,KAAK,SAAS,SAAS,GAAG,UAAU,GAAG,KAAK,CAAC,CAAC;CACvE,CAAC;AAEF,MAAM,MAAM,iBAAiB,CAC3B,UAAU,SAAS,OAAO,EAC1B,KAAK,SAAS,OAAO,GAAG,SAAS,GAAG,SAAS,EAC7C,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,IACpB,IAAI,CACN,UAAU,CACR,OAAO,UAAU,CACf,KAAK,SAAS,SAAS,GAAG,UAAU,GAAG,KAAK,EAC5C,MAAM,EACN,cAAc,CACf,CACF,CAAC,CAAC,CAAC,EACJ,YAAY,GAAG,OAAO,GAAG,OAAO,CACjC,GAAG;IACF;;;;;OAKG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;IACd;;;OAGG;IACH,UAAU,CAAC,EAAE,UAAU,CAAC,KAAK,SAAS,SAAS,GAAG,UAAU,GAAG,KAAK,CAAC,CAAC;CACvE,CAAC;AAEF,KAAK,yBAAyB,GAAG,YAAY,GAAG;IAC9C;;;OAGG;IACH,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;IACzB;;;;;OAKG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uBAAuB,CAAC,EAAE,kBAAkB,CAAC;IAC7C,sBAAsB,CAAC,EAAE,iBAAiB,CAAC;IAC3C,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,6BAA6B,CAAC,EAAE,gBAAgB,CAAC;CAClD,CAAC;AAEF,KAAK,2BAA2B,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAChE,yBAAyB,GAAG;IAC1B,MAAM,CAAC,EAAE,QAAQ,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEJ,KAAK,0BAA0B,CAAC,CAAC,IAAI,yBAAyB,GAAG;IAC/D,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,KAAK,6BAA6B,CAAC,CAAC,SAAS,MAAM,IACjD,yBAAyB,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;CACjC,CAAC;AAEJ,KAAK,6BAA6B,GAAG,yBAAyB,GAAG;IAC/D,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAIF,KAAK,kBAAkB,CAAC,CAAC,IACvB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,CAAC,SAAS,MAAM,GACd,6BAA6B,CAAC,CAAC,CAAC,GAChC,6BAA6B,CAAC;AAExC,KAAK,gBAAgB,CAAC,CAAC,IACrB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,6BAA6B,CAAC;AAEtC,MAAM,MAAM,aAAa,CAAC,CAAC,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAClD,IAAI,CAEF,UAAU,CAAC,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EACzC,yBAAyB,GAAG,aAAa,CAC1C,CAAC;AAEJ,MAAM,MAAM,mBAAmB,CAAC,CAAC,IAAI,gBAAgB,CAAC,CAAC,CAAC,GACtD,IAAI,CACF,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACrC,SAAS,GAAG,UAAU,GAAG,aAAa,CACvC,CAAC;AAEJ,KAAK,oBAAoB,GAAG,wBAAwB,GAAG;IACrD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;;GAGG;AACH,MAAM,WAAW,MAAM,CAAC,YAAY,SAAS,OAAO;IAClD;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,WAAW,EAAE,MAAM,OAAO,CAAC,SAAS,CAAC,CAAC;IACtC;;OAEG;IACH,cAAc,EAAE,CACd,KAAK,EAAE,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,KAC3C,OAAO,CAAC,SAAS,CAAC,CAAC;IACxB;;;;;;;;;OASG;IACH,YAAY,CACV,KAAK,SAAS,OAAO,GAAG,SAAS,GAAG,SAAS,EAC7C,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,IAAI,EAAE,QAAQ,CACZ,KAAK,SAAS,SAAS,GAAG,YAAY,GAAG,KAAK,EAC9C,KAAK,EACL,MAAM,EACN,cAAc,CACf,EACD,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CACR,kBAAkB,CAAC,KAAK,SAAS,SAAS,GAAG,YAAY,GAAG,KAAK,EAAE,MAAM,CAAC,GACxE,oBAAoB,CACvB,CAAC;IAEF;;;;;;;;;OASG;IACH,UAAU,CACR,KAAK,SAAS,OAAO,GAAG,SAAS,GAAG,SAAS,EAC7C,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,IAAI,EAAE,iBAAiB,CACrB,KAAK,SAAS,SAAS,GAAG,YAAY,GAAG,KAAK,EAC9C,KAAK,EACL,MAAM,EACN,cAAc,CACf,EACD,OAAO,CAAC,EAAE,OAAO,GAAG;QAClB;;;;;;;;;WASG;QACH,gBAAgB,CAAC,EAAE,OAAO,GAAG,gBAAgB,CAAC;KAC/C,GACA,OAAO,CACR,gBAAgB,CACd,KAAK,SAAS,SAAS,GAAG,YAAY,GAAG,KAAK,EAC9C,cAAc,CACf,GACC,oBAAoB,CACvB,CAAC;IACF;;;;;;;;;OASG;IACH,cAAc,CAAC,CAAC,EACd,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,EACtB,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,oBAAoB,CAAC,CAAC;IAC3D;;;;;;;;;OASG;IACH,cAAc,CACZ,IAAI,EAAE,6BAA6B,EACnC,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,oBAAoB,CAAC,CAAC;IACnE;;;;;;;;;OASG;IACH,YAAY,CAAC,CAAC,EACZ,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,EAC5B,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CACR,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,oBAAoB,CACpE,CAAC;CACH;AAED,MAAM,MAAM,OAAO,GAAG;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;OAEG;IACH,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAC9B;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,aAAa,EAAE,CAAA;CAAE,GAC3C;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,MAAM,EAAE,WAAW,EAAE,CAAA;CAAE,GACzC,SAAS,CAAC;AAGd,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,EAAE,eAAe,CAAC,gBAAgB,CAAC,CAAC,UAAU,CAAC,CAAC;CACzD,CAAC;AACF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,UAAU,CAAC,CAAC;IAC3D,WAAW,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC,CAAC;CAClE,CAAC;AACF,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,EAAE,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,UAAU,CAAC,CAAC;IACzD,WAAW,EAAE,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC,CAAC;IAC/D,SAAS,EAAE,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,CAAC;CAC5D,CAAC;AACF,MAAM,MAAM,SAAS,GAAG,YAAY,GAAG;IACrC,IAAI,EAAE,IAAI,CAAC;IACX,OAAO,EAAE,mBAAmB,CAAC;CAC9B,CAAC;AACF,MAAM,MAAM,QAAQ,GAAG,WAAW,GAAG;IACnC,OAAO,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,SAAS,CAAC,CAAC,IACrB,CAAC,SAAS,SAAS,CAAC,MAAM,EAAE,CAAC,GACzB,MAAM,GACN,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,GACnB,SAAS,CAAC,CAAC,CAAC,EAAE,GACd,CAAC,SAAS,WAAW,GACnB,WAAW,GACX,CAAC,SAAS,MAAM,GACd;KACG,CAAC,IAAI,MAAM,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAChC,GACD,CAAC,CAAC;AAEd,MAAM,MAAM,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC;KAC9B,GAAG,IAAI,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,SAAS,iBAAiB,CACpD,MAAM,KAAK,EACX,QAAQ,EACR,MAAM,KAAK,EACX,MAAM,WAAW,EACjB,MAAM,cAAc,CACrB,GACG,iBAAiB,CACf,KAAK,EACL,UAAU,EACV,SAAS,CAAC,KAAK,CAAC,EAChB,SAAS,CAAC,WAAW,CAAC,EACtB,cAAc,CACf,GACD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;CACrB,CAAC,CAAC"}
|
|
@@ -900,13 +900,13 @@ export type Mounts = {
|
|
|
900
900
|
"public",
|
|
901
901
|
{
|
|
902
902
|
beforeMessageId?: string;
|
|
903
|
+
embedding?: Array<number>;
|
|
904
|
+
embeddingModel?: string;
|
|
903
905
|
limit: number;
|
|
904
906
|
messageRange?: { after: number; before: number };
|
|
905
907
|
searchAllMessagesForUserId?: string;
|
|
906
908
|
text?: string;
|
|
907
909
|
threadId?: string;
|
|
908
|
-
vector?: Array<number>;
|
|
909
|
-
vectorModel?: string;
|
|
910
910
|
vectorScoreThreshold?: number;
|
|
911
911
|
},
|
|
912
912
|
Array<{
|
|
@@ -681,8 +681,8 @@ export declare const listMessagesByThreadId: import("convex/server").RegisteredQ
|
|
|
681
681
|
numItems: number;
|
|
682
682
|
cursor: string | null;
|
|
683
683
|
} | undefined;
|
|
684
|
-
isTool?: "use excludeToolMessages instead of this" | undefined;
|
|
685
684
|
statuses?: ("pending" | "success" | "failed")[] | undefined;
|
|
685
|
+
isTool?: "use excludeToolMessages instead of this" | undefined;
|
|
686
686
|
upToAndIncludingMessageId?: import("convex/values").GenericId<"messages"> | undefined;
|
|
687
687
|
order: "asc" | "desc";
|
|
688
688
|
threadId: import("convex/values").GenericId<"threads">;
|
|
@@ -958,14 +958,14 @@ export declare const getThreadMessages: import("convex/server").RegisteredQuery<
|
|
|
958
958
|
export declare const searchMessages: import("convex/server").RegisteredAction<"public", {
|
|
959
959
|
threadId?: import("convex/values").GenericId<"threads"> | undefined;
|
|
960
960
|
text?: string | undefined;
|
|
961
|
-
vector?: number[] | undefined;
|
|
962
|
-
vectorModel?: string | undefined;
|
|
963
961
|
vectorScoreThreshold?: number | undefined;
|
|
964
962
|
messageRange?: {
|
|
965
963
|
before: number;
|
|
966
964
|
after: number;
|
|
967
965
|
} | undefined;
|
|
968
966
|
beforeMessageId?: import("convex/values").GenericId<"messages"> | undefined;
|
|
967
|
+
embedding?: number[] | undefined;
|
|
968
|
+
embeddingModel?: string | undefined;
|
|
969
969
|
searchAllMessagesForUserId?: string | undefined;
|
|
970
970
|
limit: number;
|
|
971
971
|
}, Promise<{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../src/component/messages.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../src/component/messages.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAIL,KAAK,WAAW,EAEhB,KAAK,QAAQ,EACd,MAAM,wBAAwB,CAAC;AAiBhC,gEAAgE;AAChE,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;GAAuB,CAAC;AAExD,0CAA0C;AAC1C,eAAO,MAAM,SAAS;;;;;;;;;UAAa,CAAC;AAEpC,qDAAqD;AACrD,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;GAAgB,CAAC;AAM1C,wBAAsB,aAAa,CACjC,GAAG,EAAE,WAAW,EAChB,UAAU,EAAE,GAAG,CAAC,UAAU,CAAC,iBAS5B;AAED,eAAO,MAAM,WAAW;;4DAkBtB,CAAC;AAEH,eAAO,MAAM,eAAe,sCAE3B,CAAC;AAEF,eAAO,MAAM,aAAa;;;;;;;;;;GAsCxB,CAAC;AAYH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAMtB,CAAC;AAsGH,wBAAsB,aAAa,CACjC,GAAG,EAAE,QAAQ,EACb,QAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,EACvB,KAAK,CAAC,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAGf;AA8BD,eAAO,MAAM,eAAe;;;iBA0B1B,CAAC;AAEH,eAAO,MAAM,aAAa;eA4CI,EAAE,CAAC,UAAU,CAAC;iBAtC1C,CAAC;AAEH,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCxB,CAAC;AA6BH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2DjC,CAAC;AAEH,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aAQ3B,CAAC;AAEH,sDAAsD;AACtD,eAAO,MAAM,iBAAiB;;kBAM5B,CAAC;AAEH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA4EzB,CAAC;AAEH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA4G/B,CAAC;AAIH,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA6CrB,CAAC"}
|
|
@@ -2,7 +2,7 @@ import { assert, omit } from "convex-helpers";
|
|
|
2
2
|
import { mergedStream, stream } from "convex-helpers/server/stream";
|
|
3
3
|
import { paginationOptsValidator } from "convex/server";
|
|
4
4
|
import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, isTool, } from "../shared.js";
|
|
5
|
-
import { vMessageEmbeddings, vMessageStatus, vMessageWithMetadataInternal, vPaginationResult,
|
|
5
|
+
import { vMessageEmbeddings, vMessageStatus, vMessageWithMetadataInternal, vPaginationResult, } from "../validators.js";
|
|
6
6
|
import { api, internal } from "./_generated/api.js";
|
|
7
7
|
import { action, internalQuery, mutation, query, } from "./_generated/server.js";
|
|
8
8
|
import { schema, v, vMessageDoc } from "./schema.js";
|
|
@@ -337,7 +337,12 @@ export const searchMessages = action({
|
|
|
337
337
|
threadId: v.optional(v.id("threads")),
|
|
338
338
|
searchAllMessagesForUserId: v.optional(v.string()),
|
|
339
339
|
beforeMessageId: v.optional(v.id("messages")),
|
|
340
|
-
|
|
340
|
+
embedding: v.optional(v.array(v.number())),
|
|
341
|
+
embeddingModel: v.optional(v.string()),
|
|
342
|
+
text: v.optional(v.string()),
|
|
343
|
+
limit: v.number(),
|
|
344
|
+
vectorScoreThreshold: v.optional(v.number()),
|
|
345
|
+
messageRange: v.optional(v.object({ before: v.number(), after: v.number() })),
|
|
341
346
|
},
|
|
342
347
|
returns: v.array(vMessageDoc),
|
|
343
348
|
handler: async (ctx, args) => {
|
|
@@ -353,14 +358,14 @@ export const searchMessages = action({
|
|
|
353
358
|
beforeMessageId: args.beforeMessageId,
|
|
354
359
|
});
|
|
355
360
|
}
|
|
356
|
-
if (args.
|
|
357
|
-
const dimension = args.
|
|
361
|
+
if (args.embedding) {
|
|
362
|
+
const dimension = args.embedding.length;
|
|
358
363
|
if (!VectorDimensions.includes(dimension)) {
|
|
359
|
-
throw new Error(`Unsupported
|
|
364
|
+
throw new Error(`Unsupported embedding dimension: ${dimension}`);
|
|
360
365
|
}
|
|
361
|
-
const vectors = (await searchVectors(ctx, args.
|
|
366
|
+
const vectors = (await searchVectors(ctx, args.embedding, {
|
|
362
367
|
dimension,
|
|
363
|
-
model: args.
|
|
368
|
+
model: args.embeddingModel ?? "unknown",
|
|
364
369
|
table: "messages",
|
|
365
370
|
searchAllMessagesForUserId: args.searchAllMessagesForUserId,
|
|
366
371
|
threadId: args.threadId,
|