diff --git a/src/Vectra.Infrastructure/Semantic/Providers/InternalBert/InternalOnnxProvider.cs b/src/Vectra.Infrastructure/Semantic/Providers/InternalBert/InternalOnnxProvider.cs index a9a968d..515f7bb 100644 --- a/src/Vectra.Infrastructure/Semantic/Providers/InternalBert/InternalOnnxProvider.cs +++ b/src/Vectra.Infrastructure/Semantic/Providers/InternalBert/InternalOnnxProvider.cs @@ -56,22 +56,22 @@ public InternalOnnxProvider( } public async Task AnalyzeAsync( - string? requestBody, + string? body, string metadata, CancellationToken cancellationToken) { if (!_enabled) return new SemanticAnalysisResult { Intent = "unknown", Confidence = 0.5, FallbackSafe = true }; - if (string.IsNullOrWhiteSpace(requestBody)) + if (string.IsNullOrWhiteSpace(body)) return new SemanticAnalysisResult { Intent = "unknown", Confidence = 0.5, FallbackSafe = true }; - var cacheKey = $"semantic_internal:{ComputeHash(requestBody)}"; + var cacheKey = $"semantic_internal:{ComputeHash(body)}"; var (success, cached) = await _cacheProvider!.TryGetValueAsync(cacheKey); if (success) return cached!; - var (inputIds, attentionMask) = _tokenizer!.Tokenize(requestBody, _maxLength); + var (inputIds, attentionMask) = _tokenizer!.Tokenize(body, _maxLength); var inputTensor = new DenseTensor(inputIds, new[] { 1, _maxLength }); var maskTensor = new DenseTensor(attentionMask, new[] { 1, _maxLength });