This commit is contained in:
2023-05-21 17:37:58 -05:00
parent 22118a02b0
commit 3f7c7c98ed
3 changed files with 15 additions and 13 deletions

View File

@@ -12,14 +12,6 @@ type Controller struct {
openaiClient *openai.Client openaiClient *openai.Client
unraidAPIKey string unraidAPIKey string
unraidURI string unraidURI string
config *configStruct
}
type configStruct struct {
openaiApiKey string `json : "OPENAI_API_KEY"`
unraidAPIKey string `json : "UNRAID_API_KEY"`
unraidURI string `json : "UNRAID_URI"`
} }
func NewController() *Controller { func NewController() *Controller {
@@ -31,8 +23,7 @@ func NewController() *Controller {
unraidURI := os.Getenv("UNRAID_URI") unraidURI := os.Getenv("UNRAID_URI")
if err != nil { if err != nil {
fmt.Println(err.Error()) fmt.Println(".env file not found, using environment variables")
//return err
} }
return &Controller{ return &Controller{
openaiClient: openaiClient, openaiClient: openaiClient,

View File

@@ -2,6 +2,7 @@ package controller
import ( import (
"context" "context"
"fmt"
"net/http" "net/http"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
@@ -23,7 +24,10 @@ func (c *Controller) GeneralOpenAI(ctx *gin.Context) {
result, err := createChatCompletion(c, message) result, err := createChatCompletion(c, message)
if err != nil { if err != nil {
ctx.AbortWithError(http.StatusInternalServerError, err) err := ctx.AbortWithError(http.StatusInternalServerError, err)
if err != nil {
fmt.Println("Failed to send message")
}
} }
ctx.JSON(http.StatusOK, gin.H{"message": result}) ctx.JSON(http.StatusOK, gin.H{"message": result})
@@ -44,7 +48,10 @@ func (c *Controller) TravelAgentOpenAI(ctx *gin.Context) {
result, err := createChatCompletion(c, message) result, err := createChatCompletion(c, message)
if err != nil { if err != nil {
ctx.AbortWithError(http.StatusInternalServerError, err) err := ctx.AbortWithError(http.StatusInternalServerError, err)
if err != nil {
fmt.Println("Failed to send message")
}
} }
ctx.JSON(http.StatusOK, gin.H{"message": result}) ctx.JSON(http.StatusOK, gin.H{"message": result})

View File

@@ -1,6 +1,7 @@
package main package main
import ( import (
"fmt"
"net/http" "net/http"
"strings" "strings"
@@ -57,7 +58,10 @@ func main() {
} }
} }
r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler)) r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
r.Run(":8080") err := r.Run(":8080")
if err != nil {
fmt.Println("Failed to start server")
}
} }
func authMiddleware(allowedGroups []string) gin.HandlerFunc { func authMiddleware(allowedGroups []string) gin.HandlerFunc {