Skip to content

Commit f1ac916

Browse files
committed
fix param type
1 parent a1f353e commit f1ac916

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

  • packages/bigframes/bigframes/bigquery/_operations

packages/bigframes/bigframes/bigquery/_operations/ai.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -818,9 +818,7 @@ def if_(
818818
*,
819819
connection_id: str | None = None,
820820
endpoint: str | None = None,
821-
optimization_mode: Literal[
822-
"minimize_cost", "maximize_performance"
823-
] = "minimize_cost",
821+
optimization_mode: Literal["minimize_cost", "maximize_quality"] = "minimize_cost",
824822
max_error_ratio: float = 1.0,
825823
) -> series.Series:
826824
"""
@@ -855,10 +853,10 @@ def if_(
855853
generally available or preview Gemini model. If you specify the model name, BigQuery ML automatically identifies and
856854
uses the full endpoint of the model. If you don't specify an ENDPOINT value, BigQuery ML dynamically chooses a model based on your query to have the
857855
best cost to quality tradeoff for the task.
858-
optimization_mode (Literal["minimize_cost", "maximize_performance"]):
856+
optimization_mode (Literal["minimize_cost", "maximize_quality"]):
859857
Specifies the optimization strategy to use. Supported values are:
860858
* "minimize_cost" (default): uses a local, distilled model to process the majority of rows, reducing latency and cost.
861-
* "maximize_performance": always uses the remote LLM for inference.
859+
* "maximize_quality": always uses the remote LLM for inference.
862860
max_error_ratio (float):
863861
A float value between 0.0 and 1.0 that contains the maximum acceptable ratio of row-level inference failures to
864862
rows processed on this function. If this value is exceeded, then the query fails. The default value is 1.0.

0 commit comments

Comments
 (0)