fix: Use the correct enum for RagFile.file_status

PiperOrigin-RevId: 948937373

Author: sararob

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",

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]