Skip to content

Commit

Permalink
style: Fix formatting issues in test_openai_integration.py
Browse files Browse the repository at this point in the history
Co-Authored-By: Alex Reibman <[email protected]>
  • Loading branch information
devin-ai-integration[bot] and areibman committed Dec 22, 2024
1 parent bdc813c commit 0e0ef27
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/openai_handlers/test_openai_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,14 +7,15 @@

load_dotenv()


@pytest.mark.integration
def test_openai_integration():
"""Integration test demonstrating all four OpenAI call patterns:
1. Sync (non-streaming)
2. Sync (streaming)
3. Async (non-streaming)
4. Async (streaming)
Verifies that AgentOps correctly tracks all LLM calls via analytics.
"""
# Initialize AgentOps without auto-starting session
Expand Down Expand Up @@ -67,6 +68,6 @@ async def async_stream():

session.end_session("Success")
analytics = session.get_analytics()

# Verify that all LLM calls were tracked
assert analytics["LLM calls"] >= 4, f"Expected at least 4 LLM calls, but got {analytics['LLM calls']}"

0 comments on commit 0e0ef27

Please sign in to comment.