def fetch_all_agent_docs()

in components/webui/src/dpu/api.py [0:0]

• 13 lines of code
• 3 McCabe index (conditional complexity)

def fetch_all_agent_docs() -> list[dict]:
    """List Enterprise Search Corpus"""

    # Create request
    client = document_service_client()
    request = discoveryengine.ListDocumentsRequest(
        parent=client.branch_path(
            PROJECT_ID, LOCATION, SEARCH_DATASTORE_ID, "default_branch"
        )
    )

    # Accumulate the corpus of documents
    corpus = []  # type: ignore
    for doc in client.list_documents(request=request):
        tmp = _document_to_dict(doc)
        if isinstance(tmp, dict):  # mypy: only return valid dict results
            corpus.append(tmp)
    return corpus