mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-31 09:50:11 +00:00
fbe94dd7ed
<!-- Thank you for your contribution! Please review https://microsoft.github.io/autogen/docs/Contribute before opening a pull request. --> This PR has 3 main improvements. - Token streaming - Adds support for environment variables in the app settings - Updates AGS to persist Gallery entry in db. ## Adds Token Streaming in AGS. Agentchat now supports streaming of tokens via `ModelClientStreamingChunkEvent `. This PR is to track progress on supporting that in the AutoGen Studio UI. If `model_client_stream` is enabled in an assitant agent, then token will be streamed in UI. ```python streaming_assistant = AssistantAgent( name="assistant", model_client=model_client, system_message="You are a helpful assistant.", model_client_stream=True, # Enable streaming tokens. ) ``` https://github.com/user-attachments/assets/74d43d78-6359-40c3-a78e-c84dcb5e02a1 ## Env Variables Also adds support for env variables in AGS Settings You can set env variables that are loaded just before a team is run. Handy to set variable to be used by tools etc. <img width="1291" alt="image" src="https://github.com/user-attachments/assets/437b9d90-ccee-42f7-be5d-94ab191afd67" /> > Note: the set variables are available to the server process. <!-- Please add a reviewer to the assignee section when you create a PR. If you don't have the access to it, we will shortly find a reviewer and assign them to your PR. --> ## Why are these changes needed? <!-- Please give a short summary of the change and the problem this solves. --> ## Related issue number <!-- For example: "Closes #1234" --> Closes #5627 Closes #5662 Closes #5619 ## Checks - [ ] I've included any doc changes needed for <https://microsoft.github.io/autogen/>. See <https://github.com/microsoft/autogen/blob/main/CONTRIBUTING.md> to build and test documentation locally. - [ ] I've added tests (if relevant) corresponding to the changes introduced in this PR. - [ ] I've made sure all auto checks have passed.
37 lines
683 B
Python
37 lines
683 B
Python
from .db import Gallery, Message, Run, RunStatus, Session, Settings, Team
|
|
from .types import (
|
|
EnvironmentVariable,
|
|
GalleryComponents,
|
|
GalleryConfig,
|
|
GalleryMetadata,
|
|
LLMCallEventMessage,
|
|
MessageConfig,
|
|
MessageMeta,
|
|
Response,
|
|
SettingsConfig,
|
|
SocketMessage,
|
|
TeamResult,
|
|
)
|
|
|
|
__all__ = [
|
|
"Team",
|
|
"Run",
|
|
"RunStatus",
|
|
"Session",
|
|
"Team",
|
|
"Message",
|
|
"MessageConfig",
|
|
"MessageMeta",
|
|
"TeamResult",
|
|
"Response",
|
|
"SocketMessage",
|
|
"LLMCallEventMessage",
|
|
"GalleryConfig",
|
|
"GalleryComponents",
|
|
"GalleryMetadata",
|
|
"SettingsConfig",
|
|
"Settings",
|
|
"EnvironmentVariable",
|
|
"Gallery",
|
|
]
|