From 48af88e46c2f023780fb7083f7dca17a9e21795d Mon Sep 17 00:00:00 2001 From: = Enea_Gore Date: Wed, 4 Dec 2024 14:30:42 +0100 Subject: [PATCH] fix linting --- .../module_text_llm/retrieval_augmented_generation/__init__.py | 3 ++- .../retrieval_augmented_generation/generate_suggestions.py | 1 - .../prompt_generate_suggestions.py | 1 - 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py index 2054b2be..0dbe53c1 100644 --- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py +++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/__init__.py @@ -2,9 +2,10 @@ from pydantic import Field from typing import Literal from module_text_llm.retrieval_augmented_generation.agents import TutorAgent -tutor = TutorAgent() from module_text_llm.retrieval_augmented_generation.prompt_generate_suggestions import GenerateSuggestionsPrompt +tutor = TutorAgent() + class RAGApproachConfig(ApproachConfig): type: Literal['rag'] = 'rag' generate_suggestions_prompt: GenerateSuggestionsPrompt = Field(default=GenerateSuggestionsPrompt()) diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py index 346b5e37..521eacdd 100644 --- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py +++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/generate_suggestions.py @@ -8,7 +8,6 @@ check_prompt_length_and_omit_features_if_necessary, num_tokens_from_prompt, ) -from athena.text import Exercise, Submission, Feedback from module_text_llm.config import BasicApproachConfig from module_text_llm.helpers.utils import add_sentence_numbers, get_index_range_from_line_range, format_grading_instructions from module_text_llm.basic_approach.prompt_generate_suggestions import AssessmentModel diff --git a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py index 6599fc7d..7988b9b4 100644 --- a/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py +++ b/modules/text/module_text_llm/module_text_llm/retrieval_augmented_generation/prompt_generate_suggestions.py @@ -1,6 +1,5 @@ from pydantic import Field, BaseModel from typing import List, Optional -from pydantic import BaseModel, Field system_message = """\ You are an AI tutor for text assessment at a prestigious university.