Skip to content

Conversation

@pyek-bot
Copy link
Owner

Example of implementing using TelemetryAwarePlugin to capture metrics

Comment on lines 1199 to 1332
@Override
public String getJobType() {
return ML_COMMONS_JOBS_TYPE;
}

@Override
public String getJobIndex() {
return CommonValue.ML_JOBS_INDEX;
}

@Override
public ScheduledJobRunner getJobRunner() {
return MLJobRunner.getInstance();
}

@Override
public ScheduledJobParser getJobParser() {
return (parser, id, jobDocVersion) -> MLJobParameter.parse(parser);
}
Copy link
Owner Author

Choose a reason for hiding this comment

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

ignore: these are part of job scheduler changes and not related to telemetry

} else {
handleAsyncMLTaskComplete(mlTask);
mlModelManager.trackPredictDuration(modelId, startTime);
MLOperationalMetricsCounter.getInstance().incrementCounter(OperationalMetric.MODEL_PREDICT_COUNT, Tags.create().addTag("MODEL_ID", modelId));
Copy link
Owner Author

Choose a reason for hiding this comment

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

Example of usage

@pyek-bot pyek-bot force-pushed the otel_metrics_integration branch from 26a2ea9 to 86b569b Compare May 13, 2025 20:12
@pyek-bot pyek-bot force-pushed the otel_metrics_integration branch 2 times, most recently from 66636c9 to 12f0ad6 Compare May 23, 2025 18:33
@pyek-bot pyek-bot force-pushed the otel_metrics_integration branch from 8b16cd8 to d2969c1 Compare June 10, 2025 19:45
pyek-bot and others added 14 commits June 10, 2025 12:52
Signed-off-by: Pavan Yekbote <[email protected]>
: comment code for build

Signed-off-by: Pavan Yekbote <[email protected]>
Signed-off-by: Pavan Yekbote <[email protected]>
Signed-off-by: Pavan Yekbote <[email protected]>
Signed-off-by: Pavan Yekbote <[email protected]>
* update agent draft

Signed-off-by: Jiaping Zeng <[email protected]>

* Moved tenant check & refactored response

Signed-off-by: Jiaping Zeng <[email protected]>

* added tests

Signed-off-by: Jiaping Zeng <[email protected]>

* added agent ID in success logging

Signed-off-by: Jiaping Zeng <[email protected]>

* modified update agent request input

Signed-off-by: Jiaping Zeng <[email protected]>

* updated unit tests

Signed-off-by: Jiaping Zeng <[email protected]>

* added input validation for update agent & replaced assert with exception

Signed-off-by: Jiaping Zeng <[email protected]>

* updated and added new tests

Signed-off-by: Jiaping Zeng <[email protected]>

* updated test name prefixes

Signed-off-by: Jiaping Zeng <[email protected]>

---------

Signed-off-by: Jiaping Zeng <[email protected]>
Co-authored-by: Dhrubo Saha <[email protected]>
@pyek-bot pyek-bot force-pushed the otel_metrics_integration branch from d2969c1 to 78056ff Compare June 10, 2025 20:01
pyek-bot and others added 9 commits June 10, 2025 15:24
…, control taskPollingJob, comment out predict metrics capture

Signed-off-by: Pavan Yekbote <[email protected]>
…3888)

* function calling in agent draft

Signed-off-by: Jiaping Zeng <[email protected]>

* added function calling in runTool

Signed-off-by: Jiaping Zeng <[email protected]>

* UT fix for function calling

Signed-off-by: Jiaping Zeng <[email protected]>

* fixed openai llm interface

Signed-off-by: Jiaping Zeng <[email protected]>

* fixed test + reuse existing static where possible

Signed-off-by: Jiaping Zeng <[email protected]>

---------

Signed-off-by: Jiaping Zeng <[email protected]>
Co-authored-by: Dhrubo Saha <[email protected]>
Co-authored-by: Xun Zhang <[email protected]>
Signed-off-by: Pavan Yekbote <[email protected]>
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.

8 participants