Remove if __main__ from main.py

Author: GiovanniGiacometti

validator/main.py

@@ -134,47 +134,3 @@ def validate(self, value: Any, metadata: Dict = {}) -> ValidationResult:
             )
 
         return PassResult()
-
-
-if __name__ == "__main__":
-    from validator.prompts.prompts import RagContextRelevancePrompt
-
-    guard = Guard().use(
-        MLcubeRagContextValidator(
-            rag_context_eval_prompt=RagContextRelevancePrompt(),
-            pass_threshold=1,
-            model_name="gpt-4o-mini",
-            on_fail="noop",  # type: ignore
-            on="prompt",
-        )
-    )
-
-    metadata = {
-        "user_input": "What's the weather in Milan, today?",
-        "retrieved_context": "Milan, what a beautiful day. Sunny and warm.",
-    }
-
-    response = guard(
-        llm_api=openai.chat.completions.create,
-        prompt=metadata["user_input"],
-        model="gpt-4o-mini",
-        max_tokens=1024,
-        temperature=0,
-        metadata=metadata,
-    )
-
-    print(response)
-    assert response.validation_passed, "The validation should pass."  # type: ignore
-
-    metadata["retrieved_context"] = "The capital of Italy is Rome."
-    response = guard(
-        llm_api=openai.chat.completions.create,
-        prompt=metadata["user_input"],
-        model="gpt-4o-mini",
-        max_tokens=1024,
-        temperature=0,
-        metadata=metadata,
-    )
-
-    print(response)
-    assert not response.validation_passed, "The validation shouldn't pass."  # type: ignore