diff --git a/google/generativeai/py.typed b/google/generativeai/py.typed new file mode 100644 index 000000000..d57989efb --- /dev/null +++ b/google/generativeai/py.typed @@ -0,0 +1 @@ +# see: https://peps.python.org/pep-0561/ diff --git a/setup.py b/setup.py index b4b05e619..4d7ef32ca 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. -# You may obtain a copy of the License at +# You may obtain a copy of the License at: # # http://www.apache.org/licenses/LICENSE-2.0 # @@ -88,6 +88,7 @@ def get_version(): "Programming Language :: Python :: 3.12", "Operating System :: OS Independent", "Topic :: Scientific/Engineering :: Artificial Intelligence", + "Typing :: Typed", ], platforms="Posix; MacOS X; Windows", packages=packages, @@ -97,4 +98,5 @@ def get_version(): extras_require=extras_require, include_package_data=True, zip_safe=False, + package_data={"generativeai": ["py.typed"]}, )