omi/backend/models/chat_first_e2e.py at main · ghzhost/omi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
"""Strict, content-free contracts for the local Chat-first E2E harness."""

from enum import Enum

from pydantic import BaseModel, ConfigDict, Field


class ChatFirstE2EFixtureCase(str, Enum):
    enabled = 'enabled'
    question = 'question'
    disabled_control = 'disabled_control'
    unreachable_control = 'unreachable_control'
    cold_start = 'cold_start'


class ChatFirstE2EControlEndpointMode(str, Enum):
    reachable = 'reachable'
    unreachable = 'unreachable'


class ChatFirstE2EExpectedShell(str, Enum):
    legacy = 'legacy'
    chat_first = 'chat_first'


class _StrictHarnessModel(BaseModel):
    model_config = ConfigDict(extra='forbid', frozen=True)


class ChatFirstE2EPrepareRequest(_StrictHarnessModel):
    fixture_case: ChatFirstE2EFixtureCase


class ChatFirstE2EAdvanceRequest(_StrictHarnessModel):
    seconds: int = Field(ge=1, le=172800)


class ChatFirstE2EFixtureSnapshot(_StrictHarnessModel):
    """Shape-only observations. Product text and entity payloads never leave here."""

    fixture_case: ChatFirstE2EFixtureCase
    fixture_revision: int = Field(ge=1)
    expected_shell: ChatFirstE2EExpectedShell
    control_endpoint_mode: ChatFirstE2EControlEndpointMode
    advanced_seconds: int = Field(ge=0)
    materialized_intent_count: int = Field(ge=0)
    ready_intent_count: int = Field(ge=0)
    proactive_intent_count: int = Field(ge=0)
    pending_deferral_count: int = Field(ge=0)


__all__ = [
    'ChatFirstE2EAdvanceRequest',
    'ChatFirstE2EControlEndpointMode',
    'ChatFirstE2EExpectedShell',
    'ChatFirstE2EFixtureCase',
    'ChatFirstE2EFixtureSnapshot',
    'ChatFirstE2EPrepareRequest',
]