mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-09 01:51:16 +00:00
2cc8c73d3b
<!-- Thank you for your contribution! Please review https://microsoft.github.io/autogen/docs/Contribute before opening a pull request. --> <!-- 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? Fix termination UI in AGS, ensure it can be edited correctly <img width="1269" alt="image" src="https://github.com/user-attachments/assets/eaa7a92f-a1ea-4ab4-a679-2894ac441311" /> <img width="1273" alt="image" src="https://github.com/user-attachments/assets/6db81068-932f-4d4e-9512-279770c02bf2" /> <img width="1270" alt="image" src="https://github.com/user-attachments/assets/5ca9df7d-b968-46c9-9d62-becd78809273" /> <!-- Please give a short summary of the change and the problem this solves. --> ## Related issue number <!-- For example: "Closes #1234" --> Closes #5872 ## 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.
29 lines
824 B
Python
29 lines
824 B
Python
import os
|
|
from typing import cast
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
|
|
from ..datamodel import Response
|
|
from ..teammanager import TeamManager
|
|
|
|
app = FastAPI()
|
|
team_manager = TeamManager()
|
|
|
|
|
|
@app.get("/predict/{task}")
|
|
async def predict(task: str):
|
|
response = Response(message="Task successfully completed", status=True, data=None)
|
|
try:
|
|
team_file_path = os.environ.get("AUTOGENSTUDIO_TEAM_FILE")
|
|
|
|
# Check if team_file_path is set
|
|
if team_file_path is None:
|
|
raise ValueError("AUTOGENSTUDIO_TEAM_FILE environment variable is not set")
|
|
|
|
result_message = await team_manager.run(task=task, team_config=team_file_path)
|
|
response.data = result_message
|
|
except Exception as e:
|
|
response.message = str(e)
|
|
response.status = False
|
|
return response
|