diff --git a/src/pollux/__init__.py b/src/pollux/__init__.py index e07db9e..ee7d517 100644 --- a/src/pollux/__init__.py +++ b/src/pollux/__init__.py @@ -35,6 +35,8 @@ from pollux.source import Source if TYPE_CHECKING: + from collections.abc import Sequence + from pollux.providers.base import Provider try: @@ -79,9 +81,9 @@ async def run( async def run_many( - prompts: str | list[str | None] | tuple[str | None, ...] | None = None, + prompts: str | Sequence[str | None] | None = None, *, - sources: tuple[Source, ...] | list[Source] = (), + sources: Sequence[Source] = (), config: Config, options: Options | None = None, ) -> ResultEnvelope: @@ -160,7 +162,7 @@ async def continue_tool( async def create_cache( - sources: tuple[Source, ...] | list[Source], + sources: Sequence[Source], *, config: Config, system_instruction: str | None = None, diff --git a/src/pollux/cache.py b/src/pollux/cache.py index fe06eed..63bf47a 100644 --- a/src/pollux/cache.py +++ b/src/pollux/cache.py @@ -15,6 +15,8 @@ from pollux.retry import RetryPolicy, retry_async, should_retry_side_effect if TYPE_CHECKING: + from collections.abc import Sequence + from pollux.config import Config from pollux.providers.base import Provider from pollux.source import Source @@ -209,7 +211,7 @@ async def _upload(_fp: str = fp, _mt: str = mt) -> Any: async def create_cache_impl( - sources: tuple[Source, ...] | list[Source], + sources: Sequence[Source], *, provider: Provider, config: Config, diff --git a/src/pollux/request.py b/src/pollux/request.py index dac30c4..baf8872 100644 --- a/src/pollux/request.py +++ b/src/pollux/request.py @@ -10,6 +10,8 @@ from pollux.source import Source if TYPE_CHECKING: + from collections.abc import Sequence + from pollux.config import Config @@ -24,8 +26,8 @@ class Request: def normalize_request( - prompts: tuple[str | None, ...] | list[str | None] | str | None, - sources: tuple[Source, ...] | list[Source], + prompts: str | Sequence[str | None] | None, + sources: Sequence[Source], config: Config, *, options: Options | None = None,