-
Notifications
You must be signed in to change notification settings - Fork 504
fix: OpenAI Chat Completions tool calls #1153
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Codecov ReportAttention: Patch coverage is
📢 Thoughts on this report? Let us know! |
Dwij1704
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR updates the OpenAI chat instrumentation to create dedicated tool-call spans instead of simply recording tool-call attributes.
- Introduces a
_create_tool_spanhelper to start child spans for each tool call - Updates
handle_chat_attributesto invoke the new span helper for non-streaming calls - Modifies the streaming wrapper to import and use
_create_tool_spaninstead of setting tool-call attributes
Reviewed Changes
Copilot reviewed 2 out of 2 changed files in this pull request and generated 2 comments.
| File | Description |
|---|---|
| agentops/instrumentation/providers/openai/wrappers/chat.py | Add _create_tool_span, adjust handle_chat_attributes to call it |
| agentops/instrumentation/providers/openai/stream_wrapper.py | Import and use _create_tool_span in stream finalization |
Comments suppressed due to low confidence (1)
agentops/instrumentation/providers/openai/stream_wrapper.py:18
- [nitpick] Importing a private helper function (
_create_tool_span) across modules couples implementation details. Consider promoting it to a public utility or moving it to a shared helper module.
from agentops.instrumentation.providers.openai.wrappers.chat import handle_chat_attributes, _create_tool_span
19a989f to
f564029
Compare
📥 Pull Request
📘 Description
Fix for tool calls in OpenAI Chat Completions API
🧪 Testing
Trust me bro arc 😉