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)