fix: Default to list base models (instead of tuned models)

PiperOrigin-RevId: 722817748

Author: matthew29tang

README.md

@@ -122,12 +122,12 @@ print(response.text)
 To retrieve tuned models, see [list tuned models](#list-tuned-models).
 
 ```python
-for model in client.models.list(config={'query_base':True}):
+for model in client.models.list():
     print(model)
 ```
 
 ```python
-pager = client.models.list(config={"page_size": 10, 'query_base':True})
+pager = client.models.list(config={"page_size": 10})
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -137,12 +137,12 @@ print(pager[0])
 #### Async
 
 ```python
-async for job in await client.aio.models.list(config={'query_base':True}):
+async for job in await client.aio.models.list():
     print(job)
 ```
 
 ```python
-async_pager = await client.aio.models.list(config={"page_size": 10, 'query_base':True})
+async_pager = await client.aio.models.list(config={"page_size": 10})
 print(async_pager.page_size)
 print(async_pager[0])
 await async_pager.next_page()
@@ -801,12 +801,12 @@ print(tuned_model)
 To retrieve base models, see [list base models](#list-base-models).
 
 ```python
-for model in client.models.list(config={"page_size": 10}):
+for model in client.models.list(config={"page_size": 10, "query_base": False}):
     print(model)
 ```
 
 ```python
-pager = client.models.list(config={"page_size": 10})
+pager = client.models.list(config={"page_size": 10, "query_base": False})
 print(pager.page_size)
 print(pager[0])
 pager.next_page()
@@ -816,12 +816,12 @@ print(pager[0])
 #### Async
 
 ```python
-async for job in await client.aio.models.list(config={"page_size": 10}):
+async for job in await client.aio.models.list(config={"page_size": 10, "query_base": False}):
     print(job)
 ```
 
 ```python
-async_pager = await client.aio.models.list(config={"page_size": 10})
+async_pager = await client.aio.models.list(config={"page_size": 10, "query_base": False})
 print(async_pager.page_size)
 print(async_pager[0])
 await async_pager.next_page()

google/genai/models.py

@@ -4780,8 +4780,8 @@ def list(
   ) -> Pager[types.Model]:
     """Makes an API request to list the available models.
 
-    If `query_base` is set to True in the config, the API will return all
-    available base models. If set to False or not set (default), it will return
+    If `query_base` is set to True in the config or not set (default), the
+    API will return all available base models. If set to False, it will return
     all tuned models.
 
     Args:
@@ -4804,6 +4804,8 @@ def list(
         types._ListModelsParameters(config=config).config
         or types.ListModelsConfig()
     )
+    if config.query_base is None:
+      config.query_base = True
     if self._api_client.vertexai:
       config = config.copy()
       if not config.query_base:
@@ -4814,8 +4816,6 @@ def list(
             if filter_value
             else 'labels.tune-type:*'
         )
-    if not config.query_base:
-      config.query_base = False
     return Pager(
         'models',
         self._list,

google/genai/tests/models/test_list.py

@@ -29,11 +29,11 @@
 test_table: list[pytest_helper.TestTableItem] = [
     pytest_helper.TestTableItem(
         name='test_tuned_models',
-        parameters=types._ListModelsParameters(),
+        parameters=types._ListModelsParameters(config={'query_base': False}),
     ),
     pytest_helper.TestTableItem(
         name='test_base_models',
-        parameters=types._ListModelsParameters(config={'query_base': True}),
+        parameters=types._ListModelsParameters(),
     ),
     pytest_helper.TestTableItem(
         name='test_base_models_with_config',