Fix structure to work after install

Author: GiovanniGiacometti

benchmark/benchmark_llm_rag_context.py

@@ -45,7 +45,7 @@
 
 sys.path.append(".")
 from guardrails import Guard
-from validator.prompts.prompts import (
+from validator.prompts import (
     Ml3RagContextEvalBasePrompt,
     RagContextUsefulnessPrompt,
 )

tests/test_validator.py

@@ -3,7 +3,7 @@
 import sys
 
 sys.path.append(".")
-from validator.prompts.prompts import RagContextUsefulnessPrompt
+from validator.prompts import RagContextUsefulnessPrompt
 
 from validator.main import MLcubeRagContextValidator
 

validator/main.py

@@ -8,10 +8,10 @@
     register_validator,
 )
 import litellm
-from validator.prompts.prompts import (
+from .prompts import (
     Ml3RagContextEvalBasePrompt,
 )
-from validator.models import RagRatingResponse
+from .models import RagRatingResponse
 from langchain import chat_models
 
 logger = logging.getLogger(__name__)

validator/prompts/prompt_hub.py renamed to validator/prompt_hub.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
-from validator.enums import TextLanguage
-from validator.prompts.prompt_hub import (
+from .enums import TextLanguage
+from .prompt_hub import (
     RAG_CONTEXT_RELEVANCE_PROMPT,
     RAG_CONTEXT_USEFULNESS_PROMPT,
 )