morph_lib.aiパッケージはmorph-dataパッケージの軽量化に伴い、v0.1.5で廃止されました。今後、外部アダプターの形式で提供されることを予定しています。
プロンプトを受け取ってSQLを生成します。
# 📦 Package: morph_lib.ai.openai.code_generation
# 🛠️ Function: text_to_sql

def text_to_sql(
    prompt: str,
    api_key: str,
    table_names: List[str],
    model: Optional[str] = "gpt-4o",
    connection: Optional[str] = None,
    schema_name: Optional[str] = None,
) -> TextConversionResponse:

Parameters

OpenAIの場合のパラメータです。上記の関数定義から適宜他のモデルの場合の認証パラメータを設定してください。
prompt
string
required
プロンプト
api_key
string
required
LLMのモデルを使用するためのAPIキー
table_names
list
required
生成するために使用するテーブル名のリスト
model
string
使用するLLMのモデル
connection
string
生成をするために使用するconnection
schema_name
string
生成をするために使用するschema名

Example

from morph_lib.ai.openai.code_generation import text_to_sql
import os

@morph.func
def func_name(context):
    prompt = context.vars["prompt"]
    sql = text_to_sql(
		prompt,
		os.environ["OPENAI_API_KEY"],
		["user"],
		"gpt-4o",
	)
	return execute_sql(sql)