mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-15 23:31:29 +00:00
managed agents: add GET /v1/managed_agents/agents
Adds list_agents — returns every agent ordered by created_at desc. Needed for the agent-platform UI agents-list page.
This commit is contained in:
parent
d6b97d3b7f
commit
662b487bb4
1 changed files with 17 additions and 0 deletions
|
|
@ -1,5 +1,6 @@
|
|||
import asyncio
|
||||
import json
|
||||
from typing import List
|
||||
|
||||
from fastapi import Depends, HTTPException
|
||||
|
||||
|
|
@ -20,6 +21,7 @@ def _agent_row_to_out(row) -> AgentOut:
|
|||
model=row.model,
|
||||
template_id=row.template_id,
|
||||
branch=row.branch,
|
||||
created_at=getattr(row, "created_at", None),
|
||||
)
|
||||
|
||||
|
||||
|
|
@ -70,6 +72,21 @@ async def create_agent(
|
|||
return _agent_row_to_out(row)
|
||||
|
||||
|
||||
@router.get("/agents", response_model=List[AgentOut])
|
||||
async def list_agents(
|
||||
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
|
||||
) -> List[AgentOut]:
|
||||
from litellm.proxy.proxy_server import prisma_client
|
||||
|
||||
if prisma_client is None:
|
||||
raise HTTPException(status_code=500, detail="prisma client not available")
|
||||
|
||||
rows = await prisma_client.db.litellm_managedagenttable.find_many(
|
||||
order={"created_at": "desc"}
|
||||
)
|
||||
return [_agent_row_to_out(row) for row in rows]
|
||||
|
||||
|
||||
@router.get("/agents/{agent_id}", response_model=AgentOut)
|
||||
async def get_agent(
|
||||
agent_id: str,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue