centralize the header const values

proxy/addons/addIDToHeaders.go

@@ -1,15 +1,11 @@
 package addons
 
 import (
+	"github.com/proxati/llm_proxy/v2/schema/headers"
 	"github.com/proxati/llm_proxy/v2/version"
 	px "github.com/proxati/mitmproxy/proxy"
 )
 
-const (
-	idHeader  = "X-Llm_proxy-id"
-	idVersion = "X-Llm_proxy-version"
-)
-
 type AddIDToHeaders struct {
 	px.BaseAddon
 }
@@ -19,6 +15,6 @@ func NewAddIDToHeaders() *AddIDToHeaders {
 }
 
 func (c *AddIDToHeaders) Response(f *px.Flow) {
-	f.Response.Header.Add(idHeader, f.Id.String())
-	f.Response.Header.Add(idVersion, version.String())
+	f.Response.Header.Add(headers.ProxyID, f.Id.String())
+	f.Response.Header.Add(headers.Version, version.String())
 }

proxy/addons/scheme_upgrader.go

@@ -3,13 +3,10 @@ package addons
 import (
 	"log/slog"
 
+	"github.com/proxati/llm_proxy/v2/schema/headers"
 	px "github.com/proxati/mitmproxy/proxy"
 )
 
-const (
-	upgradedHeader = "X-Llm_proxy-scheme-upgraded"
-)
-
 type SchemeUpgrader struct {
 	px.BaseAddon
 	logger *slog.Logger
@@ -31,14 +28,14 @@ func (c *SchemeUpgrader) Request(f *px.Flow) {
 	}
 
 	// add a header to the request to indicate that the scheme was upgraded
-	f.Request.Header.Add(upgradedHeader, "true")
+	f.Request.Header.Add(headers.SchemeUpgraded, "true")
 
 	// upgrade the connection from http to https, so when sent upstream it will be encrypted
 	f.Request.URL.Scheme = "https"
 }
 
 func (c *SchemeUpgrader) Response(f *px.Flow) {
-	if f.Request.Header.Get(upgradedHeader) != "" {
-		f.Response.Header.Add(upgradedHeader, "true")
+	if f.Request.Header.Get(headers.SchemeUpgraded) != "" {
+		f.Response.Header.Add(headers.SchemeUpgraded, "true")
 	}
 }

proxy/addons/scheme_upgrader_test.go

@@ -4,6 +4,7 @@ import (
 	"log/slog"
 	"testing"
 
+	"github.com/proxati/llm_proxy/v2/schema/headers"
 	px "github.com/proxati/mitmproxy/proxy"
 	"github.com/stretchr/testify/assert"
 )
@@ -21,7 +22,7 @@ func TestSchemeUpgrader_Request(t *testing.T) {
 
 	upgrader.Request(flow)
 	assert.Equal(t, "https", flow.Request.URL.Scheme)
-	assert.Equal(t, "true", flow.Request.Header.Get(upgradedHeader))
+	assert.Equal(t, "true", flow.Request.Header.Get(headers.SchemeUpgraded))
 }
 
 func TestSchemeUpgrader_Request_HTTPS(t *testing.T) {
@@ -37,5 +38,5 @@ func TestSchemeUpgrader_Request_HTTPS(t *testing.T) {
 
 	upgrader.Request(flow)
 	assert.Equal(t, "https", flow.Request.URL.Scheme)
-	assert.Equal(t, "", flow.Request.Header.Get(upgradedHeader))
+	assert.Equal(t, "", flow.Request.Header.Get(headers.SchemeUpgraded))
 }

schema/headers/const.go

@@ -0,0 +1,7 @@
+package headers
+
+const (
+	ProxyID        = "X-Llm_proxy-id"
+	Version        = "X-Llm_proxy-version"
+	SchemeUpgraded = "X-Llm_proxy-scheme-upgraded"
+)