Skip to content

feat: populate feature_enablement_type for Workflow events

Alejandro Rodríguezrequested to merge
workflow-enablement-type into main

What does this merge request do and why?

Closes https://gitlab.com/gitlab-org/architecture/usage-billing/design-doc/-/issues/20

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
  1. Make an agentic chat request, then check on http://localhost:9091/micro/good for a duo_agent_platform_workflow_completion and make sure the field feature_enablement_type 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