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

Added attributes to query results

Browse Source
This commit is contained in:
LD-Reborn
2025-12-25 12:39:01 +01:00
parent 040d4f916a
commit 5877ebaff2
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/Server/Controllers/EntityController.cs
View File

@@ -25,7 +25,7 @@ public class EntityController : ControllerBase
} }
[HttpGet("Query")] [HttpGet("Query")]
public ActionResult<EntityQueryResults> Query(string searchdomain, string query, int? topN) public ActionResult<EntityQueryResults> Query(string searchdomain, string query, int? topN, bool returnAttributes = false)
{ {
Searchdomain searchdomain_; Searchdomain searchdomain_;
try try
@@ -44,7 +44,8 @@ public class EntityController : ControllerBase
List<EntityQueryResult> queryResults = [.. results.Select(r => new EntityQueryResult List<EntityQueryResult> queryResults = [.. results.Select(r => new EntityQueryResult
{ {
Name = r.Item2, Name = r.Item2,
Value = r.Item1 Value = r.Item1,
Attributes = returnAttributes ? (searchdomain_.entityCache.FirstOrDefault(x => x.name == r.Item2)?.attributes ?? null) : null
})]; })];
return Ok(new EntityQueryResults(){Results = queryResults, Success = true }); return Ok(new EntityQueryResults(){Results = queryResults, Success = true });
} }

3
src/Shared/Models/EntityResults.cs
View File

@@ -19,6 +19,9 @@ public class EntityQueryResult
public required string Name { get; set; } public required string Name { get; set; }
[JsonPropertyName("Value")] [JsonPropertyName("Value")]
public float Value { get; set; } public float Value { get; set; }
[JsonPropertyName("Attributes")]
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
public Dictionary<string, string>? Attributes { get; set; }
} }
public class EntityIndexResult public class EntityIndexResult