AI Agent開発日記 2025/01/25

2025年2月2日

やりたいこと

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

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

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

使用したメソッド

Google Cloud Documentation

Class DocumentServiceClient (0.15.0) | Python client libraries | Google Cloud Documentation

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

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()