first commit

core/providers/huggingface/errors.go

@@ -0,0 +1,57 @@
+package huggingface
+
+import (
+	"fmt"
+	"strings"
+
+	providerUtils "github.com/maximhq/bifrost/core/providers/utils"
+	"github.com/maximhq/bifrost/core/schemas"
+	"github.com/valyala/fasthttp"
+)
+
+// parseHuggingFaceImageError parses HuggingFace error responses
+func parseHuggingFaceImageError(resp *fasthttp.Response) *schemas.BifrostError {
+	var errorResp HuggingFaceResponseError
+	bifrostErr := providerUtils.HandleProviderAPIError(resp, &errorResp)
+
+	if strings.TrimSpace(errorResp.Type) != "" {
+		typeCopy := errorResp.Type
+		bifrostErr.Type = &typeCopy
+	}
+
+	if bifrostErr.Error == nil {
+		bifrostErr.Error = &schemas.ErrorField{}
+	}
+
+	// Handle FastAPI validation errors
+	if len(errorResp.Detail) > 0 {
+		var errorMessages []string
+		for _, detail := range errorResp.Detail {
+			msg := detail.Msg
+			if len(detail.Loc) > 0 {
+				// Build location string from loc array
+				var locParts []string
+				for _, locPart := range detail.Loc {
+					if locStr, ok := locPart.(string); ok {
+						locParts = append(locParts, locStr)
+					} else if locNum, ok := locPart.(float64); ok {
+						locParts = append(locParts, fmt.Sprintf("%.0f", locNum))
+					}
+				}
+				if len(locParts) > 0 {
+					msg = fmt.Sprintf("%s at %s", msg, strings.Join(locParts, "."))
+				}
+			}
+			errorMessages = append(errorMessages, msg)
+		}
+		if len(errorMessages) > 0 {
+			bifrostErr.Error.Message = strings.Join(errorMessages, "; ")
+		}
+	} else if strings.TrimSpace(errorResp.Message) != "" {
+		bifrostErr.Error.Message = errorResp.Message
+	} else if strings.TrimSpace(errorResp.Error) != "" {
+		bifrostErr.Error.Message = errorResp.Error
+	}
+
+	return bifrostErr
+}