Skip to content

Commit 0bd216b

Browse files
authored
Fix docs signature (#27)
1 parent 505f94b commit 0bd216b

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

cloudquery/sdk/serve/plugin.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
import argparse
2-
import structlog
32
from concurrent import futures
43

54
import grpc
5+
import structlog
66
import sys
7-
87
from cloudquery.discovery_v1 import discovery_pb2_grpc
98
from cloudquery.plugin_v3 import plugin_pb2_grpc
9+
1010
from cloudquery.sdk.docs.generator import Generator
1111
from cloudquery.sdk.internal.servers.discovery_v1.discovery import DiscoveryServicer
1212
from cloudquery.sdk.internal.servers.plugin_v3 import PluginServicer
13-
from cloudquery.sdk.plugin.plugin import Plugin
13+
from cloudquery.sdk.plugin.plugin import Plugin, TableOptions
1414

1515
DOC_FORMATS = ["json", "markdown"]
1616

@@ -105,6 +105,13 @@ def stop(self):
105105
def _generate_docs(self, args):
106106
print("Generating docs in format: " + args.format)
107107
generator = Generator(
108-
self._plugin.name(), self._plugin.get_tables(tables=["*"], skip_tables=[])
108+
self._plugin.name(),
109+
self._plugin.get_tables(
110+
options=TableOptions(
111+
tables=["*"],
112+
skip_tables=[],
113+
skip_dependent_tables=False,
114+
)
115+
),
109116
)
110117
generator.generate(args.directory, args.format)

0 commit comments

Comments
 (0)