Updated models to handle circular import issues

Author: isivaselvan

src/pytfe/models/project.py

@@ -3,12 +3,16 @@
 
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 from pydantic import BaseModel, ConfigDict, Field
 
-from .agent import AgentPool
 from .common import TagBinding
 from .organization import Organization
 
+if TYPE_CHECKING:
+    from .agent import AgentPool
+
 
 class Project(BaseModel):
     model_config = ConfigDict(populate_by_name=True, validate_by_name=True)

src/pytfe/models/workspace.py

@@ -21,7 +21,6 @@
     UnsupportedOperationsError,
 )
 from ..utils import has_tags_regex_defined, is_valid_workspace_name, valid_string
-from .agent import AgentPool
 from .common import EffectiveTagBinding, Tag, TagBinding
 from .configuration_version import ConfigurationVersion
 from .data_retention_policy import DataRetentionPolicyChoice
@@ -32,6 +31,7 @@
 from .variable import Variable
 
 if TYPE_CHECKING:
+    from .agent import AgentPool
     from .run import Run
 
 
@@ -525,6 +525,7 @@ class VCSRepoOptions(BaseModel):
 def _rebuild_workspace_model() -> None:
     """Rebuild Workspace model to resolve forward references."""
     try:
+        from .agent import AgentPool  # noqa: F401
         from .run import Run  # noqa: F401
         from .task_stage import TaskStage  # noqa: F401
 

src/pytfe/models/workspace_run_task.py

@@ -68,4 +68,4 @@ class WorkspaceRunTaskUpdateOptions(BaseModel):
 
 # WorkspaceRunTask is now fully defined; rebuild RunTask so Pydantic can
 # resolve the forward reference in RunTask.workspace_run_tasks.
-RunTask.model_rebuild()
+RunTask.model_rebuild(raise_errors=False)