fix: validation

Author: ysolanky

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

@@ -122,7 +122,7 @@ def test_generate_image_success(mock_gemini_tools, mock_agent, mock_successful_r
         result = mock_gemini_tools.generate_image(mock_agent, prompt)
 
         expected_media_id = "12345678-1234-5678-1234-567812345678"
-        assert result == f"Image generated successfully with ID: {expected_media_id}"
+        assert result == f"Image generated successfully"
         mock_gemini_tools.client.models.generate_images.assert_called_once_with(model=image_model, prompt=prompt)
 
         # 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
     with patch("agno.tools.models.gemini.uuid4", return_value=UUID("87654321-4321-8765-4321-876543214321")):
         result = mock_gemini_tools.generate_video(mock_agent, prompt)
         expected_id = "87654321-4321-8765-4321-876543214321"
-        assert result == f"Video generated successfully with ID: {expected_id}"
+        assert result == f"Video generated successfully" 
         assert mock_gemini_tools.client.models.generate_videos.called
         call_args = mock_gemini_tools.client.models.generate_videos.call_args
         assert call_args.kwargs["model"] == mock_gemini_tools.video_model