feat: [google-ai-generativelanguage] Add code execution (#12843)

- [ ] Regenerate this pull request now.

feat: Add max_temperature
docs: Minor fixes

PiperOrigin-RevId: 646960455

Source-Link:
googleapis/googleapis@cb060cb

Source-Link:
googleapis/googleapis-gen@64dcd13
Copy-Tag:
eyJwIjoicGFja2FnZXMvZ29vZ2xlLWFpLWdlbmVyYXRpdmVsYW5ndWFnZS8uT3dsQm90LnlhbWwiLCJoIjoiNjRkY2QxMzViMzdiOTE1MjYyY2M2MTBlMGVjOWQ3YTc5NzAxMGY2MCJ9

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
Co-authored-by: Victor Chudnovsky <vchudnov@google.com>

Author: gcf-owl-bot[bot]

packages/google-ai-generativelanguage/google/ai/generativelanguage/__init__.py

@@ -81,7 +81,10 @@
 )
 from google.ai.generativelanguage_v1beta.types.content import (
     Blob,
+    CodeExecution,
+    CodeExecutionResult,
     Content,
+    ExecutableCode,
     FileData,
     FunctionCall,
     FunctionCallingConfig,
@@ -254,7 +257,10 @@
     "CitationMetadata",
     "CitationSource",
     "Blob",
+    "CodeExecution",
+    "CodeExecutionResult",
     "Content",
+    "ExecutableCode",
     "FileData",
     "FunctionCall",
     "FunctionCallingConfig",

packages/google-ai-generativelanguage/google/ai/generativelanguage/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.6.5"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.6.5"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/__init__.py

@@ -47,7 +47,10 @@
 from .types.citation import CitationMetadata, CitationSource
 from .types.content import (
     Blob,
+    CodeExecution,
+    CodeExecutionResult,
     Content,
+    ExecutableCode,
     FileData,
     FunctionCall,
     FunctionCallingConfig,
@@ -220,6 +223,8 @@
     "ChunkData",
     "CitationMetadata",
     "CitationSource",
+    "CodeExecution",
+    "CodeExecutionResult",
     "Condition",
     "Content",
     "ContentEmbedding",
@@ -257,6 +262,7 @@
     "EmbedTextResponse",
     "Embedding",
     "Example",
+    "ExecutableCode",
     "File",
     "FileData",
     "FileServiceClient",

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.6.5"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/types/__init__.py

@@ -25,7 +25,10 @@
 from .citation import CitationMetadata, CitationSource
 from .content import (
     Blob,
+    CodeExecution,
+    CodeExecutionResult,
     Content,
+    ExecutableCode,
     FileData,
     FunctionCall,
     FunctionCallingConfig,
@@ -182,7 +185,10 @@
     "CitationMetadata",
     "CitationSource",
     "Blob",
+    "CodeExecution",
+    "CodeExecutionResult",
     "Content",
+    "ExecutableCode",
     "FileData",
     "FunctionCall",
     "FunctionCallingConfig",

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/types/content.py

@@ -28,7 +28,10 @@
         "Part",
         "Blob",
         "FileData",
+        "ExecutableCode",
+        "CodeExecutionResult",
         "Tool",
+        "CodeExecution",
         "ToolConfig",
         "FunctionCallingConfig",
         "FunctionDeclaration",
@@ -145,6 +148,15 @@ class Part(proto.Message):
         file_data (google.ai.generativelanguage_v1beta.types.FileData):
             URI based data.
 
+            This field is a member of `oneof`_ ``data``.
+        executable_code (google.ai.generativelanguage_v1beta.types.ExecutableCode):
+            Code generated by the model that is meant to
+            be executed.
+
+            This field is a member of `oneof`_ ``data``.
+        code_execution_result (google.ai.generativelanguage_v1beta.types.CodeExecutionResult):
+            Result of executing the ``ExecutableCode``.
+
             This field is a member of `oneof`_ ``data``.
     """
 
@@ -177,6 +189,18 @@ class Part(proto.Message):
         oneof="data",
         message="FileData",
     )
+    executable_code: "ExecutableCode" = proto.Field(
+        proto.MESSAGE,
+        number=9,
+        oneof="data",
+        message="ExecutableCode",
+    )
+    code_execution_result: "CodeExecutionResult" = proto.Field(
+        proto.MESSAGE,
+        number=10,
+        oneof="data",
+        message="CodeExecutionResult",
+    )
 
 
 class Blob(proto.Message):
@@ -228,6 +252,94 @@ class FileData(proto.Message):
     )
 
 
+class ExecutableCode(proto.Message):
+    r"""Code generated by the model that is meant to be executed, and the
+    result returned to the model.
+
+    Only generated when using the ``CodeExecution`` tool, in which the
+    code will be automatically executed, and a corresponding
+    ``CodeExecutionResult`` will also be generated.
+
+    Attributes:
+        language (google.ai.generativelanguage_v1beta.types.ExecutableCode.Language):
+            Required. Programming language of the ``code``.
+        code (str):
+            Required. The code to be executed.
+    """
+
+    class Language(proto.Enum):
+        r"""Supported programming languages for the generated code.
+
+        Values:
+            LANGUAGE_UNSPECIFIED (0):
+                Unspecified language. This value should not
+                be used.
+            PYTHON (1):
+                Python >= 3.10, with numpy and simpy
+                available.
+        """
+        LANGUAGE_UNSPECIFIED = 0
+        PYTHON = 1
+
+    language: Language = proto.Field(
+        proto.ENUM,
+        number=1,
+        enum=Language,
+    )
+    code: str = proto.Field(
+        proto.STRING,
+        number=2,
+    )
+
+
+class CodeExecutionResult(proto.Message):
+    r"""Result of executing the ``ExecutableCode``.
+
+    Only generated when using the ``CodeExecution``, and always follows
+    a ``part`` containing the ``ExecutableCode``.
+
+    Attributes:
+        outcome (google.ai.generativelanguage_v1beta.types.CodeExecutionResult.Outcome):
+            Required. Outcome of the code execution.
+        output (str):
+            Optional. Contains stdout when code execution
+            is successful, stderr or other description
+            otherwise.
+    """
+
+    class Outcome(proto.Enum):
+        r"""Enumeration of possible outcomes of the code execution.
+
+        Values:
+            OUTCOME_UNSPECIFIED (0):
+                Unspecified status. This value should not be
+                used.
+            OUTCOME_OK (1):
+                Code execution completed successfully.
+            OUTCOME_FAILED (2):
+                Code execution finished but with a failure. ``stderr``
+                should contain the reason.
+            OUTCOME_DEADLINE_EXCEEDED (3):
+                Code execution ran for too long, and was
+                cancelled. There may or may not be a partial
+                output present.
+        """
+        OUTCOME_UNSPECIFIED = 0
+        OUTCOME_OK = 1
+        OUTCOME_FAILED = 2
+        OUTCOME_DEADLINE_EXCEEDED = 3
+
+    outcome: Outcome = proto.Field(
+        proto.ENUM,
+        number=1,
+        enum=Outcome,
+    )
+    output: str = proto.Field(
+        proto.STRING,
+        number=2,
+    )
+
+
 class Tool(proto.Message):
     r"""Tool details that the model may use to generate response.
 
@@ -250,13 +362,31 @@ class Tool(proto.Message):
             [FunctionResponse][content.part.function_response] with the
             [content.role] "function" generation context for the next
             model turn.
+        code_execution (google.ai.generativelanguage_v1beta.types.CodeExecution):
+            Optional. Enables the model to execute code
+            as part of generation.
     """
 
     function_declarations: MutableSequence["FunctionDeclaration"] = proto.RepeatedField(
         proto.MESSAGE,
         number=1,
         message="FunctionDeclaration",
     )
+    code_execution: "CodeExecution" = proto.Field(
+        proto.MESSAGE,
+        number=3,
+        message="CodeExecution",
+    )
+
+
+class CodeExecution(proto.Message):
+    r"""Tool that executes code generated by the model, and automatically
+    returns the result to the model.
+
+    See also ``ExecutableCode`` and ``CodeExecutionResult`` which are
+    only generated when using this tool.
+
+    """
 
 
 class ToolConfig(proto.Message):

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/types/generative_service.py

@@ -431,7 +431,7 @@ class BlockReason(proto.Enum):
                     ``safety_ratings`` to understand which safety category
                     blocked it.
                 OTHER (2):
-                    Prompt was blocked due to unknown reaasons.
+                    Prompt was blocked due to unknown reasons.
             """
             BLOCK_REASON_UNSPECIFIED = 0
             SAFETY = 1

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta/types/model.py

@@ -76,14 +76,18 @@ class Model(proto.Message):
         temperature (float):
             Controls the randomness of the output.
 
-            Values can range over ``[0.0,2.0]``, inclusive. A higher
-            value will produce responses that are more varied, while a
-            value closer to ``0.0`` will typically result in less
-            surprising responses from the model. This value specifies
-            default to be used by the backend while making the call to
-            the model.
+            Values can range over ``[0.0,max_temperature]``, inclusive.
+            A higher value will produce responses that are more varied,
+            while a value closer to ``0.0`` will typically result in
+            less surprising responses from the model. This value
+            specifies default to be used by the backend while making the
+            call to the model.
 
             This field is a member of `oneof`_ ``_temperature``.
+        max_temperature (float):
+            The maximum temperature this model can use.
+
+            This field is a member of `oneof`_ ``_max_temperature``.
         top_p (float):
             For Nucleus sampling.
 
@@ -142,6 +146,11 @@ class Model(proto.Message):
         number=9,
         optional=True,
     )
+    max_temperature: float = proto.Field(
+        proto.FLOAT,
+        number=13,
+        optional=True,
+    )
     top_p: float = proto.Field(
         proto.FLOAT,
         number=10,

packages/google-ai-generativelanguage/google/ai/generativelanguage_v1beta2/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.6.5"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}