Skip to content

feat: Add granular telemetry signals decorator params and error classification#5963

Open
rsareddy0329 wants to merge 8 commits into
aws:masterfrom
rsareddy0329:master-granular-telemetry
Open

feat: Add granular telemetry signals decorator params and error classification#5963
rsareddy0329 wants to merge 8 commits into
aws:masterfrom
rsareddy0329:master-granular-telemetry

Conversation

@rsareddy0329

Copy link
Copy Markdown
Contributor

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

…r params and error classification

X-AI-Prompt: Implement granular telemetry signals for training, evaluation, and deployment
X-AI-Tool: Kiro IDE
@rsareddy0329 rsareddy0329 marked this pull request as ready for review June 22, 2026 02:53
@rsareddy0329 rsareddy0329 changed the title feat: Add granular telemetry signals with emit/emit_presence decorator feat: Add granular telemetry signals decorator params and error classification Jun 22, 2026
Comment on lines +80 to +81
if "validation" in error_type.lower() or "invalid" in error_msg or "must be" in error_msg:
return "validation_error"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

"must be" seems vague could be anything .
Is there like a more exhaustive map of this ? Has this problem been solved elsewhere already in another library or so ?

Comment on lines +82 to +83
if "accessdenied" in error_msg or "not authorized" in error_msg or "forbidden" in error_msg:
return "auth_error"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How does spacing/cases sensitivity work in these cases ?

If I get "access denied" the classification would be incorrect right

return "unknown"


def _attr_to_key(attr: str) -> str:

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we use existing snake to camel conversions ?

func_name="BedrockModelBuilder.deploy",
telemetry_params=[
("model_package", TelemetryParamType.ATTR_EXISTS),
("imported_model_kms_key_id", TelemetryParamType.KWARG_EXISTS),

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we store this ?

feature=Feature.MODEL_CUSTOMIZATION,
func_name="BedrockModelBuilder.deploy",
telemetry_params=[
("model_package", TelemetryParamType.ATTR_EXISTS),

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Model Package is fine .
How is Model data being stored here ?

Comment on lines +389 to +390
@_telemetry_emitter(
feature=Feature.MODEL_CUSTOMIZATION,

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think a model dimension can be added to parameters as well

@nargokul nargokul left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we add for pipelines as well ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants