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

Fixed multithreading mutation issue

Browse Source
This commit is contained in:
LD-Reborn
2025-12-31 03:43:44 +01:00
parent bc293bf7ec
commit 8d56883e7e
2 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/Server/Searchdomain.cs
View File

@@ -216,6 +216,11 @@ public class Searchdomain
return queryEmbeddings;
}
public void UpdateModelsInUse()
{
modelsInUse = GetModels([.. entityCache]);
}
private static float EvaluateEntityAgainstQueryEmbeddings(Entity entity, Dictionary<string, float[]> queryEmbeddings)
{
List<(string, float)> datapointProbs = [];
@@ -237,16 +242,19 @@ public class Searchdomain
public static List<string> GetModels(List<Entity> entities)
{
List<string> result = [];
foreach (Entity entity in entities)
lock (entities)
{
foreach (Datapoint datapoint in entity.datapoints)
foreach (Entity entity in entities)
{
foreach ((string, float[]) tuple in datapoint.embeddings)
foreach (Datapoint datapoint in entity.datapoints)
{
string model = tuple.Item1;
if (!result.Contains(model))
foreach ((string, float[]) tuple in datapoint.embeddings)
{
result.Add(model);
string model = tuple.Item1;
if (!result.Contains(model))
{
result.Add(model);
}
}
}
}