aimlapi-sdk-python 2.8.1b0__py3-none-any.whl

aimlapi/resources/videos.py

@@ -0,0 +1,267 @@
+from __future__ import annotations
+
+from typing import Any, Callable, Dict, Mapping
+
+import httpx
+
+from openai._base_client import make_request_options
+from openai._types import Body, Headers, NotGiven, Query, Timeout, not_given
+from openai.resources.videos import (
+    AsyncVideos as OpenAIAsyncVideos,
+    AsyncVideosWithRawResponse,
+    AsyncVideosWithStreamingResponse,
+    Videos as OpenAIVideos,
+    VideosWithRawResponse,
+    VideosWithStreamingResponse,
+)
+
+from .audio._polling import async_poll_job, poll_job
+
+__all__ = [
+    "Videos",
+    "AsyncVideos",
+    "VideosWithRawResponse",
+    "AsyncVideosWithRawResponse",
+    "VideosWithStreamingResponse",
+    "AsyncVideosWithStreamingResponse",
+]
+
+_VIDEO_GENERATION_CREATE_PATH = "/v2/video/generations"
+_VIDEO_GENERATION_STATUS_PATH = "/v2/video/generations?generation_id={generation_id}"
+_VIDEO_PENDING_STATUSES = frozenset({"queued", "generating"})
+_DEFAULT_POLL_INTERVAL = 10.0
+_DEFAULT_POLL_TIMEOUT = 600.0
+
+
+def _add_if_not_none(payload: Dict[str, Any], key: str, value: object | None) -> None:
+    if value is not None:
+        payload[key] = value
+
+
+def _extract_generation_id(payload: Mapping[str, Any]) -> str | None:
+    generation_id = payload.get("generation_id")
+    if isinstance(generation_id, str):
+        return generation_id
+
+    id_value = payload.get("id")
+    if isinstance(id_value, str):
+        return id_value
+
+    return None
+
+
+def _coerce_mapping(value: object, *, context: str) -> Dict[str, Any]:
+    if isinstance(value, Mapping):
+        return dict(value)
+    raise TypeError(f"Unexpected response payload from {context}")
+
+
+def _video_api_url(base_url: Any, path: str) -> str:
+    normalized_path = path if path.startswith("/") else f"/{path}"
+    normalized_base = base_url.copy_with(raw_path=b"/")
+    base = str(normalized_base).rstrip("/")
+    return f"{base}{normalized_path}"
+
+
+def _extract_video_url(payload: Mapping[str, Any]) -> str:
+    url = _find_video_url(payload)
+    if url:
+        return url
+    raise ValueError("Video generation result did not include a video URL")
+
+
+def _find_video_url(value: object) -> str | None:
+    if isinstance(value, Mapping):
+        url_value = value.get("url")
+        if isinstance(url_value, str) and url_value:
+            return url_value
+
+        for nested in value.values():
+            found = _find_video_url(nested)
+            if found:
+                return found
+
+    elif isinstance(value, list):
+        for item in value:
+            found = _find_video_url(item)
+            if found:
+                return found
+
+    elif isinstance(value, tuple):
+        for item in value:
+            found = _find_video_url(item)
+            if found:
+                return found
+
+    return None
+
+
+def _download_video_bytes(url: str) -> bytes:
+    with httpx.Client(follow_redirects=True, timeout=None) as client:
+        response = client.get(url)
+        response.raise_for_status()
+        return response.content
+
+
+async def _async_download_video_bytes(url: str) -> bytes:
+    async with httpx.AsyncClient(follow_redirects=True, timeout=None) as client:
+        response = await client.get(url)
+        response.raise_for_status()
+        return response.content
+
+
+class Videos(OpenAIVideos):
+    """AIMLAPI specific video helpers."""
+
+    def generate_with_polling(
+        self,
+        *,
+        model: str,
+        prompt: str,
+        aspect_ratio: str | None = None,
+        duration: int | None = None,
+        negative_prompt: str | None = None,
+        seed: int | None = None,
+        enhance_prompt: bool | None = None,
+        poll_interval: float = _DEFAULT_POLL_INTERVAL,
+        poll_timeout: float | None = _DEFAULT_POLL_TIMEOUT,
+        return_bytes: bool = False,
+        status_callback: Callable[[Mapping[str, Any]], None] | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | Timeout | None | NotGiven = not_given,
+    ) -> Dict[str, Any] | bytes:
+        """Create a video generation job and poll until it finishes.
+
+        Args:
+            return_bytes: When ``True``, download and return the final video bytes
+                instead of the JSON payload.
+        """
+
+        payload: Dict[str, Any] = {
+            "model": model,
+            "prompt": prompt,
+        }
+        _add_if_not_none(payload, "aspect_ratio", aspect_ratio)
+        _add_if_not_none(payload, "duration", duration)
+        _add_if_not_none(payload, "negative_prompt", negative_prompt)
+        _add_if_not_none(payload, "seed", seed)
+        _add_if_not_none(payload, "enhance_prompt", enhance_prompt)
+
+        base_url = self._client.base_url
+        create_url = _video_api_url(base_url, _VIDEO_GENERATION_CREATE_PATH)
+        job = self._post(
+            create_url,
+            body=payload,
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+            ),
+            cast_to=object,
+            stream=False,
+        )
+
+        job_payload = _coerce_mapping(job, context="video generation create")
+        if status_callback is not None:
+            status_callback(job_payload)
+
+        generation_id = _extract_generation_id(job_payload)
+        if not generation_id:
+            raise RuntimeError("Video generation response did not include a generation_id")
+
+        status_template = _video_api_url(base_url, _VIDEO_GENERATION_STATUS_PATH)
+        result = poll_job(
+            self,
+            path_template=status_template,
+            generation_id=generation_id,
+            poll_interval=poll_interval,
+            poll_timeout=poll_timeout,
+            request_timeout=timeout,
+            status_callback=status_callback,
+            pending_statuses=_VIDEO_PENDING_STATUSES,
+        )
+
+        final_payload = _coerce_mapping(result, context="video generation status")
+        if not return_bytes:
+            return final_payload
+
+        video_url = _extract_video_url(final_payload)
+        return _download_video_bytes(video_url)
+
+
+class AsyncVideos(OpenAIAsyncVideos):
+    """Async AIMLAPI specific video helpers."""
+
+    async def generate_with_polling(
+        self,
+        *,
+        model: str,
+        prompt: str,
+        aspect_ratio: str | None = None,
+        duration: int | None = None,
+        negative_prompt: str | None = None,
+        seed: int | None = None,
+        enhance_prompt: bool | None = None,
+        poll_interval: float = _DEFAULT_POLL_INTERVAL,
+        poll_timeout: float | None = _DEFAULT_POLL_TIMEOUT,
+        return_bytes: bool = False,
+        status_callback: Callable[[Mapping[str, Any]], None] | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | Timeout | None | NotGiven = not_given,
+    ) -> Dict[str, Any] | bytes:
+        payload: Dict[str, Any] = {
+            "model": model,
+            "prompt": prompt,
+        }
+        _add_if_not_none(payload, "aspect_ratio", aspect_ratio)
+        _add_if_not_none(payload, "duration", duration)
+        _add_if_not_none(payload, "negative_prompt", negative_prompt)
+        _add_if_not_none(payload, "seed", seed)
+        _add_if_not_none(payload, "enhance_prompt", enhance_prompt)
+
+        base_url = self._client.base_url
+        create_url = _video_api_url(base_url, _VIDEO_GENERATION_CREATE_PATH)
+        job = await self._post(
+            create_url,
+            body=payload,
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+            ),
+            cast_to=object,
+            stream=False,
+        )
+
+        job_payload = _coerce_mapping(job, context="video generation create")
+        if status_callback is not None:
+            status_callback(job_payload)
+
+        generation_id = _extract_generation_id(job_payload)
+        if not generation_id:
+            raise RuntimeError("Video generation response did not include a generation_id")
+
+        status_template = _video_api_url(base_url, _VIDEO_GENERATION_STATUS_PATH)
+        result = await async_poll_job(
+            self,
+            path_template=status_template,
+            generation_id=generation_id,
+            poll_interval=poll_interval,
+            poll_timeout=poll_timeout,
+            request_timeout=timeout,
+            status_callback=status_callback,
+            pending_statuses=_VIDEO_PENDING_STATUSES,
+        )
+
+        final_payload = _coerce_mapping(result, context="video generation status")
+        if not return_bytes:
+            return final_payload
+
+        video_url = _extract_video_url(final_payload)
+        return await _async_download_video_bytes(video_url)

