48 lines
977 B
Python
48 lines
977 B
Python
"""Data models for shell-speak."""
|
|
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
|
|
@dataclass
|
|
class CommandPattern:
|
|
"""A pattern for matching natural language to shell commands."""
|
|
name: str
|
|
tool: str
|
|
description: str
|
|
patterns: list[str]
|
|
template: str
|
|
explanation: str = ""
|
|
examples: list[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass
|
|
class CommandMatch:
|
|
"""A match between natural language and a shell command."""
|
|
pattern: CommandPattern
|
|
confidence: float
|
|
matched_query: str
|
|
command: str
|
|
explanation: str
|
|
|
|
|
|
@dataclass
|
|
class HistoryEntry:
|
|
"""An entry in the command history."""
|
|
query: str
|
|
command: str
|
|
tool: str
|
|
timestamp: datetime
|
|
explanation: str = ""
|
|
|
|
|
|
@dataclass
|
|
class Correction:
|
|
"""A user correction for a query."""
|
|
original_query: str
|
|
corrected_command: str
|
|
tool: str
|
|
timestamp: datetime
|
|
explanation: str = ""
|