update sth. at 2024-11-14 22:11:45

Dockerfile

@@ -26,7 +26,6 @@ WORKDIR /root/
 COPY --from=builder /app/ddg-chat-go .
 
 # 设置环境变量 
-ENV API_PREFIX="/api"  
 ENV MAX_RETRY_COUNT="3"  
 ENV RETRY_DELAY="5000"    
 ENV PORT="7860"

main.go

@@ -29,7 +29,6 @@ var config Config
 func init() {
 	godotenv.Load()
 	config = Config{
-		APIPrefix:     getEnv("API_PREFIX", "/"),
 		APIKey:        getEnv("API_KEY", ""),
 		MaxRetryCount: getIntEnv("MAX_RETRY_COUNT", 3),
 		RetryDelay:    getDurationEnv("RETRY_DELAY", 5000),
@@ -94,8 +93,8 @@ func main() {
 	authorized := r.Group("/")
 	authorized.Use(authMiddleware())
 	{
-		authorized.GET(config.APIPrefix+"/v1/models", handleModels)
-		authorized.POST(config.APIPrefix+"/v1/chat/completions", handleCompletion)
+		authorized.GET("/models", handleModels)
+		authorized.POST("/completions", handleCompletion)
 	}
 
 	port := os.Getenv("PORT")