aimlapi/resources/webhooks.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.resources.webhooks import *  # noqa: F401, F403

aimlapi/types/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types import *  # noqa: F401, F403

aimlapi/types/audio/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio import *  # noqa: F401, F403

aimlapi/types/audio/speech_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.speech_create_params import *  # noqa: F401, F403

aimlapi/types/audio/speech_model.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.speech_model import *  # noqa: F401, F403

aimlapi/types/audio/transcription.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription import *  # noqa: F401, F403

aimlapi/types/audio/transcription_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_create_params import *  # noqa: F401, F403

aimlapi/types/audio/transcription_create_response.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_create_response import *  # noqa: F401, F403

aimlapi/types/audio/transcription_diarized.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_diarized import *  # noqa: F401, F403

aimlapi/types/audio/transcription_diarized_segment.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_diarized_segment import *  # noqa: F401, F403

aimlapi/types/audio/transcription_include.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_include import *  # noqa: F401, F403

aimlapi/types/audio/transcription_segment.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_segment import *  # noqa: F401, F403

aimlapi/types/audio/transcription_stream_event.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_stream_event import *  # noqa: F401, F403

aimlapi/types/audio/transcription_text_delta_event.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_text_delta_event import *  # noqa: F401, F403

aimlapi/types/audio/transcription_text_done_event.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_text_done_event import *  # noqa: F401, F403

