Skip to content

Comments

Connector Interface & Bigquery Connector#209

Merged
wild-endeavor merged 28 commits intomainfrom
connector
Oct 29, 2025
Merged

Connector Interface & Bigquery Connector#209
wild-endeavor merged 28 commits intomainfrom
connector

Conversation

@pingsutw
Copy link
Member

@pingsutw pingsutw commented Sep 22, 2025

This pull request introduces a new async connector framework for Flyte, enabling extensible and standardized integration with external systems via connectors. It adds core connector abstractions, a registry for managing connectors, a gRPC-based connector service with metrics, and comprehensive tests for the new service

Connector Framework and Registry

  • Introduced the AsyncConnector base class, ResourceMeta, Resource, and ConnectorRegistry in src/flyte/connectors/_connector.py, providing a standardized interface for async connectors and a registry for managing them. Includes serialization/deserialization for resource metadata and support for connector metrics/logs.
  • Exposed AsyncConnector in the public API via src/flyte/connectors/__init__.py.

Connector Service Implementation

  • Added src/flyte/connectors/_server.py implementing gRPC services for connector operations (create, get, delete, metrics, logs), integrating Prometheus metrics for observability and robust error handling.

Testing and Validation

  • Added tests/flyte/connector/test_connector_service.py with a dummy connector and extensive tests for the connector service, validating all major operations and error handling.

Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
@pingsutw pingsutw changed the title [WIP] Connector Interface Connector Interface Oct 19, 2025
@pingsutw pingsutw changed the title Connector Interface Connector Interface & Bigquery Connector Oct 19, 2025
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
pyproject.toml Outdated

[tool.uv.sources]
flyteidl2 = { git = "https://github.com/flyteorg/flyte.git", subdirectory = "gen/python", rev = "v2" }
flyteidl2 = { git = "https://github.com/flyteorg/flyte.git", subdirectory = "gen/python", rev = "16e01e918fe9d4ec73955cbd707435996ddac83f" }
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Need to merge this first. flyteorg/flyte#6659

Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
wild-endeavor
wild-endeavor previously approved these changes Oct 29, 2025
Signed-off-by: Kevin Su <pingsutw@apache.org>
Signed-off-by: Kevin Su <pingsutw@apache.org>
wild-endeavor
wild-endeavor previously approved these changes Oct 29, 2025
Signed-off-by: Kevin Su <pingsutw@apache.org>
@wild-endeavor wild-endeavor merged commit 7130722 into main Oct 29, 2025
25 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants