エージェンティック AI で RAGを賢く – インデータベースAIエージェント編 #oracle

CREATE OR REPLACE FUNCTION GET_OPTIMAL_AI_PROFILE(
    p_question IN VARCHAR2
) RETURN VARCHAR2
AS
    v_profiles_info CLOB;
    v_chat_prompt CLOB;
    v_response CLOB;
    v_selected_profile VARCHAR2(128);
    v_current_profile VARCHAR2(128);
    v_profile_count NUMBER;
    
BEGIN
    -- 入力チェック
    IF p_question IS NULL OR LENGTH(TRIM(p_question)) = 0 THEN
        RETURN NULL;
    END IF;
    
    -- 現在のプロファイルを保存
    BEGIN
        SELECT DBMS_CLOUD_AI.GET_PROFILE() INTO v_current_profile FROM DUAL;
    EXCEPTION
        WHEN OTHERS THEN
            v_current_profile := NULL;
    END;
    
    -- まず有効なプロファイルが存在するかチェック
    SELECT COUNT(*) INTO v_profile_count 
    FROM USER_CLOUD_AI_PROFILES 
    WHERE STATUS = 'ENABLED';
    
    IF v_profile_count = 0 THEN
        RETURN NULL;
    END IF;
    
    -- USER_CLOUD_AI_PROFILESからプロファイル情報を取得
    BEGIN
        SELECT LISTAGG(
            'プロファイル名: ' || PROFILE_NAME || 
            ', 説明: ' || NVL(DESCRIPTION, '説明なし'),
            CHR(10)
        ) WITHIN GROUP (ORDER BY PROFILE_NAME)
        INTO v_profiles_info
        FROM USER_CLOUD_AI_PROFILES
        WHERE STATUS = 'ENABLED';
        
        -- プロファイルが見つからない場合（LISTAGGがNULLを返すケース）
        IF v_profiles_info IS NULL OR LENGTH(v_profiles_info) = 0 THEN
            -- デフォルトプロファイルを返す
            SELECT PROFILE_NAME 
            INTO v_selected_profile
            FROM USER_CLOUD_AI_PROFILES 
            WHERE STATUS = 'ENABLED' 
            AND ROWNUM = 1;
            RETURN v_selected_profile;
        END IF;
        
    EXCEPTION
        WHEN OTHERS THEN
            -- エラー時もデフォルトプロファイルを返す
            BEGIN
                SELECT PROFILE_NAME 
                INTO v_selected_profile
                FROM USER_CLOUD_AI_PROFILES 
                WHERE STATUS = 'ENABLED' 
                AND ROWNUM = 1;
                RETURN v_selected_profile;
            EXCEPTION
                WHEN OTHERS THEN
                    RETURN NULL;
            END;
    END;
    
    -- SELECT AI chatに渡すプロンプトを構築
    v_chat_prompt := 
        '以下の質問文に最も適したAI Profileを1つだけ選択してください。プロファイル名のみを回答してください。' || CHR(10) || CHR(10) ||
        '【質問文】' || CHR(10) ||
        p_question || CHR(10) || CHR(10) ||
        '【利用可能なAI Profiles】' || CHR(10) ||
        v_profiles_info || CHR(10) || CHR(10) ||
        '回答はプロファイル名のみを返してください（例：SALES_HISTORY_PROFILE）';
    
    -- 適当なプロファイルを一時的に設定して SELECT AI chat を実行
    -- SELECT AI chat では、検索エージェントと異なる LLM を使用したい場合は、別途専用の AI Profile を用意することができます。今回はすべてのAIエージェントで同じ LLM を使用するため簡易的に検索エージェント用の AI Profile を ルーターエージェントでも使用しています。
    DECLARE
        v_temp_profile VARCHAR2(128);
    BEGIN
        SELECT PROFILE_NAME 
        INTO v_temp_profile
        FROM USER_CLOUD_AI_PROFILES 
        WHERE STATUS = 'ENABLED' 
        AND ROWNUM = 1;
        
        -- 一時的にプロファイルを設定
        DBMS_CLOUD_AI.SET_PROFILE(v_temp_profile);
        
        -- SELECT AI chatを実行
        SELECT DBMS_CLOUD_AI.GENERATE(
            prompt => v_chat_prompt,
            action => 'chat'
        ) INTO v_response FROM DUAL;
        
    EXCEPTION
        WHEN OTHERS THEN
            -- エラー時はデフォルトのプロファイルを返す
            SELECT PROFILE_NAME 
            INTO v_selected_profile
            FROM USER_CLOUD_AI_PROFILES 
            WHERE STATUS = 'ENABLED' 
            AND ROWNUM = 1;
            
            -- 元のプロファイルを復元
            IF v_current_profile IS NOT NULL THEN
                BEGIN
                    DBMS_CLOUD_AI.SET_PROFILE(v_current_profile);
                EXCEPTION
                    WHEN OTHERS THEN NULL;
                END;
            END IF;
            
            RETURN v_selected_profile;
    END;
    
    -- レスポンスから選択されたプロファイル名を抽出
    BEGIN
        -- CLOBを文字列に変換（最初の4000文字まで）
        DECLARE
            v_response_str VARCHAR2(4000);
        BEGIN
            v_response_str := DBMS_LOB.SUBSTR(v_response, 4000, 1);
            -- 改行とスペースを整理
            v_response_str := TRIM(REPLACE(REPLACE(v_response_str, CHR(10), ' '), CHR(13), ' '));
            
            -- 有効なプロファイル名かチェック
            SELECT PROFILE_NAME
            INTO v_selected_profile
            FROM USER_CLOUD_AI_PROFILES
            WHERE STATUS = 'ENABLED'
            AND (UPPER(PROFILE_NAME) = UPPER(v_response_str) 
                 OR UPPER(v_response_str) LIKE '%' || UPPER(PROFILE_NAME) || '%'
                 OR UPPER(PROFILE_NAME) LIKE '%' || UPPER(v_response_str) || '%')
            AND ROWNUM = 1;
        END;
        
    EXCEPTION
        WHEN NO_DATA_FOUND THEN
            -- マッチするプロファイルが見つからない場合はデフォルトを使用
            SELECT PROFILE_NAME 
            INTO v_selected_profile
            FROM USER_CLOUD_AI_PROFILES 
            WHERE STATUS = 'ENABLED' 
            AND ROWNUM = 1;
        WHEN OTHERS THEN
            -- その他のエラー時もデフォルトを使用
            SELECT PROFILE_NAME 
            INTO v_selected_profile
            FROM USER_CLOUD_AI_PROFILES 
            WHERE STATUS = 'ENABLED' 
            AND ROWNUM = 1;
    END;
    
    -- 元のプロファイルを復元
    IF v_current_profile IS NOT NULL THEN
        BEGIN
            DBMS_CLOUD_AI.SET_PROFILE(v_current_profile);
        EXCEPTION
            WHEN OTHERS THEN NULL;
        END;
    END IF;
    
    RETURN v_selected_profile;
    
EXCEPTION
    WHEN OTHERS THEN
        -- 元のプロファイルを復元
        IF v_current_profile IS NOT NULL THEN
            BEGIN
                DBMS_CLOUD_AI.SET_PROFILE(v_current_profile);
            EXCEPTION
                WHEN OTHERS THEN NULL;
            END;
        END IF;
        
        -- エラー時はNULLを返す
        RETURN NULL;
END GET_OPTIMAL_AI_PROFILE;