coai/main.go

package main

import (
	"chat/api"
	"chat/auth"
	"chat/connection"
	"chat/conversation"
	"chat/generation"
	"chat/middleware"
	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
)

func main() {
	viper.SetConfigFile("config.yaml")
	if err := viper.ReadInConfig(); err != nil {
		panic(err)
	}

	generation.CreateGenerationWithCache("", "写一个turbowarp插件")
	return
	app := gin.Default()
	{
		app.Use(middleware.CORSMiddleware())
		app.Use(middleware.BuiltinMiddleWare(connection.ConnectMySQL(), connection.ConnectRedis()))
		app.Use(middleware.ThrottleMiddleware())
		app.Use(auth.Middleware())

		app.POST("/anonymous", api.AnonymousAPI)
		app.POST("/card", api.CardAPI)
		app.GET("/chat", api.ChatAPI)
		app.POST("/login", auth.LoginAPI)
		app.POST("/state", auth.StateAPI)
		app.GET("/package", auth.PackageAPI)
		app.GET("/quota", auth.QuotaAPI)
		app.POST("/buy", auth.BuyAPI)
		app.GET("/subscription", auth.SubscriptionAPI)
		app.POST("/subscribe", auth.SubscribeAPI)
		app.GET("/conversation/list", conversation.ListAPI)
		app.GET("/conversation/load", conversation.LoadAPI)
		app.GET("/conversation/delete", conversation.DeleteAPI)
	}
	if viper.GetBool("debug") {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}
	if err := app.Run(":" + viper.GetString("server.port")); err != nil {
		panic(err)
	}
}