Fix failing tests.

PiperOrigin-RevId: 829447863

Author: kibergus

genai_processors/content_api.py

@@ -375,7 +375,7 @@ def from_function_response(
       function_response = genai_types.FunctionResponse(
           response={response_key: response}, **function_response_args
       )
-      function_response.json()
+      function_response.model_dump_json()
     except ValueError:
       # Response is not JSON serializable. Then try to construct content.
       response_content = ProcessorContent(response)
@@ -386,8 +386,11 @@ def from_function_response(
         )
       except ValueError:
         parts = [
-            genai_types.FunctionResponsePart.from_bytes(
-                data=part.bytes, mime_type=part.mimetype
+            genai_types.FunctionResponsePart(
+                inline_data=genai_types.FunctionResponseBlob(
+                    data=part.bytes,
+                    mime_type=part.mimetype,
+                )
             )
             for part in response_content
         ]

genai_processors/tests/content_api_test.py

@@ -132,12 +132,17 @@ def test_from_function_response_with_mixed_content(self):
         genai_types.FunctionResponse(
             name='foo',
             parts=[
-                genai_types.FunctionResponsePart.from_bytes(
-                    data=b'Here is a black cat in a black room: ',
-                    mime_type='text/plain',
+                genai_types.FunctionResponsePart(
+                    inline_data=genai_types.FunctionResponseBlob(
+                        data=b'Here is a black cat in a black room: ',
+                        mime_type='text/plain',
+                    ),
                 ),
-                genai_types.FunctionResponsePart.from_bytes(
-                    data=image_bytes, mime_type='image/png'
+                genai_types.FunctionResponsePart(
+                    inline_data=genai_types.FunctionResponseBlob(
+                        data=image_bytes,
+                        mime_type='image/png',
+                    ),
                 ),
             ],
         ),

genai_processors/tests/sql_cache_test.py

@@ -30,7 +30,7 @@ def setUp(self):
   async def test_cache_put_and_lookup(self):
     """Tests basic put and lookup functionality."""
     async with sql_cache.sql_cache(self.db_url) as cache:
-      await cache.put(TEST_QUERY, TEST_VALUE)
+      await cache.put(query=TEST_QUERY, value=TEST_VALUE)
       result = await cache.lookup(TEST_QUERY)
       self.assertEqual(result, TEST_VALUE)
 
@@ -45,15 +45,15 @@ async def test_cache_ttl(self):
     async with sql_cache.sql_cache(
         self.db_url, ttl_hours=0.0001 / 60 / 60
     ) as cache:  # Very short TTL
-      await cache.put(TEST_QUERY, TEST_VALUE)
+      await cache.put(query=TEST_QUERY, value=TEST_VALUE)
       await asyncio.sleep(0.0002)  # Wait for TTL to expire
       result = await cache.lookup(TEST_QUERY)
       self.assertIs(result, cache_base.CacheMiss)
 
   async def test_cache_remove(self):
     """Tests removing an item from the cache."""
     async with sql_cache.sql_cache(self.db_url) as cache:
-      await cache.put(TEST_QUERY, TEST_VALUE)
+      await cache.put(query=TEST_QUERY, value=TEST_VALUE)
       await cache.remove(TEST_QUERY)
       result = await cache.lookup(TEST_QUERY)
       self.assertIs(result, cache_base.CacheMiss)
@@ -63,8 +63,8 @@ async def test_with_key_prefix(self):
     async with sql_cache.sql_cache(self.db_url) as cache1:
       cache2 = cache1.with_key_prefix('prefix:')
 
-      await cache1.put(TEST_QUERY, TEST_VALUE)
-      await cache2.put(TEST_QUERY, TEST_VALUE_2)
+      await cache1.put(query=TEST_QUERY, value=TEST_VALUE)
+      await cache2.put(query=TEST_QUERY, value=TEST_VALUE_2)
 
       result1 = await cache1.lookup(TEST_QUERY)
       result2 = await cache2.lookup(TEST_QUERY)
@@ -84,16 +84,16 @@ async def test_different_content_types(self):
 
       value2 = content_api.ProcessorContent({'a': 1, 'b': True})
 
-      await cache.put(query1, value1)
-      await cache.put(query2, value2)
+      await cache.put(query=query1, value=value1)
+      await cache.put(query=query2, value=value2)
 
       self.assertEqual(await cache.lookup(query1), value1)
       self.assertEqual(await cache.lookup(query2), value2)
 
   async def test_cleanup_expired(self):
     """Tests that the _cleanup_expired method removes old entries."""
     async with sql_cache.sql_cache(self.db_url, ttl_hours=0.0001) as cache:
-      await cache.put(TEST_QUERY, TEST_VALUE)
+      await cache.put(query=TEST_QUERY, value=TEST_VALUE)
       # Mock datetime to control time
       with mock.patch('genai_processors.sql_cache.datetime') as mock_datetime:
         mock_datetime.datetime.now.return_value = datetime.datetime.now(