refactor predict and parse

ls1intum · Oct 9, 2024 · b92e316 · b92e316
1 parent 8845382
commit b92e316
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modules/text/module_text_llm/module_text_llm/generate_evaluation.py b/modules/text/module_text_llm/module_text_llm/generate_evaluation.py
@@ -9,8 +9,9 @@
 from llm_core.utils.llm_utils import (
     get_chat_prompt_with_formatting_instructions,
     check_prompt_length_and_omit_features_if_necessary,
-    predict_and_parse
 )
+from llm_core.utils.predict_and_parse import predict_and_parse
+
 from module_text_llm.helpers.utils import add_sentence_numbers, get_line_range_from_index_range
 from module_text_llm.prompts.generate_evaluation import system_message, human_message
 

diff --git a/modules/text/module_text_llm/module_text_llm/generate_suggestions.py b/modules/text/module_text_llm/module_text_llm/generate_suggestions.py
@@ -10,8 +10,9 @@
     get_chat_prompt_with_formatting_instructions, 
     check_prompt_length_and_omit_features_if_necessary, 
     num_tokens_from_prompt,
-    predict_and_parse
 )
+from llm_core.utils.predict_and_parse import predict_and_parse
+
 from module_text_llm.helpers.utils import add_sentence_numbers, get_index_range_from_line_range, format_grading_instructions
 
 class FeedbackModel(BaseModel):