1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ai/ai_utils.h"
#include "third_party/blink/public/mojom/ai/model_streaming_responder.mojom.h"
using ModelExecutionError = optimization_guide::
OptimizationGuideModelExecutionError::ModelExecutionError;
blink::mojom::ModelStreamingResponseStatus AIUtils::ConvertModelExecutionError(
ModelExecutionError error) {
switch (error) {
case ModelExecutionError::kUnknown:
return blink::mojom::ModelStreamingResponseStatus::kErrorUnknown;
case ModelExecutionError::kInvalidRequest:
return blink::mojom::ModelStreamingResponseStatus::kErrorInvalidRequest;
case ModelExecutionError::kRequestThrottled:
return blink::mojom::ModelStreamingResponseStatus::kErrorRequestThrottled;
case ModelExecutionError::kPermissionDenied:
return blink::mojom::ModelStreamingResponseStatus::kErrorPermissionDenied;
case ModelExecutionError::kGenericFailure:
return blink::mojom::ModelStreamingResponseStatus::kErrorGenericFailure;
case ModelExecutionError::kRetryableError:
return blink::mojom::ModelStreamingResponseStatus::kErrorRetryableError;
case ModelExecutionError::kNonRetryableError:
return blink::mojom::ModelStreamingResponseStatus::
kErrorNonRetryableError;
case ModelExecutionError::kUnsupportedLanguage:
return blink::mojom::ModelStreamingResponseStatus::
kErrorUnsupportedLanguage;
case ModelExecutionError::kFiltered:
return blink::mojom::ModelStreamingResponseStatus::kErrorFiltered;
case ModelExecutionError::kDisabled:
return blink::mojom::ModelStreamingResponseStatus::kErrorDisabled;
case ModelExecutionError::kCancelled:
return blink::mojom::ModelStreamingResponseStatus::kErrorCancelled;
case ModelExecutionError::kResponseLowQuality:
return blink::mojom::ModelStreamingResponseStatus::
kErrorResponseLowQuality;
}
}
// static
int64_t AIUtils::NormalizeModelDownloadProgress(int64_t bytes_so_far,
int64_t total_bytes) {
double raw_progress_fraction =
bytes_so_far / static_cast<double>(total_bytes);
return raw_progress_fraction * AIUtils::kNormalizedDownloadProgressMax;
}
|