Update Query Agent response model (#16)

* Update responses types for new fetch queries

* Update response model with removed fields

---------

Co-authored-by: Dan Jones <[email protected]>

Author: danmichaeljones

src/query/response/api-response-mapping.ts

@@ -27,9 +27,6 @@ export const mapApiResponse = (
   aggregations: mapApiAggregations(response.aggregations),
   usage: mapApiUsage(response.usage),
   total_time: response.totalTime,
-  aggregation_answer: response.aggregationAnswer,
-  has_aggregation_answer: response.hasAggregationAnswer,
-  has_search_answer: response.hasSearchAnswer,
   is_partial_answer: response.isPartialAnswer,
   missing_information: response.missingInformation,
   final_answer: response.finalAnswer,

src/query/response/api-response.ts

@@ -12,9 +12,6 @@ export type ApiQueryAgentResponse = {
   aggregations: ApiAggregationResult[][];
   usage: ApiUsage;
   total_time: number;
-  aggregation_answer?: string;
-  has_aggregation_answer: boolean;
-  has_search_answer: boolean;
   is_partial_answer: boolean;
   missing_information: string[];
   final_answer: string;
@@ -23,7 +20,7 @@ export type ApiQueryAgentResponse = {
 
 export type ApiSearchResult = {
   collection: string;
-  queries: string[];
+  queries: (string | null)[];
   filters: ApiPropertyFilter[][];
   filter_operators: "AND" | "OR";
 };

src/query/response/response-mapping.ts

@@ -33,9 +33,6 @@ export const mapResponse = (
     aggregations: mapAggregations(response.aggregations),
     usage: mapUsage(response.usage),
     totalTime: response.total_time,
-    aggregationAnswer: response.aggregation_answer,
-    hasAggregationAnswer: response.has_aggregation_answer,
-    hasSearchAnswer: response.has_search_answer,
     isPartialAnswer: response.is_partial_answer,
     missingInformation: response.missing_information,
     finalAnswer: response.final_answer,
@@ -155,9 +152,6 @@ export const mapResponseFromSSE = (sse: ServerSentEvent): QueryAgentResponse =>
     aggregations: mapAggregations(data.aggregations),
     usage: mapUsage(data.usage),
     totalTime: data.total_time,
-    aggregationAnswer: data.aggregation_answer,
-    hasAggregationAnswer: data.has_aggregation_answer,
-    hasSearchAnswer: data.has_search_answer,
     isPartialAnswer: data.is_partial_answer,
     missingInformation: data.missing_information,
     finalAnswer: data.final_answer,

src/query/response/response.ts

@@ -6,9 +6,6 @@ export type QueryAgentResponse = {
   aggregations: AggregationResult[][];
   usage: Usage;
   totalTime: number;
-  aggregationAnswer?: string;
-  hasAggregationAnswer: boolean;
-  hasSearchAnswer: boolean;
   isPartialAnswer: boolean;
   missingInformation: string[];
   finalAnswer: string;
@@ -18,7 +15,7 @@ export type QueryAgentResponse = {
 
 export type SearchResult = {
   collection: string;
-  queries: string[];
+  queries: (string | null)[];
   filters: PropertyFilter[][];
   filterOperators: "AND" | "OR";
 };