main.go

package main

import (
	"flag"
	"net/http"

	"go-gpt-server/api"
	"go-gpt-server/middle"
)

func main() {
	var clearConvs bool
	flag.BoolVar(&clearConvs, "clear", false, "clear all conversations")
	flag.BoolVar(&api.MultiSession, "multi", false, "support multi conversations")
	flag.Parse()

	if clearConvs {
		api.ClearConvs()
	}

	http.Handle("/event", middle.EnableCors(middle.HandleOptions(http.HandlerFunc(api.HandleSSE))))
	http.Handle("/token", http.HandlerFunc(api.GenToken))
	http.Handle("/chat", middle.EnableCors(middle.HandleOptions(middle.AuthMiddleware(http.HandlerFunc(api.HandleChat)))))
	http.Handle("/conv", middle.EnableCors(middle.HandleOptions(middle.AuthMiddleware(http.HandlerFunc(api.HandleConv)))))
	println("Welcome to the gpt server")
	http.ListenAndServe(":8000", nil)
}