Add 'cached_content' to GenerativeModel's repr

Change-Id: I06676fad23895e3e1a6393baa938fc1f2df57d80

Author: mayureshagashe2105

google/generativeai/generative_models.py

@@ -123,6 +123,7 @@ def maybe_text(content):
                 safety_settings={self._safety_settings},
                 tools={self._tools},
                 system_instruction={maybe_text(self._system_instruction)},
+                cached_content={getattr(self, "cached_content", None)}
             )"""
         )
 

tests/test_generative_models.py

@@ -1213,6 +1213,7 @@ def test_repr_for_multi_turn_chat(self):
                     safety_settings={},
                     tools=None,
                     system_instruction=None,
+                    cached_content=None
                 ),
                 history=[glm.Content({'parts': [{'text': 'I really like fantasy books.'}], 'role': 'user'}), glm.Content({'parts': [{'text': 'first'}], 'role': 'model'}), glm.Content({'parts': [{'text': 'I also like this image.'}, {'inline_data': {'data': 'iVBORw0KGgoA...AAElFTkSuQmCC', 'mime_type': 'image/png'}}], 'role': 'user'}), glm.Content({'parts': [{'text': 'second'}], 'role': 'model'}), glm.Content({'parts': [{'text': 'What things do I like?.'}], 'role': 'user'}), glm.Content({'parts': [{'text': 'third'}], 'role': 'model'})]
             )"""
@@ -1241,6 +1242,7 @@ def test_repr_for_incomplete_streaming_chat(self):
                     safety_settings={},
                     tools=None,
                     system_instruction=None,
+                    cached_content=None
                 ),
                 history=[glm.Content({'parts': [{'text': 'I really like fantasy books.'}], 'role': 'user'}), <STREAMING IN PROGRESS>]
             )"""
@@ -1285,6 +1287,7 @@ def test_repr_for_broken_streaming_chat(self):
                     safety_settings={},
                     tools=None,
                     system_instruction=None,
+                    cached_content=None
                 ),
                 history=[glm.Content({'parts': [{'text': 'I really like fantasy books.'}], 'role': 'user'}), <STREAMING ERROR>]
             )"""
@@ -1296,6 +1299,14 @@ def test_repr_for_system_instruction(self):
         result = repr(model)
         self.assertIn("system_instruction='Be excellent.'", result)
 
+    def test_repr_for_model_created_from_cahced_content(self):
+        model = generative_models.GenerativeModel.from_cached_content(
+            cached_content="test-cached-content"
+        )
+        result = repr(model)
+        self.assertIn("cached_content=cachedContent/test-cached-content", result)
+        self.assertIn("model_name='models/gemini-1.0-pro-001'", result)
+
     def test_count_tokens_called_with_request_options(self):
         self.responses["count_tokens"].append(glm.CountTokensResponse())
         request_options = {"timeout": 120}