Update __all__ to include LLMPersonaIdentity (#14)

NeonGeckoCom · Feb 6, 2025 · 4f0b78b · 4f0b78b
1 parent 5b55b86
commit 4f0b78b
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__]