fix tests

Change-Id: I39f61012f850a82e09a7afb80b527a0f99ad0ec7

tests/test_generative_models.py

@@ -76,6 +76,20 @@ def count_tokens(
  self.observed_kwargs.append(kwargs)
  response = self.responses["count_tokens"].pop(0)
  return response
+
+ def get_cached_content(
+ self,
+ request: glm.GetCachedContentRequest,
+ **kwargs,
+ ) -> glm.CachedContent:
+ self.observed_requests.append(request)
+ return glm.CachedContent(
+ name="cachedContent/test-cached-content",
+ model="models/gemini-1.0-pro-001",
+ create_time="2000-01-01T01:01:01.123456Z",
+ update_time="2000-01-01T01:01:01.123456Z",
+ expire_time="2000-01-01T01:01:01.123456Z",
+ )
 
 
 class CUJTests(parameterized.TestCase):
@@ -98,19 +112,19 @@ def setUp(self):
  client_lib._client_manager.clients["generative"] = self.client
  client_lib._client_manager.clients["cache"] = self.client
 
- @add_client_method
- def get_cached_content(
- request: glm.GetCachedContentRequest,
- **kwargs,
- ) -> glm.CachedContent:
- self.observed_requests.append(request)
- return glm.CachedContent(
- name="cachedContent/test-cached-content",
- model="models/gemini-1.0-pro-001",
- create_time="2000-01-01T01:01:01.123456Z",
- update_time="2000-01-01T01:01:01.123456Z",
- expire_time="2000-01-01T01:01:01.123456Z",
- )
+ # @add_client_method
+ # def get_cached_content(
+ #  request: glm.GetCachedContentRequest,
+ #  **kwargs,
+ # ) -> glm.CachedContent:
+ #  self.observed_requests.append(request)
+ #  return glm.CachedContent(
+ #  name="cachedContent/test-cached-content",
+ #  model="models/gemini-1.0-pro-001",
+ #  create_time="2000-01-01T01:01:01.123456Z",
+ #  update_time="2000-01-01T01:01:01.123456Z",
+ #  expire_time="2000-01-01T01:01:01.123456Z",
+ #  )
 
  def test_hello(self):
  # Generate text from text prompt