We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 6b2b896 + 5b2be18 commit 49d61cfCopy full SHA for 49d61cf
1 file changed
inference/usage_tracking/collector.py
@@ -28,6 +28,7 @@
28
from inference.core.env import (
29
API_KEY,
30
DEDICATED_DEPLOYMENT_ID,
31
+ DEVICE_ID,
32
GCP_SERVERLESS,
33
LAMBDA,
34
REDIS_HOST,
@@ -578,6 +579,8 @@ def _extract_usage_params_from_func_kwargs(
578
579
}
580
if DEDICATED_DEPLOYMENT_ID:
581
resource_details["dedicated_deployment_id"] = DEDICATED_DEPLOYMENT_ID
582
+ if DEVICE_ID:
583
+ resource_details["device_id"] = DEVICE_ID
584
resource_id = ""
585
# TODO: add requires_api_key, True if workflow definition comes from platform or model comes from workspace
586
if category == "workflows":
0 commit comments