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

Merge pull request #523 from iceljc/features/refine-image-generation

Browse source 
add default embed dimen
This commit is contained in:
C. Oceania 2024-07-01 17:07:16 -05:00 committed by GitHub
parent a08010a8c9 446f7784e1
commit 3218b36c53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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;
}
}