style: update method signatures for consistency with async/await syntax

liblaf · Mar 16, 2024 · 09a8936 · 09a8936
1 parent 7b0169d
commit 09a8936
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git 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
@@ -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:
+        ...