diff --git a/packages/agent-client/src/query-serializer.ts b/packages/agent-client/src/query-serializer.ts index 13e52582e..a89fb48e5 100644 --- a/packages/agent-client/src/query-serializer.ts +++ b/packages/agent-client/src/query-serializer.ts @@ -7,14 +7,27 @@ export default class QuerySerializer { static serialize(query: SelectOptions, collectionName: string): Record { if (!query) return {}; + const { + fields, + sort, + filters, + shouldSearchInRelation, + pagination, + search, + segmentQuery, + connectionName, + } = query; + return { - ...query, - sort: QuerySerializer.formatSort(query.sort), - filters: QuerySerializer.formatFilters(query.filters), - searchExtended: !!query.shouldSearchInRelation, - 'page[size]': query.pagination?.size, - 'page[number]': query.pagination?.number, - ...(query.fields?.length ? QuerySerializer.formatFields(collectionName, query.fields) : {}), + search, + segmentQuery, + connectionName, + sort: QuerySerializer.formatSort(sort), + filters: QuerySerializer.formatFilters(filters), + searchExtended: !!shouldSearchInRelation, + 'page[size]': pagination?.size, + 'page[number]': pagination?.number, + ...(fields?.length ? QuerySerializer.formatFields(collectionName, fields) : {}), }; } diff --git a/packages/agent-client/src/types.ts b/packages/agent-client/src/types.ts index bd644b9b3..0bc797f5a 100644 --- a/packages/agent-client/src/types.ts +++ b/packages/agent-client/src/types.ts @@ -27,4 +27,6 @@ export type SelectOptions = BaseOptions & { size?: number; // number of items per page number?: number; // current page number }; + segmentQuery?: string; // SQL query for live query segments + connectionName?: string; // Connection name for live query segments };