mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-10-06 02:09:40 +00:00
improved stability of embedding chunks
This commit is contained in:
1 parent
1bf9328fb9
commit
759c7f9d89
3 files changed
+37
-3
No files matched your search
@@ -82,7 +82,7 @@ public sealed partial class DataSourceEmbeddingService
|
||||
|
||||
private async IAsyncEnumerable<string> SplitChunkByEmbeddingTokenLimitAsync(string chunk, EmbeddingProvider embeddingProvider, [System.Runtime.CompilerServices.EnumeratorCancellation] CancellationToken token)
|
||||
{
|
||||
var tokenLimit = embeddingProvider.EffectiveTokenLimit;
|
||||
var tokenLimit = embeddingProvider.EffectiveChunkTokenLimit;
|
||||
var tokenCount = await this.GetEmbeddingTokenCountAsync(embeddingProvider, chunk, token);
|
||||
if (tokenCount <= tokenLimit)
|
||||
{
|
||||
@@ -90,6 +90,15 @@ public sealed partial class DataSourceEmbeddingService
|
||||
yield break;
|
||||
}
|
||||
|
||||
if (embeddingProvider.UsesAssumedTokenSizing)
|
||||
{
|
||||
logger.LogDebug(
|
||||
"Using conservative embedding chunk limit {ChunkTokenLimit} for provider '{EmbeddingProviderName}' because tokenizer or token limit sizing is assumed. ConfiguredTokenLimit={ConfiguredTokenLimit}.",
|
||||
tokenLimit,
|
||||
embeddingProvider.Name,
|
||||
embeddingProvider.EffectiveTokenLimit);
|
||||
}
|
||||
|
||||
logger.LogDebug(
|
||||
"Splitting an embedding chunk for provider '{EmbeddingProviderName}' because it has {TokenCount} tokens and the configured limit is {TokenLimit}.",
|
||||
embeddingProvider.Name,
|
||||
@@ -430,7 +439,8 @@ public sealed partial class DataSourceEmbeddingService
|
||||
embeddingProvider.Host,
|
||||
embeddingProvider.Hostname,
|
||||
embeddingProvider.TokenizerPath,
|
||||
embeddingProvider.EffectiveTokenLimit);
|
||||
embeddingProvider.EffectiveTokenLimit,
|
||||
embeddingProvider.EffectiveChunkTokenLimit);
|
||||
}
|
||||
|
||||
private async Task<string> BuildFingerprintAsync(FileInfo file, CancellationToken token)
|
||||
|
||||
Reference in new issue
Block a user