missingstudio
diff --git a/‎gateway/internal/api/v1/chatcompletions.go‎
Lines changed: 4 additions & 5 deletions b/‎gateway/internal/api/v1/chatcompletions.go‎
Lines changed: 4 additions & 5 deletions
diff --git a/‎gateway/internal/api/v1/logs.go‎
Lines changed: 36 additions & 0 deletions b/‎gateway/internal/api/v1/logs.go‎
Lines changed: 36 additions & 0 deletions
diff --git a/‎gateway/internal/httpserver/server.go‎
Lines changed: 2 additions & 0 deletions b/‎gateway/internal/httpserver/server.go‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎gateway/pkg/utils/headers.go‎
Lines changed: 0 additions & 6 deletions b/‎gateway/pkg/utils/headers.go‎
Lines changed: 0 additions & 6 deletions
diff --git a/‎protos/pkg/llm/llmv1connect/service.connect.go‎
Lines changed: 51 additions & 20 deletions b/‎protos/pkg/llm/llmv1connect/service.connect.go‎
Lines changed: 51 additions & 20 deletions
@@ -60,11 +60,10 @@ func (s *V1Handler) GetChatCompletions(
 	ingesterdata["provider"] = provider.GetName()
 	ingesterdata["model"] = data.Model
 	ingesterdata["latency"] = latency
-	ingesterdata["usage"] = map[string]interface{}{
-		"total_tokens":      data.Usage.TotalTokens,
-		"prompt_tokens":     data.Usage.PromptTokens,
-		"completion_tokens": data.Usage.CompletionTokens,
-	}
+	ingesterdata["total_tokens"] = *data.Usage.TotalTokens
+	ingesterdata["prompt_tokens"] = *data.Usage.PromptTokens
+	ingesterdata["completion_tokens"] = *data.Usage.CompletionTokens
+
 	go s.ingester.Ingest(ingesterdata, "analytics")
 
 	response := connect.NewResponse(data)
 
@@ -0,0 +1,36 @@
+package v1
+
+import (
+	"context"
+
+	"connectrpc.com/connect"
+	"github.com/missingstudio/studio/common/errors"
+	llmv1 "github.com/missingstudio/studio/protos/pkg/llm"
+	"google.golang.org/protobuf/types/known/emptypb"
+	"google.golang.org/protobuf/types/known/structpb"
+)
+
+func (s *V1Handler) ListTrackingLogs(ctx context.Context, req *connect.Request[emptypb.Empty]) (*connect.Response[llmv1.LogResponse], error) {
+	response, err := s.ingester.Get("select * from analytics")
+	if err != nil {
+		return nil, errors.New(err)
+	}
+
+	logs := []*structpb.Struct{}
+	for _, log := range response {
+		point := map[string]interface{}{
+			"latency":           log["latency"],
+			"model":             log["model"],
+			"provider":          log["provider"],
+			"total_tokens":      log["total_tokens"],
+			"prompt_tokens":     log["prompt_tokens"],
+			"completion_tokens": log["completion_tokens"],
+		}
+		pointdata, _ := structpb.NewStruct(point)
+		logs = append(logs, pointdata)
+	}
+
+	return connect.NewResponse(&llmv1.LogResponse{
+		Logs: logs,
+	}), nil
+}
@@ -84,8 +84,10 @@ func newCORS() *cors.Cors {
 		ExposedHeaders: []string{
 			// Content-Type is in the default safelist.
 			"Accept",
+			"Authorization",
 			"Accept-Encoding",
 			"Accept-Post",
+			"Content-Type",
 			"Connect-Accept-Encoding",
 			"Connect-Content-Encoding",
 			"Content-Encoding",
 
@@ -95,11 +95,5 @@ func ValidateHeaders(data interface{}) error {
 }
 
 func CopyHeaders[T any](resp *http.Response, newResponse *connect.Response[T]) *connect.Response[T] {
-	for key, values := range resp.Header {
-		for _, value := range values {
-			newResponse.Header().Set(key, value)
-		}
-	}
-
 	return newResponse
 }
Original file line number	Diff line number	Diff line change
`@@ -95,11 +95,5 @@ func ValidateHeaders(data interface{}) error {`
`95`	`95`	`}`
`96`	`96`
`97`	`97`	`func CopyHeaders[T any](resp http.Response, newResponse connect.Response[T]) *connect.Response[T] {`
`98`		`- for key, values := range resp.Header {`
`99`		`- for _, value := range values {`
`100`		`- newResponse.Header().Set(key, value)`
`101`		`- }`
`102`		`- }`
`103`		`-`
`104`	`98`	`return newResponse`
`105`	`99`	`}`