Fix Model dataclass (#439)

* Fix Model dataclass

Dataclass Model misses max_temperature field which brakes list_models()

* Fix to handle max_temperature

Change-Id: I0cdacd0134c868044ae5b96322b266224d7b8a69

* Fix type error.

Change-Id: I5db22405c93b66fd268b19f1e081dc24f3246d24

---------

Co-authored-by: Mark Daoust <[email protected]>

Author: rozha

google/generativeai/types/model_types.py

@@ -117,6 +117,7 @@ class Model:
     output_token_limit: int
     supported_generation_methods: list[str]
     temperature: float | None = None
+    max_temperature: float | None = None
     top_p: float | None = None
     top_k: int | None = None
 

tests/test_models.py

@@ -44,7 +44,7 @@ def setUp(self):
 
         client._client_manager.clients["model"] = self.client
 
-        # TODO(markdaoust): Check if typechecking works better if wee define this as a
+        # TODO(markdaoust): Check if typechecking works better if we define this as a
         #                   subclass of `glm.ModelServiceClient`, would pyi files for `glm`. help?
         def add_client_method(f):
             name = f.__name__
@@ -160,6 +160,15 @@ def test_get_model(self, name):
         else:
             self.assertIsInstance(model, model_types.TunedModel)
 
+    def test_max_temperature(self):
+        name = "models/fake-bison-001"
+        max_temperature = 3.0
+        self.responses = {
+            "get_model": protos.Model(name=name, max_temperature=max_temperature),
+        }
+        model = models.get_base_model(name)
+        self.assertEqual(max_temperature, model.max_temperature)
+
     @parameterized.named_parameters(
         ["simple", "mystery-bison-001"],
         ["model-instance", protos.Model(name="how?-bison-001")],