forked from cleveragents/cleveragents-core
freemo
a6f3bd3841
This commit adds JSON/YAML schemas, Python stubs, fixture examples, and comprehensive test coverage for data contracts across all modules (auth, AI models, contexts, plans, orgs, etc.). Implements automated validation through Behave and
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""Auto-generated Python stubs from Go contracts."""
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Any, Dict, List, Optional
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
|
|
|
|
class StreamMessageType(str, Enum):
|
|
"""Enum for StreamMessageType."""
|
|
|
|
StreamMessageStart = "StreamMessageStart"
|
|
StreamMessageConnectActive = "StreamMessageConnectActive"
|
|
StreamMessageHeartbeat = "StreamMessageHeartbeat"
|
|
StreamMessageReply = "StreamMessageReply"
|
|
StreamMessageDescribing = "StreamMessageDescribing"
|
|
StreamMessageRepliesFinished = "StreamMessageRepliesFinished"
|
|
StreamMessageBuildInfo = "StreamMessageBuildInfo"
|
|
StreamMessagePromptMissingFile = "StreamMessagePromptMissingFile"
|
|
StreamMessageLoadContext = "StreamMessageLoadContext"
|
|
StreamMessageAborted = "StreamMessageAborted"
|
|
StreamMessageFinished = "StreamMessageFinished"
|
|
StreamMessageError = "StreamMessageError"
|
|
StreamMessageMulti = "StreamMessageMulti"
|
|
|
|
|
|
StreamMessageType = str
|
|
|
|
@dataclass
|
|
class BuildInfo:
|
|
"""Data contract for BuildInfo."""
|
|
|
|
path: str
|
|
numTokens: int
|
|
finished: bool
|
|
removed: bool = None
|
|
|
|
@dataclass
|
|
class StreamMessage:
|
|
"""Data contract for StreamMessage."""
|
|
|
|
type_: StreamMessageType
|
|
replyChunk: str = None
|
|
buildInfo: Optional[BuildInfo] = None
|
|
description: Optional[ConvoMessageDescription] = None
|
|
error: Optional[ApiError] = None
|
|
missingFilePath: str = None
|
|
missingFileAutoContext: bool = None
|
|
modelStreamId: str = None
|
|
loadContextFiles: List[str] = None
|
|
initPrompt: str = None
|
|
initReplies: List[str] = None
|
|
initBuildOnly: bool = None
|
|
streamMessages: List[StreamMessage] = None
|