From e3ffcf00079f1c5a631f65da76a60b2643c11e26 Mon Sep 17 00:00:00 2001 From: Zhaoyang-Chu Date: Wed, 10 Sep 2025 19:12:07 +0800 Subject: [PATCH] ruff format fix --- athena/models/__init__.py | 18 ++++++------------ athena/models/memory.py | 9 +++++---- athena/models/message.py | 2 +- 3 files changed, 12 insertions(+), 17 deletions(-) diff --git a/athena/models/__init__.py b/athena/models/__init__.py index 4730359..7d32594 100644 --- a/athena/models/__init__.py +++ b/athena/models/__init__.py @@ -1,22 +1,17 @@ -from .message import Message - from .memory import ( - # BaseModel classes for business logic - MemoryUnit, + Action, MemoryContext, MemoryTimestamp, - Task, - State, - Action, - Result, - # SQLModel classes for database operations + MemoryUnit, MemoryUnitDB, + Result, + State, + Task, ) +from .message import Message __all__ = [ - # Message exports "Message", - # Memory BaseModel exports "MemoryUnit", "MemoryContext", "MemoryTimestamp", @@ -24,6 +19,5 @@ "State", "Action", "Result", - # Memory SQLModel exports "MemoryUnitDB", ] diff --git a/athena/models/memory.py b/athena/models/memory.py index 6f6a8fb..11fe613 100644 --- a/athena/models/memory.py +++ b/athena/models/memory.py @@ -1,10 +1,11 @@ import json -from typing import Literal, Optional, Dict, Any +import uuid +from datetime import datetime +from typing import Any, Dict, Literal, Optional + from pydantic import BaseModel, Field -from sqlmodel import SQLModel, UniqueConstraint from sqlalchemy import Column, DateTime, Text -from datetime import datetime -import uuid +from sqlmodel import SQLModel, UniqueConstraint class MemoryTimestamp(BaseModel): diff --git a/athena/models/message.py b/athena/models/message.py index 5c97cd8..5e4f2c4 100644 --- a/athena/models/message.py +++ b/athena/models/message.py @@ -1,6 +1,6 @@ from typing import Dict -from pydantic import BaseModel, field_validator +from pydantic import BaseModel, field_validator ROLE_ALIASES: Dict[str, str] = { "ai": "assistant",