refine embedding model config

This commit is contained in:
Jicheng Lu 2025-08-20 09:58:45 -05:00
parent 2bf4210485
commit 7a56dd5f71
4 changed files with 12 additions and 5 deletions

View file

@ -43,9 +43,9 @@ public class LlmModelSetting
public bool ImageGeneration { get; set; }
/// <summary>
/// Embedding dimension
/// Settings for embedding
/// </summary>
public int Dimension { get; set; }
public EmbeddingSetting? Embedding { get; set; }
/// <summary>
/// Settings for reasoning model
@ -73,6 +73,11 @@ public class LlmModelSetting
}
}
public class EmbeddingSetting
{
public int Dimension { get; set; }
}
public class ReasoningSetting
{
public float Temperature { get; set; } = 1.0f;

View file

@ -126,7 +126,7 @@ public class CompletionProvider
var found = llmProviderService.GetSetting(provider, model);
completer.SetModelName(model);
completer.SetDimension(found.Dimension);
completer.SetDimension(found.Embedding?.Dimension ?? 0);
return completer;
}

View file

@ -43,6 +43,6 @@ public static class KnowledgeSettingHelper
{
var settings = services.GetRequiredService<ILlmProviderService>();
var found = settings.GetSetting(provider, model);
return found?.Dimension ?? 0;
return found?.Embedding?.Dimension ?? 0;
}
}

View file

@ -183,7 +183,9 @@
"Version": "3-small",
"ApiKey": "",
"Type": "embedding",
"Dimension": 1536
"Embedding": {
"Dimension": 1536
}
}
]
},