diff --git a/trustgraph-flow/pyproject.toml b/trustgraph-flow/pyproject.toml index c4e2651a..3d0e7554 100644 --- a/trustgraph-flow/pyproject.toml +++ b/trustgraph-flow/pyproject.toml @@ -88,12 +88,12 @@ librarian = "trustgraph.librarian:run" mcp-tool = "trustgraph.agent.mcp_tool:run" metering = "trustgraph.metering:run" objects-write-cassandra = "trustgraph.storage.objects.cassandra:run" +objects-query-cassandra = "trustgraph.query.objects.cassandra:run" oe-write-milvus = "trustgraph.storage.object_embeddings.milvus:run" pdf-decoder = "trustgraph.decoding.pdf:run" pdf-ocr-mistral = "trustgraph.decoding.mistral_ocr:run" prompt-template = "trustgraph.prompt.template:run" rev-gateway = "trustgraph.rev_gateway:run" -rows-write-cassandra = "trustgraph.storage.rows.cassandra:run" run-processing = "trustgraph.processing:run" text-completion-azure = "trustgraph.model.text_completion.azure:run" text-completion-azure-openai = "trustgraph.model.text_completion.azure_openai:run" diff --git a/trustgraph-flow/trustgraph/query/objects/cassandra/__init__.py b/trustgraph-flow/trustgraph/query/objects/cassandra/__init__.py index e69de29b..214f7272 100644 --- a/trustgraph-flow/trustgraph/query/objects/cassandra/__init__.py +++ b/trustgraph-flow/trustgraph/query/objects/cassandra/__init__.py @@ -0,0 +1,2 @@ + +from . service import * diff --git a/trustgraph-flow/trustgraph/query/objects/cassandra/__main__.py b/trustgraph-flow/trustgraph/query/objects/cassandra/__main__.py index 0bec8f9d..68122e7f 100644 --- a/trustgraph-flow/trustgraph/query/objects/cassandra/__main__.py +++ b/trustgraph-flow/trustgraph/query/objects/cassandra/__main__.py @@ -2,4 +2,5 @@ from . service import run -run() \ No newline at end of file +run() + diff --git a/trustgraph-flow/trustgraph/query/objects/cassandra/service.py b/trustgraph-flow/trustgraph/query/objects/cassandra/service.py index 44d56cfd..7b3cd9c7 100644 --- a/trustgraph-flow/trustgraph/query/objects/cassandra/service.py +++ b/trustgraph-flow/trustgraph/query/objects/cassandra/service.py @@ -243,11 +243,11 @@ class Processor(FlowProcessor): # Create resolver function for this schema def make_resolver(schema_name, row_schema, graphql_type): async def resolver( - self_query, + self_query: Any, info: Info, collection: str, limit: Optional[int] = 100, - **filters + **filters: Any ) -> List[Any]: # Get the processor instance from context processor = info.context["processor"] @@ -507,4 +507,5 @@ class Processor(FlowProcessor): def run(): """Entry point for objects-query-graphql-cassandra command""" - Processor.launch(default_ident, __doc__) \ No newline at end of file + Processor.launch(default_ident, __doc__) +