"""UpstreamProvider：组合 Auth + Client，对外暴露 duck-type stream() 供电 deps/adapters 用。

duck-type 接口：``async stream(prompt, model_id=None) -> AsyncIterator[IREvent]``。
app.deps._RetryingClient 与 app.orchestrator 均按此接口使用（不依赖具体上游）。
"""
from __future__ import annotations

from collections.abc import AsyncIterator
from typing import Any

from app.events import IREvent


class UpstreamProvider:
    def __init__(self, account: Any, settings: Any, http_client: Any, auth: Any, client: Any) -> None:
        self.account = account
        self.settings = settings
        self.http_client = http_client
        self._auth = auth
        self._client = client

    async def stream(self, prompt: str, model_id: str | None = None, **kw: Any) -> AsyncIterator[IREvent]:
        async for ir in self._client.stream(prompt, model_id=model_id, **kw):
            yield ir

    async def upload_image(self, data: bytes, mime: str, filename: str = "") -> str:
        return await self._client.upload_image(data, mime, filename)

    async def upload_file(self, data: bytes, mime: str, filename: str = "") -> str:
        return await self._client.upload_file(data, mime, filename)