Skip to content

Commit 3f05108

Browse files
committed
remove context from ask functions
1 parent c183f83 commit 3f05108

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

src/query/agent.ts

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ export class QueryAgent {
103103
*/
104104
async ask(
105105
query: string,
106-
{ collections, context }: QueryAgentRunOptions = {},
106+
{ collections }: QueryAgentRunOptions = {},
107107
): Promise<QueryAgentResponse> {
108108
const targetCollections = collections ?? this.collections;
109109
if (!targetCollections) {
@@ -369,6 +369,12 @@ export type QueryAgentRunOptions = {
369369
context?: QueryAgentResponse;
370370
};
371371

372+
/** Options for the QueryAgent ask. */
373+
export type QueryAgentAskOptions = {
374+
/** List of collections to query. Will override any collections if passed in the constructor. */
375+
collections?: (string | QueryAgentCollectionConfig)[];
376+
};
377+
372378
/** Options for the QueryAgent stream. */
373379
export type QueryAgentStreamOptions = {
374380
/** List of collections to query. Will override any collections if passed in the constructor. */
@@ -381,6 +387,16 @@ export type QueryAgentStreamOptions = {
381387
includeFinalState?: boolean;
382388
};
383389

390+
/** Options for the QueryAgent askStream. */
391+
export type QueryAgentAskStreamOptions = {
392+
/** List of collections to query. Will override any collections if passed in the constructor. */
393+
collections?: (string | QueryAgentCollectionConfig)[];
394+
/** Include progress messages in the stream. */
395+
includeProgress?: boolean;
396+
/** Include final state in the stream. */
397+
includeFinalState?: boolean;
398+
};
399+
384400
/** Options for the QueryAgent search-only run. */
385401
export type QueryAgentSearchOnlyOptions = {
386402
/** The maximum number of results to return. */

0 commit comments

Comments
 (0)