Fix/agent groups broken (#504)

* Fix non-backward-compatible agent changes

* Fix broken agents
This commit is contained in:
cybermaggedon 2025-09-08 21:17:18 +01:00 committed by GitHub
parent f22bf13aa6
commit 5867f45c3a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 31 additions and 11 deletions

View file

@ -29,7 +29,7 @@ def output(text, prefix="> ", width=78):
async def question(
url, question, flow_id, user, collection,
plan=None, state=None, verbose=False
plan=None, state=None, group=None, verbose=False
):
if not url.endswith("/"):
@ -55,19 +55,24 @@ async def question(
async with connect(url) as ws:
req = json.dumps({
req = {
"id": mid,
"service": "agent",
"flow": flow_id,
"request": {
"question": question,
"user": user,
"state": state or "",
"group": [],
"history": []
}
}
# Only add optional fields if they have values
if state is not None:
req["request"]["state"] = state
if group is not None:
req["request"]["group"] = group
})
req = json.dumps(req)
await ws.send(req)
@ -144,6 +149,12 @@ def main():
help=f'Agent initial state (default: unspecified)'
)
parser.add_argument(
'-g', '--group',
nargs='+',
help='Agent tool groups (can specify multiple)'
)
parser.add_argument(
'-v', '--verbose',
action="store_true",
@ -163,6 +174,7 @@ def main():
collection = args.collection,
plan = args.plan,
state = args.state,
group = args.group,
verbose = args.verbose,
)
)