create detection models #226

ls1intum · Jan 27, 2025 · a91ac59 · a91ac59
1 parent 6983b99
commit a91ac59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/server/intelligence-service/app/detector/bad_practice_detector.py b/server/intelligence-service/app/detector/bad_practice_detector.py
@@ -1,16 +1,19 @@
 from typing import List
 
-class PullRequest:
+from pydantic import BaseModel
+
+
+class PullRequest(BaseModel):
     id: str
     title: str
     description: str
 
-class Rule:
+class Rule(BaseModel):
     name: str
     description: str
     bad_practice_id: str
 
-class PullRequestWithBadPractices:
+class PullRequestWithBadPractices(BaseModel):
     pull_request_id: str
     bad_practice_ids: List[str]
 

diff --git a/server/intelligence-service/app/routers/detector.py b/server/intelligence-service/app/routers/detector.py
@@ -1,8 +1,7 @@
 from typing import List
 from fastapi import APIRouter
 from pydantic import BaseModel
-
-from app.detector.bad_practice_detector import PullRequestWithBadPractices, PullRequest, Rule, detectbadpractices
+from ..detector.bad_practice_detector import PullRequestWithBadPractices, PullRequest, Rule, detectbadpractices
 
 router = APIRouter(prefix="/detector", tags=["detector"])