chihlasm/resolutionflow
1
0
Fork 0
Code Issues 23 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: use correct google-genai async API and remove debug endpoint

Browse Source 
The google-genai SDK uses `client.aio.models.generate_content()` for
async calls, not `client.models.generate_content_async()` which doesn't
exist. Also removes the temporary /ai/provider-debug endpoint.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
chihlasm
2026-02-27 00:08:20 -05:00
parent dc68d992a4
commit 957f13b993
3 changed files with 4 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/tests/test_ai_provider.py
View File

@@ -137,7 +137,7 @@ class TestGeminiProvider:
mock_response.usage_metadata = mock_usage
mock_client = MagicMock()
mock_client.models.generate_content_async = AsyncMock(
mock_client.aio.models.generate_content = AsyncMock(
return_value=mock_response
)
@@ -172,7 +172,7 @@ class TestGeminiProvider:
assert input_tokens == 80
assert output_tokens == 40
mock_client.models.generate_content_async.assert_called_once()
mock_client.aio.models.generate_content.assert_called_once()
@pytest.mark.asyncio
async def test_generate_json_handles_none_usage(self):
@@ -185,7 +185,7 @@ class TestGeminiProvider:
mock_response.usage_metadata = mock_usage
mock_client = MagicMock()
mock_client.models.generate_content_async = AsyncMock(
mock_client.aio.models.generate_content = AsyncMock(
return_value=mock_response
)