Update __all__ to include LLMPersonaIdentity

NeonGeckoCom · Feb 2, 2025 · 6b0d19f · 6b0d19f
2 parents ed08eae + b7e7beb
commit 6b0d19f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neon_data_models/models/api/llm.py b/neon_data_models/models/api/llm.py
@@ -199,4 +199,4 @@ def validate_inputs(cls, values):
         return values
 
 
-__all__ = [LLMPersona.__name__, LLMRequest.__name__, LLMResponse.__name__]
+__all__ = [LLMPersonaIdentity.__name__, LLMPersona.__name__, LLMRequest.__name__, LLMResponse.__name__]