Skip to content

feat: set assignments in BillingEventContext

Alejandro Rodríguezrequested to merge
billing-context-assignments into main

What does this merge request do and why?

Follow-up to !3642 (merged). See !3642 (comment 2867214908)

How to set up and validate locally

  • Enable snowplow micro.
  • Set export GITLAB_SIMULATE_SAAS=1 in your gdk root's env.runit
  • Restart your GDK
  • Enable billing events with below env variable in .env in AI Gateway
# Billing Events
AIGW_BILLING_EVENT__ENABLED=true
AIGW_BILLING_EVENT__ENDPOINT=http://localhost:9091 # snowplow micro url
AIGW_BILLING_EVENT__NAMESPACE=gl
AIGW_BILLING_EVENT__BATCH_SIZE=1
AIGW_BILLING_EVENT__THREAD_COUNT=1
  • Make an agentic chat request, then check on http://localhost:9091/micro/good for a duo_agent_platform_workflow_completion event and make sure the field assignments exists and has a non-empty value

Merge request checklist

  • Tests added for new functionality. If not, please raise an issue to follow up.
  • Documentation added/updated, if needed.
  • If this change requires executor implementation: verified that issues/MRs exist for both Go executor and Node executor or confirmed that changes are backward-compatible and don't break existing executor functionality.

Merge request reports