From 4b102facb134d5438dfa4402d4cdef76ab8aaac5 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 2 Nov 2024 16:04:20 +0000 Subject: [PATCH] Agent schema --- trustgraph-base/trustgraph/schema/agent.py | 35 ++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 trustgraph-base/trustgraph/schema/agent.py diff --git a/trustgraph-base/trustgraph/schema/agent.py b/trustgraph-base/trustgraph/schema/agent.py new file mode 100644 index 00000000..3225985c --- /dev/null +++ b/trustgraph-base/trustgraph/schema/agent.py @@ -0,0 +1,35 @@ + +from pulsar.schema import Record, Bytes, String, Boolean, Array, Map, Integer + +from . topic import topic +from . types import Error, RowSchema + +############################################################################ + +# Prompt services, abstract the prompt generation + +class AgentStep(Record): + thought = String() + action = String() + action_input = String() + observation = String() + +class AgentRequest(Record): + question = String() + plan = String() + history = Array(AgentStep()) + +class AgentResponse(Record): + answer = String() + error = String() + thought = String() + +agent_request_queue = topic( + 'agent', kind='non-persistent', namespace='request' +) +agent_response_queue = topic( + 'agent', kind='non-persistent', namespace='response' +) + +############################################################################ +