models.py

from __future__ import annotations

# models.py
from typing import Any, Dict
from pydantic import BaseModel, Field, ConfigDict

class ArchitectureAction(BaseModel):
    model_config = ConfigDict(populate_by_name=True, extra="ignore")
    message: str = Field(default="", alias="value")


class ArchitectureObservation(BaseModel):
    echoed_message: str
    message_length: int
    done: bool
    reward: float
    metadata: Dict[str, Any] = Field(default_factory=dict)