nadeshiko-sdk 0.1.0__py3-none-any.whl

nadeshiko/api/search/fetch_sentence_context.py

@@ -0,0 +1,219 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.fetch_sentence_context_request import FetchSentenceContextRequest
+from ...models.fetch_sentence_context_response import FetchSentenceContextResponse
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: FetchSentenceContextRequest,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/v1/search/media/context",
+    }
+
+    _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Error | FetchSentenceContextResponse | None:
+    if response.status_code == 200:
+        response_200 = FetchSentenceContextResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+
+    if response.status_code == 429:
+        response_429 = Error.from_dict(response.json())
+
+        return response_429
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    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[Error | FetchSentenceContextResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: FetchSentenceContextRequest,
+) -> Response[Error | FetchSentenceContextResponse]:
+    """Fetch context for a sentence
+
+     Retrieves sentences surrounding a specific segment position within an episode.
+    Returns segments both before and after the target position, providing context for understanding how
+    a sentence is used in dialogue.
+
+    **Requirements:**
+    - Required scopes: `READ_MEDIA`
+
+    Args:
+        body (FetchSentenceContextRequest):
+
+    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[Error | FetchSentenceContextResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient | Client,
+    body: FetchSentenceContextRequest,
+) -> Error | FetchSentenceContextResponse | None:
+    """Fetch context for a sentence
+
+     Retrieves sentences surrounding a specific segment position within an episode.
+    Returns segments both before and after the target position, providing context for understanding how
+    a sentence is used in dialogue.
+
+    **Requirements:**
+    - Required scopes: `READ_MEDIA`
+
+    Args:
+        body (FetchSentenceContextRequest):
+
+    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:
+        Error | FetchSentenceContextResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: FetchSentenceContextRequest,
+) -> Response[Error | FetchSentenceContextResponse]:
+    """Fetch context for a sentence
+
+     Retrieves sentences surrounding a specific segment position within an episode.
+    Returns segments both before and after the target position, providing context for understanding how
+    a sentence is used in dialogue.
+
+    **Requirements:**
+    - Required scopes: `READ_MEDIA`
+
+    Args:
+        body (FetchSentenceContextRequest):
+
+    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[Error | FetchSentenceContextResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient | Client,
+    body: FetchSentenceContextRequest,
+) -> Error | FetchSentenceContextResponse | None:
+    """Fetch context for a sentence
+
+     Retrieves sentences surrounding a specific segment position within an episode.
+    Returns segments both before and after the target position, providing context for understanding how
+    a sentence is used in dialogue.
+
+    **Requirements:**
+    - Required scopes: `READ_MEDIA`
+
+    Args:
+        body (FetchSentenceContextRequest):
+
+    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:
+        Error | FetchSentenceContextResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

nadeshiko/api/search/internal/__init__.py

@@ -0,0 +1,6 @@
+from typing import Any
+import httpx
+from .... import errors
+from ....client import AuthenticatedClient, Client
+from ....types import Response
+

nadeshiko/api/search/search.py

