fix: GenAI SDK client(multimodal) - Fix Pydantic validation errors when using create_* in some cases

PiperOrigin-RevId: 901268856

Author: cleop-google

vertexai/_genai/_datasets_utils.py

@@ -34,15 +34,26 @@
 T = TypeVar("T", bound=BaseModel)
 
 
-def create_from_response(model_type: Type[T], response: dict[str, Any]) -> T:
-    """Creates a model from a response."""
-    model_field_names = model_type.model_fields.keys()
-    filtered_response = {}
-    for key, value in response.items():
-        snake_key = common.camel_to_snake(key)
-        if snake_key in model_field_names:
-            filtered_response[snake_key] = value
-    return model_type(**filtered_response)
+def create_from_response(
+    model_type: Type[T],
+    response: dict[str, Any],
+    config: Any | None = None,
+) -> T:
+  """Creates a model from a response."""
+  kwargs = (
+      {
+          "config": {
+              "response_schema": getattr(config, "response_schema", None),
+              "response_json_schema": getattr(
+                  config, "response_json_schema", None
+              ),
+              "include_all_fields": getattr(config, "include_all_fields", None),
+          }
+      }
+      if config
+      else {}
+  )
+  return model_type._from_response(response=response, kwargs=kwargs)  # type: ignore[attr-defined,no-any-return]
 
 
 def validate_multimodal_dataset_bigquery_uri(