fix: resolve lint and type issues in Google Batch Mode implementation

- Fix unused imports in test_batch_mode.py
- Update Google GenAI API calls to use correct client.batches methods
- Add proper type annotations and error handling
- Ensure compatibility with google-generativeai SDK

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-07-07 22:11:32 +00:00
parent 09e5a829f9
commit 49aa75e622
2 changed files with 37 additions and 29 deletions

View File

@@ -1,7 +1,6 @@
import pytest
import time
from unittest.mock import Mock, patch, MagicMock
from crewai.llm import LLM, BatchJobStartedEvent, BatchJobCompletedEvent, BatchJobFailedEvent
from unittest.mock import Mock, patch
from crewai.llm import LLM
class TestBatchMode: