From 23145f30843febc29cc607930df02b1e083ce39a Mon Sep 17 00:00:00 2001
From: Sara Robinson <sararob@google.com>
Date: Mon, 20 Jul 2026 08:20:30 -0700
Subject: [PATCH] fix: Use the correct enum for RagFile.file_status

PiperOrigin-RevId: 950851566
---
 agentplatform/_genai/types/__init__.py |  6 ++++++
 agentplatform/_genai/types/common.py   | 22 ++++++++++++++++++++--
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/agentplatform/_genai/types/__init__.py b/agentplatform/_genai/types/__init__.py
index 46f7345e2e..a731e74a0c 100644
--- a/agentplatform/_genai/types/__init__.py
+++ b/agentplatform/_genai/types/__init__.py
@@ -1312,6 +1312,9 @@
 from .common import RagFileParsingConfigLlmParserOrDict
 from .common import RagFileParsingConfigOrDict
 from .common import RagFileState
+from .common import RagFileStatus
+from .common import RagFileStatusDict
+from .common import RagFileStatusOrDict
 from .common import RagFileTransformationConfig
 from .common import RagFileTransformationConfigDict
 from .common import RagFileTransformationConfigOrDict
@@ -2806,6 +2809,9 @@
     "GetRagFileConfig",
     "GetRagFileConfigDict",
     "GetRagFileConfigOrDict",
+    "RagFileStatus",
+    "RagFileStatusDict",
+    "RagFileStatusOrDict",
     "DirectUploadSource",
     "DirectUploadSourceDict",
     "DirectUploadSourceOrDict",
diff --git a/agentplatform/_genai/types/common.py b/agentplatform/_genai/types/common.py
index 79c577c7fd..02820baf86 100644
--- a/agentplatform/_genai/types/common.py
+++ b/agentplatform/_genai/types/common.py
@@ -13363,6 +13363,24 @@ class _GetRagFileRequestParametersDict(TypedDict, total=False):
 ]
 
 
+class RagFileStatus(_common.BaseModel):
+    """RagFile status."""
+
+    state: Optional[RagFileState] = Field(
+        default=None, description="""The state of the RagFile."""
+    )
+
+
+class RagFileStatusDict(TypedDict, total=False):
+    """RagFile status."""
+
+    state: Optional[RagFileState]
+    """The state of the RagFile."""
+
+
+RagFileStatusOrDict = Union[RagFileStatus, RagFileStatusDict]
+
+
 class DirectUploadSource(_common.BaseModel):
     """The input content is encapsulated and uploaded in the request."""
 
@@ -13675,7 +13693,7 @@ class RagFile(_common.BaseModel):
         default=None,
         description="""Required. The display name of the RagFile. The name can be up to 128 characters long and can consist of any UTF-8 characters.""",
     )
-    file_status: Optional[genai_types.FileStatus] = Field(
+    file_status: Optional[RagFileStatus] = Field(
         default=None, description="""Output only. State of the RagFile."""
     )
     gcs_source: Optional[genai_types.GcsSource] = Field(
@@ -13730,7 +13748,7 @@ class RagFileDict(TypedDict, total=False):
     display_name: Optional[str]
     """Required. The display name of the RagFile. The name can be up to 128 characters long and can consist of any UTF-8 characters."""
 
-    file_status: Optional[genai_types.FileStatusDict]
+    file_status: Optional[RagFileStatusDict]
     """Output only. State of the RagFile."""
 
     gcs_source: Optional[genai_types.GcsSourceDict]