From 2000370544581e153f273e16fc9b951b9eb2c81e Mon Sep 17 00:00:00 2001 From: Alexander Thiess Date: Tue, 16 Sep 2025 15:56:48 +0200 Subject: [PATCH] renamed dataclass --- src/llmutils/prompt_manager.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/llmutils/prompt_manager.py b/src/llmutils/prompt_manager.py index d2fb410..9b28057 100644 --- a/src/llmutils/prompt_manager.py +++ b/src/llmutils/prompt_manager.py @@ -10,7 +10,7 @@ logger = logging.getLogger(__name__) @dataclass -class PromptResult: +class ManagedPrompt: """Smart result object that holds template and can fill variables on demand.""" template: str name: str @@ -181,7 +181,6 @@ class PromptManager: # Default to 'prompts' directory in the current working directory return Path.cwd() / 'prompts' - def _extract_variables(self, template: str) -> Set[str]: """Extract all {{variable}} placeholders from template""" pattern = r'\{\{(\w+)\}\}' @@ -242,7 +241,7 @@ class PromptManager: instance.prompt_variables.clear() @classmethod - def get_prompt(cls, prompt_name: str, **kwargs) -> PromptResult: + def get_prompt(cls, prompt_name: str, **kwargs) -> ManagedPrompt: """ Get a PromptResult that can be filled with variables. @@ -278,7 +277,7 @@ class PromptManager: schema = instance.schemas.get(prompt_name) # Create the result object - result = PromptResult( + result = ManagedPrompt( template=template, name=prompt_name, variables=variables, @@ -330,7 +329,7 @@ class PromptManager: return prompt_name in instance.schemas @classmethod - def get_prompt_with_schema(cls, prompt_name: str, **kwargs) -> PromptResult: + def get_prompt_with_schema(cls, prompt_name: str, **kwargs) -> ManagedPrompt: """ Get both the processed prompt and its schema (if available)