Worked on the apple/axlearn repository to migrate the Gemini client from VertexAI to the Google GenAI SDK, introducing enhanced content generation capabilities with configurable parameters to improve output quality and relevance. Leveraged Python for API integration and asynchronous programming, focusing on robust error handling throughout the migration process. Developed comprehensive unit tests and improved the testing framework, using mocking techniques to ensure reliability and strengthen continuous integration feedback. Addressed minor bugs encountered during the transition, stabilizing both the Gemini client and its associated tests. This work reduced external dependency risk and positioned the codebase for future GenAI feature expansion.