Update completions_streaming.py

- lint fix; debug exception removal
writer · Jun 28, 2024 · 521c12c · 521c12c
1 parent cb406ce
commit 521c12c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/completions_streaming.py b/examples/completions_streaming.py
@@ -8,8 +8,6 @@
 # $ rye sync --all-features
 # $ WRITERAI_API_KEY="<your api key>" rye run python examples/completions_streaming.py
 
-import random
-
 from writerai import AsyncClient, BadRequestError
 
 
@@ -36,7 +34,7 @@ async def main() -> None:
             async for response in stream_res:
                 print(f"< {response.value}")
 
-    except Exception as e:
+    except BadRequestError as e:
         print(f"Error: {e.body}")