feat: add getVectorSearchDocumentsQuery and extraBundles constants

Elastic-Suite · Dec 15, 2023 · c124f35 · c124f35
1 parent f271dc3
commit c124f35
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/packages/shared/src/constants/graphql.ts b/packages/shared/src/constants/graphql.ts
@@ -170,6 +170,29 @@ export function getSearchDocumentsQuery(
   })
 }
 
+export function getVectorSearchDocumentsQuery(
+  entityType: string,
+  filter: IDocumentFieldFilterInput | IDocumentFieldFilterInput[] = null,
+  withAggregations = false
+): string {
+  const documentQueryContent = getSearchDocumentQueryContent(
+    filter,
+    withAggregations,
+    entityType
+  )
+  return jsonToGraphQLQuery({
+    query: {
+      __name: 'getVectorSearchDocuments',
+      __variables: { ...documentQueryContent.variables },
+      vectorSearchDocuments: {
+        __aliasFor: 'vectorSearchDocuments',
+        __args: { ...documentQueryContent.args },
+        ...documentQueryContent.fields,
+      },
+    },
+  })
+}
+
 export function getSearchDocumentQueryContent(
   filter: IDocumentFieldFilterInput | IDocumentFieldFilterInput[] = null,
   withAggregations = false,

diff --git a/packages/shared/src/types/documents.ts b/packages/shared/src/types/documents.ts
@@ -14,6 +14,10 @@ export interface IGraphqlSearchDocuments {
   documents: IGraphqlSearchDocument
 }
 
+export interface IGraphqlVectorSearchDocuments {
+  vectorSearchDocuments: IGraphqlSearchDocument
+}
+
 export interface IGraphqlSearchDocument {
   collection: IGraphqlDocument[]
   paginationInfo: IGraphqlDocumentPaginationInfo

diff --git a/packages/shared/src/types/extraBundle.ts b/packages/shared/src/types/extraBundle.ts
@@ -1,5 +1,8 @@
 export enum Bundle {
   VIRTUAL_CATEGORY = 'GallyVirtualCategoryBundle',
+  BOOST = 'GallyBoostBundle',
+  THESAURUS = 'GallyThesaurusBundle',
+  VECTOR_SEARCH = 'GallyVectorSearchBundle',
 }
 
 export interface IExtraBundle {