103 lines
3.4 KiB
Go
103 lines
3.4 KiB
Go
package promptkit
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/capacity"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/generation"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/llm"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/profile"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/promptdef"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/usecase"
|
|
)
|
|
|
|
func mapPublicError(err error) error {
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
var internalCapacityError *usecase.CapacityError
|
|
if errors.As(err, &internalCapacityError) && internalCapacityError != nil &&
|
|
strings.TrimSpace(internalCapacityError.BackendID) != "" {
|
|
return &CapacityError{BackendID: internalCapacityError.BackendID}
|
|
}
|
|
var internalExhausted *generation.ExhaustedError
|
|
if errors.As(err, &internalExhausted) && internalExhausted != nil {
|
|
mapped := newRetryExhaustedError(internalExhausted.Attempts(), mapPublicError(internalExhausted.Unwrap()))
|
|
if errors.Is(err, usecase.ErrValidation) && !errors.Is(mapped, ErrValidation) {
|
|
return fmt.Errorf("%w: %w", ErrValidation, mapped)
|
|
}
|
|
if publicErr := publicErrorFor(err); publicErr != nil && !errors.Is(mapped, publicErr) {
|
|
return fmt.Errorf("%w: %w", publicErr, mapped)
|
|
}
|
|
return mapped
|
|
}
|
|
publicErr := publicErrorFor(err)
|
|
var providerHTTPError *llm.ProviderHTTPError
|
|
if errors.As(err, &providerHTTPError) && providerHTTPError != nil {
|
|
generationErr := newGenerationError(
|
|
providerHTTPError.StatusCode(),
|
|
providerHTTPError.ProviderCode(),
|
|
providerHTTPError.ProviderType(),
|
|
providerHTTPError.ProviderMessage(),
|
|
)
|
|
if publicErr != nil && !errors.Is(publicErr, ErrLLMGenerate) {
|
|
return fmt.Errorf("%w: %w", publicErr, generationErr)
|
|
}
|
|
return generationErr
|
|
}
|
|
if publicErr == nil {
|
|
return err
|
|
}
|
|
switch {
|
|
case errors.Is(err, llm.ErrTransport):
|
|
return fmt.Errorf("%w: %w: %w", publicErr, ErrTransport, err)
|
|
case errors.Is(err, llm.ErrMalformedResponse):
|
|
return fmt.Errorf("%w: %w: %w", publicErr, ErrMalformedResponse, err)
|
|
default:
|
|
return fmt.Errorf("%w: %w", publicErr, err)
|
|
}
|
|
}
|
|
|
|
func publicErrorFor(err error) error {
|
|
switch {
|
|
case errors.Is(err, promptdef.ErrPromptDefinitionNotFound):
|
|
return ErrPromptNotFound
|
|
case errors.Is(err, profile.ErrProfileNotFound):
|
|
return ErrProfileNotFound
|
|
case errors.Is(err, usecase.ErrProfileRequired):
|
|
return errors.Join(ErrInvalidRequest, ErrProfileRequired)
|
|
case errors.Is(err, usecase.ErrPromptLoad):
|
|
return ErrPromptLoad
|
|
case errors.Is(err, usecase.ErrProfileLoad):
|
|
return ErrProfileLoad
|
|
case errors.Is(err, promptdef.ErrInvalidYAML), errors.Is(err, promptdef.ErrInvalidPromptDefinition):
|
|
return ErrPromptLoad
|
|
case isProfileLoadCause(err):
|
|
return ErrProfileLoad
|
|
case errors.Is(err, usecase.ErrAPIKeyEnvMissing):
|
|
return errors.Join(ErrInvalidRequest, ErrAPIKeyEnvMissing)
|
|
case errors.Is(err, capacity.ErrCapacityExceeded):
|
|
return ErrCapacityExceeded
|
|
case errors.Is(err, usecase.ErrArtifactLoad):
|
|
return ErrArtifactLoad
|
|
case errors.Is(err, usecase.ErrPromptRender):
|
|
return ErrPromptRender
|
|
case errors.Is(err, usecase.ErrLLMGenerate):
|
|
return ErrLLMGenerate
|
|
case errors.Is(err, usecase.ErrValidation):
|
|
return ErrValidation
|
|
case errors.Is(err, usecase.ErrInvalidRequest):
|
|
return ErrInvalidRequest
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func isProfileLoadCause(err error) bool {
|
|
return errors.Is(err, profile.ErrInvalidYAML) ||
|
|
errors.Is(err, profile.ErrInvalidProfile) ||
|
|
errors.Is(err, profile.ErrRawAPIKeyNotAllowed)
|
|
}
|