Skip to content

Commit bdb3118

Browse files
committed
fix: mark gemini async test with pytest.mark.asyncio
1 parent 6721700 commit bdb3118

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎tests/clients/test_gemini.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Tests for the Gemini client."""
22

33
import os
4-
from unittest.mock import MagicMock, patch, AsyncMock
4+
from unittest.mock import MagicMock, patch
55

66
import pytest
77
from dotenv import load_dotenv
@@ -163,8 +163,9 @@ def test_message_list_completion(self):
163163
not os.environ.get("GEMINI_API_KEY"),
164164
reason="GEMINI_API_KEY not set",
165165
)
166+
@pytest.mark.asyncio
166167
async def test_async_completion(self):
167168
"""Test async completion."""
168169
client = GeminiClient(model_name="gemini-2.5-flash")
169170
result = await client.acompletion("What is 3+3? Reply with just the number.")
170-
assert "6" in result
171+
assert "6" in result

0 commit comments

Comments
 (0)