update
This commit is contained in:
@@ -5,12 +5,13 @@ import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"git.nix13.pw/scuroneko/laniakea"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"git.nix13.pw/scuroneko/slog"
|
||||
)
|
||||
|
||||
type OpenAIResponse struct {
|
||||
@@ -55,15 +56,15 @@ type OpenAIAPI struct {
|
||||
Token string
|
||||
Model string
|
||||
BaseURL string
|
||||
Logger *laniakea.Logger
|
||||
Logger *slog.Logger
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI {
|
||||
logger := laniakea.CreateLogger()
|
||||
level := laniakea.FATAL
|
||||
logger := slog.CreateLogger()
|
||||
level := slog.FATAL
|
||||
if os.Getenv("DEBUG") == "true" {
|
||||
level = laniakea.DEBUG
|
||||
level = slog.DEBUG
|
||||
}
|
||||
logger = logger.Prefix("AI").Level(level)
|
||||
proxy, err := url.Parse(os.Getenv("HTTPS_PROXY"))
|
||||
|
||||
Reference in New Issue
Block a user