adk-python/contributing/samples/adk_answering_agent/README.md

ADK Answering Agent

The ADK Answering Agent is a Python-based agent designed to help answer questions in GitHub discussions for the google/adk-python repository. It uses a large language model to analyze open discussions, retrieve information from document store, generate response, and post a comment in the github discussion.

This agent can be operated in three distinct modes:

• An interactive mode for local use.
• A batch script mode for oncall use.
• A fully automated GitHub Actions workflow (TBD).

---

Interactive Mode

This mode allows you to run the agent locally to review its recommendations in real-time before any changes are made to your repository's issues.

Features

• Web Interface: The agent's interactive mode can be rendered in a web browser using the ADK's adk web command.
• User Approval: In interactive mode, the agent is instructed to ask for your confirmation before posting a comment to a GitHub issue.
• Question & Answer: You can ask ADK related questions, and the agent will provide answers based on its knowledge on ADK.

Running in Interactive Mode

To run the agent in interactive mode, first set the required environment variables. Then, execute the following command in your terminal:

adk web

This will start a local server and provide a URL to access the agent's web interface in your browser.

---

Batch Script Mode

The answer_discussions.py is created for ADK oncall team to batch process discussions.

Features

• Batch Process: Taken either a number as the count of the recent discussions or a list of discussion numbers, the script will invoke the agent to answer all the specified discussions in one single run.

Running in Interactive Mode

To run the agent in batch script mode, first set the required environment variables. Then, execute the following command in your terminal:

export PYTHONPATH=contributing/samples
python -m adk_answering_agent.answer_discussions --numbers 27 36 # Answer specified discussions

Or python -m adk_answering_agent.answer_discussions --recent 10 to answer the 10 most recent updated discussions.

---

GitHub Workflow Mode

The main.py is reserved for the Github Workflow. The detailed setup for the automatic workflow is TBD.

---

Update the Knowledge Base

The upload_docs_to_vertex_ai_search.py is a script to upload ADK related docs to Vertex AI Search datastore to update the knowledge base. It can be executed with the following command in your terminal:

export PYTHONPATH=contributing/samples # If not already exported
python -m adk_answering_agent.upload_docs_to_vertex_ai_search

Setup and Configuration

Whether running in interactive or workflow mode, the agent requires the following setup.

Dependencies

The agent requires the following Python libraries.

pip install --upgrade pip
pip install google-adk

The agent also requires gcloud login:

gcloud auth application-default login

The upload script requires the following additional Python libraries.

pip install google-cloud-storage google-cloud-discoveryengine

Environment Variables

The following environment variables are required for the agent to connect to the necessary services.

• GITHUB_TOKEN=YOUR_GITHUB_TOKEN: (Required) A GitHub Personal Access Token with issues:write permissions. Needed for both interactive and workflow modes.
• GOOGLE_GENAI_USE_VERTEXAI=TRUE: (Required) Use Google Vertex AI for the authentication.
• GOOGLE_CLOUD_PROJECT=YOUR_PROJECT_ID: (Required) The Google Cloud project ID.
• GOOGLE_CLOUD_LOCATION=LOCATION: (Required) The Google Cloud region.
• VERTEXAI_DATASTORE_ID=YOUR_DATASTORE_ID: (Required) The full Vertex AI datastore ID for the document store (i.e. knowledge base), with the format of projects/{project_number}/locations/{location}/collections/{collection}/dataStores/{datastore_id}.
• OWNER: The GitHub organization or username that owns the repository (e.g., google). Needed for both modes.
• REPO: The name of the GitHub repository (e.g., adk-python). Needed for both modes.
• INTERACTIVE: Controls the agent's interaction mode. For the automated workflow, this is set to 0. For interactive mode, it should be set to 1 or left unset.

The following environment variables are required to upload the docs to update the knowledge base.

• GCS_BUCKET_NAME=YOUR_GCS_BUCKET_NAME: (Required) The name of the GCS bucket to store the documents.
• ADK_DOCS_ROOT_PATH=YOUR_ADK_DOCS_ROOT_PATH: (Required) Path to the root of the downloaded adk-docs repo.
• ADK_PYTHON_ROOT_PATH=YOUR_ADK_PYTHON_ROOT_PATH: (Required) Path to the root of the downloaded adk-python repo.

For local execution in interactive mode, you can place these variables in a .env file in the project's root directory. For the GitHub workflow, they should be configured as repository secrets.