diff --git a/cloudquery/sdk/internal/servers/plugin_v3/plugin.py b/cloudquery/sdk/internal/servers/plugin_v3/plugin.py index 419bdd3..8bc561f 100644 --- a/cloudquery/sdk/internal/servers/plugin_v3/plugin.py +++ b/cloudquery/sdk/internal/servers/plugin_v3/plugin.py @@ -14,6 +14,7 @@ ) from cloudquery.sdk.plugin.plugin import Plugin, SyncOptions, TableOptions from cloudquery.sdk.schema import tables_to_arrow_schemas, Table +from cloudquery.sdk.schema.table import flatten_tables class PluginServicer(plugin_pb2_grpc.PluginServicer): @@ -39,7 +40,8 @@ def GetTables(self, request: plugin_pb2.GetTables.Request, context): skip_dependent_tables=request.skip_dependent_tables, ) ) - schema = tables_to_arrow_schemas(tables) + flattened_tables = flatten_tables(tables) + schema = tables_to_arrow_schemas(flattened_tables) tablesBytes = [] for s in schema: sink = pa.BufferOutputStream()