feat: add endpoint to list aibridge interceptions (#19929)

Co-authored-by: Dean Sheather <dean@deansheather.com>
2026-09-24 15:04:27 +08:00 · 2025-09-27 00:20:33 +10:00
parent d70e26d2e3
commit 0a6ba5d51a
29 changed files with 2646 additions and 144 deletions
@@ -0,0 +1,125 @@
+package codersdk
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type AIBridgeInterception struct {
+	ID          uuid.UUID            `json:"id" format:"uuid"`
+	InitiatorID uuid.UUID            `json:"initiator_id" format:"uuid"`
+	Provider    string               `json:"provider"`
+	Model       string               `json:"model"`
+	Metadata    map[string]any       `json:"metadata"`
+	StartedAt   time.Time            `json:"started_at" format:"date-time"`
+	TokenUsages []AIBridgeTokenUsage `json:"token_usages"`
+	UserPrompts []AIBridgeUserPrompt `json:"user_prompts"`
+	ToolUsages  []AIBridgeToolUsage  `json:"tool_usages"`
+}
+
+type AIBridgeTokenUsage struct {
+	ID                 uuid.UUID      `json:"id" format:"uuid"`
+	InterceptionID     uuid.UUID      `json:"interception_id" format:"uuid"`
+	ProviderResponseID string         `json:"provider_response_id"`
+	InputTokens        int64          `json:"input_tokens"`
+	OutputTokens       int64          `json:"output_tokens"`
+	Metadata           map[string]any `json:"metadata"`
+	CreatedAt          time.Time      `json:"created_at" format:"date-time"`
+}
+
+type AIBridgeUserPrompt struct {
+	ID                 uuid.UUID      `json:"id" format:"uuid"`
+	InterceptionID     uuid.UUID      `json:"interception_id" format:"uuid"`
+	ProviderResponseID string         `json:"provider_response_id"`
+	Prompt             string         `json:"prompt"`
+	Metadata           map[string]any `json:"metadata"`
+	CreatedAt          time.Time      `json:"created_at" format:"date-time"`
+}
+
+type AIBridgeToolUsage struct {
+	ID                 uuid.UUID      `json:"id" format:"uuid"`
+	InterceptionID     uuid.UUID      `json:"interception_id" format:"uuid"`
+	ProviderResponseID string         `json:"provider_response_id"`
+	ServerURL          string         `json:"server_url"`
+	Tool               string         `json:"tool"`
+	Input              string         `json:"input"`
+	Injected           bool           `json:"injected"`
+	InvocationError    string         `json:"invocation_error"`
+	Metadata           map[string]any `json:"metadata"`
+	CreatedAt          time.Time      `json:"created_at" format:"date-time"`
+}
+
+type AIBridgeListInterceptionsResponse struct {
+	Results []AIBridgeInterception `json:"results"`
+}
+
+// @typescript-ignore AIBridgeListInterceptionsFilter
+type AIBridgeListInterceptionsFilter struct {
+	// Limit defaults to 100, max is 1000.
+	// Offset based pagination is not supported for AIBridge interceptions. Use
+	// cursor pagination instead with after_id.
+	Pagination Pagination `json:"pagination,omitempty"`
+
+	// Initiator is a user ID, username, or "me".
+	Initiator     string    `json:"initiator,omitempty"`
+	StartedBefore time.Time `json:"started_before,omitempty" format:"date-time"`
+	StartedAfter  time.Time `json:"started_after,omitempty" format:"date-time"`
+	Provider      string    `json:"provider,omitempty"`
+	Model         string    `json:"model,omitempty"`
+
+	FilterQuery string `json:"q,omitempty"`
+}
+
+// asRequestOption returns a function that can be used in (*Client).Request.
+// It modifies the request query parameters.
+func (f AIBridgeListInterceptionsFilter) asRequestOption() RequestOption {
+	return func(r *http.Request) {
+		var params []string
+		// Make sure all user input is quoted to ensure it's parsed as a single
+		// string.
+		if f.Initiator != "" {
+			params = append(params, fmt.Sprintf("initiator:%q", f.Initiator))
+		}
+		if !f.StartedBefore.IsZero() {
+			params = append(params, fmt.Sprintf("started_before:%q", f.StartedBefore.Format(time.RFC3339Nano)))
+		}
+		if !f.StartedAfter.IsZero() {
+			params = append(params, fmt.Sprintf("started_after:%q", f.StartedAfter.Format(time.RFC3339Nano)))
+		}
+		if f.Provider != "" {
+			params = append(params, fmt.Sprintf("provider:%q", f.Provider))
+		}
+		if f.Model != "" {
+			params = append(params, fmt.Sprintf("model:%q", f.Model))
+		}
+		if f.FilterQuery != "" {
+			// If custom stuff is added, just add it on here.
+			params = append(params, f.FilterQuery)
+		}
+
+		q := r.URL.Query()
+		q.Set("q", strings.Join(params, " "))
+		r.URL.RawQuery = q.Encode()
+	}
+}
+
+// AIBridgeListInterceptions returns AIBridge interceptions with the given
+// filter.
+func (c *ExperimentalClient) AIBridgeListInterceptions(ctx context.Context, filter AIBridgeListInterceptionsFilter) (AIBridgeListInterceptionsResponse, error) {
+	res, err := c.Request(ctx, http.MethodGet, "/api/experimental/aibridge/interceptions", nil, filter.asRequestOption(), filter.Pagination.asRequestOption(), filter.Pagination.asRequestOption())
+	if err != nil {
+		return AIBridgeListInterceptionsResponse{}, err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		return AIBridgeListInterceptionsResponse{}, ReadBodyAsError(res)
+	}
+	var resp AIBridgeListInterceptionsResponse
+	return resp, json.NewDecoder(res.Body).Decode(&resp)
+}