fix adding sqlModel

Author: dori

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
     "pydantic>=2.0.0",
     "jinja2>=3.1.0",
     "litellm>=1.0.0",
+    "sqlmodel>=0.0.24",
 ]
 
 [project.urls]

src/mcp_as_a_judge/db/factory.py

@@ -16,8 +16,8 @@ class DatabaseFactory:
     """Factory for creating database providers."""
 
     _providers: ClassVar[dict[str, type[ConversationHistoryDB]]] = {
-        "in_memory": SQLiteProvider,  # SQLite in-memory (:memory: or empty URL)
-        "sqlite": SQLiteProvider,  # SQLite file-based storage
+        "in_memory": SQLiteProvider,  # SQLModel SQLite in-memory (:memory: or empty URL)
+        "sqlite": SQLiteProvider,  # SQLModel SQLite file-based storage
         # Future providers can be added here:
         # "postgresql": PostgreSQLProvider,
         # "mysql": MySQLProvider,
@@ -52,7 +52,7 @@ def create_provider(cls, config: Config) -> ConversationHistoryDB:
 
         # Create provider instance based on provider type
         if provider_name in ["in_memory", "sqlite"]:
-            # Both use SQLiteProvider but with different URLs
+            # SQLite-based providers (both SQLModel and legacy)
             return provider_class(
                 max_context_records=config.database.max_context_records,
                 retention_days=config.database.record_retention_days,

src/mcp_as_a_judge/db/interface.py

@@ -8,19 +8,21 @@
 from abc import ABC, abstractmethod
 from datetime import datetime
 
-from pydantic import BaseModel, Field
+from sqlmodel import Field, SQLModel
 
 
-class ConversationRecord(BaseModel):
-    """Model for conversation history records."""
+class ConversationRecord(SQLModel, table=True):
+    """SQLModel for conversation history records."""
 
-    id: str
-    session_id: str
+    __tablename__ = "conversation_history"
+
+    id: str | None = Field(default=None, primary_key=True)
+    session_id: str = Field(index=True)
     source: str  # tool name
     input: str  # tool input query
     output: str  # tool output string
     timestamp: datetime = Field(
-        default_factory=datetime.utcnow
+        default_factory=datetime.utcnow, index=True
     )  # when the record was created