@@ -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. */
373379export 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. */
385401export type QueryAgentSearchOnlyOptions = {
386402 /** The maximum number of results to return. */
0 commit comments