ci(pre-commit): auto fixes from pre-commit hooks

liblaf · Apr 1, 2024 · 03555bf · 03555bf
1 parent 9ece890
commit 03555bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/src/aic/prompt/base.py b/src/aic/prompt/base.py
@@ -27,8 +27,7 @@ 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
@@ -70,8 +70,7 @@ 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
@@ -85,8 +84,7 @@ 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
@@ -120,9 +118,7 @@ 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: ...