Skip to content

Commit 4c303d1

Browse files
shawn-yang-googlecopybara-github
authored andcommitted
fix: forward reference resolution in Pydantic schema generation.
FUTURE_COPYBARA_INTEGRATE_REVIEW=#6099 from googleapis:release-please--branches--main 4597d87 PiperOrigin-RevId: 829577742
1 parent ed53c69 commit 4c303d1

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

vertexai/_genai/_agent_engines_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -590,6 +590,11 @@ def _is_pydantic_serializable(param: inspect.Parameter) -> bool:
590590
if param.annotation == inspect.Parameter.empty:
591591
return True
592592

593+
# Forward references can't be resolved by Pydantic if the type is not
594+
# available at runtime (e.g. inside `if TYPE_CHECKING:`).
595+
if "ForwardRef" in repr(param.annotation):
596+
return False
597+
593598
if isinstance(param.annotation, str):
594599
return False
595600

0 commit comments

Comments
 (0)