Skip to content

Commit

Permalink
Add LLMPersonaIdentity to __all__
Browse files Browse the repository at this point in the history
Add docstrings to describe `LLMPersona`/`LLMPersonaIdentity`
  • Loading branch information
NeonDaniel committed Jan 28, 2025
1 parent ed08eae commit 3422320
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion neon_data_models/models/api/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,9 @@


class LLMPersonaIdentity(BaseModel):
"""
Defines metadata for a unique persona.
"""
name: str = Field(description="Unique name for this persona")
user_id: Optional[str] = Field(
None, description="`user_id` of the user who created this persona.")
Expand All @@ -47,6 +50,10 @@ def id(self) -> str:


class LLMPersona(LLMPersonaIdentity):
"""
Complete persona definition that may be applied to LLM inference or
committed to a database.
"""
description: Optional[str] = Field(
None, description="Human-readable description of this persona")
system_prompt: str = Field(
Expand Down Expand Up @@ -199,4 +206,5 @@ def validate_inputs(cls, values):
return values


__all__ = [LLMPersona.__name__, LLMRequest.__name__, LLMResponse.__name__]
__all__ = [LLMPersonaIdentity.__name__, LLMPersona.__name__,
LLMRequest.__name__, LLMResponse.__name__]

0 comments on commit 3422320

Please sign in to comment.