From 62a3d0175f539eddb219684238d739e97c0177cb Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 16 Sep 2025 21:36:35 +0100 Subject: [PATCH] Added entry point --- trustgraph-flow/pyproject.toml | 1 + .../trustgraph/retrieval/structured_diag/__init__.py | 3 ++- .../trustgraph/retrieval/structured_diag/service.py | 7 ++++++- 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/trustgraph-flow/pyproject.toml b/trustgraph-flow/pyproject.toml index 8b10748b..c1ecd346 100644 --- a/trustgraph-flow/pyproject.toml +++ b/trustgraph-flow/pyproject.toml @@ -96,6 +96,7 @@ prompt-template = "trustgraph.prompt.template:run" rev-gateway = "trustgraph.rev_gateway:run" run-processing = "trustgraph.processing:run" structured-query = "trustgraph.retrieval.structured_query:run" +structured-diag = "trustgraph.retrieval.structured_diag:run" text-completion-azure = "trustgraph.model.text_completion.azure:run" text-completion-azure-openai = "trustgraph.model.text_completion.azure_openai:run" text-completion-claude = "trustgraph.model.text_completion.claude:run" diff --git a/trustgraph-flow/trustgraph/retrieval/structured_diag/__init__.py b/trustgraph-flow/trustgraph/retrieval/structured_diag/__init__.py index 359dbf96..c4e9c7e7 100644 --- a/trustgraph-flow/trustgraph/retrieval/structured_diag/__init__.py +++ b/trustgraph-flow/trustgraph/retrieval/structured_diag/__init__.py @@ -1 +1,2 @@ -# Structured data diagnosis service \ No newline at end of file +# Structured data diagnosis service +from .service import * \ No newline at end of file diff --git a/trustgraph-flow/trustgraph/retrieval/structured_diag/service.py b/trustgraph-flow/trustgraph/retrieval/structured_diag/service.py index db632bcf..75af6dc3 100644 --- a/trustgraph-flow/trustgraph/retrieval/structured_diag/service.py +++ b/trustgraph-flow/trustgraph/retrieval/structured_diag/service.py @@ -386,4 +386,9 @@ class Processor(FlowProcessor): except Exception as e: logger.error(f"Error calling prompt service: {e}", exc_info=True) - return None \ No newline at end of file + return None + + +def run(): + """Entry point for structured-diag command""" + Processor.launch(default_ident, __doc__) \ No newline at end of file