add X-Llm_proxy-version header to response

robbyt · Sep 3, 2024 · f4274e3 · f4274e3
1 parent 6670120
commit f4274e3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/proxy/addons/addIDToHeaders.go b/proxy/addons/addIDToHeaders.go
@@ -1,11 +1,13 @@
 package addons
 
 import (
+	"github.com/proxati/llm_proxy/v2/version"
 	px "github.com/proxati/mitmproxy/proxy"
 )
 
 const (
-	idHeader = "X-Llm_proxy-id"
+	idHeader  = "X-Llm_proxy-id"
+	idVersion = "X-Llm_proxy-version"
 )
 
 type AddIDToHeaders struct {
@@ -18,4 +20,5 @@ 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())
 }