schema2dwh
1.0.0
schema2dwh
由 AI - Google 的 Gemini 提供支援 - 請確保在將其註冊到生產環境之前檢查所有程式碼輸出。另請注意,透過此腳本進行處理將利用 Gemini API,其中資料由 Google 處理。 schema2dwh
是一個開源框架,旨在使用其資訊模式簡化並自動生成基於資料庫的資料模型,利用人工智慧以及您輸入的一些問題,它將快速為您生成一個 SQL DDL 文件,準備創建新資料倉儲背後的骨架。
請參閱下面的範例以了解其工作原理。
my_schema.csv
運行腳本: output_schema.sql
: google-generativeai
)克隆儲存庫:
git clone https://github.com/cqllum/schema2dwh.git
cd schema2dwh
安裝所需的軟體包:
pip install google-generativeai
設定您的 Google Generative AI API 金鑰:將腳本中的佔位符your-api-key
替換為您的實際 Google Generative AI API 金鑰。在這裡產生金鑰:https://aistudio.google.com/app/apikey
替換範例輸入架構檔案 (my_schema.csv):請注意,為了使其盡可能準確,您需要從資料庫中取得資訊架構。在大多數情況下,它就像SELECT * FROM information_schema.columns
一樣簡單 - 否則,請參閱資料庫軟體文件。
運行腳本:
python schema2dwh.py
依照提示操作:該腳本將提示您輸入以下資訊:
輸出:該腳本將產生 DDL SQL 語句並將其儲存到output_schema.sql
。如果無法產生模式,它將提示您提供正確的資訊模式。
configure_genai(api_key)
:使用提供的 API 金鑰配置生成式 AI 模型。create_model()
:使用指定的設定建立並配置生成式 AI 模型。start_chat_session(model)
:使用預先定義的指令啟動與模型的聊天會話。gather_user_input()
:以互動方式收集使用者的必要輸入。format_response(db_software, db_name, industry, case_type, input_schema, additional_value)
:將使用者輸入格式化為聊天會話的回應。save_schema_to_file(output_file, schema_output)
:將產生的模式輸出儲存到指定檔案。main()
:驅動腳本的主函數。