SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

add default embed dimen

Browse source
This commit is contained in:
Jicheng Lu 2024-07-01 17:05:32 -05:00
parent 7d31df375c
commit 446f7784e1
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Infrastructure/BotSharp.OpenAPI/ViewModels/Embeddings/EmbeddingInputModel.cs
View file

@ -8,5 +8,5 @@ public class EmbeddingInputModel : MessageConfig
public IEnumerable<string> Texts { get; set; } = new List<string>();
[JsonPropertyName("dimension")]
public int Dimension { get; set; }
public int Dimension { get; set; } = 1536;
}

10
src/Plugins/BotSharp.Plugin.AzureOpenAI/Providers/Embedding/TextEmbeddingProvider.cs
View file

@ -8,11 +8,12 @@ public class TextEmbeddingProvider : ITextEmbedding
protected readonly IServiceProvider _services;
protected readonly ILogger<TextEmbeddingProvider> _logger;
private const int DEFAULT_DIMENSION = 1536;
protected string _model;
public virtual string Provider => "azure-openai";
public int Dimension { get; set; } = 4096;
public int Dimension { get; set; }
public TextEmbeddingProvider(
AzureOpenAiSettings settings,
@ -59,13 +60,12 @@ public class TextEmbeddingProvider : ITextEmbedding
private int GetDimension()
{
var defaultDimension = 4096;
var state = _services.GetRequiredService<IConversationStateService>();
var stateDimension = state.GetState("embedding_state");
var stateDimension = state.GetState("embedding_dimension");
if (int.TryParse(stateDimension, out var dimension))
{
return dimension > 0 ? dimension : defaultDimension;
return dimension > 0 ? dimension :(Dimension > 0 ? Dimension: DEFAULT_DIMENSION);
}
return Dimension > 0 ? Dimension : defaultDimension;
return Dimension > 0 ? Dimension : DEFAULT_DIMENSION;
}
}