fixes from rebase

Author: yrobla

src/codegate/db/connection.py

@@ -4,7 +4,7 @@
 import sqlite3
 import uuid
 from pathlib import Path
-from typing import Dict, List, Optional, Type
+from typing import List, Optional, Type
 
 import numpy as np
 import sqlite_vec_sl_tmp
@@ -28,7 +28,6 @@
     GetMessagesRow,
     GetWorkspaceByNameConditions,
     Instance,
-    IntermediatePromptWithOutputUsageAlerts,
     MuxRule,
     Output,
     Persona,

src/codegate/db/models.py

@@ -337,3 +337,19 @@ class PersonaDistance(Persona):
     """
 
     distance: float
+
+
+class GetMessagesRow(BaseModel):
+    id: Any
+    timestamp: Any
+    provider: Optional[Any]
+    request: Any
+    type: Any
+    output_id: Optional[Any]
+    output: Optional[Any]
+    output_timestamp: Optional[Any]
+    input_tokens: Optional[int]
+    output_tokens: Optional[int]
+    input_cost: Optional[float]
+    output_cost: Optional[float]
+    alerts: List[Alert] = []