Match docs (#197)

* Make naming corpus and document also optional.

* Update logic to naming conditions

* reformat

* Update to add document_name variable

* Update to add corpus_name variable

* Updated CustomMetadata

* Updated CustomMetadata notation

* Updating time fields on chunk object

* Testing time fields on chunk

* Try making times optional params

* none otpion to time

* Updated batch create chunks, updated time format for list corpora

* Reformat

* Can accept full name or partial name in get and delete methods

* Updated functionality of update methods

* Added tests for additional changes

* Add custom metadata handling for document

* These should be ints.

---------

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

Author: shilpakancharla

google/generativeai/retriever.py

@@ -28,7 +28,7 @@
 
 
 def create_corpus(
-    name: str,
+    name: Optional[str] = None,
     display_name: Optional[str] = None,
     client: glm.RetrieverServiceClient | None = None,
 ) -> retriever_types.Corpus:
@@ -53,8 +53,10 @@ def create_corpus(
     if client is None:
         client = get_default_retriever_client()
 
-    corpus = None
-    if retriever_types.valid_name(name):
+    corpus, corpus_name = None, None
+    if name is None:
+        corpus = glm.Corpus(name=corpus_name, display_name=display_name)
+    elif retriever_types.valid_name(name):
         corpus_name = "corpora/" + name  # Construct the name
         corpus = glm.Corpus(name=corpus_name, display_name=display_name)
     else:
@@ -70,16 +72,18 @@ def create_corpus(
 
 
 async def create_corpus_async(
-    name: str,
+    name: Optional[str] = None,
     display_name: Optional[str] = None,
     client: glm.RetrieverServiceAsyncClient | None = None,
 ) -> retriever_types.Corpus:
     """This is the async version of `retriever.create_corpus`."""
     if client is None:
         client = get_default_retriever_async_client()
 
-    corpus = None
-    if retriever_types.valid_name(name):
+    corpus, corpus_name = None, None
+    if name is None:
+        corpus = glm.Corpus(name=corpus_name, display_name=display_name)
+    elif retriever_types.valid_name(name):
         corpus_name = "corpora/" + name  # Construct the name
         corpus = glm.Corpus(name=corpus_name, display_name=display_name)
     else:
@@ -107,6 +111,9 @@ def get_corpus(name: str, client: glm.RetrieverServiceClient | None = None) -> r
     if client is None:
         client = get_default_retriever_client()
 
+    if "/" not in name:
+        name = "corpora/" + name
+
     request = glm.GetCorpusRequest(name=name)
     response = client.get_corpus(request)
     response = type(response).to_dict(response)
@@ -121,6 +128,9 @@ async def get_corpus_async(name: str, client: glm.RetrieverServiceAsyncClient |
     if client is None:
         client = get_default_retriever_async_client()
 
+    if "/" not in name:
+        name = "corpora/" + name
+
     request = glm.GetCorpusRequest(name=name)
     response = await client.get_corpus(request)
     response = type(response).to_dict(response)
@@ -141,6 +151,9 @@ def delete_corpus(name: str, force: bool = False, client: glm.RetrieverServiceCl
     if client is None:
         client = get_default_retriever_client()
 
+    if "/" not in name:
+        name = "corpora/" + name
+
     request = glm.DeleteCorpusRequest(name=name, force=force)
     client.delete_corpus(request)
 
@@ -150,6 +163,9 @@ async def delete_corpus_async(name: str, force: bool = False, client: glm.Retrie
     if client is None:
         client = get_default_retriever_async_client()
 
+    if "/" not in name:
+        name = "corpora/" + name
+
     request = glm.DeleteCorpusRequest(name=name, force=force)
     await client.delete_corpus(request)
 
@@ -175,6 +191,8 @@ def list_corpora(
     request = glm.ListCorporaRequest(page_size=page_size)
     for corpus in client.list_corpora(request):
         corpus = type(corpus).to_dict(corpus)
+        idecode_time(corpus, "create_time")
+        idecode_time(corpus, "update_time")
         yield retriever_types.Corpus(**corpus)
 
 
@@ -190,4 +208,6 @@ async def list_corpora_async(
     request = glm.ListCorporaRequest(page_size=page_size)
     async for corpus in await client.list_corpora(request):
         corpus = type(corpus).to_dict(corpus)
+        idecode_time(corpus, "create_time")
+        idecode_time(corpus, "update_time")
         yield retriever_types.Corpus(**corpus)