chore(examples): lint

groq · Apr 11, 2024 · 0077c52 · 0077c52
1 parent 6a03509
commit 0077c52
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/examples/chat_completion_async_streaming.py b/examples/chat_completion_async_streaming.py
@@ -46,7 +46,11 @@ async def main() -> None:
     # Print the incremental deltas returned by the LLM.
     async for chunk in stream:
         print(chunk.choices[0].delta.content, end="")
+
+        # Usage information is available on the final chunk
         if chunk.choices[0].finish_reason:
+            assert chunk.x_groq is not None
+            assert chunk.x_groq.usage is not None
             print(f"\n\nUsage stats: {chunk.x_groq.usage}")
 
 

diff --git a/examples/chat_completion_streaming.py b/examples/chat_completion_streaming.py
@@ -44,5 +44,9 @@
 # Print the incremental deltas returned by the LLM.
 for chunk in stream:
     print(chunk.choices[0].delta.content, end="")
+
     if chunk.choices[0].finish_reason:
+        # Usage information is available on the final chunk
+        assert chunk.x_groq is not None
+        assert chunk.x_groq.usage is not None
         print(f"\n\nUsage stats: {chunk.x_groq.usage}")