From ac7c014b394700230ff65de90594e8bd2d2867bb Mon Sep 17 00:00:00 2001 From: Cezary Bartoszuk Date: Tue, 13 Feb 2024 12:51:50 +0100 Subject: [PATCH] Quickfix for attribution panics (#60465) * Quickfix for attribution panics * Return uninitialized gateway if client is nil --- cmd/frontend/internal/guardrails/init.go | 3 +++ internal/codygateway/client.go | 3 +++ 2 files changed, 6 insertions(+) diff --git a/cmd/frontend/internal/guardrails/init.go b/cmd/frontend/internal/guardrails/init.go index 2124a302094..38c64b26f20 100644 --- a/cmd/frontend/internal/guardrails/init.go +++ b/cmd/frontend/internal/guardrails/init.go @@ -81,6 +81,9 @@ func (e *enterpriseInitialization) Service() attribution.Service { if e.endpoint == "" || e.token == "" { return attribution.Uninitialized{} } + if e.client == nil { + return attribution.Uninitialized{} + } return attribution.NewGatewayProxy(e.observationCtx, e.client) } diff --git a/internal/codygateway/client.go b/internal/codygateway/client.go index ce2da78bcc9..729962a9e58 100644 --- a/internal/codygateway/client.go +++ b/internal/codygateway/client.go @@ -141,6 +141,9 @@ func (c *client) Attribution(ctx context.Context, snippet string, limit int) (At return Attribution{}, err } req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.accessToken)) + if c.cli == nil { + return Attribution{}, errors.New("no http client") + } resp, err := c.cli.Do(req) if err != nil { return Attribution{}, err