Canner / WrenAI

Wren AI makes your database RAG-ready. Implement Text-to-SQL more accurately and securely.
https://getwren.ai
GNU Affero General Public License v3.0
1.48k stars 119 forks source link

Feature: Support Data Source - Trino #492

Open andreashimin opened 2 weeks ago

andreashimin commented 2 weeks ago

Description This feature allows users to connect Trino as a data source in connection setup. In setup connection UI, it needs to provide these inputs below:

Refer to: https://github.com/Canner/wren-engine/pull/662

Additional context

onlyjackfrost commented 2 weeks ago

Things to do in the Wren-UI Backend will be

Allow create Trino data source from the client-side

Integrate with the Ibis server

ClickHouse data source PR: https://github.com/Canner/WrenAI/pull/481

Things to do in the Wren-Engine will be

Implement (POST) /trino/metadata/tables API

This API should fetch all the tables in the schema specified by the user

Implement (POST) /trino/metadata/constraints API

This API should fetch all the constraints in the schema specified by the user Currently, we use constraints as our recommended relation between models (Note): Since the relation type (one-to-many, many-to-one, ...) can not be detected without fetching the user's data, we hard-code the relation type to one-to-many and let the user change it

ClickHouse data source PR: https://github.com/Canner/wren-engine/pull/648

himanshu634 commented 2 weeks ago

@andreashimin Can you assign it to me?