diff --git a/trustgraph-cli/scripts/tg-invoke-agent b/trustgraph-cli/scripts/tg-invoke-agent index 79760fec..40111fa1 100755 --- a/trustgraph-cli/scripts/tg-invoke-agent +++ b/trustgraph-cli/scripts/tg-invoke-agent @@ -75,7 +75,7 @@ async def question( obj = json.loads(msg) - if obj["error"]: + if "error" in obj: raise RuntimeError(obj["error"]) if obj["id"] != mid: diff --git a/trustgraph-cli/scripts/tg-invoke-llm b/trustgraph-cli/scripts/tg-invoke-llm index 21915a00..a64a8c26 100755 --- a/trustgraph-cli/scripts/tg-invoke-llm +++ b/trustgraph-cli/scripts/tg-invoke-llm @@ -57,7 +57,7 @@ def main(): query( url=args.url, - flow = args.flow_id, + flow_id = args.flow_id, system=args.system[0], prompt=args.prompt[0], )