Add cache_from_name and cache_chat (#413)

* Add cache_from_name and cache_chat

* Resolve comments.

Change-Id: I74aa097499fc426e4e39327c2ffcdcf3f68534dd

* update cache sample

Change-Id: I1b261cdcca6e564471bb8ca5a59e9138d4a5f253

---------

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

Author: shilpakancharla

samples/cache.py

@@ -32,6 +32,10 @@ def test_cache_create(self):
             contents=[document],
         )
         print(cache)
+
+        model = genai.GenerativeModel.from_cached_content(cache)
+        response = model.generate_content("Please summarize this transcript")
+        print(response.text)
         # [END cache_create]
         cache.delete()
 
@@ -92,10 +96,61 @@ def test_cache_update(self):
         print(f"After update:\n {cache}")
 
         # Or you can update the expire_time
-        cache.update(expire_time=datetime.now() + datetime.timedelta(minutes=15))
+        cache.update(expire_time=datetime.datetime.now() + datetime.timedelta(minutes=15))
         # [END cache_update]
         cache.delete()
 
+    def test_cache_create_from_name(self):
+        # [START cache_create_from_name]
+        document = genai.upload_file(path=media / "a11.txt")
+        model_name = "gemini-1.5-flash-001"
+        cache = genai.caching.CachedContent.create(
+            model=model_name,
+            system_instruction="You are an expert analyzing transcripts.",
+            contents=[document],
+        )
+        cache_name = cache.name  # Save the name for later
+
+        # Later
+        cache = genai.caching.CachedContent.get(cache_name)
+        apollo_model = genai.GenerativeModel.from_cached_content(cache)
+        response = apollo_model.generate_content("Find a lighthearted moment from this transcript")
+        print(response.text)
+        # [END cache_create_from_name]
+        cache.delete()
+
+    def test_cache_chat(self):
+        # [START cache_chat]
+        model_name = "gemini-1.5-flash-001"
+        system_instruction = "You are an expert analyzing transcripts."
+
+        model = genai.GenerativeModel(model_name=model_name, system_instruction=system_instruction)
+        chat = model.start_chat()
+        document = genai.upload_file(path=media / "a11.txt")
+        response = chat.send_message(["Hi, could you summarize this transcript?", document])
+        print("\n\nmodel:  ", response.text)
+        response = chat.send_message(
+            ["Okay, could you tell me more about the trans-lunar injection"]
+        )
+        print("\n\nmodel:  ", response.text)
+
+        # To cache the conversation so far, pass the chat history as the list of "contents".
+        cache = genai.caching.CachedContent.create(
+            model=model_name,
+            system_instruction=system_instruction,
+            contents=chat.history,
+        )
+        model = genai.GenerativeModel.from_cached_content(cached_content=cache)
+
+        # Continue the chat where you left off.
+        chat = model.start_chat()
+        response = chat.send_message(
+            "I didn't understand that last part, could you explain it in simpler language?"
+        )
+        print("\n\nmodel:  ", response.text)
+        # [END cache_chat]
+        cache.delete()
+
 
 if __name__ == "__main__":
     absltest.main()