fix import

worker/orca-agent/requirements.txt

@@ -16,5 +16,5 @@ tenacity>=9.0.0
 sqlmodel>=0.0.22
 openai>=0.28.0
 colorama>=0.4.6
-prometheus-swarm>=0.1.0
+prometheus-swarm>=0.1.2
 prometheus-test>=0.1.1

worker/orca-agent/src/database/models.py

@@ -1,11 +1,9 @@
 """Database models."""
 
-from datetime import datetime
-from typing import Optional, List
-from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional
+from sqlmodel import SQLModel, Field
 from sqlalchemy import JSON
 from sqlalchemy import Column
-from prometheus_swarm.database.models import Conversation, Message, Log
 
 
 class Submission(SQLModel, table=True):
@@ -19,4 +17,4 @@ class Submission(SQLModel, table=True):
     repo_urls: Optional[dict] = Field(
         default=None, sa_column=Column(JSON)
     )  # Store as JSON type
-    repo_url: Optional[str] = None
+    repo_url: Optional[str] = None

worker/orca-agent/src/server/routes/submission.py

@@ -1,6 +1,6 @@
 from flask import Blueprint, jsonify
 from prometheus_swarm.database import get_db
-from src.dababase.models import Submission
+from src.database.models import Submission
 import logging
 import os