atolat30 commited on
Commit
9726556
·
1 Parent(s): d7b1291

Fix type hints for Python 3.9 compatibility

Browse files
Files changed (1) hide show
  1. prompt_manager.py +4 -3
prompt_manager.py CHANGED
@@ -1,5 +1,5 @@
1
  from dataclasses import dataclass
2
- from typing import Dict, List
3
  from logger_config import logger
4
 
5
  @dataclass
@@ -150,8 +150,9 @@ Adapt your communication style to match the user's level of expertise.""",
150
  Try asking something like:
151
  {examples}"""
152
 
153
- def get_templates(self, aspect_name: str | None) -> tuple[str, str]:
154
- """Get system and user templates for an aspect."""
 
155
  if aspect_name is None:
156
  logger.info("No aspect selected, using default template")
157
  info = self.default_aspect
 
1
  from dataclasses import dataclass
2
+ from typing import Dict, List, Union, Tuple
3
  from logger_config import logger
4
 
5
  @dataclass
 
150
  Try asking something like:
151
  {examples}"""
152
 
153
+ def get_templates(self, aspect_name: Union[str, None]) -> Tuple[str, str]:
154
+ """Get system and user templates for the specified aspect."""
155
+ logger.debug(f"Getting templates for aspect: {aspect_name}")
156
  if aspect_name is None:
157
  logger.info("No aspect selected, using default template")
158
  info = self.default_aspect