apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-21 19:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fixed prompt service

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-17 22:06:11 +01:00
parent e2d7778cd3
commit 79f8d84e10
2 changed files with 32 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/trustgraph/prompt/template/service.py
View file

@ -33,7 +33,7 @@ class Processor(FlowProcessor):
super(Processor, self).__init__(
**params | {
"id": id,
"config-type": config_key,
"config-type": self.config_key,
"concurrency": concurrency,
}
)

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

@ -21,29 +21,19 @@ class PromptManager:
def __init__(self):
self.config = PromptConfiguration("", {}, {})
self.terms = config.global_terms
self.prompts = config.prompts
try:
self.system_template = ibis.Template(config.system_template)
except:
raise RuntimeError("Error in system template")
self.templates = {}
for k, v in self.prompts.items():
try:
self.templates[k] = ibis.Template(v.template)
except:
raise RuntimeError(f"Error in template: {k}")
if v.terms is None:
v.terms = {}
self.load_config({})
def load_config(self, config):
system = json.loads(config["system"])
ix = json.loads(config["template-index"])
try:
system = json.loads(config["system"])
except:
system = "Be helpful."
try:
ix = json.loads(config["template-index"])
except:
ix = []
prompts = {}
@ -69,6 +59,24 @@ class PromptManager:
prompts
)
self.terms = self.config.global_terms
self.prompts = self.config.prompts
try:
self.system_template = ibis.Template(self.config.system_template)
except:
raise RuntimeError("Error in system template")
self.templates = {}
for k, v in self.prompts.items():
try:
self.templates[k] = ibis.Template(v.template)
except:
raise RuntimeError(f"Error in template: {k}")
if v.terms is None:
v.terms = {}
def parse_json(self, text):
json_match = re.search(r'```(?:json)?(.*?)```', text, re.DOTALL)
@ -80,7 +88,7 @@ class PromptManager:
return json.loads(json_str)
async def render(self, id, input):
def render(self, id, input):
if id not in self.prompts:
raise RuntimeError("ID invalid")
@ -95,6 +103,8 @@ class PromptManager:
print("Invoke...", flush=True)
terms = self.terms | self.prompts[id].terms | input
resp_type = self.prompts[id].response_type
prompt = {