プロンプトを受け取ってSQLを生成します。

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)