-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Added support for AG-UI Multi-modal Messages #3715
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
base: main
Are you sure you want to change the base?
Added support for AG-UI Multi-modal Messages #3715
Conversation
Refactored code for better readability. Bumped version of ag-ui libs.
|
@bojan2501 Can you have a look at the missing test coverage please? |
| return cast('dict[str, Any]', state) | ||
|
|
||
| @classmethod | ||
| def _load_binary_part( |
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.
Please move private helper methods below the public ones that use them like load_messages
| url=part.url, | ||
| media_type=part.mime_type, | ||
| identifier=part.id, | ||
| vendor_metadata={'filename': part.filename}, |
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.
We can't do this this, this will be interpreted as video_metadata by GoogleModel
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.
You mean we should drop this line:
vendor_metadata={'filename': part.filename}
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.
@bojan2501 Yep
…Modal-Messages-AG-UI
…Modal-Messages-AG-UI
Moved private helper methods below public ones. Removed part where vendor metadata is used to populate filename together with url.
Also refactored code, made helper function public. As the type checker was complaining.
|
I think this is it. |
Addresses 3558
Refactored code for better readability.
Bumped version of ag-ui libs.