middleware

package

v0.1.4 Latest Latest Go to latest Published: Apr 23, 2026 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

func LocalhostOnly() gin.HandlerFunc

LocalhostOnly returns middleware that restricts access to loopback addresses.

func UsageMiddleware(usageStore *store.UsageStore) gin.HandlerFunc

UsageMiddleware records API usage statistics via a Gin middleware.

type StreamUsageAccumulator struct {
	Model        string
	InputTokens  int64
	OutputTokens int64
}

StreamUsageAccumulator accumulates token usage from SSE data lines.

func (a *StreamUsageAccumulator) Sniff(data string, format string)

Sniff parses a single SSE data line and accumulates usage fields.