Fix pydantic issue (elevenlabs#541)

PaulAsjes · web-flow · commit a5fa070c62b0 · 2025-05-07T15:45:50.000+02:00
diff --git a/.fernignore b/.fernignore
@@ -7,6 +7,7 @@ src/elevenlabs/conversational_ai/default_audio_interface.py
 src/elevenlabs/play.py
 src/elevenlabs/realtime_tts.py
 src/elevenlabs/types/get_agent_response_model.py
+src/elevenlabs/types/prompt_agent.py
 
 # Ignore CI files
 .github/
diff --git a/src/elevenlabs/types/prompt_agent.py b/src/elevenlabs/types/prompt_agent.py
@@ -10,7 +10,7 @@
 from .knowledge_base_locator import KnowledgeBaseLocator
 from .custom_llm import CustomLlm
 from .rag_config import RagConfig
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, update_forward_refs
 
 
 class PromptAgent(UncheckedBaseModel):
@@ -72,3 +72,10 @@ class Config:
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+from .array_json_schema_property_output import ArrayJsonSchemaPropertyOutput  # noqa: E402
+
+update_forward_refs(ArrayJsonSchemaPropertyOutput)
+
+if IS_PYDANTIC_V2:
+    PromptAgent.model_rebuild()
