Fix non-backward-compatible agent changes

This commit is contained in:
Cyber MacGeddon 2025-09-08 20:49:29 +01:00
parent f22bf13aa6
commit fed54795ac
2 changed files with 14 additions and 6 deletions

View file

@ -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",

View file

@ -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")
)