Project search assets

packages/client/hmi-client/src/types/Types.ts

@@ -795,9 +795,9 @@ export interface ModelUnit {
 }
 
 export interface GroundedSemantic {
-    grounding?: ModelGrounding;
     id: string;
     name?: string;
+    grounding?: ModelGrounding;
     description?: string;
 }
 
@@ -1024,6 +1024,7 @@ export enum AssetType {
     ModelConfiguration = "model-configuration",
     Artifact = "artifact",
     InterventionPolicy = "intervention-policy",
+    Project = "project",
 }
 
 export enum EvaluationScenarioStatus {

packages/server/src/main/java/software/uncharted/terarium/hmiserver/controller/dataservice/ProjectController.java

@@ -66,6 +66,7 @@
 import software.uncharted.terarium.hmiserver.service.data.ProjectAssetService;
 import software.uncharted.terarium.hmiserver.service.data.ProjectPermissionsService;
 import software.uncharted.terarium.hmiserver.service.data.ProjectSearchService;
+import software.uncharted.terarium.hmiserver.service.data.ProjectSearchService.ProjectSearchResponse;
 import software.uncharted.terarium.hmiserver.service.data.ProjectService;
 import software.uncharted.terarium.hmiserver.service.data.TerariumAssetServices;
 import software.uncharted.terarium.hmiserver.service.data.WorkflowService;
@@ -929,7 +930,7 @@ public ResponseEntity<PermissionRelationships> getProjectPermissions(@PathVariab
 		try {
 			for (final RebacPermissionRelationship permissionRelationship : rebacProject.getPermissionRelationships()) {
 				if (permissionRelationship.getSubjectType().equals(Schema.Type.USER)) {
-					PermissionUser user = reBACService.getUser(permissionRelationship.getSubjectId());
+					final PermissionUser user = reBACService.getUser(permissionRelationship.getSubjectId());
 					if (user != null) {
 						permissions.addUser(user, permissionRelationship.getRelationship());
 					}
@@ -948,6 +949,55 @@ public ResponseEntity<PermissionRelationships> getProjectPermissions(@PathVariab
 		return ResponseEntity.ok(permissions);
 	}
 
+	@GetMapping("/knn")
+	@Secured(Roles.USER)
+	@Operation(summary = "Executes a knn search against the provided asset type")
+	@ApiResponses(
+		value = {
+			@ApiResponse(
+				responseCode = "200",
+				description = "Query results",
+				content = @Content(
+					mediaType = "application/json",
+					schema = @io.swagger.v3.oas.annotations.media.Schema(implementation = JsonNode.class)
+				)
+			),
+			@ApiResponse(responseCode = "204", description = "There was no concept found", content = @Content),
+			@ApiResponse(
+				responseCode = "500",
+				description = "There was an issue retrieving the concept from the data store",
+				content = @Content
+			)
+		}
+	)
+	public ResponseEntity<List<ProjectSearchResponse>> projectKnnSearch(
+		@RequestParam(value = "page-size", defaultValue = "100", required = false) final Integer pageSize,
+		@RequestParam(value = "page", defaultValue = "0", required = false) final Integer page,
+		@RequestParam(value = "text", defaultValue = "") final String text,
+		@RequestParam(value = "k", defaultValue = "100") final int k,
+		@RequestParam(value = "num-candidates", defaultValue = "1000") final int numCandidates
+	) {
+		try {
+			final String userId = currentUserService.get().getId();
+
+			final List<ProjectSearchResponse> res = projectSearchService.searchProjectsKNN(
+				userId,
+				pageSize,
+				page,
+				text,
+				k,
+				numCandidates,
+				null
+			);
+
+			return ResponseEntity.ok(res);
+		} catch (final Exception e) {
+			final String error = "Unable to get execute knn search";
+			log.error(error, e);
+			throw new ResponseStatusException(org.springframework.http.HttpStatus.INTERNAL_SERVER_ERROR, error);
+		}
+	}
+
 	// --------------------------------------------------------------------------
 	// Project Permissions
 	// --------------------------------------------------------------------------

packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/TerariumAsset.java

@@ -1,6 +1,7 @@
 package software.uncharted.terarium.hmiserver.models;
 
 import com.fasterxml.jackson.annotation.JsonAlias;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.MapperFeature;
 import com.fasterxml.jackson.databind.ObjectMapper;
@@ -19,6 +20,7 @@
 import lombok.EqualsAndHashCode;
 import lombok.experimental.Accessors;
 import org.hibernate.annotations.Type;
+import software.uncharted.terarium.hmiserver.annotations.TSIgnore;
 import software.uncharted.terarium.hmiserver.annotations.TSModel;
 import software.uncharted.terarium.hmiserver.annotations.TSOptional;
 
@@ -80,14 +82,14 @@ protected TerariumAsset cloneSuperFields(final TerariumAsset asset) {
 		return asset;
 	}
 
-	public static void removeFieldsWithKeys(ObjectNode objectNode, List<String> keys) {
-		Iterator<String> keysIterator = objectNode.fieldNames();
+	public static void removeFieldsWithKeys(final ObjectNode objectNode, final List<String> keys) {
+		final Iterator<String> keysIterator = objectNode.fieldNames();
 		while (keysIterator.hasNext()) {
-			String key = keysIterator.next();
+			final String key = keysIterator.next();
 			if (keys.contains(key)) {
 				keysIterator.remove();
 			} else {
-				JsonNode node = objectNode.get(key);
+				final JsonNode node = objectNode.get(key);
 				if (node.isObject()) {
 					removeFieldsWithKeys((ObjectNode) node, keys);
 				}
@@ -96,22 +98,23 @@ public static void removeFieldsWithKeys(ObjectNode objectNode, List<String> keys
 	}
 
 	/**
-	 * Serialize the asset to a JSON string, removing the fields that are not needed.
+	 * Serialize the asset to a JSON string, removing the fields that are not
+	 * needed.
 	 *
-	 * @param keepFields              A list of fields that should not be removed.
-	 * @param additionalDeleteFields  Additional fields to remove.
+	 * @param keepFields             A list of fields that should not be removed.
+	 * @param additionalDeleteFields Additional fields to remove.
 	 * @return The JSON string.
 	 */
 	public String serializeWithoutTerariumFields(
-		@Nullable String[] keepFields,
-		@Nullable String[] additionalDeleteFields
+		@Nullable final String[] keepFields,
+		@Nullable final String[] additionalDeleteFields
 	) {
 		final ObjectMapper mapper = new ObjectMapper();
 		mapper.setConfig(mapper.getSerializationConfig().with(MapperFeature.SORT_PROPERTIES_ALPHABETICALLY));
 		final ObjectNode objectNode = mapper.convertValue(this, ObjectNode.class);
 
 		// Fields to delete
-		String[] deleteFields = new String[] {
+		final String[] deleteFields = new String[] {
 			"id",
 			"createdOn",
 			"updatedOn",
@@ -123,15 +126,15 @@ public String serializeWithoutTerariumFields(
 			"fileNames",
 			"userId"
 		};
-		List<String> deleteFieldsList = new ArrayList<>(Arrays.asList(deleteFields));
+		final List<String> deleteFieldsList = new ArrayList<>(Arrays.asList(deleteFields));
 		if (keepFields != null) {
-			for (String field : keepFields) {
+			for (final String field : keepFields) {
 				deleteFieldsList.removeIf(key -> key.equals(field));
 			}
 		}
 
 		// Remove the fields that are not needed
-		for (String key : deleteFieldsList) {
+		for (final String key : deleteFieldsList) {
 			objectNode.remove(key);
 		}
 
@@ -149,4 +152,10 @@ public String serializeWithoutTerariumFields(
 
 		return objectNode.toString();
 	}
+
+	@JsonIgnore
+	@TSIgnore
+	public String getEmbeddingSourceText() {
+		return name + " " + description;
+	}
 }

packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/TerariumAssetEmbeddings.java

@@ -12,7 +12,7 @@ public static class Embeddings {
 
 		private String embeddingId;
 		private double[] vector;
-		private long[] spans;
+		private long[] span;
 	}
 
 	private List<Embeddings> embeddings = new ArrayList<>();

packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/dataservice/AssetType.java

@@ -2,6 +2,7 @@
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import java.util.List;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.http.HttpStatus;
@@ -13,6 +14,7 @@
 import software.uncharted.terarium.hmiserver.models.dataservice.document.DocumentAsset;
 import software.uncharted.terarium.hmiserver.models.dataservice.model.Model;
 import software.uncharted.terarium.hmiserver.models.dataservice.model.configurations.ModelConfiguration;
+import software.uncharted.terarium.hmiserver.models.dataservice.project.Project;
 import software.uncharted.terarium.hmiserver.models.dataservice.simulation.Simulation;
 import software.uncharted.terarium.hmiserver.models.dataservice.workflow.Workflow;
 import software.uncharted.terarium.hmiserver.models.simulationservice.interventions.InterventionPolicy;
@@ -46,7 +48,10 @@ public enum AssetType {
 	ARTIFACT,
 
 	@JsonProperty("intervention-policy")
-	INTERVENTION_POLICY;
+	INTERVENTION_POLICY,
+
+	@JsonProperty("project")
+	PROJECT;
 
 	public static AssetType getAssetType(final String assetTypeName, final ObjectMapper objectMapper)
 		throws ResponseStatusException {
@@ -58,6 +63,32 @@ public static AssetType getAssetType(final String assetTypeName, final ObjectMap
 		}
 	}
 
+	public static AssetType getAssetType(final Class<? extends TerariumAsset> clazz) {
+		if (clazz == Artifact.class) {
+			return ARTIFACT;
+		} else if (clazz == Code.class) {
+			return CODE;
+		} else if (clazz == Dataset.class) {
+			return DATASET;
+		} else if (clazz == DocumentAsset.class) {
+			return DOCUMENT;
+		} else if (clazz == Model.class) {
+			return MODEL;
+		} else if (clazz == ModelConfiguration.class) {
+			return MODEL_CONFIGURATION;
+		} else if (clazz == Simulation.class) {
+			return SIMULATION;
+		} else if (clazz == Workflow.class) {
+			return WORKFLOW;
+		} else if (clazz == InterventionPolicy.class) {
+			return INTERVENTION_POLICY;
+		} else if (clazz == Project.class) {
+			return PROJECT;
+		} else {
+			throw new IllegalArgumentException("Unrecognized asset type: " + clazz);
+		}
+	}
+
 	public Class<? extends TerariumAsset> getAssetClass() {
 		switch (this) {
 			case ARTIFACT:
@@ -78,8 +109,25 @@ public Class<? extends TerariumAsset> getAssetClass() {
 				return Workflow.class;
 			case INTERVENTION_POLICY:
 				return InterventionPolicy.class;
+			case PROJECT:
+				return Project.class;
 			default:
 				throw new IllegalArgumentException("Unrecognized asset type: " + this);
 		}
 	}
+
+	public static List<AssetType> getAllAssetTypes() {
+		return List.of(
+			ARTIFACT,
+			CODE,
+			DATASET,
+			DOCUMENT,
+			MODEL,
+			MODEL_CONFIGURATION,
+			SIMULATION,
+			WORKFLOW,
+			PROJECT,
+			INTERVENTION_POLICY
+		);
+	}
 }

packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/dataservice/document/DocumentAsset.java

@@ -1,7 +1,9 @@
 package software.uncharted.terarium.hmiserver.models.dataservice.document;
 
 import com.fasterxml.jackson.annotation.JsonAlias;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import io.hypersistence.utils.hibernate.type.json.JsonType;
 import jakarta.persistence.CascadeType;
 import jakarta.persistence.Column;
@@ -21,6 +23,7 @@
 import lombok.experimental.Accessors;
 import org.hibernate.annotations.JdbcTypeCode;
 import org.hibernate.annotations.Type;
+import software.uncharted.terarium.hmiserver.annotations.TSIgnore;
 import software.uncharted.terarium.hmiserver.annotations.TSModel;
 import software.uncharted.terarium.hmiserver.annotations.TSOptional;
 import software.uncharted.terarium.hmiserver.models.TerariumAsset;
@@ -135,4 +138,20 @@ public DocumentAsset clone() {
 
 		return clone;
 	}
+
+	@JsonIgnore
+	@TSIgnore
+	public String getEmbeddingSourceText() {
+		try {
+			if (getMetadata() != null && getMetadata().containsKey("gollmCard")) {
+				// update embeddings
+				final JsonNode card = getMetadata().get("gollmCard");
+				final ObjectMapper objectMapper = new ObjectMapper();
+				return objectMapper.writeValueAsString(card);
+			}
+			return null;
+		} catch (final Exception e) {
+			throw new RuntimeException("Failed to serialize model embedding text into JSON", e);
+		}
+	}
 }

packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/dataservice/model/Model.java

@@ -354,4 +354,18 @@ public boolean isPetrinet() {
 		}
 		return this.getHeader().getSchemaName().equalsIgnoreCase("petrinet");
 	}
+
+	@JsonIgnore
+	@TSIgnore
+	public String getEmbeddingSourceText() {
+		try {
+			final ObjectMapper objectMapper = new ObjectMapper();
+			if (getMetadata() != null && getMetadata().getGollmCard() != null) {
+				return objectMapper.writeValueAsString(getMetadata().getGollmCard());
+			}
+			return objectMapper.writeValueAsString(this);
+		} catch (final Exception e) {
+			throw new RuntimeException("Failed to serialize model embedding text into JSON", e);
+		}
+	}
 }