| | |
|---|
| | | router = APIRouter() |
|---|
| | | |
|---|
| | | |
|---|
| | | # Pydantic 模型用于响应 |
|---|
| | | class AgentResponse(BaseModel): |
|---|
| | | id: str |
|---|
| | | name: str |
|---|
| | | agent_type: AgentType |
|---|
| | | |
|---|
| | | class Config: |
|---|
| | | orm_mode = True |
|---|
| | | |
|---|
| | | |
|---|
| | | @router.get("/list", response_model=ResponseList) |
|---|
| | | async def agent_list(db: Session = Depends(get_db)): |
|---|
| | | agents = db.query(AgentModel).order_by(AgentModel.sort.asc()).all() |
|---|
| | |
|---|
| | | |
|---|
| | | |
|---|
| | | @router.get("/get-chat-id/{agent_id}", response_model=Response) |
|---|
| | | async def agent_list(agent_id: str, db: Session = Depends(get_db)): |
|---|
| | | async def get_chat_id(agent_id: str, db: Session = Depends(get_db)): |
|---|
| | | agent = db.query(AgentModel).filter(AgentModel.id == agent_id).first() |
|---|
| | | if not agent: |
|---|
| | | return Response(code=404, msg="Agent not found") |
|---|
