Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Anton Pirker <anton.pirker@sentry.io>
  • Loading branch information
colin-sentry and antonpirker committed May 10, 2024
1 parent 12ba48b commit 36ba134
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/integrations/cohere/test_cohere.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def test_streaming_chat(sentry_init, capture_events, send_default_pii, include_p
)
)

with start_transaction(name="openai tx"):
with start_transaction(name="cohere tx"):
responses = list(
client.chat_stream(
model="some-model",
Expand Down Expand Up @@ -184,7 +184,7 @@ def test_embed(sentry_init, capture_events, send_default_pii, include_prompts):
)
)

with start_transaction(name="openai tx"):
with start_transaction(name="cohere tx"):
response = client.embed(texts=["hello"], model="text-embedding-3-large")

assert len(response.embeddings[0]) == 3
Expand Down

0 comments on commit 36ba134

Please sign in to comment.