diff --git a/src/aic/prompt/base.py b/src/aic/prompt/base.py index 093d3b49..ffc17c42 100644 --- a/src/aic/prompt/base.py +++ b/src/aic/prompt/base.py @@ -27,7 +27,8 @@ async def init( pathspec = [*pathspec, ":!.cspell.json", ":!*-lock.*", ":!*.lock"] self._diff = await _git.diff(pathspec) - async def ask(self) -> None: ... + async def ask(self) -> None: + ... @property @abc.abstractmethod diff --git a/src/aic/provider/base.py b/src/aic/provider/base.py index ce2f1494..02564609 100644 --- a/src/aic/provider/base.py +++ b/src/aic/provider/base.py @@ -70,7 +70,8 @@ async def init(self, models: Sequence[str] | None = None) -> None: if models is not None: self.models = models - async def list_models(self) -> AsyncGenerator[Model, None]: ... + async def list_models(self) -> AsyncGenerator[Model, None]: + ... async def quota(self) -> tuple[float, str] | None: return None @@ -84,7 +85,8 @@ def pricing(self, model: str) -> Pricing | None: @abc.abstractmethod async def generate( self, prompt: _prompt.Prompt, *, truncate: bool = True - ) -> Response: ... + ) -> Response: + ... async def generate_stream( self, prompt: _prompt.Prompt, *, truncate: bool = True @@ -118,7 +120,9 @@ def select_model(self, prompt: _prompt.Prompt) -> str: return self.models[0] @abc.abstractmethod - def count_tokens(self, model: str, prompt: _prompt.Prompt) -> int: ... + def count_tokens(self, model: str, prompt: _prompt.Prompt) -> int: + ... @abc.abstractmethod - def truncate(self, model: str, prompt: _prompt.Prompt) -> _prompt.Prompt: ... + def truncate(self, model: str, prompt: _prompt.Prompt) -> _prompt.Prompt: + ...