AI Agent開発日記 2025/01/25

やりたいこと

データストア内の指定ファイルを削除するコードを作成するのと、
データストアのファイル数増加による回答率低下の問題への対処を進める。

データストア内の指定ファイルを削除するコードを作成する

• やりたいこと
  • データストア内の指定ファイルを削除するサンプルコードが見当たらないため、対応するメソッドを調べて作成したい。
• 試したこと
  • 発見したメソッドでデータストアファイル削除コードを作成。
• 結果
  • 作成したコードを使った結果無事動作した。

使用したメソッド

Google Cloud

Class DocumentServiceClient (0.13.5)  |  Python client library  |  Google Cloud

データストアの指定ファイルを削除するコード

from google.api_core.client_options import ClientOptions
from google.cloud import discoveryengine


PROJECT_ID = ""
LOCATION = "global"
DATASTORE_ID = ""
BRANCH_ID = "default_branch"
DOCUMENT_ID = ""

def get_client():
    client_options = (
        ClientOptions(api_endpoint=f"{LOCATION}-discoveryengine.googleapis.com")
        if LOCATION != "global"
        else None
    )
    return discoveryengine.DocumentServiceClient(client_options=client_options)

def delete_document():
    client = get_client()

    document_path = f"projects/{PROJECT_ID}/locations/{LOCATION}/dataStores/{DATASTORE_ID}/branches/{BRANCH_ID}/documents/{DOCUMENT_ID}"

    try:
        client.delete_document(name=document_path)
        print("削除成功")
    except Exception as e:
        print("削除失敗")
        print(e)

if __name__ == "__main__":
    delete_document()

データストアのファイル数増加による回答率低下の問題への対処を進める。

1100個ファイルが入っているデータストアにさらにファイルを追加し、
動作が変わるか検証する。

• 試したこと
  • 100個ずつファイルを追加し、都度エージェントに10回質問して回答率を測定した。
• 結果
  • 1500個まで検証したが、回答率は下がり続けた。

回答が可能な境界線に対して、
ファイル数を何度か調整してもその境界線が変わらないかを確認する。

• 試したこと
  • 回答が可能なラインが変わらないかを検証するために、
    ファイルを追加し削除する作業を5回往復して確認する。
• 結果
  • 回答率は変わらなかった

まとめ

• データストア内の指定ファイルを削除するコードを作成し、正常に動作することを確認した。
• データストアのファイル数が増えると回答率が低下する傾向があることを確認した。
• データストアにファイルを追加・削除しても元の回答率と比べて変化は見られなかった。
• 次回の検証では、回答率低下の問題を解決するために別のアプローチを試す予定。