Merge pull request #10 from krflorian/feature/stackoverflow

refactoring

app.py

@@ -5,10 +5,10 @@
 from pydantic import BaseModel, Field
 from sentence_transformers import SentenceTransformer, CrossEncoder
 
-from src.objects import Card, Document
-from src.vector_db import VectorDB
-from src.util import load_config
-from src.hallucination import validate_answer
+from objects import Card, Document
+from vector_db import VectorDB
+from util import load_config
+from hallucination import validate_answer
 
 config: dict = load_config(Path("configs/config.yaml"))
 vector_model: SentenceTransformer = SentenceTransformer(

src/etl/create_rules_db.py

@@ -1,11 +1,11 @@
 # %%
-from src.etl.extractors import (
+from etl.extractors import (
     RulesGuruExtractor,
     ComprehensiveRulesExtractor,
     StackExchangeExtractor,
     WikipediaExtractor,
 )
-from src.etl.loaders import DocumentLoader
+from etl.loaders import DocumentLoader
 from pathlib import Path
 import logging
 

src/etl/extractors/comprehensive_rules.py

@@ -7,7 +7,7 @@
 from pathlib import Path
 from datetime import datetime
 
-from src.objects import Document
+from objects import Document
 from .data_extractor import DataExtractor
 
 

src/etl/extractors/data_extractor.py

@@ -4,7 +4,7 @@
 import json
 import logging
 
-from src.objects import Document
+from objects import Document
 
 
 class DataExtractor(BaseModel):

src/etl/extractors/rulesguru.py

@@ -7,7 +7,7 @@
 from pydantic import Field
 from pathlib import Path
 
-from src.objects import Document
+from objects import Document
 from .data_extractor import DataExtractor
 
 

src/etl/extractors/stackexchange.py

@@ -8,7 +8,7 @@
 import logging
 
 from .data_extractor import DataExtractor
-from src.objects import Document
+from objects import Document
 
 
 class StackExchangeExtractor(DataExtractor):

src/etl/loaders/document_loader.py

@@ -3,8 +3,8 @@
 from pathlib import Path
 from sentence_transformers import SentenceTransformer
 
-from src.objects import Document
-from src.vector_db import VectorDB
+from objects import Document
+from vector_db import VectorDB
 
 
 class DocumentLoader(BaseModel):

tests/conftest.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from sentence_transformers import SentenceTransformer
 
-from src.util import load_config
+from util import load_config
 
 
 @pytest.fixture(scope="session")

tests/test_card_db.py

@@ -1,5 +1,5 @@
 import pytest
-from src.vector_db import VectorDB
+from vector_db import VectorDB
 
 
 @pytest.fixture(scope="module")

tests/test_rules_db.py

@@ -1,5 +1,5 @@
 import pytest
-from src.vector_db import VectorDB
+from vector_db import VectorDB
 
 
 @pytest.fixture(scope="module")