@@ -0,0 +1,295 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.search_request import SearchRequest
+from ...models.search_response import SearchResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    body: SearchRequest | Unset = UNSET,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/v1/search/media/sentence",
+    }
+
+    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
+) -> Error | SearchResponse | None:
+    if response.status_code == 200:
+        response_200 = SearchResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+
+    if response.status_code == 429:
+        response_429 = Error.from_dict(response.json())
+
+        return response_429
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    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[Error | SearchResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: SearchRequest | Unset = UNSET,
+) -> Response[Error | SearchResponse]:
+    r"""Search by query
+
+     The primary search endpoint for finding Japanese sentences and their translations across indexed
+    media (anime, J-Drama, audiobooks).
+
+    This endpoint uses Elasticsearch with advanced Japanese text analysis supporting multiple input
+    types (romaji, kanji, kana) and providing intelligent field-based boosting.
+
+    **Query Features**
+    - **Multi-language Support:** Search using Japanese (kanji/kana), Romaji, or English/Spanish
+    - **Boolean Operators:** `AND`, `OR`, `NOT` supported (e.g., `(cat OR dog) AND bird`)
+    - **Phrase Matching:** Use quotes for exact phrases (e.g., `\"good morning\"`), or pass
+    `exact_match: true` to the request body
+    - **Wildcards:** `te*t` format (leading wildcards not supported)
+    - **Smart Field Selection:** Automatically chooses optimal search fields based on input type
+
+    **Input Type Handling**
+    | Input Type | Search Strategy |
+    |------------|-----------------|
+    | **Romaji** (`go`, `taberu`) | Boosts EN/ES translations, reading form (pronunciation), and base
+    form |
+    | **Kanji** (`食べる`, `彼女`) | Searches content and base form (dictionary form). Ignores matches by
+    reading (homophones). |
+    | **Kana** (`たべる`, `かのじょ`) | Standard search across content, base form, and reading form |
+    | **English/Spanish** | Direct translation search |
+
+    **Requirements**
+    - API Key with `READ_MEDIA` permission
+
+    Args:
+        body (SearchRequest | Unset):
+
+    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[Error | SearchResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient | Client,
+    body: SearchRequest | Unset = UNSET,
+) -> Error | SearchResponse | None:
+    r"""Search by query
+
+     The primary search endpoint for finding Japanese sentences and their translations across indexed
+    media (anime, J-Drama, audiobooks).
+
+    This endpoint uses Elasticsearch with advanced Japanese text analysis supporting multiple input
+    types (romaji, kanji, kana) and providing intelligent field-based boosting.
+
+    **Query Features**
+    - **Multi-language Support:** Search using Japanese (kanji/kana), Romaji, or English/Spanish
+    - **Boolean Operators:** `AND`, `OR`, `NOT` supported (e.g., `(cat OR dog) AND bird`)
+    - **Phrase Matching:** Use quotes for exact phrases (e.g., `\"good morning\"`), or pass
+    `exact_match: true` to the request body
+    - **Wildcards:** `te*t` format (leading wildcards not supported)
+    - **Smart Field Selection:** Automatically chooses optimal search fields based on input type
+
+    **Input Type Handling**
+    | Input Type | Search Strategy |
+    |------------|-----------------|
+    | **Romaji** (`go`, `taberu`) | Boosts EN/ES translations, reading form (pronunciation), and base
+    form |
+    | **Kanji** (`食べる`, `彼女`) | Searches content and base form (dictionary form). Ignores matches by
+    reading (homophones). |
+    | **Kana** (`たべる`, `かのじょ`) | Standard search across content, base form, and reading form |
+    | **English/Spanish** | Direct translation search |
+
+    **Requirements**
+    - API Key with `READ_MEDIA` permission
+
+    Args:
+        body (SearchRequest | Unset):
+
+    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:
+        Error | SearchResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: SearchRequest | Unset = UNSET,
+) -> Response[Error | SearchResponse]:
+    r"""Search by query
+
+     The primary search endpoint for finding Japanese sentences and their translations across indexed
+    media (anime, J-Drama, audiobooks).
+
+    This endpoint uses Elasticsearch with advanced Japanese text analysis supporting multiple input
+    types (romaji, kanji, kana) and providing intelligent field-based boosting.
+
+    **Query Features**
+    - **Multi-language Support:** Search using Japanese (kanji/kana), Romaji, or English/Spanish
+    - **Boolean Operators:** `AND`, `OR`, `NOT` supported (e.g., `(cat OR dog) AND bird`)
+    - **Phrase Matching:** Use quotes for exact phrases (e.g., `\"good morning\"`), or pass
+    `exact_match: true` to the request body
+    - **Wildcards:** `te*t` format (leading wildcards not supported)
+    - **Smart Field Selection:** Automatically chooses optimal search fields based on input type
+
+    **Input Type Handling**
+    | Input Type | Search Strategy |
+    |------------|-----------------|
+    | **Romaji** (`go`, `taberu`) | Boosts EN/ES translations, reading form (pronunciation), and base
+    form |
+    | **Kanji** (`食べる`, `彼女`) | Searches content and base form (dictionary form). Ignores matches by
+    reading (homophones). |
+    | **Kana** (`たべる`, `かのじょ`) | Standard search across content, base form, and reading form |
+    | **English/Spanish** | Direct translation search |
+
+    **Requirements**
+    - API Key with `READ_MEDIA` permission
+
+    Args:
+        body (SearchRequest | Unset):
+
+    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[Error | SearchResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient | Client,
+    body: SearchRequest | Unset = UNSET,
+) -> Error | SearchResponse | None:
+    r"""Search by query
+
+     The primary search endpoint for finding Japanese sentences and their translations across indexed
+    media (anime, J-Drama, audiobooks).
+
+    This endpoint uses Elasticsearch with advanced Japanese text analysis supporting multiple input
+    types (romaji, kanji, kana) and providing intelligent field-based boosting.
+
+    **Query Features**
+    - **Multi-language Support:** Search using Japanese (kanji/kana), Romaji, or English/Spanish
+    - **Boolean Operators:** `AND`, `OR`, `NOT` supported (e.g., `(cat OR dog) AND bird`)
+    - **Phrase Matching:** Use quotes for exact phrases (e.g., `\"good morning\"`), or pass
+    `exact_match: true` to the request body
+    - **Wildcards:** `te*t` format (leading wildcards not supported)
+    - **Smart Field Selection:** Automatically chooses optimal search fields based on input type
+
+    **Input Type Handling**
+    | Input Type | Search Strategy |
+    |------------|-----------------|
+    | **Romaji** (`go`, `taberu`) | Boosts EN/ES translations, reading form (pronunciation), and base
+    form |
+    | **Kanji** (`食べる`, `彼女`) | Searches content and base form (dictionary form). Ignores matches by
+    reading (homophones). |
+    | **Kana** (`たべる`, `かのじょ`) | Standard search across content, base form, and reading form |
+    | **English/Spanish** | Direct translation search |
+
+    **Requirements**
+    - API Key with `READ_MEDIA` permission
+
+    Args:
+        body (SearchRequest | Unset):
+
+    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:
+        Error | SearchResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

