"""
This type stub file was generated by pyright.
"""

from dataclasses import dataclass
from typing import Type
from anthropic import AsyncAnthropic
from openai import AsyncOpenAI

class NotGiven:
  ...


NOT_GIVEN = ...
@dataclass(kw_only=True)
class ConvContext:
  model: str
  oai_client: AsyncOpenAI
  ds_client: AsyncOpenAI | None = ...
  anthropic_client: AsyncAnthropic | None = ...
  perplexity_client: AsyncOpenAI | None = ...
  xai_client: AsyncOpenAI | None = ...
  temperature: float | NotGiven = ...
  reasoning_effort: str | NotGiven = ...


conv_context = ...
def get_conv_context() -> ConvContext:
  ...

def set_conv_context(ctx: ConvContext): # -> None:
  ...

async def oneshot[ResponseType](msg: str, response_format: Type[ResponseType] | None = ...) -> ResponseType | str | None:
  ...

