"""@private"""

import typing
from datetime import datetime, timezone

from langfuse.model import PromptClient


def _get_timestamp() -> datetime:
    return datetime.now(timezone.utc)


def _create_prompt_context(
    prompt: typing.Optional[PromptClient] = None,
) -> typing.Dict[str, typing.Optional[typing.Union[str, int]]]:
    if prompt is not None and not prompt.is_fallback:
        return {"prompt_version": prompt.version, "prompt_name": prompt.name}

    return {"prompt_version": None, "prompt_name": None}