nadeshiko/api/search/search_health_check.py

@@ -0,0 +1,168 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.search_health_check_response import SearchHealthCheckResponse
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": "/v1/search/health",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Error | SearchHealthCheckResponse | None:
+    if response.status_code == 200:
+        response_200 = SearchHealthCheckResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+
+    if response.status_code == 429:
+        response_429 = Error.from_dict(response.json())
+
+        return response_429
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    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[Error | SearchHealthCheckResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Error | SearchHealthCheckResponse]:
+    """Health check for search
+
+     Tests the search service connectivity and functionality by performing a simple search query.
+
+    **Requirements:**
+    - Required Scopes: `READ_MEDIA`
+
+    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[Error | SearchHealthCheckResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+) -> Error | SearchHealthCheckResponse | None:
+    """Health check for search
+
+     Tests the search service connectivity and functionality by performing a simple search query.
+
+    **Requirements:**
+    - Required Scopes: `READ_MEDIA`
+
+    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:
+        Error | SearchHealthCheckResponse
+    """
+
+    return sync_detailed(
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Error | SearchHealthCheckResponse]:
+    """Health check for search
+
+     Tests the search service connectivity and functionality by performing a simple search query.
+
+    **Requirements:**
+    - Required Scopes: `READ_MEDIA`
+
+    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[Error | SearchHealthCheckResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+) -> Error | SearchHealthCheckResponse | None:
+    """Health check for search
+
+     Tests the search service connectivity and functionality by performing a simple search query.
+
+    **Requirements:**
+    - Required Scopes: `READ_MEDIA`
+
+    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:
+        Error | SearchHealthCheckResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+        )
+    ).parsed