aimlapi/types/audio/transcription_text_segment_event.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_text_segment_event import *  # noqa: F401, F403

aimlapi/types/audio/transcription_verbose.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_verbose import *  # noqa: F401, F403

aimlapi/types/audio/transcription_word.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.transcription_word import *  # noqa: F401, F403

aimlapi/types/audio/translation.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.translation import *  # noqa: F401, F403

aimlapi/types/audio/translation_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.translation_create_params import *  # noqa: F401, F403

aimlapi/types/audio/translation_create_response.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.translation_create_response import *  # noqa: F401, F403

aimlapi/types/audio/translation_verbose.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio.translation_verbose import *  # noqa: F401, F403

aimlapi/types/audio_model.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio_model import *  # noqa: F401, F403

aimlapi/types/audio_response_format.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.audio_response_format import *  # noqa: F401, F403

aimlapi/types/auto_file_chunking_strategy_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.auto_file_chunking_strategy_param import *  # noqa: F401, F403

aimlapi/types/batch.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch import *  # noqa: F401, F403

aimlapi/types/batch_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch_create_params import *  # noqa: F401, F403

aimlapi/types/batch_error.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch_error import *  # noqa: F401, F403

aimlapi/types/batch_list_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch_list_params import *  # noqa: F401, F403

aimlapi/types/batch_request_counts.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch_request_counts import *  # noqa: F401, F403

aimlapi/types/batch_usage.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.batch_usage import *  # noqa: F401, F403

aimlapi/types/beta/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta import *  # noqa: F401, F403

aimlapi/types/beta/assistant.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant import *  # noqa: F401, F403

aimlapi/types/beta/assistant_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_create_params import *  # noqa: F401, F403

aimlapi/types/beta/assistant_deleted.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_deleted import *  # noqa: F401, F403

aimlapi/types/beta/assistant_list_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_list_params import *  # noqa: F401, F403

aimlapi/types/beta/assistant_response_format_option.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_response_format_option import *  # noqa: F401, F403

aimlapi/types/beta/assistant_response_format_option_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_response_format_option_param import *  # noqa: F401, F403

aimlapi/types/beta/assistant_stream_event.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_stream_event import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice_function.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice_function import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice_function_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice_function_param import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice_option.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice_option import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice_option_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice_option_param import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_choice_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_choice_param import *  # noqa: F401, F403

aimlapi/types/beta/assistant_tool_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_tool_param import *  # noqa: F401, F403

aimlapi/types/beta/assistant_update_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.assistant_update_params import *  # noqa: F401, F403

aimlapi/types/beta/chat/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chat import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_automatic_thread_titling.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_automatic_thread_titling import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_chatkit_configuration.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_chatkit_configuration import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_chatkit_configuration_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_chatkit_configuration_param import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_expires_after_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_expires_after_param import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_file_upload.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_file_upload import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_history.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_history import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_rate_limits.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_rate_limits import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_rate_limits_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_rate_limits_param import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_status.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_status import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chat_session_workflow_param.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chat_session_workflow_param import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_attachment.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_attachment import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_response_output_text.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_response_output_text import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_thread.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_thread import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_thread_assistant_message_item.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_thread_assistant_message_item import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_thread_item_list.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_thread_item_list import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_thread_user_message_item.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_thread_user_message_item import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/chatkit_widget_item.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.chatkit_widget_item import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/session_create_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.session_create_params import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/thread_delete_response.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.thread_delete_response import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/thread_list_items_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.thread_list_items_params import *  # noqa: F401, F403

aimlapi/types/beta/chatkit/thread_list_params.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from openai.types.beta.chatkit.thread_list_params import *  # noqa: F401, F403