rename get_cached_content to get

mayureshagashe2105 · May 22, 2024 · 6fafe6b · 6fafe6b
1 parent a4ac7a5
commit 6fafe6b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/google/generativeai/caching.py b/google/generativeai/caching.py
@@ -36,7 +36,7 @@ def get_cached_content(name: str, client: glm.CacheServiceClient | None = None)
  Returns:
  `CachedContent` resource with specified name.
  """
- return CachedContent.get_cached_content(name=name, client=client)
+ return CachedContent.get(name=name, client=client)
 
 
 def delete_cached_content(name: str, client: glm.CacheServiceClient | None = None) -> None:

diff --git a/google/generativeai/types/caching_types.py b/google/generativeai/types/caching_types.py
@@ -193,9 +193,7 @@ def create(
  return decode_cached_content(response)
 
  @classmethod
- def get_cached_content(
- cls, name: str, client: glm.CacheServiceClient | None = None
- ) -> CachedContent:
+ def get(cls, name: str, client: glm.CacheServiceClient | None = None) -> CachedContent:
  """Gets a `CachedContent` resource."""
  if client is None:
  client = get_default_cache_client()