diff --git a/trustgraph-base/trustgraph/messaging/__init__.py b/trustgraph-base/trustgraph/messaging/__init__.py index 6b1aedd2..0c805967 100644 --- a/trustgraph-base/trustgraph/messaging/__init__.py +++ b/trustgraph-base/trustgraph/messaging/__init__.py @@ -24,6 +24,7 @@ from .translators.embeddings_query import ( from .translators.objects_query import ObjectsQueryRequestTranslator, ObjectsQueryResponseTranslator from .translators.nlp_query import QuestionToStructuredQueryRequestTranslator, QuestionToStructuredQueryResponseTranslator from .translators.structured_query import StructuredQueryRequestTranslator, StructuredQueryResponseTranslator +from .translators.diagnosis import StructuredDataDiagnosisRequestTranslator, StructuredDataDiagnosisResponseTranslator # Register all service translators TranslatorRegistry.register_service( @@ -123,11 +124,17 @@ TranslatorRegistry.register_service( ) TranslatorRegistry.register_service( - "structured-query", - StructuredQueryRequestTranslator(), + "structured-query", + StructuredQueryRequestTranslator(), StructuredQueryResponseTranslator() ) +TranslatorRegistry.register_service( + "structured-diag", + StructuredDataDiagnosisRequestTranslator(), + StructuredDataDiagnosisResponseTranslator() +) + # Register single-direction translators for document loading TranslatorRegistry.register_request("document", DocumentTranslator()) TranslatorRegistry.register_request("text-document", TextDocumentTranslator()) diff --git a/trustgraph-base/trustgraph/messaging/translators/__init__.py b/trustgraph-base/trustgraph/messaging/translators/__init__.py index 1bed3020..9ce2730e 100644 --- a/trustgraph-base/trustgraph/messaging/translators/__init__.py +++ b/trustgraph-base/trustgraph/messaging/translators/__init__.py @@ -18,3 +18,4 @@ from .embeddings_query import ( GraphEmbeddingsRequestTranslator, GraphEmbeddingsResponseTranslator ) from .objects_query import ObjectsQueryRequestTranslator, ObjectsQueryResponseTranslator +from .diagnosis import StructuredDataDiagnosisRequestTranslator, StructuredDataDiagnosisResponseTranslator