usage

type ACPExtension struct {
	ServiceID           string
	AgentType           string
	Operation           string
	ThreadID            string
	SessionID           string
	PermissionRequestID string
	FreshSession        *bool
	EventCounts         map[string]int
	UsageJSON           string
	ResultStatus        string
}

type ACPSummary struct {
	RequestCount int64 `json:"request_count"`
	TurnCount    int64 `json:"turn_count"`
	SuccessCount int64 `json:"success_count"`
	FailureCount int64 `json:"failure_count"`
	AvgLatencyMS int64 `json:"avg_latency_ms"`
}

type ACPUsageEvent struct {
	InteractionEvent
	ServiceID           string
	AgentType           string
	Operation           string
	ThreadID            string
	SessionID           string
	PermissionRequestID string
	FreshSession        *bool
	EventCounts         map[string]int
	UsageJSON           string
	ResultStatus        string
}

type AgentAttribution struct {
	// contains filtered or unexported fields
}

type AgentAttributor interface {
	ResolveAgentID(routeID, serviceID, sessionID string) (string, bool)
}

type AttributionFilter struct {
	AgentID       string
	RouteIDs      []string
	ACPServiceIDs []string
}

type BreakdownOptions struct {
	From        string
	To          string
	GroupBy     string
	OrderBy     string
	Limit       int
	Filters     map[string]string
	Attribution *AttributionFilter
}

type BreakdownResponse struct {
	GroupBy string           `json:"group_by,omitempty"`
	Items   []map[string]any `json:"items"`
	Limit   int              `json:"limit"`
}

type Config struct {
	RetentionDays int `json:"retention_days,omitempty"`
	MaxAgentDepth int `json:"max_agent_depth,omitempty"`
}

type EventListOptions struct {
	From        string
	To          string
	Limit       int
	Filters     map[string]string
	Success     *bool
	Attribution *AttributionFilter
}

type EventListResponse struct {
	Items []map[string]any `json:"items"`
	Limit int              `json:"limit"`
}

type EventSink interface {
	Enqueue(any) bool
}

type InMemorySink struct {
	Dropped atomic.Uint64
	Events  []any
}

type InteractionDimensions struct {
	TraceID       string
	SpanID        string
	ParentSpanID  string
	AgentDepth    int
	RouteID       string
	RouteKind     string
	RouteProtocol string
	VirtualKeyID  string
	// AgentID, when set by the caller, is stamped verbatim. When empty the
	// observer attempts attribution from RouteID via the wired AgentAttributor.
	AgentID string
}

type InteractionEvent struct {
	EventID       string
	TraceID       string
	SpanID        string
	ParentSpanID  string
	AgentDepth    int
	StartedAt     time.Time
	FinishedAt    time.Time
	RouteID       string
	RouteKind     string
	RouteProtocol string
	VirtualKeyID  string
	Success       bool
	StatusCode    int
	ErrorType     string
	LatencyMS     int64
	// AgentID is the write-time attribution tag. It is empty for non-agent
	// traffic and for events whose route/service maps to zero or more than one
	// agent (stamp only when unambiguous).
	AgentID string
}

type InteractionObserver interface {
	Begin(ctx context.Context, dims InteractionDimensions) (InteractionSpan, context.Context)
}

type InteractionOutcome struct {
	Success    bool
	StatusCode int
	ErrorType  string
	FinishedAt time.Time
}

type InteractionSpan interface {
	SetExtension(v any)
	AddAnnotation(key, value string)
	Finish(outcome InteractionOutcome)
}

type LLMExtension struct {
	LLMAPI           string
	APIOperation     string
	ProviderID       string
	ProviderType     string
	LogicalModel     string
	UpstreamModel    string
	CredentialSource string
	CredentialID     string
	Stream           *bool
	InputTokens      *int
	OutputTokens     *int
	TotalTokens      *int
	UsageFinalized   *bool
	RequestToolCount *int
	RequestToolNames []string
	ToolCallCount    *int
	ToolNames        []string
}

type LLMSummary struct {
	RequestCount int64 `json:"request_count"`
	SuccessCount int64 `json:"success_count"`
	FailureCount int64 `json:"failure_count"`
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
	TotalTokens  int64 `json:"total_tokens"`
	AvgLatencyMS int64 `json:"avg_latency_ms"`
}

