Fix error, putting flow stuff into API

This commit is contained in:
Cyber MacGeddon 2025-05-03 10:00:16 +01:00
parent 031da4842d
commit b212dd3aae
11 changed files with 155 additions and 134 deletions

View file

@ -30,7 +30,7 @@ def output(text, prefix="> ", width=78):
print(out)
async def question(
url, question, user, collection,
url, question, flow_id, user, collection,
plan=None, state=None, verbose=False
):
@ -60,6 +60,7 @@ async def question(
req = json.dumps({
"id": mid,
"service": "agent",
"flow": flow_id,
"request": {
"question": question,
}
@ -74,6 +75,9 @@ async def question(
obj = json.loads(msg)
if obj["error"]:
raise RuntimeError(obj["error"])
if obj["id"] != mid:
print("Ignore message")
continue
@ -143,12 +147,6 @@ def main():
action="store_true",
help=f'Output thinking/observations'
)
# parser.add_argument(
# '--pulsar-api-key',
# default=default_pulsar_api_key,
# help=f'Pulsar API key',
# )
args = parser.parse_args()
@ -156,13 +154,14 @@ def main():
asyncio.run(
question(
url=args.url,
question=args.question,
user=args.user,
collection=args.collection,
plan=args.plan,
state=args.state,
verbose=args.verbose,
url = args.url,
flow_id = args.flow_id,
question = args.question,
user = args.user,
collection = args.collection,
plan = args.plan,
state = args.state,
verbose = args.verbose,
)
)