diff --git a/.pi-lens/turn-state.json b/.pi-lens/turn-state.json index 8d71670b..8c0820bb 100644 --- a/.pi-lens/turn-state.json +++ b/.pi-lens/turn-state.json @@ -2,5 +2,5 @@ "files": {}, "turnCycles": 0, "maxCycles": 3, - "lastUpdated": "2026-04-26T18:49:43.830Z" + "lastUpdated": "2026-04-26T18:55:13.038Z" } \ No newline at end of file diff --git a/internal/providers/openai.go b/internal/providers/openai.go index 16feb0aa..840502b8 100644 --- a/internal/providers/openai.go +++ b/internal/providers/openai.go @@ -2,26 +2,26 @@ package providers import ( "context" - "time" "encoding/json" "fmt" "strings" + "time" + "github.com/go-resty/resty/v2" "gophergate/internal/config" "gophergate/internal/models" - "github.com/go-resty/resty/v2" ) type OpenAIProvider struct { - client *resty.Client - config config.OpenAIConfig - apiKey string + client *resty.Client + config config.OpenAIConfig + apiKey string } func NewOpenAIProvider(cfg config.OpenAIConfig, apiKey string) *OpenAIProvider { return &OpenAIProvider{ - client: resty.New(), - config: cfg, + client: resty.New().SetTimeout(30 * time.Second), + config: cfg, apiKey: apiKey, } } @@ -100,7 +100,7 @@ func (p *OpenAIProvider) ChatCompletionStream(ctx context.Context, req *models.U } ch := make(chan *models.ChatCompletionStreamResponse) - + go func() { defer close(ch) err := StreamOpenAI(resp.RawBody(), ch) diff --git a/internal/server/clients.go b/internal/server/clients.go index cc21f644..fcc7d02e 100644 --- a/internal/server/clients.go +++ b/internal/server/clients.go @@ -17,6 +17,7 @@ import ( "log/slog" "github.com/shirou/gopsutil/v3/cpu" +) func (s *Server) handleGetClients(c *gin.Context) { var clients []db.Client diff --git a/internal/server/providers_admin.go b/internal/server/providers_admin.go index bf86969a..2c83aeff 100644 --- a/internal/server/providers_admin.go +++ b/internal/server/providers_admin.go @@ -17,6 +17,7 @@ import ( "log/slog" "github.com/shirou/gopsutil/v3/cpu" +) func (s *Server) handleGetProviders(c *gin.Context) { var dbConfigs []db.ProviderConfig diff --git a/internal/server/system.go b/internal/server/system.go index 67f00f8f..ff94c0e1 100644 --- a/internal/server/system.go +++ b/internal/server/system.go @@ -17,6 +17,7 @@ import ( "log/slog" "github.com/shirou/gopsutil/v3/cpu" +) func (s *Server) handleSystemHealth(c *gin.Context) { c.JSON(http.StatusOK, SuccessResponse(gin.H{ diff --git a/internal/server/users.go b/internal/server/users.go index ecd7568d..abbb2385 100644 --- a/internal/server/users.go +++ b/internal/server/users.go @@ -17,6 +17,7 @@ import ( "log/slog" "github.com/shirou/gopsutil/v3/cpu" +) func (s *Server) handleGetUsers(c *gin.Context) { var users []db.User