chore(examples): lint

groq · Mar 8, 2024 · fdcb94e · fdcb94e
1 parent 7dec9b7
commit fdcb94e
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}")