mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-23 00:42:54 +00:00
32d2a18bf1
<!-- 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? https://github.com/user-attachments/assets/e160f16d-f42d-49e2-a6c6-687e4e6786f4 Enable file upload/paste as a task in AGS. Enables tasks like - Can you research and fact check the ideas in this screenshot? - Summarize this file Only text and images supported for now Underneath, it constructs TextMessage and Multimodal messages as the task. <!-- Please give a short summary of the change and the problem this solves. --> ## Related issue number <!-- For example: "Closes #1234" --> Closes #5773 ## 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. --------- Co-authored-by: Jack Gerrits <jackgerrits@users.noreply.github.com>
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# api/routes/teams.py
|
|
from typing import Dict
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
|
|
from ...datamodel import Team
|
|
from ...gallery.builder import create_default_gallery
|
|
from ..deps import get_db
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/")
|
|
async def list_teams(user_id: str, db=Depends(get_db)) -> Dict:
|
|
"""List all teams for a user"""
|
|
response = db.get(Team, filters={"user_id": user_id})
|
|
|
|
if not response.data or len(response.data) == 0:
|
|
default_gallery = create_default_gallery()
|
|
default_team = Team(user_id=user_id, component=default_gallery.components.teams[0].model_dump())
|
|
|
|
db.upsert(default_team)
|
|
response = db.get(Team, filters={"user_id": user_id})
|
|
|
|
return {"status": True, "data": response.data}
|
|
|
|
|
|
@router.get("/{team_id}")
|
|
async def get_team(team_id: int, user_id: str, db=Depends(get_db)) -> Dict:
|
|
"""Get a specific team"""
|
|
response = db.get(Team, filters={"id": team_id, "user_id": user_id})
|
|
if not response.status or not response.data:
|
|
raise HTTPException(status_code=404, detail="Team not found")
|
|
return {"status": True, "data": response.data[0]}
|
|
|
|
|
|
@router.post("/")
|
|
async def create_team(team: Team, db=Depends(get_db)) -> Dict:
|
|
"""Create a new team"""
|
|
response = db.upsert(team)
|
|
if not response.status:
|
|
raise HTTPException(status_code=400, detail=response.message)
|
|
return {"status": True, "data": response.data}
|
|
|
|
|
|
@router.delete("/{team_id}")
|
|
async def delete_team(team_id: int, user_id: str, db=Depends(get_db)) -> Dict:
|
|
"""Delete a team"""
|
|
db.delete(filters={"id": team_id, "user_id": user_id}, model_class=Team)
|
|
return {"status": True, "message": "Team deleted successfully"}
|