Skip to content

Commit 1dea1b5

Browse files
ysolankyLockeysama
authored andcommitted
fix: validation
1 parent 72fff55 commit 1dea1b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

libs/agno/tests/unit/tools/models/test_gemini.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def test_generate_image_success(mock_gemini_tools, mock_agent, mock_successful_r
122122
result = mock_gemini_tools.generate_image(mock_agent, prompt)
123123

124124
expected_media_id = "12345678-1234-5678-1234-567812345678"
125-
assert result == f"Image generated successfully"
125+
assert result == "Image generated successfully"
126126
mock_gemini_tools.client.models.generate_images.assert_called_once_with(model=image_model, prompt=prompt)
127127

128128
# Verify agent.add_image was called with the correct ImageArtifact
@@ -208,7 +208,7 @@ def test_generate_video_success(mock_gemini_tools, mock_agent, mock_video_operat
208208
with patch("agno.tools.models.gemini.uuid4", return_value=UUID("87654321-4321-8765-4321-876543214321")):
209209
result = mock_gemini_tools.generate_video(mock_agent, prompt)
210210
expected_id = "87654321-4321-8765-4321-876543214321"
211-
assert result == f"Video generated successfully"
211+
assert result == "Video generated successfully"
212212
assert mock_gemini_tools.client.models.generate_videos.called
213213
call_args = mock_gemini_tools.client.models.generate_videos.call_args
214214
assert call_args.kwargs["model"] == mock_gemini_tools.video_model

0 commit comments

Comments
 (0)