added embeddings experiment

Author: theorm

src/services/experiments/experiments/entityProfiles.ts

@@ -0,0 +1,33 @@
+import { SolrNamespaces } from '../../../solr'
+import { ImpressoApplication } from '../../../types'
+import { ExperimentBase } from './base'
+
+interface RequestBody {
+  solrPayload: Record<string, any>
+}
+
+interface ResponseBody {
+  solrResponse: Record<string, any>
+}
+
+export class EntityProfilesExperiment implements ExperimentBase<RequestBody, ResponseBody> {
+  id = 'entity-profiles'
+  name = 'Experiment with entity profiles and their embeddings'
+  description = `
+  Generates embeddings for subdocuments using a specified model.
+  The body must contain a 'solrPayload' field with a 
+  JSON API Solr query (see https://solr.apache.org/guide/solr/latest/query-guide/json-request-api.html).
+  `
+
+  async execute(body: RequestBody, app: ImpressoApplication): Promise<ResponseBody> {
+    if (body.solrPayload == null) {
+      throw new Error("Request body must contain a 'solrPayload' field with a Solr JSON API query.")
+    }
+    const solrResponse = await app.service('simpleSolrClient').select(SolrNamespaces.EntityProfiles, {
+      body: body.solrPayload as any,
+    })
+    return {
+      solrResponse,
+    }
+  }
+}

src/services/experiments/experiments/index.ts

@@ -1,3 +1,4 @@
+import { EntityProfilesExperiment } from './entityProfiles'
 import { SubdocEmbeddingsExperiment } from './subdocEmbeddings'
 
-export const experiments = [new SubdocEmbeddingsExperiment()]
+export const experiments = [new SubdocEmbeddingsExperiment(), new EntityProfilesExperiment()]

src/solr.ts

@@ -16,6 +16,7 @@ export type SolrNamespace =
   | 'entities_mentions'
   | 'collection_items'
   | 'subdoc_embeddings_experiment'
+  | 'entity_profiles'
 
 export const SolrNamespaces = Object.freeze({
   Search: 'search',
@@ -32,6 +33,7 @@ export const SolrNamespaces = Object.freeze({
   EntitiesMentions: 'entities_mentions',
   CollectionItems: 'collection_items',
   SubdocEmbeddingsExperiment: 'subdoc_embeddings_experiment',
+  EntityProfiles: 'entity_profiles',
 }) satisfies Record<string, SolrNamespace>
 
 /**