backend changes extracted from codex/jon/SKY-5016 (#2508)

This commit is contained in:
Shuchang Zheng
2025-05-29 06:15:04 -07:00
committed by GitHub
parent 4ad65b579c
commit 08bf2bc428
13 changed files with 122 additions and 9 deletions

View File

@@ -48,6 +48,7 @@ from skyvern.forge.sdk.schemas.task_generations import GenerateTaskRequest, Task
from skyvern.forge.sdk.schemas.task_v2 import TaskV2Request
from skyvern.forge.sdk.schemas.tasks import (
CreateTaskResponse,
ModelsResponse,
OrderBy,
SortDirection,
Task,
@@ -794,6 +795,24 @@ async def heartbeat() -> Response:
return Response(content="Server is running.", status_code=200, headers={"X-Skyvern-API-Version": __version__})
@legacy_base_router.get(
"/models",
tags=["agent"],
openapi_extra={
"x-fern-sdk-group-name": "agent",
},
)
@legacy_base_router.get("/models/", include_in_schema=False)
async def models() -> ModelsResponse:
"""
Get a list of available models.
"""
mapping = settings.get_model_name_to_llm_key()
models = list(mapping.keys())
return ModelsResponse(models=models)
@legacy_base_router.post(
"/tasks",
tags=["agent"],