support telemetry for cua agent #694
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
🦋 Changeset detectedLatest commit: 822db17 The changes in this PR will be included in the next version bump. This PR includes changesets to release 1 package
Not sure what this means? Click here to learn what changesets are. Click here if you're a maintainer who wants to add another changeset to this PR |
seanmcguire12
changed the title
seanmcguire12
changed the title
![greptile-apps[bot]](https://avatars.githubusercontent.com/in/867647?s=60&v=4){kind=small}
There was a problem hiding this comment.
PR Summary
Added telemetry tracking capabilities for Computer Use Assistant (CUA) agents, implementing token usage and inference time metrics across the Stagehand system.
- Added
usageproperty inAgentResultinterface to track input/output tokens and inference time - Extended
StagehandMetricswithagentPromptTokens,agentCompletionTokens, andagentInferenceTimeMs - Implemented token tracking in
AnthropicCUAClientandOpenAICUAClientwith proper error handling - Added
AGENTtoStagehandFunctionNameenum and updatedupdateMetricsmethod inlib/index.ts - Modified
StagehandAgentHandlerto pass Stagehand instance for metrics access
6 file(s) reviewed, 1 comment(s)
Edit PR Review Bot Settings | Greptile
Comment on lines
+449
to
+453
| const startTime = Date.now(); | ||
| // Create the message using the Anthropic Messages API | ||
| // @ts-expect-error - The Anthropic SDK types are stricter than what we need | ||
| const response = await this.client.beta.messages.create(requestParams); | ||
| const endTime = Date.now(); |
There was a problem hiding this comment.
style: Consider wrapping API call in try/catch to ensure inference time is always measured correctly, even if API call fails
Suggested change
| const startTime = Date.now(); | |
| // Create the message using the Anthropic Messages API | |
| // @ts-expect-error - The Anthropic SDK types are stricter than what we need | |
| const response = await this.client.beta.messages.create(requestParams); | |
| const endTime = Date.now(); | |
| const startTime = Date.now(); | |
| let response; | |
| try { | |
| // Create the message using the Anthropic Messages API | |
| // @ts-expect-error - The Anthropic SDK types are stricter than what we need | |
| response = await this.client.beta.messages.create(requestParams); | |
| } finally { | |
| const endTime = Date.now(); |
miguelg719
approved these changes
Apr 24, 2025
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
why
what changed
stagehand.metricsfor cua agenttest plan