lucretia/embeddingsearch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added optional returnEmbeddings parameter to /Entity/List

Browse Source
This commit is contained in:
EzFeDezy
2025-04-27 20:00:18 +02:00
parent c955182933
commit 36d126c63f
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/server/Controllers/EntityController.cs
View File

@@ -55,7 +55,7 @@ public class EntityController : ControllerBase
}
[HttpGet("List")]
public ActionResult<EntityListResults> List(string searchdomain)
public ActionResult<EntityListResults> List(string searchdomain, bool returnEmbeddings = false)
{
EntityListResults entityListResults = new() {Results = []};
Searchdomain searchdomain_ = _domainManager.GetSearchdomain(searchdomain);
@@ -69,12 +69,19 @@ public class EntityController : ControllerBase
List<DatapointResult> datapointResults = [];
foreach (Datapoint datapoint in entity.datapoints)
{
List<EmbeddingResult> embeddingResults = [];
foreach ((string, float[]) embedding in datapoint.embeddings)
if (returnEmbeddings)
{
embeddingResults.Add(new EmbeddingResult() {Model = embedding.Item1, Embeddings = embedding.Item2});
List<EmbeddingResult> embeddingResults = [];
foreach ((string, float[]) embedding in datapoint.embeddings)
{
embeddingResults.Add(new EmbeddingResult() {Model = embedding.Item1, Embeddings = embedding.Item2});
}
datapointResults.Add(new DatapointResult() {Name = datapoint.name, ProbMethod = datapoint.probMethod.Method.Name, Embeddings = embeddingResults});
}
else
{
datapointResults.Add(new DatapointResult() {Name = datapoint.name, ProbMethod = datapoint.probMethod.Method.Name, Embeddings = null});
}
datapointResults.Add(new DatapointResult() {Name = datapoint.name, ProbMethod = datapoint.probMethod.Method.Name, Embeddings = embeddingResults});
}
EntityListResult entityListResult = new()
{

2
src/server/Models/EntityModels.cs
View File

@@ -39,7 +39,7 @@ public class DatapointResult
{
public required string Name { get; set; }
public required string ProbMethod { get; set; }
public required List<EmbeddingResult> Embeddings { get; set; }
public required List<EmbeddingResult>? Embeddings { get; set; }
}
public class EmbeddingResult