Spaces:
Running
Running
File size: 1,859 Bytes
e70c280 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | from threading import Lock
from typing import Optional
class PromptRegistry:
_instance: Optional["PromptRegistry"] = None
_lock: Lock = Lock()
def __init__(self):
self._overrides: dict[str, str] = {}
self._extensions: dict[str, list[str]] = {}
self._defaults: dict[str, str] = {}
@classmethod
def get_instance(cls) -> "PromptRegistry":
if cls._instance is None:
with cls._lock:
if cls._instance is None:
cls._instance = cls()
return cls._instance
def register_default(self, prompt_name: str, default_value: str) -> None:
self._defaults[prompt_name] = default_value
def override(self, prompt_name: str, new_prompt: str) -> None:
self._overrides[prompt_name] = new_prompt
def extend(self, prompt_name: str, suffix: str) -> None:
self._extensions.setdefault(prompt_name, []).append(suffix)
def get(self, prompt_name: str, default: Optional[str] = None) -> str:
if prompt_name in self._overrides:
return self._overrides[prompt_name]
base = default if default is not None else self._defaults.get(prompt_name, "")
extensions = self._extensions.get(prompt_name, [])
if extensions:
return base + "\n" + "\n".join(extensions)
return base
def reset(self, prompt_name: Optional[str] = None) -> None:
if prompt_name:
self._overrides.pop(prompt_name, None)
self._extensions.pop(prompt_name, None)
else:
self._overrides.clear()
self._extensions.clear()
def list_overrides(self) -> list[str]:
return list(self._overrides.keys())
def list_extensions(self) -> list[str]:
return list(self._extensions.keys())
prompt_registry = PromptRegistry.get_instance()
|