| from enum import Enum |
| import openai |
| from pydantic import BaseModel, Field |
| from typing import Optional |
| from typing import List |
| import json |
| import random |
|
|
|
|
| class TackleboxCard(BaseModel): |
| description: str |
| share_with_other_player: bool = False |
|
|
| class TackleboxDeck(BaseModel): |
| cards: List[TackleboxCard] = Field(default_factory=list) |
|
|
| def draw_card(self) -> Optional[TackleboxCard]: |
| |
| if self.cards and len(self.cards) > 0: |
| return self.cards.pop(random.randint(0, len(self.cards) - 1)) |
| return None |
|
|
| class Hand(BaseModel): |
| cards: List[TackleboxCard] = Field(default_factory=list) |
|
|
| def add_card(self, card: TackleboxCard): |
| self.cards.append(card) |
|
|
| class Personality(BaseModel): |
| baseline: str |
| traits: list[str] |
|
|
|
|
| class SpeechStyle(BaseModel): |
| average_sentence_count: int |
| allows_small_action_descriptions: bool |
| forbidden_topics: list[str] |
|
|
|
|
| class Character(BaseModel): |
| name: str |
| age: int |
| gender: str |
| relationship_to_robin: str |
| recent_event: str |
| shared_past_event: str |
| reason_for_fishing_today: Optional[str] = None |
| relationship_type: Optional[str] = None |
| pronouns: Optional[str] = None |
| personality: Personality |
| speech_style: SpeechStyle |
|
|
