diff --git a/controller/relay.go b/controller/relay.go
index 761b2d8..1797084 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -170,7 +170,8 @@ type OpenAIEmbeddingResponse struct {
 type ImageResponse struct {
 	Created int `json:"created"`
 	Data    []struct {
-		Url string `json:"url"`
+		Url     string `json:"url"`
+		B64Json string `json:"b64_json"`
 	}
 }