Skip to content

Commit

Permalink
Refactor model names to match associated SIO event names
Browse files Browse the repository at this point in the history
  • Loading branch information
NeonDaniel committed Jan 9, 2025
1 parent cc1e501 commit 2a9fc22
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions neon_data_models/models/api/klat/socketio.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ class GetTtsRequest(BaseModel):
lang: str = Field(description="BCP-47 Language code associated with `text`")


class UserMessage(BaseModel):
class NewMessage(BaseModel):
cid: str = Field(description="Conversation ID associated with the message")
userID: str = Field(description="User ID associated with the message")
prompt_id: Optional[str] = Field(
Expand Down Expand Up @@ -103,7 +103,7 @@ class PromptData(BaseModel):
request_id = Field()


class GetTranslationsRequest(BaseModel):
class RequestNeonTranslations(BaseModel):
class CidTranslationRequest(BaseModel):
lang: str = Field(description="Translation target language")
source_lang: str = Field(description="Source language of shouts")
Expand All @@ -121,13 +121,19 @@ class AuthExpired(BaseModel):
handler: str = Field(description="Name of the requested function")


class PersonasChanged(BaseModel):
class ConfiguredPersonasChanged(BaseModel):
personas: Dict[str, List[LLMPersona]] = (
Field(description="Dict of LLM name to list of supported personas"))
update_time: int = Field(
description="Unix timestamp when the change occurred")


__all__ = [GetSttRequest.__name__, GetTtsRequest.__name__, UserMessage.__name__,
PromptData.__name__, GetTranslationsRequest.__name__,
AuthExpired.__name__]
class BanSubmindFromConversation(BaseModel):
# This model is used to both ban and revoke a ban
cid: str = Field(description="Conversation ID to (un-)ban submind from")
nick: str = Field(description="Username of the submind to (un-)ban")


__all__ = [GetSttRequest.__name__, GetTtsRequest.__name__, NewMessage.__name__,
PromptData.__name__, RequestNeonTranslations.__name__,
AuthExpired.__name__, BanSubmindFromConversation.__name__]

0 comments on commit 2a9fc22

Please sign in to comment.