lockfiles updated

Author: aabmass

instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/Dockerfile

@@ -0,0 +1,6 @@
+FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim
+RUN apt update && apt install -y git
+# Creates the virtualenv and populate uv cache
+RUN DRY_RUN=true uv run --upgrade \
+    'https://raw.githubusercontent.com/aabmass/opentelemetry-python-contrib/refs/heads/vertex-langgraph/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/run_streamlit.py'
+ENTRYPOINT ["uv", "run", "https://raw.githubusercontent.com/aabmass/opentelemetry-python-contrib/refs/heads/vertex-langgraph/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/run_streamlit.py"]

instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/README.md

@@ -1,33 +1,32 @@
-This sample contains part of the LangGraph chatbot demo taken from
-https://python.langchain.com/docs/tutorials/chatbot, running with OTel instrumentation. It
-sends traces and logs to the OTel collector which sends them to GCP. Docker compose wraps
-everything to make it easy to run.
+This sample contains a Streamlit + LangGraph chatbot demo. It sends traces and logs to the GCP
+with the OTLP exporter and opentelemetry-exporter-gcp-logging exporters.
+
+The `run_streamlit.py` script allows you to easily run the sample with auto instrumentation
+enabled and sending telemetry to GCP. It just sets some environment variables and runs with
+`opentelemetry-instrument.
 
 ## Running the example
 
-I recommend running in Cloud Shell, it's super simple. You will see GenAI spans in trace
-explorer right away. Make sure the Vertex and Trace APIs are enabled in the project.
+First, make sure you have `uv` installed: https://docs.astral.sh/uv/getting-started/installation/.
+
+Optionally, set a project with `export GOOGLE_CLOUD_PROJECT=...`. The app respects ADC.
 
-### Cloud Shell or GCE
+### Without cloning
 
 ```sh
-git clone --branch=vertex-langgraph https://github.com/aabmass/opentelemetry-python-contrib.git
-cd opentelemetry-python-contrib/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo
-docker compose up --build --abort-on-container-exit
+uv run --upgrade https://raw.githubusercontent.com/aabmass/opentelemetry-python-contrib/refs/heads/vertex-langgraph/inst rumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/run_streamlit.py
 ```
 
-### Locally with Application Default Credentials
+### With cloned repo
 
 ```sh
 git clone --branch=vertex-langgraph https://github.com/aabmass/opentelemetry-python-contrib.git
 cd opentelemetry-python-contrib/instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo
+uv run run_streamlit.py
+```
+
+### Without auto instrumentation
 
-# Export the credentials to `GOOGLE_APPLICATION_CREDENTIALS` environment variable so it is
-# available inside the docker containers
-export GOOGLE_APPLICATION_CREDENTIALS=$HOME/.config/gcloud/application_default_credentials.json
-# Lets collector read mounted config
-export USERID="$(id -u)"
-# Specify the project ID
-export GOOGLE_CLOUD_PROJECT=<your project id>
-docker compose up --build --abort-on-container-exit
+```sh
+uv run streamlit run src/langgraph_chatbot_demo/langchain_history.py
 ```

instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/app.yaml

@@ -0,0 +1,11 @@
+runtime: custom
+env: flex
+env_variables:
+  STREAMLIT_SERVER_PORT: "8080"
+  PYTHONUNBUFFERED: "1"
+
+network:
+  session_affinity: true
+
+resources:
+  memory_gb: 2

instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/docker-compose.yaml

@@ -11,25 +11,28 @@ dependencies = [
     "langchain-google-vertexai>=2.0.7",
     "langgraph>0.2.27",
     "opentelemetry-distro>=0.50b0",
-    "opentelemetry-exporter-gcp-logging",
+    "opentelemetry-exporter-gcp-logging>=1.9.0a0",
     "opentelemetry-exporter-gcp-trace>=1.8.0",
     "opentelemetry-exporter-otlp-proto-grpc>=1.29.0",
     "opentelemetry-instrumentation-aiohttp-client>=0.50b0",
     "opentelemetry-instrumentation-httpx>=0.50b0",
     "opentelemetry-instrumentation-requests>=0.50b0",
+    "opentelemetry-instrumentation-sqlite3>=0.51b0",
     "opentelemetry-instrumentation-urllib>=0.50b0",
     "opentelemetry-instrumentation-urllib3>=0.50b0",
     "opentelemetry-instrumentation-vertexai",
     "streamlit>=1.41.1",
 ]
 
 [tool.uv.sources]
-opentelemetry-instrumentation-vertexai = { git = "https://github.com/aabmass/opentelemetry-python-contrib.git", subdirectory = "instrumentation-genai/opentelemetry-instrumentation-vertexai", branch = "vertex-langgraph" }
-opentelemetry-exporter-gcp-logging = { git = "https://github.com/DylanRussell/opentelemetry-operations-python.git", subdirectory = "opentelemetry-exporter-gcp-logging", branch = "logging_exporter" }
+opentelemetry-instrumentation-vertexai = { path = "../../" }
 
 [dependency-groups]
 dev = ["ipython>=8.18.1", "ruff>=0.9.2"]
 
+[project.entry-points.opentelemetry_traces_exporter]
+otlp_google_auth = "langgraph_chatbot_demo._otlp_google_auth:OTLPSpanExporterGoogleAuth"
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"

instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo/otel-collector-config.yaml

@@ -1,55 +1,14 @@
 # /// script
 # requires-python = ">=3.13"
 # dependencies = [
-#     "langgraph-chatbot-demo",
+#  "langgraph-chatbot-demo",
 # ]
 #
 # [tool.uv.sources]
-# langgraph-chatbot-demo = { path = "." }
+# langgraph-chatbot-demo = { git = "https://github.com/aabmass/opentelemetry-python-contrib.git", subdirectory = "instrumentation-genai/opentelemetry-instrumentation-vertexai/examples/langgraph-chatbot-demo", branch = "vertex-langgraph" }
+#
 # ///
 
-import os
-import importlib.util
-import subprocess
-
-import google.auth
-import google.auth.transport
-import google.auth.transport.requests
-
-creds, project_id = google.auth.default()
-creds.refresh(google.auth.transport.requests.Request())
-
-
-def setenv_default(k: str, v: str) -> None:
-    if k not in os.environ:
-        os.environ[k] = v
-
-
-setenv_default(
-    "OTEL_EXPORTER_OTLP_ENDPOINT", "https://telemetry.googleapis.com:443"
-)
-setenv_default("OTEL_SERVICE_NAME", "langgraph-chatbot-demo")
-setenv_default("OTEL_PYTHON_LOGGING_AUTO_INSTRUMENTATION_ENABLED", "true")
-setenv_default("OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT", "true")
-setenv_default("OTEL_LOGS_EXPORTER", "gcp_logging")
-setenv_default("OTEL_RESOURCE_ATTRIBUTES", f"gcp.project_id={project_id}")
-setenv_default(
-    "OTEL_EXPORTER_OTLP_HEADERS",
-    f"authorization=Bearer {creds.token},x-goog-user-project={project_id}",
-)
-
-langchain_app_spec = importlib.util.find_spec(
-    "langgraph_chatbot_demo.langchain_history"
-)
-if not (langchain_app_spec and langchain_app_spec.origin):
-    raise Exception("Could not find langchain_history.py")
+from langgraph_chatbot_demo.run_streamlit import run_streamlit
 
-subprocess.run(
-    [
-        "opentelemetry-instrument",
-        "streamlit",
-        "run",
-        langchain_app_spec.origin,
-    ],
-    check=True,
-)
+run_streamlit()