refine embedding model config
This commit is contained in:
parent
2bf4210485
commit
7a56dd5f71
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -183,7 +183,9 @@
|
|||
"Version": "3-small",
|
||||
"ApiKey": "",
|
||||
"Type": "embedding",
|
||||
"Dimension": 1536
|
||||
"Embedding": {
|
||||
"Dimension": 1536
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
|
|
|
|||
Loading…
Reference in a new issue