mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-09-30 21:39:52 +02:00
add env var for prompt template dir; error handling
This commit is contained in:
1 parent
41afb99622
commit
296669ac49
3 files changed
+9
-2
No files matched your search
@@ -13,10 +13,11 @@ class PromptTemplateRepository(AbstractPromptTemplateRepository):
|
||||
return os.path.join(self.templates_dir, template_filename)
|
||||
|
||||
def get(self, id: str) -> PromptTemplate:
|
||||
path = self._create_path_from_id(id)
|
||||
try:
|
||||
return load_prompt(self._create_path_from_id(id))
|
||||
return load_prompt(path)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print(f'Failed to load template from path "{path}":\n{e}')
|
||||
return None
|
||||
|
||||
def add(self, id: str, prompt_template: PromptTemplate) -> None:
|
||||
|
||||
Reference in new issue
Block a user