apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-30 09:45:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix: avoid swallowing prompt manager interrupts

Browse source
This commit is contained in:
Jacob Molz 2026-05-26 10:37:21 -04:00
parent f8e8c0e9e6
commit 8a8e496acf
2 changed files with 43 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
trustgraph-flow/trustgraph/template/prompt_manager.py
View file

@ -31,12 +31,12 @@ class PromptManager:
try:
system = json.loads(config["system"])
except:
except (KeyError, TypeError, json.JSONDecodeError):
system = "Be helpful."
try:
ix = json.loads(config["template-index"])
except:
except (KeyError, TypeError, json.JSONDecodeError):
ix = []
prompts = {}
@ -68,8 +68,8 @@ class PromptManager:
try:
self.system_template = ibis.Template(self.config.system_template)
except:
raise RuntimeError("Error in system template")
except Exception as e:
raise RuntimeError(f"Error in system template: {e}")
self.templates = {}
for k, v in self.prompts.items():
@ -136,8 +136,6 @@ class PromptManager:
terms = self.terms | self.prompts[id].terms | input
resp_type = self.prompts[id].response_type
return self.templates[id].render(terms)
async def invoke(self, id, input, llm):
@ -161,7 +159,7 @@ class PromptManager:
if resp_type == "json":
try:
obj = self.parse_json(resp)
except:
except (json.JSONDecodeError, TypeError):
logger.error(f"JSON parse failed: {resp}")
raise RuntimeError("JSON parse fail")
@ -195,4 +193,3 @@ class PromptManager:
return objects
raise RuntimeError(f"Response type {resp_type} not known")