diff --git a/google/genai/batches.py b/google/genai/batches.py index a83371ede..1bbdc8f8c 100644 --- a/google/genai/batches.py +++ b/google/genai/batches.py @@ -1077,10 +1077,7 @@ def _GenerateContentConfig_to_mldev( ) if getv(from_object, ['labels']) is not None: - raise ValueError( - 'labels parameter is only supported in Gemini Enterprise Agent Platform' - ' mode, not in Gemini Developer API mode.' - ) + setv(parent_object, ['labels'], getv(from_object, ['labels'])) if getv(from_object, ['cached_content']) is not None: setv( diff --git a/google/genai/models.py b/google/genai/models.py index 0e965a839..8bd6e8655 100644 --- a/google/genai/models.py +++ b/google/genai/models.py @@ -1296,10 +1296,7 @@ def _GenerateContentConfig_to_mldev( ) if getv(from_object, ['labels']) is not None: - raise ValueError( - 'labels parameter is only supported in Gemini Enterprise Agent Platform' - ' mode, not in Gemini Developer API mode.' - ) + setv(parent_object, ['labels'], getv(from_object, ['labels'])) if getv(from_object, ['cached_content']) is not None: setv( diff --git a/google/genai/tests/models/test_generate_content.py b/google/genai/tests/models/test_generate_content.py index 655742931..9ee982ce9 100644 --- a/google/genai/tests/models/test_generate_content.py +++ b/google/genai/tests/models/test_generate_content.py @@ -148,9 +148,6 @@ class InstrumentEnum(Enum): ), pytest_helper.TestTableItem( name='test_labels', - exception_if_mldev=( - 'only supported in Gemini Enterprise Agent Platform mode' - ), parameters=types._GenerateContentParameters( model=GEMINI_FLASH_LATEST, contents=t.t_contents('What is your name?'), diff --git a/google/genai/types.py b/google/genai/types.py index d4bae8c99..40073ff6f 100644 --- a/google/genai/types.py +++ b/google/genai/types.py @@ -6654,7 +6654,7 @@ class GenerateContentConfig(_common.BaseModel): ) labels: Optional[dict[str, str]] = Field( default=None, - description="""Labels with user-defined metadata to break down billed charges.""", + description="""Labels with user-defined metadata for the request.""", ) cached_content: Optional[str] = Field( default=None, @@ -6887,7 +6887,7 @@ class GenerateContentConfigDict(TypedDict, total=False): """ labels: Optional[dict[str, str]] - """Labels with user-defined metadata to break down billed charges.""" + """Labels with user-defined metadata for the request.""" cached_content: Optional[str] """Resource name of a context cache that can be used in subsequent