Function update_testset

api/oss/src/routers/testset_router.py:293–341

Update a testset with given id, update the testset in Postgres. Args: testset_id (str): id of the test set to be updated. csvdata (NewTestset): New data to replace the old testset. Returns: str: The id of the test set updated.

(
    testset_id: str,
    csvdata: NewTestset,
    request: Request,
)

@router.put("/{testset_id}/", operation_id="update_testset")
async def update_testset(
    testset_id: str,
    csvdata: NewTestset,
    request: Request,
):
    """
    Update a testset with given id, update the testset in Postgres.

    Args:
    testset_id (str): id of the test set to be updated.
    csvdata (NewTestset): New data to replace the old testset.

    Returns:
    str: The id of the test set updated.
    """

    testset = await db_manager.fetch_testset_by_id(testset_id=testset_id)
    if testset is None:
        raise HTTPException(status_code=404, detail="testset not found")

    if is_ee():
        has_permission = await check_action_access(
            user_uid=request.state.user_id,
            project_id=str(testset.project_id),
            permission=Permission.EDIT_TESTSET,
        )
        if not has_permission:
            error_msg = f"You do not have permission to perform this action. Please contact your organization admin."
            log.error(error_msg)
            return JSONResponse(
                {"detail": error_msg},
                status_code=403,
            )

    _validate_testset_limits(csvdata.csvdata)

    testset_update = {
        "name": csvdata.name,
        "csvdata": csvdata.csvdata,
        "updated_at": datetime.now(timezone.utc),
    }
    await db_manager.update_testset(
        testset_id=str(testset.id), values_to_update=testset_update
    )
    return {
        "status": "success",
        "message": "testset updated successfully",
        "_id": testset_id,
    }


@router.get("/", operation_id="get_testsets")