Преглед на файлове

Merge pull request #4 from luxlzz6/main

添加了图生文 混图路由
Calcium-Ion преди 2 години
родител
ревизия
8b2e4ec22d
променени са 2 файла, в които са добавени 6 реда и са изтрити 0 реда
  1. 4 0
      controller/relay.go
  2. 2 0
      router/relay-router.go

+ 4 - 0
controller/relay.go

@@ -255,6 +255,10 @@ func RelayMidjourney(c *gin.Context) {
 	relayMode := RelayModeUnknown
 	if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/imagine") {
 		relayMode = RelayModeMidjourneyImagine
+	} else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/blend") {
+		relayMode = RelayModeMidjourneyBlend
+	} else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/describe") {
+		relayMode = RelayModeMidjourneyDescribe
 	} else if strings.HasPrefix(c.Request.URL.Path, "/mj/notify") {
 		relayMode = RelayModeMidjourneyNotify
 	} else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/change") {

+ 2 - 0
router/relay-router.go

@@ -48,6 +48,8 @@ func SetRelayRouter(router *gin.Engine) {
 	{
 		relayMjRouter.POST("/submit/imagine", controller.RelayMidjourney)
 		relayMjRouter.POST("/submit/change", controller.RelayMidjourney)
+		relayMjRouter.POST("/submit/describe", controller.RelayMidjourney)
+		relayMjRouter.POST("/submit/blend", controller.RelayMidjourney)
 		relayMjRouter.POST("/notify", controller.RelayMidjourney)
 		relayMjRouter.GET("/task/:id/fetch", controller.RelayMidjourney)
 	}