apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-26 15:55:16 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Agent tweak (#271)

Browse source 
- Removed unused LLM client configuration from agent-manager-react
- Change agent-manager-react template to use prompt-rag instead of
  prompt
- Changed TextCompletion tool to use 'question' instead of 'computation'
  for its parameter.
This commit is contained in:
cybermaggedon 2025-01-20 14:04:48 +00:00 • committed by GitHub
parent acdd3efe51
commit 2aa0586d38
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 5 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

29
trustgraph-flow/trustgraph/agent/react/service.py
View file

@ -14,8 +14,6 @@ from ... schema import AgentRequest, AgentResponse, AgentStep
from ... schema import agent_request_queue, agent_response_queue
from ... schema import prompt_request_queue as pr_request_queue
from ... schema import prompt_response_queue as pr_response_queue
from ... schema import text_completion_request_queue as tc_request_queue
from ... schema import text_completion_response_queue as tc_response_queue
from ... schema import graph_rag_request_queue as gr_request_queue
from ... schema import graph_rag_response_queue as gr_response_queue
from ... clients.prompt_client import PromptClient
@ -133,12 +131,6 @@ class Processor(ConsumerProducer):
prompt_response_queue = params.get(
"prompt_response_queue", pr_response_queue
)
text_completion_request_queue = params.get(
"text_completion_request_queue", tc_request_queue
)
text_completion_response_queue = params.get(
"text_completion_response_queue", tc_response_queue
)
graph_rag_request_queue = params.get(
"graph_rag_request_queue", gr_request_queue
)
@ -155,8 +147,6 @@ class Processor(ConsumerProducer):
"output_schema": AgentResponse,
"prompt_request_queue": prompt_request_queue,
"prompt_response_queue": prompt_response_queue,
"text_completion_request_queue": tc_request_queue,
"text_completion_response_queue": tc_response_queue,
"graph_rag_request_queue": gr_request_queue,
"graph_rag_response_queue": gr_response_queue,
}
@ -169,13 +159,6 @@ class Processor(ConsumerProducer):
pulsar_host = self.pulsar_host
)
self.llm = LlmClient(
subscriber=subscriber,
input_queue=text_completion_request_queue,
output_queue=text_completion_response_queue,
pulsar_host = self.pulsar_host
)
self.graph_rag = GraphRagClient(
subscriber=subscriber,
input_queue=graph_rag_request_queue,
@ -340,18 +323,6 @@ class Processor(ConsumerProducer):
help=f'Prompt response queue (default: {pr_response_queue})',
)
parser.add_argument(
'--text-completion-request-queue',
default=tc_request_queue,
help=f'Text completion request queue (default: {tc_request_queue})',
)
parser.add_argument(
'--text-completion-response-queue',
default=tc_response_queue,
help=f'Text completion response queue (default: {tc_response_queue})',
)
parser.add_argument(
'--graph-rag-request-queue',
default=gr_request_queue,

2
trustgraph-flow/trustgraph/agent/react/tools.py
View file

@ -14,6 +14,6 @@ class TextCompletionImpl:
self.context = context
def invoke(self, **arguments):
return self.context.prompt.request(
"question", { "question": arguments.get("computation") }
"question", { "question": arguments.get("question") }
)