diff --git a/src/langbot/pkg/pipeline/process/handlers/chat.py b/src/langbot/pkg/pipeline/process/handlers/chat.py index 203a3612..87f8d8ce 100644 --- a/src/langbot/pkg/pipeline/process/handlers/chat.py +++ b/src/langbot/pkg/pipeline/process/handlers/chat.py @@ -208,6 +208,7 @@ class ChatMessageHandler(handler.MessageHandler): 'model_name': model_name, 'version': constants.semantic_version, 'instance_id': constants.instance_id, + 'edition': constants.edition, 'pipeline_plugins': pipeline_plugins, 'error': locals().get('error_info', None), 'timestamp': datetime.utcnow().isoformat(), diff --git a/src/langbot/pkg/telemetry/telemetry.py b/src/langbot/pkg/telemetry/telemetry.py index d0849a3d..19cc8bb4 100644 --- a/src/langbot/pkg/telemetry/telemetry.py +++ b/src/langbot/pkg/telemetry/telemetry.py @@ -60,7 +60,7 @@ class TelemetryManager: except Exception: sanitized['query_id'] = str(sanitized.get('query_id', '')) - for sfield in ('adapter', 'runner', 'runner_category', 'model_name', 'version', 'error', 'timestamp'): + for sfield in ('adapter', 'runner', 'runner_category', 'model_name', 'version', 'edition', 'error', 'timestamp'): v = sanitized.get(sfield) sanitized[sfield] = '' if v is None else str(v)