| 1996 | } |
| 1997 | |
| 1998 | func AIBridgeUserPrompt(t testing.TB, db database.Store, seed database.InsertAIBridgeUserPromptParams) database.AIBridgeUserPrompt { |
| 1999 | prompt, err := db.InsertAIBridgeUserPrompt(genCtx, database.InsertAIBridgeUserPromptParams{ |
| 2000 | ID: takeFirst(seed.ID, uuid.New()), |
| 2001 | InterceptionID: takeFirst(seed.InterceptionID, uuid.New()), |
| 2002 | ProviderResponseID: takeFirst(seed.ProviderResponseID, "provider_response_id"), |
| 2003 | Prompt: takeFirst(seed.Prompt, "prompt"), |
| 2004 | Metadata: takeFirstSlice(seed.Metadata, json.RawMessage("{}")), |
| 2005 | CreatedAt: takeFirst(seed.CreatedAt, dbtime.Now()), |
| 2006 | }) |
| 2007 | require.NoError(t, err, "insert aibridge user prompt") |
| 2008 | return prompt |
| 2009 | } |
| 2010 | |
| 2011 | func AIBridgeToolUsage(t testing.TB, db database.Store, seed database.InsertAIBridgeToolUsageParams) database.AIBridgeToolUsage { |
| 2012 | serverURL := sql.NullString{} |