text_to_sql

morph_lib.aiパッケージはmorph-dataパッケージの軽量化に伴い、v0.1.5で廃止されました。

今後、外部アダプターの形式で提供されることを予定しています。

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

OpenAI

# 📦 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)