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
unraidAPIKey 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 {
@@ -31,8 +23,7 @@ func NewController() *Controller {
unraidURI := os.Getenv("UNRAID_URI")
if err != nil {
fmt.Println(err.Error())
//return err
fmt.Println(".env file not found, using environment variables")
}
return &Controller{
openaiClient: openaiClient,

View File

@@ -2,6 +2,7 @@ package controller
import (
"context"
"fmt"
"net/http"
"github.com/gin-gonic/gin"
@@ -23,7 +24,10 @@ func (c *Controller) GeneralOpenAI(ctx *gin.Context) {
result, err := createChatCompletion(c, message)
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})
@@ -44,7 +48,10 @@ func (c *Controller) TravelAgentOpenAI(ctx *gin.Context) {
result, err := createChatCompletion(c, message)
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})