Create a 0.2.1 release. (#72)

* Update release notes.

* Update to `google.ai.generativelanguage` v0.3.3

Author: MarkDaoust

RELEASE.md

@@ -1,3 +1,7 @@
+## v0.2.1
+
+- Bugfix: Add missing enum names to `saftey_types.py`
+- Update to `google.ai.generativelanguage` v0.3.3 
 
 ## v0.2
 

docs/build_docs.py

@@ -30,7 +30,7 @@
 
 import google
 from google import generativeai as palm
-from google.ai import generativelanguage_v1beta3 as glm
+from google.ai import generativelanguage as glm
 
 from tensorflow_docs.api_generator import generate_lib
 from tensorflow_docs.api_generator import public_api
@@ -54,7 +54,7 @@
 when initializing a client:
 
 ```
-from google.ai import generativelanguage_v1beta3 as glm
+from google.ai import generativelanguage as glm
 
 client = glm.DiscussServiceClient(
     client_options={'api_key':'YOUR_API_KEY'})
@@ -147,9 +147,6 @@ def __call__(self, path, parent, children):
             if "ServiceClient" in path[-1] or "ServiceAsyncClient" in path[-1]:
                 children = list(self.drop_staticmethods(parent, children))
 
-        if "generativelanguage" in path[-1]:
-            children.append(('types', glm.types))
-
         return children
 
 
@@ -207,8 +204,9 @@ def gen_api_docs():
     # Fixup the toc file.
     toc_path = out_path / "google/_toc.yaml"
     toc = yaml.safe_load(toc_path.read_text())
+    assert toc["toc"][0]["title"] == "google"
     toc["toc"] = toc["toc"][1:]
-    toc["toc"][0]["title"] = "google.ai.generativelanguage_v1beta3"
+    toc["toc"][0]["title"] = "google.ai.generativelanguage"
     toc["toc"][0]["section"] = toc["toc"][0]["section"][1]["section"]
     toc["toc"][0], toc["toc"][1] = toc["toc"][1], toc["toc"][0]
     toc_path.write_text(yaml.dump(toc))
@@ -230,6 +228,11 @@ def gen_api_docs():
         new_content = re.sub(r"\.\. _oneof:.*?\n", "", new_content)
         new_content = re.sub(r"`oneof`_.*?\n", "", new_content)
         new_content = re.sub(r"\.\. code-block:: python.*?\n", "", new_content)
+
+        new_content = re.sub(
+            r"generativelanguage_\w+.types", "generativelanguage", new_content
+        )
+
         if new_content != old_content:
             fpath.write_text(new_content)
 

google/generativeai/client.py

@@ -17,7 +17,7 @@
 import os
 from typing import cast, Optional, Union
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 
 from google.auth import credentials as ga_credentials
 from google.api_core import client_options as client_options_lib

google/generativeai/discuss.py

@@ -20,7 +20,7 @@
 
 from typing import Iterable, List, Optional, Union
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 
 from google.generativeai.client import get_default_discuss_client
 from google.generativeai.client import get_default_discuss_async_client

google/generativeai/models.py

@@ -17,7 +17,7 @@
 import typing
 from typing import Any, Literal
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 from google.generativeai import operations
 from google.generativeai.client import get_default_model_client
 from google.generativeai.types import model_types

google/generativeai/operations.py

@@ -17,7 +17,7 @@
 import functools
 from typing import Iterator
 
-from google.ai import generativelanguage_v1beta3 as glm
+from google.ai import generativelanguage as glm
 from google.generativeai import client as client_lib
 from google.generativeai.types import model_types
 from google.api_core import operation as operation_lib

google/generativeai/text.py

@@ -18,7 +18,7 @@
 from collections.abc import Sequence
 from typing import Iterable, overload
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 
 from google.generativeai.client import get_default_text_client
 from google.generativeai.types import text_types

google/generativeai/types/citation_types.py

@@ -16,7 +16,7 @@
 
 from typing import Optional, List
 
-from google.ai import generativelanguage_v1beta3 as glm
+from google.ai import generativelanguage as glm
 from google.generativeai import docstring_utils
 from typing import TypedDict
 

google/generativeai/types/discuss_types.py

@@ -18,7 +18,7 @@
 import dataclasses
 from typing import Any, Dict, TypedDict, Union, Iterable, Optional, Tuple, List
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 from google.generativeai.types import safety_types
 from google.generativeai.types import citation_types
 

google/generativeai/types/model_types.py

@@ -20,7 +20,7 @@
 import re
 from typing import Any, Iterable, TypedDict, Union
 
-import google.ai.generativelanguage_v1beta3 as glm
+import google.ai.generativelanguage as glm
 
 __all__ = [
     "Model",