diff --git a/database/autoheal_models.py b/database/autoheal_models.py
index 8fc2f9d0b..4bb914069 100644
--- a/database/autoheal_models.py
+++ b/database/autoheal_models.py
@@ -1,4 +1,4 @@
-from sqlalchemy import Column, Integer, String, DateTime, Text, Boolean, ForeignKey, Index
+from sqlalchemy import Column, Integer, String, DateTime, Text, Boolean, ForeignKey, Index, Float
 from sqlalchemy.orm import relationship
 from database.models import Base
 from datetime import datetime
diff --git a/services/ai_orchestrator.py b/services/ai_orchestrator.py
index c69120d82..de48971b4 100644
--- a/services/ai_orchestrator.py
+++ b/services/ai_orchestrator.py
@@ -1,3 +1,4 @@
+import asyncio
 import logging
 from typing import Any, Dict, Optional