Fix new lint errors showing up after change

Author: aabmass

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_async_nonstreaming.py

@@ -65,9 +65,11 @@ def configure_valid_response(
             )
         )
 
-    def generate_content(self, *args, **kwargs):
+    def generate_content(self, *, model, contents):
         return asyncio.run(
-            self.client.aio.models.generate_content(*args, **kwargs)
+            self.client.aio.models.generate_content(
+                model=model, contents=contents
+            )
         )
 
     def test_async_generate_content_not_broken_by_instrumentation(self):

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_async_streaming.py

@@ -67,18 +67,20 @@ def configure_valid_response(
             )
         )
 
-    async def _generate_content_helper(self, *args, **kwargs):
+    async def _generate_content_helper(self, *, model, contents):
         result = []
         async for (
             response
         ) in await self.client.aio.models.generate_content_stream(
-            *args, **kwargs
+            model=model, contents=contents
         ):
             result.append(response)
         return result
 
-    def generate_content(self, *args, **kwargs):
-        return asyncio.run(self._generate_content_helper(*args, **kwargs))
+    def generate_content(self, *, model, contents):
+        return asyncio.run(
+            self._generate_content_helper(model=model, contents=contents)
+        )
 
     def test_async_generate_content_not_broken_by_instrumentation(self):
         self.configure_valid_response(response_text="Yep, it works!")

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_sync_streaming.py

@@ -64,10 +64,10 @@ def configure_valid_response(
             )
         )
 
-    def generate_content(self, *args, **kwargs):
+    def generate_content(self, *, model, contents):
         result = []
         for response in self.client.models.generate_content_stream(
-            *args, **kwargs
+            model=model, contents=contents
         ):
             result.append(response)
         return result