type LLMUsageEvent struct {
	InteractionEvent
	LLMAPI           string
	APIOperation     string
	ProviderID       string
	ProviderType     string
	LogicalModel     string
	UpstreamModel    string
	CredentialSource string
	CredentialID     string
	Stream           bool
	InputTokens      int
	OutputTokens     int
	TotalTokens      int
	UsageFinalized   bool
	RequestToolCount int
	RequestToolNames []string
	ToolCallCount    int
	ToolNames        []string
}

type MCPExtension struct {
	RequestID          string
	ServiceID          string
	Method             string
	ToolName           string
	PresentedToolName  string
	ExecutedToolName   string
	ExecutionMode      string
	PolicyAction       string
	ResourceURI        string
	PromptName         string
	CompletionRefType  string
	CompletionArgument string
	ArgCount           *int
	ResultStatus       string
	Cancelled          *bool
	ToolArgsJSON       string
}

type MCPSummary struct {
	RequestCount   int64 `json:"request_count"`
	SuccessCount   int64 `json:"success_count"`
	FailureCount   int64 `json:"failure_count"`
	ToolsCallCount int64 `json:"tools_call_count"`
	AvgLatencyMS   int64 `json:"avg_latency_ms"`
}

type MCPUsageEvent struct {
	InteractionEvent
	RequestID          string
	ServiceID          string
	Method             string
	ToolName           string
	PresentedToolName  string
	ExecutedToolName   string
	ExecutionMode      string
	PolicyAction       string
	ResourceURI        string
	PromptName         string
	CompletionRefType  string
	CompletionArgument string
	ArgCount           int
	ResultStatus       string
	Cancelled          bool
	ToolArgsJSON       string
}

type NoopObserver struct{}

type NoopSpan struct{}

type Observer struct {
	// contains filtered or unexported fields
}

type Pipeline interface {
	EventSink
	Start()
	Close() error
	DroppedEvents() uint64
	WriteFailures() uint64
}

type PrometheusProvider interface {
	PrometheusSnapshot() PrometheusSnapshot
}

type PrometheusSnapshot struct {
	RequestsByKind map[string]int64
	FailuresByKind map[string]int64
	TokensByKind   map[string]int64
}

type QueryService interface {
	Summary() (Summary, error)
	ListEvents(kind string, opts EventListOptions) (EventListResponse, error)
	ListInteractions(opts EventListOptions) (EventListResponse, error)
	LLMTimeseries(opts TimeseriesOptions) (SeriesResponse, error)
	LLMBreakdown(opts BreakdownOptions) (BreakdownResponse, error)
	MCPTimeseries(opts TimeseriesOptions) (SeriesResponse, error)
	MCPBreakdown(opts BreakdownOptions) (BreakdownResponse, error)
	MCPToolsSummary(opts SummaryOptions) (BreakdownResponse, error)
	ACPTimeseries(opts TimeseriesOptions) (SeriesResponse, error)
	ACPBreakdown(opts BreakdownOptions) (BreakdownResponse, error)
	ACPSummary(opts BreakdownOptions) (BreakdownResponse, error)
	InteractionsSummary(opts BreakdownOptions) (BreakdownResponse, error)
}

type RuntimeStats interface {
	DroppedEvents() uint64
	WriteFailures() uint64
}

type SQLDBProvider interface {
	UsageDB() *gorm.DB
}

type SeriesResponse struct {
	Bucket  string           `json:"bucket"`
	GroupBy string           `json:"group_by,omitempty"`
	Items   []map[string]any `json:"items"`
}

type Summary struct {
	LLM LLMSummary `json:"llm"`
	MCP MCPSummary `json:"mcp"`
	ACP ACPSummary `json:"acp"`
}

type SummaryOptions struct {
	From        string
	To          string
	Limit       int
	Filters     map[string]string
	Attribution *AttributionFilter
}

type TimeseriesOptions struct {
	From        string
	To          string
	Bucket      string
	GroupBy     string
	Filters     map[string]string
	Attribution *AttributionFilter
}

type UsageService struct {
	// contains filtered or unexported fields
}