diff --git a/.github/workflows/core_contrib_test_0.yml b/.github/workflows/core_contrib_test_0.yml
index 67bda629f..aa5ca12b5 100644
--- a/.github/workflows/core_contrib_test_0.yml
+++ b/.github/workflows/core_contrib_test_0.yml
@@ -63,6 +63,50 @@ jobs:
- name: Run tests
run: tox -e py38-test-instrumentation-openai-v2-1 -- -ra
+ py38-test-instrumentation-vertexai-0:
+ name: instrumentation-vertexai-0
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout contrib repo @ SHA - ${{ env.CONTRIB_REPO_SHA }}
+ uses: actions/checkout@v4
+ with:
+ repository: open-telemetry/opentelemetry-python-contrib
+ ref: ${{ env.CONTRIB_REPO_SHA }}
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+ architecture: "x64"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-vertexai-0 -- -ra
+
+ py38-test-instrumentation-vertexai-1:
+ name: instrumentation-vertexai-1
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout contrib repo @ SHA - ${{ env.CONTRIB_REPO_SHA }}
+ uses: actions/checkout@v4
+ with:
+ repository: open-telemetry/opentelemetry-python-contrib
+ ref: ${{ env.CONTRIB_REPO_SHA }}
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+ architecture: "x64"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-vertexai-1 -- -ra
+
py38-test-resource-detector-container:
name: resource-detector-container
runs-on: ubuntu-latest
diff --git a/.github/workflows/lint_0.yml b/.github/workflows/lint_0.yml
index 9d77ef5e2..80c7902e6 100644
--- a/.github/workflows/lint_0.yml
+++ b/.github/workflows/lint_0.yml
@@ -34,6 +34,24 @@ jobs:
- name: Run tests
run: tox -e lint-instrumentation-openai-v2
+ lint-instrumentation-vertexai:
+ name: instrumentation-vertexai
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e lint-instrumentation-vertexai
+
lint-resource-detector-container:
name: resource-detector-container
runs-on: ubuntu-latest
diff --git a/.github/workflows/test_0.yml b/.github/workflows/test_0.yml
index 47c9a19cf..92352959e 100644
--- a/.github/workflows/test_0.yml
+++ b/.github/workflows/test_0.yml
@@ -232,6 +232,186 @@ jobs:
- name: Run tests
run: tox -e pypy3-test-instrumentation-openai-v2-1 -- -ra
+ py38-test-instrumentation-vertexai-0_ubuntu-latest:
+ name: instrumentation-vertexai-0 3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-vertexai-0 -- -ra
+
+ py38-test-instrumentation-vertexai-1_ubuntu-latest:
+ name: instrumentation-vertexai-1 3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-vertexai-1 -- -ra
+
+ py39-test-instrumentation-vertexai-0_ubuntu-latest:
+ name: instrumentation-vertexai-0 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-instrumentation-vertexai-0 -- -ra
+
+ py39-test-instrumentation-vertexai-1_ubuntu-latest:
+ name: instrumentation-vertexai-1 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-instrumentation-vertexai-1 -- -ra
+
+ py310-test-instrumentation-vertexai-0_ubuntu-latest:
+ name: instrumentation-vertexai-0 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-instrumentation-vertexai-0 -- -ra
+
+ py310-test-instrumentation-vertexai-1_ubuntu-latest:
+ name: instrumentation-vertexai-1 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-instrumentation-vertexai-1 -- -ra
+
+ py311-test-instrumentation-vertexai-0_ubuntu-latest:
+ name: instrumentation-vertexai-0 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-instrumentation-vertexai-0 -- -ra
+
+ py311-test-instrumentation-vertexai-1_ubuntu-latest:
+ name: instrumentation-vertexai-1 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-instrumentation-vertexai-1 -- -ra
+
+ py312-test-instrumentation-vertexai-0_ubuntu-latest:
+ name: instrumentation-vertexai-0 3.12 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py312-test-instrumentation-vertexai-0 -- -ra
+
+ py312-test-instrumentation-vertexai-1_ubuntu-latest:
+ name: instrumentation-vertexai-1 3.12 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py312-test-instrumentation-vertexai-1 -- -ra
+
py38-test-resource-detector-container_ubuntu-latest:
name: resource-detector-container 3.8 Ubuntu
runs-on: ubuntu-latest
@@ -4335,183 +4515,3 @@ jobs:
- name: Run tests
run: tox -e py38-test-instrumentation-mysqlclient -- -ra
-
- py39-test-instrumentation-mysqlclient_ubuntu-latest:
- name: instrumentation-mysqlclient 3.9 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.9
- uses: actions/setup-python@v5
- with:
- python-version: "3.9"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py39-test-instrumentation-mysqlclient -- -ra
-
- py310-test-instrumentation-mysqlclient_ubuntu-latest:
- name: instrumentation-mysqlclient 3.10 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.10
- uses: actions/setup-python@v5
- with:
- python-version: "3.10"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py310-test-instrumentation-mysqlclient -- -ra
-
- py311-test-instrumentation-mysqlclient_ubuntu-latest:
- name: instrumentation-mysqlclient 3.11 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.11
- uses: actions/setup-python@v5
- with:
- python-version: "3.11"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py311-test-instrumentation-mysqlclient -- -ra
-
- py312-test-instrumentation-mysqlclient_ubuntu-latest:
- name: instrumentation-mysqlclient 3.12 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.12
- uses: actions/setup-python@v5
- with:
- python-version: "3.12"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py312-test-instrumentation-mysqlclient -- -ra
-
- pypy3-test-instrumentation-mysqlclient_ubuntu-latest:
- name: instrumentation-mysqlclient pypy-3.8 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python pypy-3.8
- uses: actions/setup-python@v5
- with:
- python-version: "pypy-3.8"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e pypy3-test-instrumentation-mysqlclient -- -ra
-
- py38-test-instrumentation-psycopg2_ubuntu-latest:
- name: instrumentation-psycopg2 3.8 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.8
- uses: actions/setup-python@v5
- with:
- python-version: "3.8"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py38-test-instrumentation-psycopg2 -- -ra
-
- py39-test-instrumentation-psycopg2_ubuntu-latest:
- name: instrumentation-psycopg2 3.9 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.9
- uses: actions/setup-python@v5
- with:
- python-version: "3.9"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py39-test-instrumentation-psycopg2 -- -ra
-
- py310-test-instrumentation-psycopg2_ubuntu-latest:
- name: instrumentation-psycopg2 3.10 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.10
- uses: actions/setup-python@v5
- with:
- python-version: "3.10"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py310-test-instrumentation-psycopg2 -- -ra
-
- py311-test-instrumentation-psycopg2_ubuntu-latest:
- name: instrumentation-psycopg2 3.11 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.11
- uses: actions/setup-python@v5
- with:
- python-version: "3.11"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py311-test-instrumentation-psycopg2 -- -ra
-
- py312-test-instrumentation-psycopg2_ubuntu-latest:
- name: instrumentation-psycopg2 3.12 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.12
- uses: actions/setup-python@v5
- with:
- python-version: "3.12"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py312-test-instrumentation-psycopg2 -- -ra
diff --git a/.github/workflows/test_1.yml b/.github/workflows/test_1.yml
index 9c5d48aea..e6aa293f9 100644
--- a/.github/workflows/test_1.yml
+++ b/.github/workflows/test_1.yml
@@ -16,6 +16,186 @@ env:
jobs:
+ py39-test-instrumentation-mysqlclient_ubuntu-latest:
+ name: instrumentation-mysqlclient 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-instrumentation-mysqlclient -- -ra
+
+ py310-test-instrumentation-mysqlclient_ubuntu-latest:
+ name: instrumentation-mysqlclient 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-instrumentation-mysqlclient -- -ra
+
+ py311-test-instrumentation-mysqlclient_ubuntu-latest:
+ name: instrumentation-mysqlclient 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-instrumentation-mysqlclient -- -ra
+
+ py312-test-instrumentation-mysqlclient_ubuntu-latest:
+ name: instrumentation-mysqlclient 3.12 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py312-test-instrumentation-mysqlclient -- -ra
+
+ pypy3-test-instrumentation-mysqlclient_ubuntu-latest:
+ name: instrumentation-mysqlclient pypy-3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python pypy-3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "pypy-3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e pypy3-test-instrumentation-mysqlclient -- -ra
+
+ py38-test-instrumentation-psycopg2_ubuntu-latest:
+ name: instrumentation-psycopg2 3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-psycopg2 -- -ra
+
+ py39-test-instrumentation-psycopg2_ubuntu-latest:
+ name: instrumentation-psycopg2 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-instrumentation-psycopg2 -- -ra
+
+ py310-test-instrumentation-psycopg2_ubuntu-latest:
+ name: instrumentation-psycopg2 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-instrumentation-psycopg2 -- -ra
+
+ py311-test-instrumentation-psycopg2_ubuntu-latest:
+ name: instrumentation-psycopg2 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-instrumentation-psycopg2 -- -ra
+
+ py312-test-instrumentation-psycopg2_ubuntu-latest:
+ name: instrumentation-psycopg2 3.12 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py312-test-instrumentation-psycopg2 -- -ra
+
py38-test-instrumentation-psycopg_ubuntu-latest:
name: instrumentation-psycopg 3.8 Ubuntu
runs-on: ubuntu-latest
@@ -4335,183 +4515,3 @@ jobs:
- name: Run tests
run: tox -e py312-test-instrumentation-asyncio -- -ra
-
- py38-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra 3.8 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.8
- uses: actions/setup-python@v5
- with:
- python-version: "3.8"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py38-test-instrumentation-cassandra -- -ra
-
- py39-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra 3.9 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.9
- uses: actions/setup-python@v5
- with:
- python-version: "3.9"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py39-test-instrumentation-cassandra -- -ra
-
- py310-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra 3.10 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.10
- uses: actions/setup-python@v5
- with:
- python-version: "3.10"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py310-test-instrumentation-cassandra -- -ra
-
- py311-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra 3.11 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.11
- uses: actions/setup-python@v5
- with:
- python-version: "3.11"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py311-test-instrumentation-cassandra -- -ra
-
- py312-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra 3.12 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.12
- uses: actions/setup-python@v5
- with:
- python-version: "3.12"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py312-test-instrumentation-cassandra -- -ra
-
- pypy3-test-instrumentation-cassandra_ubuntu-latest:
- name: instrumentation-cassandra pypy-3.8 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python pypy-3.8
- uses: actions/setup-python@v5
- with:
- python-version: "pypy-3.8"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e pypy3-test-instrumentation-cassandra -- -ra
-
- py38-test-processor-baggage_ubuntu-latest:
- name: processor-baggage 3.8 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.8
- uses: actions/setup-python@v5
- with:
- python-version: "3.8"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py38-test-processor-baggage -- -ra
-
- py39-test-processor-baggage_ubuntu-latest:
- name: processor-baggage 3.9 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.9
- uses: actions/setup-python@v5
- with:
- python-version: "3.9"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py39-test-processor-baggage -- -ra
-
- py310-test-processor-baggage_ubuntu-latest:
- name: processor-baggage 3.10 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.10
- uses: actions/setup-python@v5
- with:
- python-version: "3.10"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py310-test-processor-baggage -- -ra
-
- py311-test-processor-baggage_ubuntu-latest:
- name: processor-baggage 3.11 Ubuntu
- runs-on: ubuntu-latest
- steps:
- - name: Checkout repo @ SHA - ${{ github.sha }}
- uses: actions/checkout@v4
-
- - name: Set up Python 3.11
- uses: actions/setup-python@v5
- with:
- python-version: "3.11"
-
- - name: Install tox
- run: pip install tox
-
- - name: Run tests
- run: tox -e py311-test-processor-baggage -- -ra
diff --git a/.github/workflows/test_2.yml b/.github/workflows/test_2.yml
index c23866ffa..7614c8988 100644
--- a/.github/workflows/test_2.yml
+++ b/.github/workflows/test_2.yml
@@ -16,6 +16,186 @@ env:
jobs:
+ py38-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra 3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-instrumentation-cassandra -- -ra
+
+ py39-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-instrumentation-cassandra -- -ra
+
+ py310-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-instrumentation-cassandra -- -ra
+
+ py311-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-instrumentation-cassandra -- -ra
+
+ py312-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra 3.12 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.12
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.12"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py312-test-instrumentation-cassandra -- -ra
+
+ pypy3-test-instrumentation-cassandra_ubuntu-latest:
+ name: instrumentation-cassandra pypy-3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python pypy-3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "pypy-3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e pypy3-test-instrumentation-cassandra -- -ra
+
+ py38-test-processor-baggage_ubuntu-latest:
+ name: processor-baggage 3.8 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.8
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.8"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py38-test-processor-baggage -- -ra
+
+ py39-test-processor-baggage_ubuntu-latest:
+ name: processor-baggage 3.9 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.9
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.9"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py39-test-processor-baggage -- -ra
+
+ py310-test-processor-baggage_ubuntu-latest:
+ name: processor-baggage 3.10 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.10
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.10"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py310-test-processor-baggage -- -ra
+
+ py311-test-processor-baggage_ubuntu-latest:
+ name: processor-baggage 3.11 Ubuntu
+ runs-on: ubuntu-latest
+ steps:
+ - name: Checkout repo @ SHA - ${{ github.sha }}
+ uses: actions/checkout@v4
+
+ - name: Set up Python 3.11
+ uses: actions/setup-python@v5
+ with:
+ python-version: "3.11"
+
+ - name: Install tox
+ run: pip install tox
+
+ - name: Run tests
+ run: tox -e py311-test-processor-baggage -- -ra
+
py312-test-processor-baggage_ubuntu-latest:
name: processor-baggage 3.12 Ubuntu
runs-on: ubuntu-latest
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/CHANGELOG.md b/instrumentation-genai/opentelemetry-instrumentation-vertexai/CHANGELOG.md
new file mode 100644
index 000000000..4d786c784
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/CHANGELOG.md
@@ -0,0 +1,11 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## Unreleased
+
+- Initial VertexAI instrumentation
+ ([#3123](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3123))
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/LICENSE b/instrumentation-genai/opentelemetry-instrumentation-vertexai/LICENSE
new file mode 100644
index 000000000..261eeb9e9
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/LICENSE
@@ -0,0 +1,201 @@
+ Apache License
+ Version 2.0, January 2004
+ http://www.apache.org/licenses/
+
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+ 1. Definitions.
+
+ "License" shall mean the terms and conditions for use, reproduction,
+ and distribution as defined by Sections 1 through 9 of this document.
+
+ "Licensor" shall mean the copyright owner or entity authorized by
+ the copyright owner that is granting the License.
+
+ "Legal Entity" shall mean the union of the acting entity and all
+ other entities that control, are controlled by, or are under common
+ control with that entity. For the purposes of this definition,
+ "control" means (i) the power, direct or indirect, to cause the
+ direction or management of such entity, whether by contract or
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
+ outstanding shares, or (iii) beneficial ownership of such entity.
+
+ "You" (or "Your") shall mean an individual or Legal Entity
+ exercising permissions granted by this License.
+
+ "Source" form shall mean the preferred form for making modifications,
+ including but not limited to software source code, documentation
+ source, and configuration files.
+
+ "Object" form shall mean any form resulting from mechanical
+ transformation or translation of a Source form, including but
+ not limited to compiled object code, generated documentation,
+ and conversions to other media types.
+
+ "Work" shall mean the work of authorship, whether in Source or
+ Object form, made available under the License, as indicated by a
+ copyright notice that is included in or attached to the work
+ (an example is provided in the Appendix below).
+
+ "Derivative Works" shall mean any work, whether in Source or Object
+ form, that is based on (or derived from) the Work and for which the
+ editorial revisions, annotations, elaborations, or other modifications
+ represent, as a whole, an original work of authorship. For the purposes
+ of this License, Derivative Works shall not include works that remain
+ separable from, or merely link (or bind by name) to the interfaces of,
+ the Work and Derivative Works thereof.
+
+ "Contribution" shall mean any work of authorship, including
+ the original version of the Work and any modifications or additions
+ to that Work or Derivative Works thereof, that is intentionally
+ submitted to Licensor for inclusion in the Work by the copyright owner
+ or by an individual or Legal Entity authorized to submit on behalf of
+ the copyright owner. For the purposes of this definition, "submitted"
+ means any form of electronic, verbal, or written communication sent
+ to the Licensor or its representatives, including but not limited to
+ communication on electronic mailing lists, source code control systems,
+ and issue tracking systems that are managed by, or on behalf of, the
+ Licensor for the purpose of discussing and improving the Work, but
+ excluding communication that is conspicuously marked or otherwise
+ designated in writing by the copyright owner as "Not a Contribution."
+
+ "Contributor" shall mean Licensor and any individual or Legal Entity
+ on behalf of whom a Contribution has been received by Licensor and
+ subsequently incorporated within the Work.
+
+ 2. Grant of Copyright License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ copyright license to reproduce, prepare Derivative Works of,
+ publicly display, publicly perform, sublicense, and distribute the
+ Work and such Derivative Works in Source or Object form.
+
+ 3. Grant of Patent License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ (except as stated in this section) patent license to make, have made,
+ use, offer to sell, sell, import, and otherwise transfer the Work,
+ where such license applies only to those patent claims licensable
+ by such Contributor that are necessarily infringed by their
+ Contribution(s) alone or by combination of their Contribution(s)
+ with the Work to which such Contribution(s) was submitted. If You
+ institute patent litigation against any entity (including a
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
+ or a Contribution incorporated within the Work constitutes direct
+ or contributory patent infringement, then any patent licenses
+ granted to You under this License for that Work shall terminate
+ as of the date such litigation is filed.
+
+ 4. Redistribution. You may reproduce and distribute copies of the
+ Work or Derivative Works thereof in any medium, with or without
+ modifications, and in Source or Object form, provided that You
+ meet the following conditions:
+
+ (a) You must give any other recipients of the Work or
+ Derivative Works a copy of this License; and
+
+ (b) You must cause any modified files to carry prominent notices
+ stating that You changed the files; and
+
+ (c) You must retain, in the Source form of any Derivative Works
+ that You distribute, all copyright, patent, trademark, and
+ attribution notices from the Source form of the Work,
+ excluding those notices that do not pertain to any part of
+ the Derivative Works; and
+
+ (d) If the Work includes a "NOTICE" text file as part of its
+ distribution, then any Derivative Works that You distribute must
+ include a readable copy of the attribution notices contained
+ within such NOTICE file, excluding those notices that do not
+ pertain to any part of the Derivative Works, in at least one
+ of the following places: within a NOTICE text file distributed
+ as part of the Derivative Works; within the Source form or
+ documentation, if provided along with the Derivative Works; or,
+ within a display generated by the Derivative Works, if and
+ wherever such third-party notices normally appear. The contents
+ of the NOTICE file are for informational purposes only and
+ do not modify the License. You may add Your own attribution
+ notices within Derivative Works that You distribute, alongside
+ or as an addendum to the NOTICE text from the Work, provided
+ that such additional attribution notices cannot be construed
+ as modifying the License.
+
+ You may add Your own copyright statement to Your modifications and
+ may provide additional or different license terms and conditions
+ for use, reproduction, or distribution of Your modifications, or
+ for any such Derivative Works as a whole, provided Your use,
+ reproduction, and distribution of the Work otherwise complies with
+ the conditions stated in this License.
+
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
+ any Contribution intentionally submitted for inclusion in the Work
+ by You to the Licensor shall be under the terms and conditions of
+ this License, without any additional terms or conditions.
+ Notwithstanding the above, nothing herein shall supersede or modify
+ the terms of any separate license agreement you may have executed
+ with Licensor regarding such Contributions.
+
+ 6. Trademarks. This License does not grant permission to use the trade
+ names, trademarks, service marks, or product names of the Licensor,
+ except as required for reasonable and customary use in describing the
+ origin of the Work and reproducing the content of the NOTICE file.
+
+ 7. Disclaimer of Warranty. Unless required by applicable law or
+ agreed to in writing, Licensor provides the Work (and each
+ Contributor provides its Contributions) on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ implied, including, without limitation, any warranties or conditions
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+ PARTICULAR PURPOSE. You are solely responsible for determining the
+ appropriateness of using or redistributing the Work and assume any
+ risks associated with Your exercise of permissions under this License.
+
+ 8. Limitation of Liability. In no event and under no legal theory,
+ whether in tort (including negligence), contract, or otherwise,
+ unless required by applicable law (such as deliberate and grossly
+ negligent acts) or agreed to in writing, shall any Contributor be
+ liable to You for damages, including any direct, indirect, special,
+ incidental, or consequential damages of any character arising as a
+ result of this License or out of the use or inability to use the
+ Work (including but not limited to damages for loss of goodwill,
+ work stoppage, computer failure or malfunction, or any and all
+ other commercial damages or losses), even if such Contributor
+ has been advised of the possibility of such damages.
+
+ 9. Accepting Warranty or Additional Liability. While redistributing
+ the Work or Derivative Works thereof, You may choose to offer,
+ and charge a fee for, acceptance of support, warranty, indemnity,
+ or other liability obligations and/or rights consistent with this
+ License. However, in accepting such obligations, You may act only
+ on Your own behalf and on Your sole responsibility, not on behalf
+ of any other Contributor, and only if You agree to indemnify,
+ defend, and hold each Contributor harmless for any liability
+ incurred by, or claims asserted against, such Contributor by reason
+ of your accepting any such warranty or additional liability.
+
+ END OF TERMS AND CONDITIONS
+
+ APPENDIX: How to apply the Apache License to your work.
+
+ To apply the Apache License to your work, attach the following
+ boilerplate notice, with the fields enclosed by brackets "[]"
+ replaced with your own identifying information. (Don't include
+ the brackets!) The text should be enclosed in the appropriate
+ comment syntax for the file format. We also recommend that a
+ file or class name and description of purpose be included on the
+ same "printed page" as the copyright notice for easier
+ identification within third-party archives.
+
+ Copyright [yyyy] [name of copyright owner]
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/README.rst b/instrumentation-genai/opentelemetry-instrumentation-vertexai/README.rst
new file mode 100644
index 000000000..ecb404cb7
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/README.rst
@@ -0,0 +1,79 @@
+OpenTelemetry VertexAI Instrumentation
+====================================
+
+|pypi|
+
+.. |pypi| image:: https://badge.fury.io/py/opentelemetry-instrumentation-vertexai.svg
+ :target: https://pypi.org/project/opentelemetry-instrumentation-vertexai/
+
+This library allows tracing LLM requests and logging of messages made by the
+`VertexAI Python API library `_.
+
+
+Installation
+------------
+
+If your application is already instrumented with OpenTelemetry, add this
+package to your requirements.
+::
+
+ pip install opentelemetry-instrumentation-vertexai
+
+If you don't have an VertexAI application, yet, try our `examples `_.
+
+Check out `zero-code example `_ for a quick start.
+
+Usage
+-----
+
+This section describes how to set up VertexAI instrumentation if you're setting OpenTelemetry up manually.
+Check out the `manual example `_ for more details.
+
+Instrumenting all clients
+*************************
+
+When using the instrumentor, all clients will automatically trace VertexAI chat completion operations.
+You can also optionally capture prompts and completions as log events.
+
+Make sure to configure OpenTelemetry tracing, logging, and events to capture all telemetry emitted by the instrumentation.
+
+.. code-block:: python
+
+ from opentelemetry.instrumentation.vertexai import VertexAIInstrumentor
+ from vertexai.generative_models import GenerativeModel
+
+ VertexAIInstrumentor().instrument()
+
+
+ vertexai.init()
+ model = GenerativeModel("gemini-1.5-flash-002")
+ response = model.generate_content("Write a short poem on OpenTelemetry.")
+
+Enabling message content
+*************************
+
+Message content such as the contents of the prompt, completion, function arguments and return values
+are not captured by default. To capture message content as log events, set the environment variable
+`OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT` to `true`.
+
+Uninstrument
+************
+
+To uninstrument clients, call the uninstrument method:
+
+.. code-block:: python
+
+ from opentelemetry.instrumentation.vertexai import VertexAIInstrumentor
+
+ VertexAIInstrumentor().instrument()
+ # ...
+
+ # Uninstrument all clients
+ VertexAIInstrumentor().uninstrument()
+
+References
+----------
+* `OpenTelemetry VertexAI Instrumentation `_
+* `OpenTelemetry Project `_
+* `OpenTelemetry Python Examples `_
+
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/.env b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/.env
new file mode 100644
index 000000000..9ca033983
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/.env
@@ -0,0 +1,8 @@
+# Uncomment and change to your OTLP endpoint
+# OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4317
+# OTEL_EXPORTER_OTLP_PROTOCOL=grpc
+
+OTEL_SERVICE_NAME=opentelemetry-python-vertexai
+
+# Change to 'false' to hide prompt and completion content
+OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT=true
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/README.rst b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/README.rst
new file mode 100644
index 000000000..ab5e7d1c5
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/README.rst
@@ -0,0 +1,43 @@
+OpenTelemetry VertexAI Instrumentation Example
+============================================
+
+This is an example of how to instrument VertexAI calls when configuring OpenTelemetry SDK and Instrumentations manually.
+
+When `main.py `_ is run, it exports traces and logs to an OTLP
+compatible endpoint. Traces include details such as the model used and the
+duration of the chat request. Logs capture the chat request and the generated
+response, providing a comprehensive view of the performance and behavior of
+your VertexAI requests.
+
+Note: `.env <.env>`_ file configures additional environment variables:
+
+- `OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT=true` configures
+VertexAI instrumentation to capture prompt and completion contents on
+events.
+
+Setup
+-----
+
+An OTLP compatible endpoint should be listening for traces and logs on
+http://localhost:4317. If not, update "OTEL_EXPORTER_OTLP_ENDPOINT" as well.
+
+Next, set up a virtual environment like this:
+
+::
+
+ python3 -m venv .venv
+ source .venv/bin/activate
+ pip install "python-dotenv[cli]"
+ pip install -r requirements.txt
+
+Run
+---
+
+Run the example like this:
+
+::
+
+ dotenv run -- python main.py
+
+You should see a poem generated by VertexAI while traces and logs export to your
+configured observability tool.
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/main.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/main.py
new file mode 100644
index 000000000..5d329aad8
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/main.py
@@ -0,0 +1,47 @@
+# pylint: skip-file
+import vertexai
+from vertexai.generative_models import GenerativeModel
+
+# NOTE: OpenTelemetry Python Logs and Events APIs are in beta
+from opentelemetry import _events, _logs, trace
+from opentelemetry.exporter.otlp.proto.grpc._log_exporter import (
+ OTLPLogExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+ OTLPSpanExporter,
+)
+from opentelemetry.instrumentation.vertexai import VertexAIInstrumentor
+from opentelemetry.sdk._events import EventLoggerProvider
+from opentelemetry.sdk._logs import LoggerProvider
+from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+# configure tracing
+trace.set_tracer_provider(TracerProvider())
+trace.get_tracer_provider().add_span_processor(
+ BatchSpanProcessor(OTLPSpanExporter())
+)
+
+# configure logging and events
+_logs.set_logger_provider(LoggerProvider())
+_logs.get_logger_provider().add_log_record_processor(
+ BatchLogRecordProcessor(OTLPLogExporter())
+)
+_events.set_event_logger_provider(EventLoggerProvider())
+
+# instrument VertexAI
+VertexAIInstrumentor().instrument()
+
+
+def main():
+ vertexai.init()
+ model = GenerativeModel("gemini-1.5-flash-002")
+ chat_completion = model.generate_content(
+ "Write a short poem on OpenTelemetry."
+ )
+ print(chat_completion.text)
+
+
+if __name__ == "__main__":
+ main()
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/requirements.txt b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/requirements.txt
new file mode 100644
index 000000000..aa0203750
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/manual/requirements.txt
@@ -0,0 +1,5 @@
+google-cloud-aiplatform>=1.64
+
+opentelemetry-sdk~=1.29.0
+opentelemetry-exporter-otlp-proto-grpc~=1.29.0
+opentelemetry-instrumentation-vertexai~=2.0b0
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/.env b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/.env
new file mode 100644
index 000000000..8874bc5e2
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/.env
@@ -0,0 +1,13 @@
+# Uncomment and change to your OTLP endpoint
+# OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4317
+# OTEL_EXPORTER_OTLP_PROTOCOL=grpc
+
+OTEL_SERVICE_NAME=opentelemetry-python-vertexai
+
+# Change to 'false' to disable logging
+OTEL_PYTHON_LOGGING_AUTO_INSTRUMENTATION_ENABLED=true
+# Change to 'console' if your OTLP endpoint doesn't support logs
+# TODO: this should not be necessary once https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3042 is released
+OTEL_LOGS_EXPORTER=otlp
+# Change to 'false' to hide prompt and completion content
+OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT=true
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/README.rst b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/README.rst
new file mode 100644
index 000000000..6fe161f82
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/README.rst
@@ -0,0 +1,38 @@
+OpenTelemetry VertexAI Instrumentation Example
+============================================
+
+This is an example of how to instrument VertexAI calls with zero code changes,
+using `opentelemetry-instrument`.
+
+When `main.py `_ is run, it exports traces and logs to an OTLP
+compatible endpoint. Traces include details such as the model used and the
+duration of the chat request. Logs capture the chat request and the generated
+response, providing a comprehensive view of the performance and behavior of
+your VertexAI requests.
+
+Setup
+-----
+
+An OTLP compatible endpoint should be listening for traces and logs on http://localhost:4318.
+If not, update "OTEL_EXPORTER_OTLP_ENDPOINT" as well.
+
+Next, set up a virtual environment like this:
+
+::
+
+ python3 -m venv .venv
+ source .venv/bin/activate
+ pip install "python-dotenv[cli]"
+ pip install -r requirements.txt
+
+Run
+---
+
+Run the example like this:
+
+::
+
+ dotenv run -- opentelemetry-instrument python main.py
+
+You should see a poem generated by VertexAI while traces and logs export to your
+configured observability tool.
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/main.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/main.py
new file mode 100644
index 000000000..0f9086bb8
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/main.py
@@ -0,0 +1,15 @@
+import vertexai
+from vertexai.generative_models import GenerativeModel
+
+
+def main():
+ vertexai.init()
+ model = GenerativeModel("gemini-1.5-flash-002")
+ chat_completion = model.generate_content(
+ "Write a short poem on OpenTelemetry."
+ )
+ print(chat_completion.text)
+
+
+if __name__ == "__main__":
+ main()
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/requirements.txt b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/requirements.txt
new file mode 100644
index 000000000..a44caf976
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/zero-code/requirements.txt
@@ -0,0 +1,6 @@
+google-cloud-aiplatform>=1.64
+
+opentelemetry-sdk~=1.28.2
+opentelemetry-exporter-otlp-proto-http~=1.28.2
+opentelemetry-distro~=0.49b2
+opentelemetry-instrumentation-vertexai~=2.0b0
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/pyproject.toml b/instrumentation-genai/opentelemetry-instrumentation-vertexai/pyproject.toml
new file mode 100644
index 000000000..910ebc232
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/pyproject.toml
@@ -0,0 +1,50 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "opentelemetry-instrumentation-vertexai"
+dynamic = ["version"]
+description = "OpenTelemetry Official VertexAI instrumentation"
+readme = "README.rst"
+license = "Apache-2.0"
+requires-python = ">=3.8"
+authors = [
+ { name = "OpenTelemetry Authors", email = "cncf-opentelemetry-contributors@lists.cncf.io" },
+]
+classifiers = [
+ "Development Status :: 4 - Beta",
+ "Intended Audience :: Developers",
+ "License :: OSI Approved :: Apache Software License",
+ "Programming Language :: Python",
+ "Programming Language :: Python :: 3",
+ "Programming Language :: Python :: 3.8",
+ "Programming Language :: Python :: 3.9",
+ "Programming Language :: Python :: 3.10",
+ "Programming Language :: Python :: 3.11",
+ "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+ "opentelemetry-api ~= 1.28",
+ "opentelemetry-instrumentation ~= 0.49b0",
+ "opentelemetry-semantic-conventions ~= 0.49b0",
+]
+
+[project.optional-dependencies]
+instruments = ["google-cloud-aiplatform >= 1.64"]
+
+[project.entry-points.opentelemetry_instrumentor]
+vertexai = "opentelemetry.instrumentation.vertexai:VertexAIInstrumentor"
+
+[project.urls]
+Homepage = "https://github.com/open-telemetry/opentelemetry-python-contrib/tree/main/instrumentation-genai/opentelemetry-instrumentation-vertexai"
+Repository = "https://github.com/open-telemetry/opentelemetry-python-contrib"
+
+[tool.hatch.version]
+path = "src/opentelemetry/instrumentation/vertexai/version.py"
+
+[tool.hatch.build.targets.sdist]
+include = ["/src", "/tests"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/opentelemetry"]
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/__init__.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/__init__.py
new file mode 100644
index 000000000..b2011513a
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/__init__.py
@@ -0,0 +1,74 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+VertexAI client instrumentation supporting `google-cloud-aiplatform` SDK, it can be enabled by
+using ``VertexAIInstrumentor``.
+
+.. _vertexai: https://pypi.org/project/google-cloud-aiplatform/
+
+Usage
+-----
+
+.. code:: python
+
+ import vertexai
+ from vertexai.generative_models import GenerativeModel
+ from opentelemetry.instrumentation.vertexai import VertexAIInstrumentor
+
+ VertexAIInstrumentor().instrument()
+
+ vertexai.init()
+ model = GenerativeModel("gemini-1.5-flash-002")
+ chat_completion = model.generate_content(
+ "Write a short poem on OpenTelemetry."
+ )
+
+API
+---
+"""
+
+from typing import Collection
+
+from opentelemetry._events import get_event_logger
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from opentelemetry.instrumentation.vertexai.package import _instruments
+from opentelemetry.semconv.schemas import Schemas
+from opentelemetry.trace import get_tracer
+
+
+class VertexAIInstrumentor(BaseInstrumentor):
+ def instrumentation_dependencies(self) -> Collection[str]:
+ return _instruments
+
+ def _instrument(self, **kwargs):
+ """Enable VertexAI instrumentation."""
+ tracer_provider = kwargs.get("tracer_provider")
+ _tracer = get_tracer(
+ __name__,
+ "",
+ tracer_provider,
+ schema_url=Schemas.V1_28_0.value,
+ )
+ event_logger_provider = kwargs.get("event_logger_provider")
+ _event_logger = get_event_logger(
+ __name__,
+ "",
+ schema_url=Schemas.V1_28_0.value,
+ event_logger_provider=event_logger_provider,
+ )
+ # TODO: implemented in later PR
+
+ def _uninstrument(self, **kwargs) -> None:
+ """TODO: implemented in later PR"""
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/package.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/package.py
new file mode 100644
index 000000000..c5f776ac5
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/package.py
@@ -0,0 +1,16 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+_instruments = ("google-cloud-aiplatform >= 1.64",)
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/patch.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/patch.py
new file mode 100644
index 000000000..b0a6f4284
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/patch.py
@@ -0,0 +1,13 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/version.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/version.py
new file mode 100644
index 000000000..5b77207d9
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/version.py
@@ -0,0 +1,15 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__version__ = "2.1b0.dev"
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-0.txt b/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-0.txt
new file mode 100644
index 000000000..3450e980a
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-0.txt
@@ -0,0 +1,57 @@
+annotated-types==0.7.0
+cachetools==5.5.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+Deprecated==1.2.14
+docstring_parser==0.16
+exceptiongroup==1.2.2
+google-api-core==2.23.0
+google-auth==2.36.0
+google-cloud-aiplatform==1.74.0
+google-cloud-bigquery==3.27.0
+google-cloud-core==2.4.1
+google-cloud-resource-manager==1.13.1
+google-cloud-storage==2.19.0
+google-crc32c==1.5.0
+google-resumable-media==2.7.2
+googleapis-common-protos==1.66.0
+grpc-google-iam-v1==0.13.1
+grpcio==1.68.1
+grpcio-status==1.68.1
+idna==3.10
+importlib-metadata==6.11.0
+iniconfig==2.0.0
+multidict==6.1.0
+packaging==24.0
+pluggy==1.5.0
+propcache==0.2.0
+proto-plus==1.25.0
+protobuf==5.29.1
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pydantic==2.8.2
+pydantic_core==2.20.1
+pytest==7.4.4
+pytest-asyncio==0.21.0
+pytest-vcr==1.0.2
+python-dateutil==2.9.0.post0
+PyYAML==6.0.2
+requests==2.32.3
+rsa==4.9
+shapely==2.0.6
+six==1.17.0
+tomli==2.2.1
+typing_extensions==4.12.2
+urllib3==1.26.20
+vcrpy==6.0.2
+wrapt==1.16.0
+yarl==1.15.2
+zipp==3.20.2
+
+# when updating, also update in pyproject.toml
+opentelemetry-api==1.28
+opentelemetry-sdk==1.28
+opentelemetry-semantic-conventions==0.49b0
+opentelemetry-instrumentation==0.49b0
+
+-e instrumentation-genai/opentelemetry-instrumentation-vertexai[instruments]
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-1.txt b/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-1.txt
new file mode 100644
index 000000000..9980561fb
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-1.txt
@@ -0,0 +1,54 @@
+annotated-types==0.7.0
+asgiref==3.8.1
+cachetools==5.5.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+Deprecated==1.2.15
+docstring_parser==0.16
+exceptiongroup==1.2.2
+google-api-core==2.23.0
+google-auth==2.36.0
+google-cloud-aiplatform==1.74.0
+google-cloud-bigquery==3.27.0
+google-cloud-core==2.4.1
+google-cloud-resource-manager==1.13.1
+google-cloud-storage==2.19.0
+google-crc32c==1.5.0
+google-resumable-media==2.7.2
+googleapis-common-protos==1.66.0
+grpc-google-iam-v1==0.13.1
+grpcio==1.68.1
+grpcio-status==1.68.1
+idna==3.10
+importlib_metadata==8.5.0
+iniconfig==2.0.0
+multidict==6.1.0
+packaging==24.2
+pluggy==1.5.0
+propcache==0.2.0
+proto-plus==1.25.0
+protobuf==5.29.1
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pydantic==2.10.3
+pydantic_core==2.27.1
+pytest==7.4.4
+pytest-asyncio==0.21.0
+pytest-vcr==1.0.2
+python-dateutil==2.9.0.post0
+PyYAML==6.0.2
+requests==2.32.3
+rsa==4.9
+shapely==2.0.6
+six==1.17.0
+tomli==2.2.1
+typing_extensions==4.12.2
+urllib3==1.26.20
+vcrpy==6.0.2
+wrapt==1.17.0
+yarl==1.15.2
+zipp==3.20.2
+# test with the latest version of opentelemetry-api, sdk, and semantic conventions
+
+-e opentelemetry-instrumentation
+-e instrumentation-genai/opentelemetry-instrumentation-vertexai[instruments]
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/__init__.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/conftest.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/conftest.py
new file mode 100644
index 000000000..8337188ec
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/conftest.py
@@ -0,0 +1,136 @@
+"""Unit tests configuration module."""
+
+import json
+
+import pytest
+import yaml
+
+from opentelemetry.sdk._events import EventLoggerProvider
+from opentelemetry.sdk._logs import LoggerProvider
+from opentelemetry.sdk._logs.export import (
+ InMemoryLogExporter,
+ SimpleLogRecordProcessor,
+)
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
+ InMemorySpanExporter,
+)
+
+
+@pytest.fixture(scope="function", name="span_exporter")
+def fixture_span_exporter():
+ exporter = InMemorySpanExporter()
+ yield exporter
+
+
+@pytest.fixture(scope="function", name="log_exporter")
+def fixture_log_exporter():
+ exporter = InMemoryLogExporter()
+ yield exporter
+
+
+@pytest.fixture(scope="function", name="tracer_provider")
+def fixture_tracer_provider(span_exporter):
+ provider = TracerProvider()
+ provider.add_span_processor(SimpleSpanProcessor(span_exporter))
+ return provider
+
+
+@pytest.fixture(scope="function", name="event_logger_provider")
+def fixture_event_logger_provider(log_exporter):
+ provider = LoggerProvider()
+ provider.add_log_record_processor(SimpleLogRecordProcessor(log_exporter))
+ event_logger_provider = EventLoggerProvider(provider)
+
+ return event_logger_provider
+
+
+@pytest.fixture(scope="module")
+def vcr_config():
+ return {
+ "filter_headers": [
+ ("cookie", "test_cookie"),
+ ("authorization", "Bearer test_vertexai_api_key"),
+ ("vertexai-organization", "test_vertexai_org_id"),
+ ("vertexai-project", "test_vertexai_project_id"),
+ ],
+ "decode_compressed_response": True,
+ "before_record_response": scrub_response_headers,
+ }
+
+
+class LiteralBlockScalar(str):
+ """Formats the string as a literal block scalar, preserving whitespace and
+ without interpreting escape characters"""
+
+
+def literal_block_scalar_presenter(dumper, data):
+ """Represents a scalar string as a literal block, via '|' syntax"""
+ return dumper.represent_scalar("tag:yaml.org,2002:str", data, style="|")
+
+
+yaml.add_representer(LiteralBlockScalar, literal_block_scalar_presenter)
+
+
+def process_string_value(string_value):
+ """Pretty-prints JSON or returns long strings as a LiteralBlockScalar"""
+ try:
+ json_data = json.loads(string_value)
+ return LiteralBlockScalar(json.dumps(json_data, indent=2))
+ except (ValueError, TypeError):
+ if len(string_value) > 80:
+ return LiteralBlockScalar(string_value)
+ return string_value
+
+
+def convert_body_to_literal(data):
+ """Searches the data for body strings, attempting to pretty-print JSON"""
+ if isinstance(data, dict):
+ for key, value in data.items():
+ # Handle response body case (e.g., response.body.string)
+ if key == "body" and isinstance(value, dict) and "string" in value:
+ value["string"] = process_string_value(value["string"])
+
+ # Handle request body case (e.g., request.body)
+ elif key == "body" and isinstance(value, str):
+ data[key] = process_string_value(value)
+
+ else:
+ convert_body_to_literal(value)
+
+ elif isinstance(data, list):
+ for idx, choice in enumerate(data):
+ data[idx] = convert_body_to_literal(choice)
+
+ return data
+
+
+class PrettyPrintJSONBody:
+ """This makes request and response body recordings more readable."""
+
+ @staticmethod
+ def serialize(cassette_dict):
+ cassette_dict = convert_body_to_literal(cassette_dict)
+ return yaml.dump(
+ cassette_dict, default_flow_style=False, allow_unicode=True
+ )
+
+ @staticmethod
+ def deserialize(cassette_string):
+ return yaml.load(cassette_string, Loader=yaml.Loader)
+
+
+@pytest.fixture(scope="module", autouse=True)
+def fixture_vcr(vcr):
+ vcr.register_serializer("yaml", PrettyPrintJSONBody)
+ return vcr
+
+
+def scrub_response_headers(response):
+ """
+ This scrubs sensitive response headers. Note they are case-sensitive!
+ """
+ response["headers"]["vertexai-organization"] = "test_vertexai_org_id"
+ response["headers"]["Set-Cookie"] = "test_set_cookie"
+ return response
diff --git a/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/test_placeholder.py b/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/test_placeholder.py
new file mode 100644
index 000000000..c910bfa0b
--- /dev/null
+++ b/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/test_placeholder.py
@@ -0,0 +1,20 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# TODO: adapt tests from OpenLLMetry here along with tests from
+# instrumentation-genai/opentelemetry-instrumentation-openai-v2/tests/test_chat_completions.py
+
+
+def test_placeholder():
+ assert True
diff --git a/opentelemetry-instrumentation/src/opentelemetry/instrumentation/bootstrap_gen.py b/opentelemetry-instrumentation/src/opentelemetry/instrumentation/bootstrap_gen.py
index 4e497b820..5825e8f7b 100644
--- a/opentelemetry-instrumentation/src/opentelemetry/instrumentation/bootstrap_gen.py
+++ b/opentelemetry-instrumentation/src/opentelemetry/instrumentation/bootstrap_gen.py
@@ -20,6 +20,10 @@ libraries = [
"library": "openai >= 1.26.0",
"instrumentation": "opentelemetry-instrumentation-openai-v2==2.1b0.dev",
},
+ {
+ "library": "google-cloud-aiplatform >= 1.64",
+ "instrumentation": "opentelemetry-instrumentation-vertexai==2.1b0.dev",
+ },
{
"library": "aio_pika >= 7.2.0, < 10.0.0",
"instrumentation": "opentelemetry-instrumentation-aio-pika==0.51b0.dev",
diff --git a/tox.ini b/tox.ini
index 563f6cc34..369ccaaf7 100644
--- a/tox.ini
+++ b/tox.ini
@@ -11,6 +11,12 @@ envlist =
pypy3-test-instrumentation-openai-v2-{0,1}
lint-instrumentation-openai-v2
+ ; instrumentation-vertexai
+ py3{8,9,10,11,12}-test-instrumentation-vertexai-{0,1}
+ # Disabling pypy3 as shapely does not have wheels and fails to compile
+ # pypy3-test-instrumentation-vertexai-{0,1}
+ lint-instrumentation-vertexai
+
; opentelemetry-resource-detector-container
py3{8,9,10,11,12}-test-resource-detector-container
pypy3-test-resource-detector-container
@@ -418,6 +424,11 @@ deps =
openai-1: -r {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-openai-v2/test-requirements-1.txt
lint-instrumentation-openai-v2: -r {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-openai-v2/test-requirements-0.txt
+ vertexai-0: -r {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-0.txt
+ vertexai-1: {[testenv]test_deps}
+ vertexai-1: -r {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-1.txt
+ lint-instrumentation-vertexai: -r {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-vertexai/test-requirements-0.txt
+
asgi: {[testenv]test_deps}
asgi: -r {toxinidir}/instrumentation/opentelemetry-instrumentation-asgi/test-requirements.txt
@@ -769,6 +780,9 @@ commands =
test-instrumentation-openai-v2: pytest {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-openai-v2/tests {posargs}
lint-instrumentation-openai-v2: sh -c "cd instrumentation-genai && pylint --rcfile ../.pylintrc opentelemetry-instrumentation-openai-v2"
+ test-instrumentation-vertexai: pytest {toxinidir}/instrumentation-genai/opentelemetry-instrumentation-vertexai/tests {posargs}
+ lint-instrumentation-vertexai: sh -c "cd instrumentation-genai && pylint --rcfile ../.pylintrc opentelemetry-instrumentation-vertexai"
+
test-instrumentation-sio-pika: pytest {toxinidir}/instrumentation/opentelemetry-instrumentation-pika/tests {posargs}
lint-instrumentation-sio-pika: sh -c "cd instrumentation && pylint --rcfile ../.pylintrc opentelemetry-instrumentation-pika"