from http import HTTPStatus
from typing import Any, cast
from urllib.parse import quote

import httpx

from ...client import AuthenticatedClient, Client
from ...types import Response, UNSET
from ... import errors

from ...models.agentic_prompt_run_response import AgenticPromptRunResponse
from ...models.http_validation_error import HTTPValidationError
from ...models.prompt_completion_request import PromptCompletionRequest
from ...types import UNSET, Unset
from typing import cast
from uuid import UUID



def _get_kwargs(
    task_id: UUID,
    prompt_name: str,
    prompt_version: str,
    *,
    body: PromptCompletionRequest | Unset = UNSET,

) -> dict[str, Any]:
    headers: dict[str, Any] = {}


    

    

    _kwargs: dict[str, Any] = {
        "method": "post",
        "url": "/api/v1/tasks/{task_id}/prompts/{prompt_name}/versions/{prompt_version}/completions".format(task_id=quote(str(task_id), safe=""),prompt_name=quote(str(prompt_name), safe=""),prompt_version=quote(str(prompt_version), safe=""),),
    }

    
    if not isinstance(body, Unset):
        _kwargs["json"] = body.to_dict()


    headers["Content-Type"] = "application/json"

    _kwargs["headers"] = headers
    return _kwargs



def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> AgenticPromptRunResponse | HTTPValidationError | None:
    if response.status_code == 200:
        response_200 = AgenticPromptRunResponse.from_dict(response.json())



        return response_200

    if response.status_code == 422:
        response_422 = HTTPValidationError.from_dict(response.json())



        return response_422

    if client.raise_on_unexpected_status:
        raise errors.UnexpectedStatus(response.status_code, response.content)
    else:
        return None


def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[AgenticPromptRunResponse | HTTPValidationError]:
    return Response(
        status_code=HTTPStatus(response.status_code),
        content=response.content,
        headers=response.headers,
        parsed=_parse_response(client=client, response=response),
    )


def sync_detailed(
    task_id: UUID,
    prompt_name: str,
    prompt_version: str,
    *,
    client: AuthenticatedClient,
    body: PromptCompletionRequest | Unset = UNSET,

) -> Response[AgenticPromptRunResponse | HTTPValidationError]:
    """ Run/Stream a specific version of an agentic prompt

     Run or stream a specific version of an existing agentic prompt

    Args:
        task_id (UUID):
        prompt_name (str): The name of the prompt to run.
        prompt_version (str): The version of the prompt to run. Can be 'latest', a version number
            (e.g. '1', '2', etc.), an ISO datetime string (e.g. '2025-01-01T00:00:00'), or a tag.
        body (PromptCompletionRequest | Unset): Request schema for running an agentic prompt

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        Response[AgenticPromptRunResponse | HTTPValidationError]
     """


    kwargs = _get_kwargs(
        task_id=task_id,
prompt_name=prompt_name,
prompt_version=prompt_version,
body=body,

    )

    response = client.get_httpx_client().request(
        **kwargs,
    )

    return _build_response(client=client, response=response)

def sync(
    task_id: UUID,
    prompt_name: str,
    prompt_version: str,
    *,
    client: AuthenticatedClient,
    body: PromptCompletionRequest | Unset = UNSET,

) -> AgenticPromptRunResponse | HTTPValidationError | None:
    """ Run/Stream a specific version of an agentic prompt

     Run or stream a specific version of an existing agentic prompt

    Args:
        task_id (UUID):
        prompt_name (str): The name of the prompt to run.
        prompt_version (str): The version of the prompt to run. Can be 'latest', a version number
            (e.g. '1', '2', etc.), an ISO datetime string (e.g. '2025-01-01T00:00:00'), or a tag.
        body (PromptCompletionRequest | Unset): Request schema for running an agentic prompt

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        AgenticPromptRunResponse | HTTPValidationError
     """


    return sync_detailed(
        task_id=task_id,
prompt_name=prompt_name,
prompt_version=prompt_version,
client=client,
body=body,

    ).parsed

async def asyncio_detailed(
    task_id: UUID,
    prompt_name: str,
    prompt_version: str,
    *,
    client: AuthenticatedClient,
    body: PromptCompletionRequest | Unset = UNSET,

) -> Response[AgenticPromptRunResponse | HTTPValidationError]:
    """ Run/Stream a specific version of an agentic prompt

     Run or stream a specific version of an existing agentic prompt

    Args:
        task_id (UUID):
        prompt_name (str): The name of the prompt to run.
        prompt_version (str): The version of the prompt to run. Can be 'latest', a version number
            (e.g. '1', '2', etc.), an ISO datetime string (e.g. '2025-01-01T00:00:00'), or a tag.
        body (PromptCompletionRequest | Unset): Request schema for running an agentic prompt

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        Response[AgenticPromptRunResponse | HTTPValidationError]
     """


    kwargs = _get_kwargs(
        task_id=task_id,
prompt_name=prompt_name,
prompt_version=prompt_version,
body=body,

    )

    response = await client.get_async_httpx_client().request(
        **kwargs
    )

    return _build_response(client=client, response=response)

async def asyncio(
    task_id: UUID,
    prompt_name: str,
    prompt_version: str,
    *,
    client: AuthenticatedClient,
    body: PromptCompletionRequest | Unset = UNSET,

) -> AgenticPromptRunResponse | HTTPValidationError | None:
    """ Run/Stream a specific version of an agentic prompt

     Run or stream a specific version of an existing agentic prompt

    Args:
        task_id (UUID):
        prompt_name (str): The name of the prompt to run.
        prompt_version (str): The version of the prompt to run. Can be 'latest', a version number
            (e.g. '1', '2', etc.), an ISO datetime string (e.g. '2025-01-01T00:00:00'), or a tag.
        body (PromptCompletionRequest | Unset): Request schema for running an agentic prompt

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        AgenticPromptRunResponse | HTTPValidationError
     """


    return (await asyncio_detailed(
        task_id=task_id,
prompt_name=prompt_name,
prompt_version=prompt_version,
client=client,
body=body,

    )).parsed
