apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Simplify the prompt spec invocation

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-04 15:39:55 +01:00
parent 5244e86615
commit 83536b2942
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
trustgraph-flow/trustgraph/retrieval/nlp_query/service.py
View file

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