| 1980 | } |
| 1981 | |
| 1982 | func AIBridgeTokenUsage(t testing.TB, db database.Store, seed database.InsertAIBridgeTokenUsageParams) database.AIBridgeTokenUsage { |
| 1983 | usage, err := db.InsertAIBridgeTokenUsage(genCtx, database.InsertAIBridgeTokenUsageParams{ |
| 1984 | ID: takeFirst(seed.ID, uuid.New()), |
| 1985 | InterceptionID: takeFirst(seed.InterceptionID, uuid.New()), |
| 1986 | ProviderResponseID: takeFirst(seed.ProviderResponseID, "provider_response_id"), |
| 1987 | InputTokens: takeFirst(seed.InputTokens, 100), |
| 1988 | OutputTokens: takeFirst(seed.OutputTokens, 100), |
| 1989 | CacheReadInputTokens: seed.CacheReadInputTokens, |
| 1990 | CacheWriteInputTokens: seed.CacheWriteInputTokens, |
| 1991 | Metadata: takeFirstSlice(seed.Metadata, json.RawMessage("{}")), |
| 1992 | CreatedAt: takeFirst(seed.CreatedAt, dbtime.Now()), |
| 1993 | }) |
| 1994 | require.NoError(t, err, "insert aibridge token usage") |
| 1995 | return usage |
| 1996 | } |
| 1997 | |
| 1998 | func AIBridgeUserPrompt(t testing.TB, db database.Store, seed database.InsertAIBridgeUserPromptParams) database.AIBridgeUserPrompt { |
| 1999 | prompt, err := db.InsertAIBridgeUserPrompt(genCtx, database.InsertAIBridgeUserPromptParams{ |