diff --git a/pkg/providers/openai/chat.go b/pkg/providers/openai/chat.go index 02f3d101..00fdd0ab 100644 --- a/pkg/providers/openai/chat.go +++ b/pkg/providers/openai/chat.go @@ -13,6 +13,8 @@ import ( "strings" "glide/pkg/providers" + + "glide/pkg/telemetry" ) const ( @@ -24,6 +26,7 @@ const ( type ProviderClient struct { BaseURL string `json:"baseURL"` HTTPClient *http.Client `json:"httpClient"` + Telemetry *telemetry.Telemetry `json:"telemetry"` } // ChatRequest is a request to complete a chat completion.. @@ -92,7 +95,7 @@ type ChatResponse struct { func (c *ProviderClient) Chat(u *providers.UnifiedAPIData) (*ChatResponse, error) { // Create a new chat request - slog.Info("creating chat request") + c.Telemetry.Logger.Info("creating new chat request") chatRequest := CreateChatRequest(u) diff --git a/pkg/providers/openai/openaiclient.go b/pkg/providers/openai/openaiclient.go index ddbabff0..6988c2a4 100644 --- a/pkg/providers/openai/openaiclient.go +++ b/pkg/providers/openai/openaiclient.go @@ -8,6 +8,8 @@ import ( "errors" "glide/pkg/providers" + + "glide/pkg/telemetry" ) const ( @@ -30,10 +32,20 @@ var ( // - *Client: A pointer to the created client. // - error: An error if the client creation failed. func Client() (*ProviderClient, error) { + + tel, err := telemetry.NewTelemetry(&telemetry.Config{LogConfig: telemetry.NewLogConfig()}) + if err != nil { + return nil, err + } + + tel.Logger.Info("init openai provider client") + + // Create a new client c := &ProviderClient{ BaseURL: defaultBaseURL, HTTPClient: providers.HTTPClient, + Telemetry: tel, } return c, nil