models.py

"""
Pydantic models for the OneNote Omi plugin.
"""
from datetime import datetime
from typing import List, Optional, Any, Dict
from pydantic import BaseModel, Field


# Omi Chat Tool Models
class ChatToolRequest(BaseModel):
    """Base request model for Omi chat tools."""
    uid: str
    app_id: str
    tool_name: str


class CreateNoteRequest(ChatToolRequest):
    """Request model for creating a note."""
    title: str = "Untitled Note"
    content: str  # Note content (text/HTML)
    notebook: Optional[str] = None  # Notebook name (fuzzy matched)
    section: Optional[str] = None  # Section name (fuzzy matched)


class SearchNotesRequest(ChatToolRequest):
    """Request model for searching notes."""
    query: str  # Search query
    notebook: Optional[str] = None  # Filter by notebook
    limit: int = 10


class ChatToolResponse(BaseModel):
    """Response model for Omi chat tools."""
    result: Optional[str] = None
    error: Optional[str] = None


# OneNote Data Models
class OneNoteNotebook(BaseModel):
    """OneNote notebook information."""
    id: str
    display_name: str
    created_datetime: Optional[datetime] = None
    last_modified_datetime: Optional[datetime] = None
    is_default: bool = False
    is_shared: bool = False


class OneNoteSection(BaseModel):
    """OneNote section information."""
    id: str
    display_name: str
    notebook_id: Optional[str] = None
    created_datetime: Optional[datetime] = None
    last_modified_datetime: Optional[datetime] = None


class OneNotePage(BaseModel):
    """OneNote page information."""
    id: str
    title: str
    section_id: Optional[str] = None
    created_datetime: Optional[datetime] = None
    last_modified_datetime: Optional[datetime] = None
    content_url: Optional[str] = None
    preview_text: Optional[str] = None


class OneNoteUser(BaseModel):
    """Current OneNote/Microsoft user information."""
    id: str
    display_name: str
    email: Optional[str] = None
    user_principal_name: Optional[str] = None


# Omi Conversation Models (for future memory/webhook integrations)
class TranscriptSegment(BaseModel):
    """Transcript segment from Omi conversation."""
    text: str
    speaker: Optional[str] = "SPEAKER_00"
    is_user: bool
    start: float
    end: float


class Structured(BaseModel):
    """Structured conversation data."""
    title: str
    overview: str
    emoji: str = ""
    category: str = "other"


class Conversation(BaseModel):
    """Omi conversation model."""
    created_at: datetime
    started_at: Optional[datetime] = None
    finished_at: Optional[datetime] = None
    transcript_segments: List[TranscriptSegment] = []
    structured: Structured
    discarded: bool


class EndpointResponse(BaseModel):
    """Standard endpoint response for Omi webhooks."""
    message: str = Field(description="A short message to be sent as notification to the user, if needed.", default="")