From fed54795ac6742922e2cc59f3d6cae32b63d2e7c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 8 Sep 2025 20:49:29 +0100 Subject: [PATCH] Fix non-backward-compatible agent changes --- trustgraph-base/trustgraph/api/flow.py | 16 ++++++++++++---- .../trustgraph/messaging/translators/agent.py | 4 ++-- 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/trustgraph-base/trustgraph/api/flow.py b/trustgraph-base/trustgraph/api/flow.py index 681696c3..74b7a117 100644 --- a/trustgraph-base/trustgraph/api/flow.py +++ b/trustgraph-base/trustgraph/api/flow.py @@ -132,16 +132,24 @@ class FlowInstance: input )["response"] - def agent(self, question, user="trustgraph", state="", group=None, history=None): + def agent(self, question, user="trustgraph", state=None, group=None, history=None): # The input consists of a question and optional context input = { "question": question, "user": user, - "state": state, - "group": group or [], - "history": history or [] } + + # Only include state if it has a value + if state is not None: + input["state"] = state + + # Only include group if it has a value + if group is not None: + input["group"] = group + + # Always include history (empty list if None) + input["history"] = history or [] return self.request( "service/agent", diff --git a/trustgraph-base/trustgraph/messaging/translators/agent.py b/trustgraph-base/trustgraph/messaging/translators/agent.py index 4408fea3..d6ce8bbb 100644 --- a/trustgraph-base/trustgraph/messaging/translators/agent.py +++ b/trustgraph-base/trustgraph/messaging/translators/agent.py @@ -9,8 +9,8 @@ class AgentRequestTranslator(MessageTranslator): def to_pulsar(self, data: Dict[str, Any]) -> AgentRequest: return AgentRequest( question=data["question"], - state=data.get("state", ""), - group=data.get("group", []), + state=data.get("state", None), + group=data.get("group", None), history=data.get("history", []), user=data.get("user", "trustgraph") )