Spaces:
Sleeping
Sleeping
Fix type hints for Python 3.9 compatibility
Browse files- 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
|
154 |
-
"""Get system and user templates for
|
|
|
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
|