Sync with downstream

road-core · Jan 28, 2025 · 7efabaf · 7efabaf
1 parent 895ebe5
commit 7efabaf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ols/src/query_helpers/question_validator.py b/ols/src/query_helpers/question_validator.py
@@ -9,7 +9,7 @@
 from ols import config
 from ols.app.metrics import TokenMetricUpdater
 from ols.constants import SUBJECT_REJECTED, GenericLLMParameters
-from ols.customize import prompts
+from ols.customize.prompts import prompts QUESTION_VALIDATOR_PROMPT_TEMPLATE
 from ols.src.query_helpers.query_helper import QueryHelper
 from ols.utils.token_handler import TokenHandler
 
@@ -54,7 +54,7 @@ def validate_question(
         logger.info("%s call settings: %s", conversation_id, settings_string)
 
         prompt_instructions = PromptTemplate.from_template(
-            prompts.QUESTION_VALIDATOR_PROMPT_TEMPLATE
+            QUESTION_VALIDATOR_PROMPT_TEMPLATE
         )
 
         bare_llm = self.llm_loader(