From 83536b2942ccdf169b68b76957dacddf121a9a1b Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 4 Sep 2025 15:39:55 +0100 Subject: [PATCH] Simplify the prompt spec invocation --- trustgraph-flow/trustgraph/retrieval/nlp_query/service.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py b/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py index 1e962e0a..f5100be3 100644 --- a/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py +++ b/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py @@ -8,10 +8,10 @@ import logging from typing import Dict, Any, Optional, List from ...schema import QuestionToStructuredQueryRequest, QuestionToStructuredQueryResponse -from ...schema import PromptRequest, PromptResponse +from ...schema import PromptRequest from ...schema import Error, RowSchema, Field as SchemaField -from ...base import FlowProcessor, ConsumerSpec, ProducerSpec, ClientSpec +from ...base import FlowProcessor, ConsumerSpec, ProducerSpec, PromptClientSpec # Module logger logger = logging.getLogger(__name__) @@ -57,11 +57,9 @@ class Processor(FlowProcessor): # Client spec for calling prompt service self.register_specification( - ClientSpec( + PromptClientSpec( request_name = "prompt-request", response_name = "prompt-response", - request_schema = PromptRequest, - response_schema = PromptResponse ) )