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

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