"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from duckyai.types import BaseModel
from typing import Optional
from typing_extensions import NotRequired, TypedDict


class GenerateResponseRequestQueryTranslationStrategyTypedDict(TypedDict):
    r"""Configuration for query translation"""

    multiquery: NotRequired[bool]
    r"""Whether to break the query into multiple subqueries"""
    semantic_expansion: NotRequired[bool]
    r"""Expand the query semantically to retrieve related documents"""
    decomposition: NotRequired[bool]
    r"""Break down a complex query into subqueries"""
    step_back: NotRequired[bool]
    r"""Re-evaluate the query when results seem insufficient"""


class GenerateResponseRequestQueryTranslationStrategy(BaseModel):
    r"""Configuration for query translation"""

    multiquery: Optional[bool] = None
    r"""Whether to break the query into multiple subqueries"""

    semantic_expansion: Optional[bool] = None
    r"""Expand the query semantically to retrieve related documents"""

    decomposition: Optional[bool] = None
    r"""Break down a complex query into subqueries"""

    step_back: Optional[bool] = None
    r"""Re-evaluate the query when results seem insufficient"""
