├── tests
    ├── __init__.py
    ├── test_apmtelemetry.py
    ├── integration_snapshots
    │   ├── test_trace_stats_tracestats.json
    │   ├── test_single_trace.json
    │   ├── test_trace_distributed_propagated.json
    │   ├── test_trace_missing_received.json
    │   ├── test_multi_trace.json
    │   ├── test_trace_distributed_same_payload.json
    │   └── test_trace_stats.json
    ├── test_client.py
    ├── test_tracestats.py
    ├── test_tracerflare.py
    ├── test_container.py
    └── trace_utils.py
├── ddapm_test_agent
    ├── context.py
    ├── __init__.py
    ├── integration.py
    ├── templates
    │   ├── sessions.html
    │   ├── traces.html
    │   ├── base.html
    │   ├── macros.html
    │   ├── session_detail.html
    │   ├── trace_detail.html
    │   ├── snapshots.html
    │   └── dashboard.html
    ├── apmtelemetry.py
    ├── tracerflare.py
    ├── tracestats.py
    ├── cmd.py
    ├── logs.py
    ├── tracestats_snapshot.py
    ├── remoteconfig.py
    ├── fmt.py
    └── metrics.py
├── .dockerignore
├── releasenotes
    ├── config.yaml
    ├── notes
    │   ├── wheels-68a7dc2f4b00e0c4.yaml
    │   ├── python310-025fbc190363de98.yaml
    │   ├── v0.7-traces-7ea3ebbed7502d01.yaml
    │   ├── args-88f664c83a96075d.yaml
    │   ├── info-34dcd9cbcc382487.yaml
    │   ├── remove-nix-ec1c4aca7dc002c4.yaml
    │   ├── add-datadog-proxy-7e2e608c101a564b.yaml
    │   ├── add-python-3.13-support-ae7b96f4a10f20f4.yaml
    │   ├── add-span-events-306e697168373899.yaml
    │   ├── support-ant-vcr-1eff9a40caa0aba0.yaml
    │   ├── info-drop-p0s-c207889f590340fb.yaml
    │   ├── span-parent-id-a4d7d63fa623361c.yaml
    │   ├── vcr-support-querystring-b06555fb9a7c90ae.yaml
    │   ├── fix-fmt-logging-6a323fd8d5d4eb28.yaml
    │   ├── image-optimization-d614eccb59441217.yaml
    │   ├── new-process-id-tag-f0ab0daf652e05cc.yaml
    │   ├── python310-docker-image-d4e329cfecf6510e.yaml
    │   ├── remove-old-pythons-4667b182aacf0ce6.yaml
    │   ├── reuse-session-1670aa569907cf68.yaml
    │   ├── rc-encoding-length-fix-cc011edea7f8f5b2.yaml
    │   ├── uds-permissions-903266ac6445b873.yaml
    │   ├── add-session-to-rcm-endpoints-d97e925a04a99a65.yaml
    │   ├── check-http-header-sensitivity-daaa9d9595ae86d6.yaml
    │   ├── fix-tracerflare-request-parsing-fae18ff9997c216d.yaml
    │   ├── named-pipe-ded0ebbbeee2e7fa.yaml
    │   ├── version-0c32d2752f154eb9.yaml
    │   ├── windows-docker-image-1f5d5e894558bb4b.yaml
    │   ├── addAgentStateToInfo-f9a179ce88f2ab9e.yaml
    │   ├── tracestats-post-65ce788f81862532.yaml
    │   ├── llmobs-evp-proxy-1383f4b4962ac7c9.yaml
    │   ├── sample-rate-1e06ae4cdc933b14.yaml
    │   ├── traces-endpoint-f4f9c1d94d6ceda2.yaml
    │   ├── addPeerTagsToInfo-4ccc2539d22c2993.yaml
    │   ├── docker-arm64-images-a4b6b2cc527327d6.yaml
    │   ├── genai-vcr-support-b499ec54df84eaa3.yaml
    │   ├── pipeline_stats_endpoint-e8ad6da735dc2647.yaml
    │   ├── snapshot-parent-id-7abb860008702e70.yaml
    │   ├── add-fmt-command-cc31769942a5fec3.yaml
    │   ├── fix-missing-slim-deps-fa3b0fc0292a09b3.yaml
    │   ├── fix-partial-chunks-0db14ce3017e6173.yaml
    │   ├── tracestats-service-2a0a178cbd21b07d.yaml
    │   ├── vcr-cassettes-for-vercel-ai-7a8275f9c2b48bc7.yaml
    │   ├── add-tracer_flare-endpoint-b2c81510920974d7.yaml
    │   ├── evp-proxy-on-info-d65cd077575306ca.yaml
    │   ├── remoteconfig-requests-1af7abb99c61c069.yaml
    │   ├── snapshot-empty-maps-2cbe8cea3e7112fa.yaml
    │   ├── snapshot-error-messages-de41d9e3207fa30a.yaml
    │   ├── Add-metrics.process_id-to-ignored-attributes-62b264f0ea591326.yaml
    │   ├── python-support-1c447acc4607abc7.yaml
    │   ├── oai-incomplete-cass-795ad19b6a226c1f.yaml
    │   ├── snapshot-filename-c068de1bc3e50c04.yaml
    │   ├── add-support-for-plain-text-responses-00d20bf7b335e75b.yaml
    │   ├── exception-messages-21ee64461020edd1.yaml
    │   ├── span-links-1ec76bd2b47a3880.yaml
    │   ├── add-feature-flags-evp-endpoint-da7786735ac02f50.yaml
    │   ├── otlp-logs-support-d5a150d9f5304d7c.yaml
    │   ├── SNAPSHOT_REGEX_PLACEHOLDERS-1d2c5e2711d1953b.yaml
    │   ├── ignore-dsm-pathway-hash-during-trace-snapshots-f0b99ea267fbad48.yaml
    │   ├── fix-racecondition-9bd0ef15b7da60fa.yaml
    │   ├── add-support-telemetry-requests-10737cd3da802ca6.yaml
    │   ├── client-173dbc6655e42337.yaml
    │   ├── session-requests-b8057811f976220e.yaml
    │   ├── vcr-cassettes-for-openai-v5-6e9d78091de1f0c2.yaml
    │   ├── SNAPSHOT_REGEX_PLACEHOLDERS-compare-c829e7a4df4fa1f4.yaml
    │   ├── add-errors-intake-proxy-endpoint-7e0f31d72a130f1a.yaml
    │   ├── fix-matching-319744b82d2514c1.yaml
    │   ├── more-oai-cass-908eb5ea46bd2985.yaml
    │   ├── remove-agent-state-header-from-info-87f920bf4ce5785c.yaml
    │   ├── snapshot-order-type-2d6424e380b2c88f.yaml
    │   ├── uds-c99ea5d099e67156.yaml
    │   ├── vcr-proxy-aws-recalculation-fix-2184c10270a7bca5.yaml
    │   ├── fix-docker-image-tags-785d40f5bfdb8003.yaml
    │   ├── session-tracerflares-400ce9b3d3838b40.yaml
    │   ├── uds-container-8138f8b1438b1a0a.yaml
    │   ├── uds-64b11960931d8b77.yaml
    │   ├── add-nix-build-support-29dc8a347e3bd7c1.yaml
    │   ├── add-rcm-endpoints-7e1f949e83e21039.yaml
    │   ├── vcr-prefix-instead-of-suffix-4c56dcb5038d222e.yaml
    │   ├── v0.5-traces-cf469b0b71398d97.yaml
    │   ├── add-distributed-trace-c35a314698a3b966.yaml
    │   ├── ignore-transfer-encoding-header-3b22699615356f91.yaml
    │   ├── rc-get-endpoint-e502f55b7df544e2.yaml
    │   ├── acceptV1TracePayloads-5435cff37cfb6a85.yaml
    │   ├── fix-otlp-endpoint-response-type-18822a3ac85e1cb7.yaml
    │   ├── profiling-endpoint-0cd209e4e5b6cfc0.yaml
    │   ├── snapshot-cmds-de4a561911afee11.yaml
    │   ├── controllable-test-agent-version-a12e61934bbce5fe.yaml
    │   ├── content_length-e1bab75580aa1dcb.yaml
    │   ├── Telemetry-requests-delay-2586180b286ce393.yaml
    │   ├── add-request-forwarding-middleware-d3f325166b0f8e9e.yaml
    │   ├── python-3.12-952e06c6158af67d.yaml
    │   ├── vcr-proxy-use-test-name-e47891994c961c52.yaml
    │   ├── improve-test-agent-debugging-5759413f1fc5f7fa.yaml
    │   ├── llmobs-eval-metric-endpoint-8b5222491969ab4f.yaml
    │   ├── vcr-cassette-name-middleware-fix-9e47c395892a9fe7.yaml
    │   ├── add-web-ui-304c84316f2b4818.yaml
    │   ├── Add-values-handling-to-span_event-array_value-to-mirror-agent-ee7483c90c263ff4.yaml
    │   ├── otlp-logs-grpc-45365879c22969aa.yaml
    │   ├── Settings-endpoint-928ae5b6e8ddc625.yaml
    │   ├── trace-stats-snapshot-718ca37385c93e1e.yaml
    │   ├── vcr-proxy-use-test-name-bugfix-be758294ba30ac78.yaml
    │   ├── tracestats-fixes-07152c708e73ef96.yaml
    │   ├── span-ordering-1a7b383aa015bddc.yaml
    │   ├── proxy-37c94e2fb428c6fc.yaml
    │   ├── throttling-590bf58ecb3cc4a5.yaml
    │   ├── change-trace-proxy-headers-f9fcbb4efc4fcb95.yaml
    │   ├── fix-getting-tested-integrations-multiple-session-tokens-4ee41d545754f87f.yaml
    │   ├── trace-stats-97da3a478dabfa99.yaml
    │   ├── vcr-ci-mode-99dbd2940fff8d7a.yaml
    │   ├── Use-default-value-for-null-fields-in-tracestats-55d465f489efb2ba.yaml
    │   ├── meta-struct-2cce08475cb05470.yaml
    │   ├── improve-trace-check-results-endpoints-6ab4132628520793.yaml
    │   ├── vcr-aws-bedrock-proxy-3bf018b1712d5105.yaml
    │   ├── remove-vcr-dependancy-ea09889df42b3975.yaml
    │   ├── allow-attribute-removal-snapshots-314b744b043b97c5.yaml
    │   ├── provider-snapshot-server-f6a307acbf11d214.yaml
    │   ├── add-service-naming-checks-2b68ff51d1b5496d.yaml
    │   ├── improve-test-agent-proxying-a207f0fc81b3ab2d.yaml
    │   ├── vcr-custom-providers-map-a43b7e2d62d02015.yaml
    │   ├── add-optional-json-parser-10aac54bd1cbe322.yaml
    │   ├── add-otel-metrics-support-a2ebeb28cae2f0ba.yaml
    │   ├── support-tracking-of-tested-integrations-ac19ccd1f8333c00.yaml
    │   ├── disable-error-responses-and-collect-trace-failures-in-memory-4a199bc7be416709.yaml
    │   ├── Checks-changed-to-opt-in-8716cac4ecdbb1c2.yaml
    │   ├── clearing-sessions-now-only-removes-6b0311673d8b37aa.yaml
    │   └── 1.0-e192f5064e917083.yaml
    └── check-releasenotes
├── test_deps.txt
├── pyproject.toml
├── Dockerfile
├── .gitignore
├── Dockerfile.windows
├── setup.cfg
├── .github
    └── workflows
    │   ├── releasenote.yml
    │   ├── publish_pypi.yml
    │   ├── main.yml
    │   └── docker.yml
├── LICENSE.BSD3
├── vcr-cassettes
    ├── openai
    │   ├── openai_models_gpt-4_get_b13c5b23.yaml
    │   ├── openai_files_file-RpTpuvRVtnKpdKZb7DDGto_delete_3ee48409.yaml
    │   ├── openai_models_ft_gpt-4.1-mini-2025-04-14_datadog-staging__BkaILRSh_delete_c70f2992.yaml
    │   ├── openai_files_file-RpTpuvRVtnKpdKZb7DDGto_get_1afefb93.yaml
    │   ├── openai_files_get_68efa25d.yaml
    │   ├── openai_completions_post_6224acff.yaml
    │   ├── openai_chat_completions_post_917478d0.yaml
    │   ├── openai_images_generations_post_736070ea.yaml
    │   ├── openai_completions_post_9bb49464.yaml
    │   ├── openai_moderations_post_532b3315.yaml
    │   ├── openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_get_265489e7.yaml
    │   ├── openai_fine_tuning_jobs_get_ebc57c6f.yaml
    │   ├── openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_cancel_post_d7c16b38.yaml
    │   ├── openai_fine_tuning_jobs_post_c3b6e4a9.yaml
    │   ├── openai_completions_post_33681ea4.yaml
    │   ├── openai_files_file-RpTpuvRVtnKpdKZb7DDGto_content_get_60bd10ef.yaml
    │   ├── openai_chat_completions_post_57612363.yaml
    │   ├── openai_chat_completions_post_655a8ef2.yaml
    │   ├── openai_chat_completions_post_649d8162.yaml
    │   ├── openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_events_get_557d820e.yaml
    │   ├── openai_chat_completions_post_4dce8db6.yaml
    │   ├── openai_chat_completions_post_23558eed.yaml
    │   ├── openai_chat_completions_post_d8fb132a.yaml
    │   └── openai_chat_completions_post_ce69ed91.yaml
    └── deepseek
    │   └── deepseek_chat_completions_post_afb6ccab.yaml
├── riotfile.py
└── setup.py


/tests/__init__.py:
--------------------------------------------------------------------------------
1 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/context.py:
--------------------------------------------------------------------------------
1 | 


--------------------------------------------------------------------------------
/.dockerignore:
--------------------------------------------------------------------------------
1 | .venv
2 | result
3 | *.Dockerfile


--------------------------------------------------------------------------------
/releasenotes/config.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | unreleased_version_title: Unreleased
3 | 


--------------------------------------------------------------------------------
/test_deps.txt:
--------------------------------------------------------------------------------
1 | ddtrace==3.15.0
2 | pytest
3 | riot==0.20.1
4 | PyYAML==6.0.3


--------------------------------------------------------------------------------
/releasenotes/notes/wheels-68a7dc2f4b00e0c4.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes: Publish wheels to PyPI.
3 | 


--------------------------------------------------------------------------------
/releasenotes/notes/python310-025fbc190363de98.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add Python 3.10 support.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/v0.7-traces-7ea3ebbed7502d01.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for v0.7 traces.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/args-88f664c83a96075d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix command line arguments not being parsed.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/info-34dcd9cbcc382487.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for the ``/info`` endpoint.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/remove-nix-ec1c4aca7dc002c4.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     This change removes nix build support.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-datadog-proxy-7e2e608c101a564b.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     Add datadog vcr cassette support.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-python-3.13-support-ae7b96f4a10f20f4.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for Python 3.13
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-span-events-306e697168373899.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add agent support for Span Events.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/support-ant-vcr-1eff9a40caa0aba0.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Support anthropic vcr cassettes.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/info-drop-p0s-c207889f590340fb.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Add ClientDropP0s setting to /info response.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/span-parent-id-a4d7d63fa623361c.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Handle span ``parent_id`` being undefined.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-support-querystring-b06555fb9a7c90ae.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     VCR: add support for query strings.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-fmt-logging-6a323fd8d5d4eb28.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix log messages in ``ddapm-test-agent-fmt``.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/image-optimization-d614eccb59441217.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     The image size has been cut roughly in half.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/new-process-id-tag-f0ab0daf652e05cc.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Add `metrics.process_id` tag to ignored tags.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/python310-docker-image-d4e329cfecf6510e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Use Python 3.11 in the Docker image.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/remove-old-pythons-4667b182aacf0ce6.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     Removes support for Python 3.8, 3.9, and 3.10.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/reuse-session-1670aa569907cf68.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix sessions returning traces from all sessions.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/rc-encoding-length-fix-cc011edea7f8f5b2.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix message length for remote config requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/uds-permissions-903266ac6445b873.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Use 722 permissions for UDS socket to match the agent.
5 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/__init__.py:
--------------------------------------------------------------------------------
1 | def _get_version():
2 |     import pkg_resources
3 | 
4 |     return pkg_resources.get_distribution(__name__).version
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-session-to-rcm-endpoints-d97e925a04a99a65.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add session token to Remote Config endpoint.


--------------------------------------------------------------------------------
/releasenotes/notes/check-http-header-sensitivity-daaa9d9595ae86d6.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix case sensitive headers for trace checks.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-tracerflare-request-parsing-fae18ff9997c216d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix the parsing logic for tracer flare requests


--------------------------------------------------------------------------------
/releasenotes/notes/named-pipe-ded0ebbbeee2e7fa.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for running the test agent over a Windows named pipe.


--------------------------------------------------------------------------------
/releasenotes/notes/version-0c32d2752f154eb9.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add ``-v/--version`` command line option to display the version.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/windows-docker-image-1f5d5e894558bb4b.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add Windows container support to Docker images.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/addAgentStateToInfo-f9a179ce88f2ab9e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `Datadog-Agent-State` header to `/info` endpoint.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/tracestats-post-65ce788f81862532.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Handle POST /v0.6/stats requests. Only PUT was supported before.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/llmobs-evp-proxy-1383f4b4962ac7c9.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `/evp_proxy/v2/api/v2/llmobs` endpoint to collect requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/sample-rate-1e06ae4cdc933b14.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add the ability to set the trace sample rates returned by the agent.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/traces-endpoint-f4f9c1d94d6ceda2.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix /test/traces endpoint attempting to decode non-trace requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/addPeerTagsToInfo-4ccc2539d22c2993.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `peer_tags` field to `/info` endpoint for testing client side stats.


--------------------------------------------------------------------------------
/releasenotes/notes/docker-arm64-images-a4b6b2cc527327d6.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Build an ARM64 version of the Docker image as a multi-arch image.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/genai-vcr-support-b499ec54df84eaa3.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     vcr: adds support for recording requests from google genai/gemini.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/pipeline_stats_endpoint-e8ad6da735dc2647.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add new pipeline stats endpoint for data streams monitoring.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-parent-id-7abb860008702e70.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix not present parent_id in snapshots. It should be normalized to 0.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-fmt-command-cc31769942a5fec3.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Added ``ddapm-test-agent-fmt`` to format or validate snapshot json files.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-missing-slim-deps-fa3b0fc0292a09b3.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Ensures ``runc`` and ``containerd`` are installed in the docker image.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-partial-chunks-0db14ce3017e6173.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix partial trace chunks (referring to spans in other chunks) causing 500s.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/tracestats-service-2a0a178cbd21b07d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Decode service field in trace stats requests, it was previously omitted.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-cassettes-for-vercel-ai-7a8275f9c2b48bc7.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Support a vcr cassette for OpenAI from Vercel AI SDK requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-tracer_flare-endpoint-b2c81510920974d7.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add ``/tracer_flare/v1`` endpoint with basic form-field validation.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/evp-proxy-on-info-d65cd077575306ca.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Adds `/evp_proxy/v2/` as one of the endpoints listed from the `/info` endpoint.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/remoteconfig-requests-1af7abb99c61c069.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Remote config requests are now returned via ``/test/session/requests``.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-empty-maps-2cbe8cea3e7112fa.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     Snapshots will no longer have empty meta/metrics maps included in the output.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-error-messages-de41d9e3207fa30a.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Better snapshot error messages for unexpected number of traces received.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/Add-metrics.process_id-to-ignored-attributes-62b264f0ea591326.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `metrics.process_id` to default ignored attributes
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/python-support-1c447acc4607abc7.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Specify required Python version to >=3.8. It was mistakenly set to >=3.7
5 |     in setup.py.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/oai-incomplete-cass-795ad19b6a226c1f.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Support a vcr cassette for openai responses api streams that end with an incomplete chunk.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-filename-c068de1bc3e50c04.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix custom snapshot filenames. Custom filenames are not supposed to include the file extension.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-support-for-plain-text-responses-00d20bf7b335e75b.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Adds support for plain/text response content-types when forwarding requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/exception-messages-21ee64461020edd1.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     Exceptions will now result in a 400 status code with a body containing the exception that occurred.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/span-links-1ec76bd2b47a3880.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Added ability to parse and verify span links in payloads as well as
5 |     comparing them in snapshots.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-feature-flags-evp-endpoint-da7786735ac02f50.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `/evp_proxy/v2/api/v2/exposures` endpoint to collect feature flag exposures requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/otlp-logs-support-d5a150d9f5304d7c.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     otel: Adds OpenTelemetry Protocol (OTLP) v1.7.0 logs support via HTTP endpoint ``/v1/logs`` on port 4318.


--------------------------------------------------------------------------------
/releasenotes/notes/SNAPSHOT_REGEX_PLACEHOLDERS-1d2c5e2711d1953b.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add a SNAPSHOT_REGEX_PLACEHOLDERS option to allow infrastructure independent
5 |     snapshots.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/ignore-dsm-pathway-hash-during-trace-snapshots-f0b99ea267fbad48.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Adds dsm `pathway.hash` span tag to the ignored attributes for a trace snapshot.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-racecondition-9bd0ef15b7da60fa.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix race condition when the test agent would simultaneously handle requests
5 |     which would read and write data.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-support-telemetry-requests-10737cd3da802ca6.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for ``POST /telemetry/proxy/api/v2/apmtelemetry`` endpoint and tracking telemetry requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/client-173dbc6655e42337.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add a `client` module which provides an API client that can be used to
5 |     programmatically interface with the test agent.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/session-requests-b8057811f976220e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add ``/test/session/requests`` endpoint to return all requests made to the
5 |     test agent for a given session token.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-cassettes-for-openai-v5-6e9d78091de1f0c2.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Provider VCR: Fixes normalization support for a different form data boundary format used by the Node.js runtime.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/SNAPSHOT_REGEX_PLACEHOLDERS-compare-c829e7a4df4fa1f4.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Actually make the SNAPSHOT_REGEX_PLACEHOLDERS feature also work during comparison, and not only generation.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-errors-intake-proxy-endpoint-7e0f31d72a130f1a.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for ``/evp_proxy/v4/api/v2/errorsintake`` endpoint to collect errors intake payload requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-matching-319744b82d2514c1.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix the snapshot matching algorithm which had previously assumed that trace
5 |     ids would be unique across both sets of traces.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/more-oai-cass-908eb5ea46bd2985.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     Add OpenAI vcr cassettes for responses api prompt caching, responses api streaming prompt caching, and chat completions with tool calls.


--------------------------------------------------------------------------------
/releasenotes/notes/remove-agent-state-header-from-info-87f920bf4ce5785c.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Remove `Datadog-Agent-State` header from `/info` endpoint response to match actual Datadog agent behavior.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-order-type-2d6424e380b2c88f.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Move span attributes ``type``, ``meta`` and ``metrics`` up in the JSON
5 |     snapshot output to help with readability.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/uds-c99ea5d099e67156.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Remove any existing UDS socket. If a socket was left behind by a previous
5 |     instance of the test agent then it would fail to start.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-proxy-aws-recalculation-fix-2184c10270a7bca5.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     vcr: fixes a bug where the aws signature recalculation would always happen, even if the cassette already existed.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-docker-image-tags-785d40f5bfdb8003.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fixes windows docker image tagging to avoid having only linux or windows images published for ``latest`` and ``vx.y.z`` tags.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/session-tracerflares-400ce9b3d3838b40.yaml:
--------------------------------------------------------------------------------
1 | features:
2 |   - |
3 |     Add ``/test/session/tracerflares`` endpoint to return all tracer-flares received
4 |     by the test agent for a given session token.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/uds-container-8138f8b1438b1a0a.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Workaround an issue where `chmod`ing the UDS socket in containers causes
5 |     the testagent to crash. Instead, a warning is emitted.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/uds-64b11960931d8b77.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add unix domain socket support.
5 |     Enabled via environment variable `DD_APM_RECEIVER_SOCKET` or command line argument `--trace-uds-socket`.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-nix-build-support-29dc8a347e3bd7c1.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | other:
3 |   - |
4 |     This change gives the repository Nix build support. You can now build, test and run the code using nix as a build and dependency provider
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-rcm-endpoints-7e1f949e83e21039.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add Remote Config endpoint (``/v0.7/config``) and endpoints to create/update Remote Config payloads 
5 |     (``/test/session/responses/config`)


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-prefix-instead-of-suffix-4c56dcb5038d222e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | upgrade:
3 |   - |
4 |     vcr: ``test_name`` specified in ``/vcr/test/start`` is now used as a prefix for generated cassette names instead of a suffix.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/v0.5-traces-cf469b0b71398d97.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for the /v0.5/traces endpoint. Traces sent to this endpoint can
5 |     be queried and snapshotted just as they can be for /v0.4/traces.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-distributed-trace-c35a314698a3b966.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for distributed traces where an instrumented service sends a trace chunk where the root span has a parent not in the trace chunk.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/ignore-transfer-encoding-header-3b22699615356f91.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Ignores the Transfer-Encoding header in proxied vcr requests, forwarding this header can lead to failures in 
5 |     downstream services. 
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/rc-get-endpoint-e502f55b7df544e2.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Add GET method support for the ``/v0.7/config`` endpoint. The Go library
5 |     uses ``GET`` requests and the Datadog Agent supports multiple HTTP methods.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/acceptV1TracePayloads-5435cff37cfb6a85.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for the /v1.0/traces endpoint on the trace-agent, for now these payloads are converted automatically to the "v4" format for ease of adoption.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-otlp-endpoint-response-type-18822a3ac85e1cb7.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     otlp: Fix OTLP endpoint response format for logs and metrics to return proper protobuf content type and serialized response body for HTTP requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/profiling-endpoint-0cd209e4e5b6cfc0.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add basic support for profiling. Profiling requests are now handled by the
5 |     test agent. Nothing is done with the requests other than accepting them.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/snapshot-cmds-de4a561911afee11.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add ``ddapm-test-agent-session-start`` and ``ddapm-test-agent-snapshot`` commands
5 |     to provide a more convenient way to perform snapshots manually or in CI.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/controllable-test-agent-version-a12e61934bbce5fe.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for controlling the test agent version via the TEST_AGENT_VERSION environment variable. The version is returned in the /info endpoint response.


--------------------------------------------------------------------------------
/releasenotes/notes/content_length-e1bab75580aa1dcb.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Fix content length check assuming that the Content-Length header is always
5 |     included in trace http requests. Some clients (like go) don't submit this
6 |     header.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/Telemetry-requests-delay-2586180b286ce393.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     --trace-request-delay now also affects telemetry requests.
5 |     The delay can be set independently for each request by setting the X-Datadog-Test-Stall-Seconds header.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-request-forwarding-middleware-d3f325166b0f8e9e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Adds middleware to forward all received requests to the Agent URL, if set. Requests are 
5 |     only forwarded for the `/traces`, `/stats`, and `/telemetry` paths. 
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/python-3.12-952e06c6158af67d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for Python 3.12. The interpreter used in the Docker image is
5 |     now also Python 3.12. This should lead to some better error messages and
6 |     some performance improvement.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-proxy-use-test-name-e47891994c961c52.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     vcr: adds two new ``/vcr/test/start`` and ``/vcr/test/stop``, which capture all proxy requests made in between these two requests with a test name specified in ``/vcr/test/start``.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/improve-test-agent-debugging-5759413f1fc5f7fa.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Improves APM Test-Agent debugging by improving Assertion error messages when verifying
5 |     decoded traces. Also adds debug logging for the received request and request headers.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/llmobs-eval-metric-endpoint-8b5222491969ab4f.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add `/evp_proxy/v2/api/intake/llm-obs/v1/eval-metric` and `/evp_proxy/v2/api/intake/llm-obs/v2/eval-metric` endpoints to collect LLM Observability evaluation metric payload requests.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-cassette-name-middleware-fix-9e47c395892a9fe7.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     vcr: fixes an issue where the vcr cassette name middleware would not properly ignore decoding errors for request bodies that were not associated with then ``/vcr/test/start`` path.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-web-ui-304c84316f2b4818.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Adds a new web ui that provides a dashboard for inspecting agent
5 |     configuration, viewing received requests, exploring traces and snapshots,
6 |     and managing tracer-flare and remote configuration.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/Add-values-handling-to-span_event-array_value-to-mirror-agent-ee7483c90c263ff4.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     The agent requires any span_event array_value to be wrapped in a {values: [...]} object.
5 |     This reflects this requirement from the SpanEvent prototype here.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/otlp-logs-grpc-45365879c22969aa.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     OTLP: Added GRPC server support for OTLP logs on port 4317 that forwards requests to the OTLP HTTP server for validation and storage.
5 |     This enables testing of applications that send OTLP logs via GRPC protocol.


--------------------------------------------------------------------------------
/releasenotes/notes/Settings-endpoint-928ae5b6e8ddc625.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Added a /test/settings endpoint to change the settings on the fly.
5 |     It takes a json encoded list of key/values, that will be applied to the app dict.
6 |     Example:
7 |     { "trace_request_delay": 5 }
8 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/integration.py:
--------------------------------------------------------------------------------
1 | class Integration:
2 |     def __init__(self, integration_name: str, integration_version: str, dependency_name: str):
3 |         self.integration_name = integration_name
4 |         self.integration_version = integration_version
5 |         self.dependency_name = dependency_name
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/trace-stats-snapshot-718ca37385c93e1e.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Implement trace stats snapshotting.
5 | 
6 |     Trace stats are now included in the snapshot behaviour provided by the testagent.
7 |  
8 |     Similar to traces, trace stats snapshots are output to a json file.
9 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-proxy-use-test-name-bugfix-be758294ba30ac78.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     vcr: fixes a bug where the vcr cassette would cause unnecessary 400s from proxied providers.
5 |   - |
6 |     vcr: fixes a bug where google genai client keys would not be omitted from generated vcr cassettes.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/tracestats-fixes-07152c708e73ef96.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Normalize trace stats snapshots using all the aggregation fields. This
5 |     fixes an issue where snapshots could fail due to ordering of buckets.
6 |   - |
7 |     Use HTTPStatusCode in trace stats snapshots. This field was missing.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/span-ordering-1a7b383aa015bddc.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Spans output in snapshots are now ordered to be more human readable with
5 |     the important attributes (name, service, resource) moved to the top of the
6 |     JSON object. This affects the output JSON but does not break existing
7 |     snapshots.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/proxy-37c94e2fb428c6fc.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Trace requests can now be proxied to an actual agent by passing an agent
5 |     url to the test agent. This can be done by passing the ``--agent-url``
6 |     command-line option or via the ``DD_AGENT_URL`` or ``DD_TRACE_AGENT_URL``
7 |     environment variables.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/throttling-590bf58ecb3cc4a5.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for artificially throttling requests to simulate an agent under
5 |     duress.  Enabled dynamically via request header
6 |     ``X-Datadog-Test-Trace-Stall-Seconds`` with /v0.4/trace requests or
7 |     statically via the ``--trace-request-delay`` flag.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/change-trace-proxy-headers-f9fcbb4efc4fcb95.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     This change adds support for disabling / enabling proxying for a trace by sending 
5 |     the X-Datadog-Agent-Proxy-Disabled with a boolean value within a trace's headers. By default,
6 |     proxying is enabled if the Test-Agent has an agent_url set.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/fix-getting-tested-integrations-multiple-session-tokens-4ee41d545754f87f.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | fixes:
3 |   - |
4 |     Add fix for getting tested integrations when multiple session tokens are used. When no session token is included for a [GET] request to 
5 |     ``/test/integrations/tested_versions``, all tested integrations are returned now.
6 | 


--------------------------------------------------------------------------------
/releasenotes/notes/trace-stats-97da3a478dabfa99.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for the ``/v0.6/stats`` endpoint. All requests made to the
5 |     endpoint will be stored and can be retrieved via the
6 |     ``/test/session/requests`` or ``/test/session/stats`` endpoints.
7 |     Note that snapshotting for stats is not yet implemented.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-ci-mode-99dbd2940fff8d7a.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     vcr: adds a new ``--vcr-ci-mode`` flag and accompanying ``VCR_CI_MODE`` environment variable that, when set to ``true`` (defaults to ``false``), will throw a 404 error if a cassette is not found in CI mode to ensure that all cassettes are generated locally and committed.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/Use-default-value-for-null-fields-in-tracestats-55d465f489efb2ba.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Use default value for null fields in tracestats. This aligns with the trace agent behaviors.
5 | upgrade:
6 |   - |
7 |     BREAKING CHANGE: Test snapshots must be updated as field comparisons on stats payload will fail between "" and None values.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/meta-struct-2cce08475cb05470.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for parsing the `meta_struct` field in traces. This field just like the `meta` field map but the values
5 |     are arbitrary inner msgpack-encoded values. When the `meta_struct` field is present, its inner messages will be
6 |     parsed and added to the trace as a dictionary.


--------------------------------------------------------------------------------
/releasenotes/notes/improve-trace-check-results-endpoints-6ab4132628520793.yaml:
--------------------------------------------------------------------------------
1 | features:
2 |   - |
3 |     Adds new endpoints to the Test-Agent API: `/test/trace_check/failures`, `/test/trace_check/clear`, and `/test/trace_check/summary`. 
4 |     These endpoints allow users to retrieve trace check failures, clear trace check failures, and get trace check summary results respectively. 
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-aws-bedrock-proxy-3bf018b1712d5105.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     vcr: adds support for proxying aws bedrock runtime. to record cassettes for the first time, the `AWS_SECRET_ACCESS_KEY` environment variable must be set for the container running the test agent, for request signature recalculation. Additionally, `AWS_REGION` can be set, defaulting to `us-east-1`.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/remove-vcr-dependancy-ea09889df42b3975.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | deprecations:
3 |   - |
4 |     VCR has been removed as a dependency, and instead JSON cassettes are produced. Legacy VCR cassettes are converted to JSON on first use, and the old VCR cassettes are removed. This functionality will be removed in ddapm-test-agent==2.0.0, at which point only JSON cassettes will be written and read.
5 | 


--------------------------------------------------------------------------------
/tests/test_apmtelemetry.py:
--------------------------------------------------------------------------------
 1 | from ddapm_test_agent.apmtelemetry import v2_decode
 2 | 
 3 | 
 4 | def test_decode_v2(
 5 |     v2_reference_http_apmtelemetry_payload_data_raw,
 6 |     v2_reference_http_apmtelemetry_payload_data,
 7 | ):
 8 |     # decode_v2 is just json.loads for now
 9 |     assert v2_decode(v2_reference_http_apmtelemetry_payload_data) == v2_reference_http_apmtelemetry_payload_data_raw
10 | 


--------------------------------------------------------------------------------
/releasenotes/notes/allow-attribute-removal-snapshots-314b744b043b97c5.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     This change adds a new feature that allows desired attributes to be removed from spans in snapshots. By default, no
5 |     attributes will be removed. The attributes to be removed can be set through the optional environment variable
6 |     `SNAPSHOT_REMOVED_ATTRS`, or with the optional query parameter `?removes`.


--------------------------------------------------------------------------------
/releasenotes/notes/provider-snapshot-server-f6a307acbf11d214.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Provider snapshot server: Introduces a provider snapshot server that can be used to record and play back provider API requests. The first unique request is forwarded to the provider API, and the response is recorded to a cassette that can be played back later in the session, or at any time if persisted to the test agent.
5 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-service-naming-checks-2b68ff51d1b5496d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Adds two new Trace Checks for service naming and peer service. `trace_dd_service` check ensures that handled traces
5 |     have a correctly configured service name according to the trace's Span Attribute Schema Version. `trace_peer_service`
6 |     ensures the that the correct `peer.service` value is set for spans according to the supplied span precursors.
7 | 


--------------------------------------------------------------------------------
/pyproject.toml:
--------------------------------------------------------------------------------
 1 | [tool.isort]
 2 | force_single_line = true
 3 | lines_after_imports = 2
 4 | force_sort_within_sections = true
 5 | known_first_party = "ddapm_test_agent"
 6 | default_section = "THIRDPARTY"
 7 | skip = [".riot/", ".venv/", "public_api/"]
 8 | line_length = 120
 9 | profile = "black"
10 | 
11 | [tool.black]
12 | line_length = 120
13 | exclude = '''
14 | (
15 |   /(
16 |     \.riot
17 |   | \.venv.*
18 |   | \.eggs
19 |   | public_api
20 |   )/
21 | )
22 | '''
23 | 


--------------------------------------------------------------------------------
/releasenotes/notes/improve-test-agent-proxying-a207f0fc81b3ab2d.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     This change improves APM Test Agent proxying in a few ways. Proxying is now done first when a trace is 
5 |     received, instead of waiting for Test Agent checks to complete. This means that a failing check will 
6 |     no longer prevent the trace(s) being proxied to the final destination. Additionally, proxying now 
7 |     has better handling of response types including byte, json and text/html responses.
8 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_trace_stats_tracestats.json:
--------------------------------------------------------------------------------
 1 | [
 2 |   {
 3 |     "Start": 0,
 4 |     "Duration": 10000000000,
 5 |     "Stats": [
 6 |       {
 7 |         "Name": "http.request",
 8 |         "Resource": "/users/view",
 9 |         "Service": "",
10 |         "Type": "",
11 |         "HTTPStatusCode": 0,
12 |         "Synthetics": false,
13 |         "Hits": 5,
14 |         "TopLevelHits": 5,
15 |         "Duration": 118000,
16 |         "Errors": 1,
17 |         "OkSummary": 1046,
18 |         "ErrorSummary": 1046
19 |       }
20 |     ]
21 |   }
22 | ]
23 | 


--------------------------------------------------------------------------------
/releasenotes/notes/vcr-custom-providers-map-a43b7e2d62d02015.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     vcr: Adds support for specifying a list of custom providers in the ``--vcr-provider-map`` flag or ``VCR_PROVIDER_MAP`` environment variable. The list should take the form of ``provider1=http://provider1.com/,provider2=http://provider2.com/``.
5 |   - |
6 |     vcr: Adds support for specifying a list of headers to ignore when recording VCR cassettes in the ``--vcr-ignore-headers`` flag or ``VCR_IGNORE_HEADERS`` environment variable. The list should take the form of ``header1,header2,header3``.
7 | 


--------------------------------------------------------------------------------
/releasenotes/notes/add-optional-json-parser-10aac54bd1cbe322.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     This change adds support for an optional trace JSON decoder that will attempt to parse a
5 |     trace, even if the trace is partially misconformed, such as one having port numbers within
6 |     the meta field, which would previously throw an error. Additionally, the change also adds 
7 |     a default parser of json.loads, which was originally used. The new parser can be used by
8 |     specifying the `DD_SUPPRESS_TRACE_PARSE_ERRORS` env variable or by setting the flag `--suppress-trace-parse-errors`.


--------------------------------------------------------------------------------
/releasenotes/notes/add-otel-metrics-support-a2ebeb28cae2f0ba.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     OTLP: Added OpenTelemetry Protocol (OTLP) metrics support via HTTP endpoint ``/v1/metrics`` on port 4318 and GRPC server on port 4317.
5 |     Supports all OTLP metric types including Gauge, Sum, Histogram, ExponentialHistogram, and Summary.
6 |     The GRPC server forwards requests to the OTLP HTTP server for validation and storage, enabling testing of applications that send metrics via either protocol.
7 |     Session endpoint ``/test/session/metrics`` allows retrieval of captured metrics for test validation.
8 | 


--------------------------------------------------------------------------------
/releasenotes/notes/support-tracking-of-tested-integrations-ac19ccd1f8333c00.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Add support for tracking the integrations being tested. Tracked integrations can be recorded by a [PUT] request to the Test Agent
5 |     at `/test/session/integrations`. To get data about which integrations the Test Agent encountered, make a [GET] request to `/test/integrations/tested_versions`.
6 |     Tested integrations include information such as the integration name, the tested integration version, the tracer language, the tracer version and the 
7 |     dependency name of the integration.
8 | 


--------------------------------------------------------------------------------
/Dockerfile:
--------------------------------------------------------------------------------
 1 | FROM python:3.13-slim
 2 | 
 3 | WORKDIR /src
 4 | EXPOSE 8126
 5 | 
 6 | ENV SNAPSHOT_CI=1
 7 | ENV LOG_LEVEL=INFO
 8 | ENV SNAPSHOT_DIR=/snapshots
 9 | ENV VCR_CASSETTES_DIRECTORY=/vcr-cassettes
10 | 
11 | RUN apt update && \
12 |     apt install -y --no-install-recommends git curl runc containerd && \
13 |     rm -rf /var/lib/apt/lists/*
14 | 
15 | ADD vcr-cassettes /vcr-cassettes
16 | 
17 | # Add only necessary files to speed up development builds
18 | ADD README.md setup.py test_deps.txt ./
19 | ADD ddapm_test_agent ./ddapm_test_agent
20 | ADD .git ./.git
21 | RUN pip install /src && \
22 |     rm -rf /root/.cache/pip
23 | 
24 | CMD ["ddapm-test-agent"]
25 | 


--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
 1 | __pycache__/
 2 | .mypy_cache/
 3 | .pytest_cache/
 4 | .riot/
 5 | 
 6 | # Distribution / packaging
 7 | .Python
 8 | env/
 9 | build/
10 | develop-eggs/
11 | dist/
12 | downloads/
13 | eggs/
14 | .eggs/
15 | lib64/
16 | parts/
17 | sdist/
18 | var/
19 | *.egg-info/
20 | .installed.cfg
21 | *.egg
22 | *.whl
23 | 
24 | # Common virtual environments
25 | .venv/
26 | venv/
27 | .python-version
28 | 
29 | # Editors
30 | .idea/
31 | .vscode/
32 | *.swp
33 | 
34 | 
35 | # Unit test / coverage reports
36 | htmlcov/
37 | .coverage
38 | .coverage.*
39 | .cache
40 | 
41 | # Automated dev env setup - see .envrc.example
42 | /.envrc
43 | /.direnv/
44 | 
45 | # nix 
46 | /result


--------------------------------------------------------------------------------
/releasenotes/notes/disable-error-responses-and-collect-trace-failures-in-memory-4a199bc7be416709.yaml:
--------------------------------------------------------------------------------
1 | ---
2 | features:
3 |   - |
4 |     Adds a feature to disable Test-Agent Failure <Response 400> using the ``DISABLE_ERROR_RESPONSES`` env variable or 
5 |     ``--disable-error-responses`` command line option. Adds a feature to pool Test-Agent Trace Check Failures in-memory using the 
6 |     ``DD_POOL_TRACE_CHECK_FAILURES`` env variable or ``--pool-trace-check-failures`` option. These failures can then be queried by the 
7 |     ``/test/trace_check/failures`` GET endpoint, which returns a <Response 200> if no failures occurred or <Response 400> with 
8 |     the Check Trace Failure messages in the response body if they exist. 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_single_trace.json:
--------------------------------------------------------------------------------
 1 | [[
 2 |   {
 3 |     "name": "root",
 4 |     "service": "custom_service",
 5 |     "resource": "/url/endpoint",
 6 |     "trace_id": 0,
 7 |     "span_id": 1,
 8 |     "parent_id": 0,
 9 |     "type": "web",
10 |     "error": 0,
11 |     "meta": {
12 |       "_dd.base_service": "",
13 |       "_dd.p.dm": "-0",
14 |       "_dd.p.tid": "68e68f3d00000000",
15 |       "language": "python",
16 |       "runtime-id": "fed87db4f462440e8279440b11a394e4"
17 |     },
18 |     "metrics": {
19 |       "_dd.top_level": 1,
20 |       "_dd.tracer_kr": 1.0,
21 |       "_sampling_priority_v1": 1,
22 |       "process_id": 44128
23 |     },
24 |     "duration": 38000,
25 |     "start": 1759940413390843000
26 |   }]]
27 | 


--------------------------------------------------------------------------------
/releasenotes/notes/Checks-changed-to-opt-in-8716cac4ecdbb1c2.yaml:
--------------------------------------------------------------------------------
 1 | ---
 2 | features:
 3 |   - |
 4 |     Checks have been changed to opt-in. This allows adding new checks before
 5 |     all languages have implemented the feature.
 6 | upgrade:
 7 |   - |
 8 |     Checks have been changed to opt-in. In order to maintain current
 9 |     functionality users should update either their command or their 
10 |     environment. 
11 |     `ENABLED_CHECKS=trace_count_header,meta_tracer_version_header,trace_content_length`
12 |     or
13 |     `--enabled-checks=trace_count_header,meta_tracer_version_header,trace_content_length`
14 | deprecations:
15 |   - |
16 |     `DISABLED_CHECKS` and `--disabled-checks` have been removed in favor of
17 |     making checks opt-in.
18 | 


--------------------------------------------------------------------------------
/releasenotes/notes/clearing-sessions-now-only-removes-6b0311673d8b37aa.yaml:
--------------------------------------------------------------------------------
 1 | ---
 2 | features:
 3 |   - |
 4 |     Updates the logic for associating requests with a session.  Prior to this change:
 5 |     (1) non-existent tokens were permitted and returned associations with all requests, 
 6 |     (2) existent tokens returned associations with all matching requests + all untokenized requests after the session was created
 7 | 
 8 |     After this change:
 9 |     (1) Requests with non-existent tokens return 400 error codes
10 |     (2) Requests with existent tokens return all matching requests + all untokenized requests after the session was created but 
11 |     only up to the next session creation.
12 | 
13 |     Requests without a token continue to return all requests.


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/sessions.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | {% from "macros.html" import page_layout, empty_state, action_button %}
 3 | 
 4 | {% block content %}
 5 | {% call page_layout("Sessions") %}
 6 |     {% if sessions %}
 7 |         <div class="sessions-list">
 8 |             {% for session in sessions %}
 9 |             <div class="session-item">
10 |                 <div class="session-header">
11 |                     <h3>{{ session }}</h3>
12 |                 </div>
13 |                 <div class="session-actions">
14 |                     {{ action_button("/sessions/" + session, "View Details") }}
15 |                 </div>
16 |             </div>
17 |             {% endfor %}
18 |         </div>
19 |     {% else %}
20 |         {{ empty_state("No active sessions") }}
21 |     {% endif %}
22 | {% endcall %}
23 | {% endblock %}


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/traces.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | {% from "macros.html" import page_layout, empty_state, action_button %}
 3 | 
 4 | {% block content %}
 5 | {% call page_layout("Traces") %}
 6 |     {% if traces %}
 7 |         <div class="traces-list">
 8 |             {% for trace_id, spans in traces.items() %}
 9 |             <div class="trace-item">
10 |                 <div class="trace-header">
11 |                     <h3>Trace {{ trace_id }}</h3>
12 |                     <span class="span-count">{{ spans|length }} spans</span>
13 |                 </div>
14 |                 <div class="trace-actions">
15 |                     {{ action_button("/traces/" + trace_id, "View Details") }}
16 |                 </div>
17 |             </div>
18 |             {% endfor %}
19 |         </div>
20 |     {% else %}
21 |         {{ empty_state("No traces available") }}
22 |     {% endif %}
23 | {% endcall %}
24 | {% endblock %}


--------------------------------------------------------------------------------
/tests/test_client.py:
--------------------------------------------------------------------------------
 1 | from ddtrace.trace import Tracer
 2 | import pytest
 3 | 
 4 | from ddapm_test_agent.client import TestAgentClient
 5 | 
 6 | 
 7 | @pytest.fixture
 8 | async def client(testagent, testagent_url):
 9 |     return TestAgentClient(testagent_url)
10 | 
11 | 
12 | @pytest.fixture
13 | async def tracer(testagent_url):
14 |     t = Tracer()
15 |     t._agent_url = testagent_url
16 |     yield t
17 |     t.shutdown()
18 | 
19 | 
20 | async def test_client_traces(client: TestAgentClient, tracer: Tracer) -> None:
21 |     assert client.traces() == []
22 |     with tracer.trace("test"):
23 |         pass
24 |     traces = client.wait_for_num_traces(1)
25 |     assert len(traces) == 1
26 | 
27 | 
28 | async def test_client_requests(client: TestAgentClient, tracer: Tracer) -> None:
29 |     assert client.requests() == []
30 |     with tracer.trace("test"):
31 |         pass
32 |     tracer.flush()
33 |     assert len(client.requests()) == 1
34 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_trace_distributed_propagated.json:
--------------------------------------------------------------------------------
 1 | [[
 2 |   {
 3 |     "name": "root",
 4 |     "service": "",
 5 |     "resource": "root",
 6 |     "trace_id": 0,
 7 |     "span_id": 1,
 8 |     "parent_id": 5678,
 9 |     "type": "",
10 |     "error": 0,
11 |     "meta": {
12 |       "_dd.p.dm": "-0",
13 |       "language": "python",
14 |       "runtime-id": "f41a72c378a44fa19936a90858bdcf4b"
15 |     },
16 |     "metrics": {
17 |       "_dd.top_level": 1,
18 |       "_dd.tracer_kr": 1.0,
19 |       "_sampling_priority_v1": 1,
20 |       "process_id": 44248
21 |     },
22 |     "duration": 49000,
23 |     "start": 1759940420748099000
24 |   },
25 |      {
26 |        "name": "child",
27 |        "service": "",
28 |        "resource": "child",
29 |        "trace_id": 0,
30 |        "span_id": 2,
31 |        "parent_id": 1,
32 |        "type": "",
33 |        "error": 0,
34 |        "duration": 9000,
35 |        "start": 1759940420748129000
36 |      }]]
37 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_trace_missing_received.json:
--------------------------------------------------------------------------------
 1 | [[
 2 |   {
 3 |     "name": "root0",
 4 |     "service": "",
 5 |     "resource": "root0",
 6 |     "trace_id": 0,
 7 |     "span_id": 1,
 8 |     "parent_id": 0,
 9 |     "type": "",
10 |     "error": 0,
11 |     "meta": {
12 |       "_dd.p.dm": "-0",
13 |       "_dd.p.tid": "68e68f4500000000",
14 |       "language": "python",
15 |       "runtime-id": "c9fafd4ce8e5490291a720c8c8ef660b"
16 |     },
17 |     "metrics": {
18 |       "_dd.top_level": 1,
19 |       "_dd.tracer_kr": 1.0,
20 |       "_sampling_priority_v1": 1,
21 |       "process_id": 44278
22 |     },
23 |     "duration": 68000,
24 |     "start": 1759940421492184000
25 |   },
26 |      {
27 |        "name": "child0",
28 |        "service": "",
29 |        "resource": "child0",
30 |        "trace_id": 0,
31 |        "span_id": 2,
32 |        "parent_id": 1,
33 |        "type": "",
34 |        "error": 0,
35 |        "duration": 15000,
36 |        "start": 1759940421492225000
37 |      }]]
38 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/apmtelemetry.py:
--------------------------------------------------------------------------------
 1 | import asyncio
 2 | import json
 3 | import logging
 4 | from typing import Any
 5 | from typing import Dict
 6 | from typing import cast
 7 | 
 8 | from aiohttp.web import Request
 9 | 
10 | 
11 | log = logging.getLogger(__name__)
12 | TelemetryEvent = Dict[str, Any]
13 | 
14 | 
15 | async def v2_decode_request(request: Request, data: bytes) -> TelemetryEvent:
16 |     headers = request.headers
17 | 
18 |     if "X-Datadog-Test-Stall-Seconds" in headers:
19 |         duration = float(headers["X-Datadog-Test-Stall-Seconds"])
20 |     else:
21 |         duration = request.app["trace_request_delay"]
22 |     if duration > 0:
23 |         log.info("Stalling for %r seconds.", duration)
24 |         await asyncio.sleep(duration)
25 |     return v2_decode(data)
26 | 
27 | 
28 | def v2_decode(data: bytes) -> TelemetryEvent:
29 |     """Decode v2 apm telemetry request data as a dict"""
30 |     # TODO: Handle decoding into a telemetry payload object
31 |     return cast(TelemetryEvent, json.loads(data))
32 | 


--------------------------------------------------------------------------------
/tests/test_tracestats.py:
--------------------------------------------------------------------------------
 1 | async def test_stats_basic(agent, do_reference_v06_http_stats):
 2 |     resp = await do_reference_v06_http_stats(token="test_stats_decode")
 3 |     assert resp.status == 200, await resp.text()
 4 | 
 5 |     resp = await agent.get(
 6 |         "/test/session/stats",
 7 |         headers={"X-Datadog-Test-Session-Token": "test_stats_decode"},
 8 |     )
 9 |     assert resp.status == 200, await resp.text()
10 |     body = await resp.json()
11 |     assert isinstance(body, list)
12 |     assert len(body) == 1
13 |     assert body[0]["Hostname"] == "Host-1234"
14 |     assert body[0]["Stats"][0]["Stats"][0]["Hits"] == 100
15 | 
16 | 
17 | async def test_stats_post(
18 |     agent,
19 |     v06_reference_http_stats_payload_headers,
20 |     v06_reference_http_stats_payload_data,
21 | ):
22 |     resp = await agent.post(
23 |         "/v0.6/stats",
24 |         headers=v06_reference_http_stats_payload_headers,
25 |         data=v06_reference_http_stats_payload_data,
26 |     )
27 |     assert resp.status == 200, await resp.text()
28 | 


--------------------------------------------------------------------------------
/releasenotes/check-releasenotes:
--------------------------------------------------------------------------------
 1 | #!/usr/bin/env bash
 2 | set -e
 3 | 
 4 | # If we are running outside a GitHub action, default to `master`
 5 | BASE_REF="${GITHUB_BASE_REF:-master}"
 6 | 
 7 | # Print input data
 8 | echo "Base ref: origin/${BASE_REF}"
 9 | echo "GitHub event path: ${GITHUB_EVENT_PATH}"
10 | echo "JQ: $(which jq)"
11 | 
12 | 
13 | # Skip the label check if we do not have a GitHub event path
14 | if [[ -f "${GITHUB_EVENT_PATH}" ]] && jq -e '.pull_request?.labels[]?.name | select(. == "no-releasenote")' "${GITHUB_EVENT_PATH}";
15 | then
16 |     echo "PR has label 'no-releasenote', skipping validation"
17 |     exit 0
18 | fi
19 | 
20 | # Check if they added a new file to releasenotes/notes
21 | if git diff --name-only --diff-filter=A "origin/${BASE_REF}" | grep releasenotes/notes;
22 | then
23 |     echo "New release note found, success"
24 |     exit 0
25 | else
26 |     echo "Release note not found."
27 |     echo "Use 'reno new <slug>' to add a new note to 'releasenotes/notes', or add the label 'no-releasenote' to skip this validation"
28 |     exit 1
29 | fi
30 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/tracerflare.py:
--------------------------------------------------------------------------------
 1 | import base64
 2 | from typing import Dict
 3 | 
 4 | from aiohttp import BodyPartReader
 5 | from aiohttp import MultipartReader
 6 | from aiohttp import StreamReader
 7 | from aiohttp.web import Request
 8 | 
 9 | 
10 | TracerFlareEvent = Dict[str, str]
11 | 
12 | 
13 | async def v1_decode(request: Request, data: bytes) -> TracerFlareEvent:
14 |     """Decode v1 tracer flare form as a dict"""
15 |     tracer_flare: TracerFlareEvent = {}
16 | 
17 |     try:
18 |         stream = StreamReader(request.protocol, len(data))
19 |         stream.feed_data(data)
20 |         stream.feed_eof()
21 |         async for part in MultipartReader(request.headers, stream):
22 |             if isinstance(part, BodyPartReader) and part.name:
23 |                 if part.name == "flare_file":
24 |                     tracer_flare[part.name] = base64.b64encode(await part.read()).decode("ascii")
25 |                 else:
26 |                     tracer_flare[part.name] = await part.text()
27 |     except Exception as err:
28 |         tracer_flare["error"] = str(err)
29 |     return tracer_flare
30 | 


--------------------------------------------------------------------------------
/releasenotes/notes/1.0-e192f5064e917083.yaml:
--------------------------------------------------------------------------------
 1 | ---
 2 | prelude: >
 3 |     The test agent has been rewritten in Python and the API being cleaned up
 4 |     substantially.
 5 | features:
 6 |   - |
 7 |     `/test/traces/` has been added to retrieve traces collected by the agent.
 8 |   - |
 9 |     `/test/session/traces/` has been added to retrieve traces collected by the
10 |     agent for a specific session.
11 |   - |
12 |     The test agent is now published to PyPI. It can be installed with `pip install ddapm-test-agent`.
13 | upgrade:
14 |   - |
15 |     `TEST_AGENT_PORT` has been renamed to `PORT`.
16 |   - |
17 |     `X-Datadog-Test-Token` has been renamed to `X-Datadog-Test-Session-Token`.
18 |   - |
19 |     `token` has been renamed to `test_session_token`.
20 |     `X-Datadog-Test-Token` has been renamed to `X-Datadog-Test-Session-Token`.
21 |   - |
22 |     The default snapshot directory has been changed from `snaps` to `snapshots`.
23 |   - |
24 |     The `/test/start` endpoint has been renamed to `/test/session/start`.
25 |   - |
26 |     The `/test/snapshot` endpoint has been renamed to `/test/session/snapshot`.
27 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/base.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html lang="en">
 3 | <head>
 4 |     <meta charset="UTF-8">
 5 |     <meta name="viewport" content="width=device-width, initial-scale=1.0">
 6 |     <title>{% block title %}Test Agent - {{ title }}{% endblock %}</title>
 7 |     <link rel="stylesheet" href="/static/style.css">
 8 | </head>
 9 | <body>
10 |     <nav class="navbar">
11 |         <div class="nav-container">
12 |             <div class="nav-brand">
13 |                 <h1>Test Agent</h1>
14 |             </div>
15 |             <ul class="nav-links">
16 |                 <li><a href="/" class="nav-link">Dashboard</a></li>
17 |                 <li><a href="/requests" class="nav-link">Requests</a></li>
18 |                 <li><a href="/config" class="nav-link">Remote Config</a></li>
19 |                 <li><a href="/tracerflares" class="nav-link">Tracer Flares</a></li>
20 |                 <li><a href="/snapshots" class="nav-link">Snapshots</a></li>
21 |             </ul>
22 |         </div>
23 |     </nav>
24 | 
25 |     <main class="main-content">
26 |         {% block content %}
27 |         {% endblock %}
28 |     </main>
29 | 
30 | </body>
31 | </html>


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/macros.html:
--------------------------------------------------------------------------------
 1 | {# Reusable template components #}
 2 | 
 3 | {# Page layout with consistent header and structure #}
 4 | {% macro page_layout(page_title, subtitle=None, extra_header=None) -%}
 5 | <div class="{{ page_title.lower() }}-page">
 6 |     <div class="page-header">
 7 |         <h2>{{ page_title }}</h2>
 8 |         {% if subtitle %}
 9 |             <div class="page-subtitle">
10 |                 <span>{{ subtitle }}</span>
11 |             </div>
12 |         {% endif %}
13 |         {% if extra_header %}
14 |             {{ extra_header | safe }}
15 |         {% endif %}
16 |     </div>
17 |     <div class="{{ page_title.lower() }}-container">
18 |         {{ caller() }}
19 |     </div>
20 | </div>
21 | {%- endmacro %}
22 | 
23 | {# Empty state component #}
24 | {% macro empty_state(message) -%}
25 | <div class="empty-state">
26 |     <p>{{ message }}</p>
27 | </div>
28 | {%- endmacro %}
29 | 
30 | {# Error message component #}
31 | {% macro error_message(message) -%}
32 | <div class="error-message">
33 |     <p>{{ message }}</p>
34 | </div>
35 | {%- endmacro %}
36 | 
37 | {# Action button component #}
38 | {% macro action_button(url, text, class_name="action-button") -%}
39 | <a href="{{ url }}" class="{{ class_name }}">{{ text }}</a>
40 | {%- endmacro %}


--------------------------------------------------------------------------------
/Dockerfile.windows:
--------------------------------------------------------------------------------
 1 | # escape=`
 2 | FROM python:3.13-windowsservercore
 3 | 
 4 | WORKDIR /src
 5 | EXPOSE 8126
 6 | 
 7 | ENV SNAPSHOT_CI=1
 8 | ENV LOG_LEVEL=INFO
 9 | ENV SNAPSHOT_DIR=C:\snapshots
10 | ENV VCR_CASSETTES_DIRECTORY=C:\vcr-cassettes
11 | 
12 | # Install git (required for setuptools_scm)
13 | RUN powershell -Command `
14 |     Invoke-WebRequest -Uri 'https://github.com/git-for-windows/git/releases/download/v2.47.1.windows.1/MinGit-2.47.1-64-bit.zip' -OutFile 'mingit.zip'; `
15 |     Expand-Archive -Path 'mingit.zip' -DestinationPath 'C:\git'; `
16 |     Remove-Item 'mingit.zip'; `
17 |     [Environment]::SetEnvironmentVariable('PATH', $env:PATH + ';C:\git\cmd', [EnvironmentVariableTarget]::Machine)
18 | 
19 | # Create directories
20 | RUN powershell -Command `
21 |     New-Item -ItemType Directory -Force -Path C:\snapshots; `
22 |     New-Item -ItemType Directory -Force -Path C:\vcr-cassettes
23 | 
24 | ADD vcr-cassettes C:\vcr-cassettes
25 | 
26 | # Add only necessary files to speed up development builds
27 | ADD README.md setup.py test_deps.txt ./
28 | ADD ddapm_test_agent ./ddapm_test_agent
29 | ADD .git ./.git
30 | 
31 | RUN pip install /src
32 | RUN powershell -Command "Remove-Item -Path $env:LOCALAPPDATA\pip\Cache -Recurse -Force -ErrorAction SilentlyContinue"
33 | 
34 | CMD ["ddapm-test-agent"]


--------------------------------------------------------------------------------
/setup.cfg:
--------------------------------------------------------------------------------
 1 | [flake8]
 2 | application-import-name=ddapm_test_agent
 3 | exclude=
 4 |   .riot,
 5 |   .git,
 6 |   .venv,
 7 |   .eggs,
 8 |   __pycache__,
 9 |   *.eggs-info,
10 |   build,
11 | # E501,E231,W503: not respected by black
12 | ignore = E501,W503,E203,E231,D100,D101,,D102,D103,D104,D105,D107,D202,D205,D400,D402,B902,I100,I202,A005
13 | enable-extensions=G
14 | import-order-style=google
15 | 
16 | [mypy]
17 | python_version = 3.11
18 | ignore_missing_imports = true
19 | disallow_incomplete_defs = true
20 | disallow_untyped_decorators = true
21 | warn_unused_ignores = true
22 | warn_unused_configs = true
23 | disallow_any_generics = true
24 | warn_redundant_casts = true
25 | warn_return_any = true
26 | strict_equality = true
27 | no_implicit_optional = true
28 | files = ddapm_test_agent,tests
29 | 
30 | [tool:pytest]
31 | addopts =
32 |     --doctest-modules
33 |     --doctest-glob="*.py"
34 |     --color=yes
35 |     --show-capture=all
36 |     --durations=10
37 |     # Pytest generates very long temp file paths (too long to be used by Unix sockets)
38 |     # So use a custom tmp directory instead.
39 |     --basetemp=/tmp/pytest
40 | testpaths =
41 |     ddapm_test_agent
42 |     tests
43 | log_level = DEBUG
44 | filterwarnings =
45 |     ignore::aiohttp.web_exceptions.NotAppKeyWarning
46 | # log_cli = 1
47 | # log_cli_level = DEBUG
48 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/session_detail.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | 
 3 | {% block content %}
 4 | <div class="session-detail-page">
 5 |     <div class="page-header">
 6 |         <h2>Session {{ token }}</h2>
 7 |     </div>
 8 | 
 9 |     {% if error %}
10 |         <div class="error-message">
11 |             <p>{{ error }}</p>
12 |         </div>
13 |     {% else %}
14 |         <div class="session-details">
15 |             <div class="session-info">
16 |                 <h3>Session Information</h3>
17 |                 <p><strong>Token:</strong> {{ token }}</p>
18 |                 <p><strong>Traces:</strong> {{ traces|length }}</p>
19 |             </div>
20 | 
21 |             {% if traces %}
22 |                 <div class="session-traces">
23 |                     <h3>Traces in this Session</h3>
24 |                     {% for trace in traces %}
25 |                     <div class="trace-summary">
26 |                         <p><strong>Spans:</strong> {{ trace|length }}</p>
27 |                         {% if trace %}
28 |                             <p><strong>Root Service:</strong> {{ trace[0].get('service', 'N/A') }}</p>
29 |                         {% endif %}
30 |                     </div>
31 |                     {% endfor %}
32 |                 </div>
33 |             {% endif %}
34 |         </div>
35 |     {% endif %}
36 | </div>
37 | {% endblock %}


--------------------------------------------------------------------------------
/.github/workflows/releasenote.yml:
--------------------------------------------------------------------------------
 1 | name: Release notes
 2 | on:
 3 |   push:
 4 |     branches:
 5 |       - master
 6 |   pull_request:
 7 |     # Important that we run on `labeled` and `unlabeled` to pick up `no-releasenote` being added/removed
 8 |     # DEV: [opened, reopened, synchronize] is the default
 9 |     types: [opened, reopened, synchronize, labeled, unlabeled, ready_for_review]
10 | jobs:
11 |   validate:
12 |     name: Validate release notes
13 |     runs-on: ubuntu-latest
14 |     steps:
15 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
16 |         # Include all history and tags
17 |         with:
18 |           fetch-depth: 0
19 | 
20 |       # Ensure a new reno release note was added in this PR.
21 |       # Use `reno new <slug>` to add a new note to `no-releasenote`,
22 |       #   or add `no-releasenote` label if no release note is needed.
23 |       - name: Ensure release note added
24 |         # Only run this on pull requests
25 |         if: github.event_name == 'pull_request'
26 |         run: releasenotes/check-releasenotes
27 | 
28 |       - uses: actions/setup-python@e9aba2c848f5ebd159c070c61ea2c4e2b122355e # v2.3.4
29 |         name: Install Python
30 |         with:
31 |           python-version: '3.12'
32 | 
33 |       - name: Install Dependencies
34 |         run: pip install reno docutils
35 | 
36 |       - name: Lint release notes
37 |         run: reno lint
38 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/trace_detail.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | 
 3 | {% block content %}
 4 | <div class="trace-detail-page">
 5 |     <div class="page-header">
 6 |         <h2>Trace {{ trace_id }}</h2>
 7 |     </div>
 8 | 
 9 |     {% if error %}
10 |         <div class="error-message">
11 |             <p>{{ error }}</p>
12 |         </div>
13 |     {% elif trace %}
14 |         <div class="trace-details">
15 |             <div class="spans-list">
16 |                 {% for span in trace %}
17 |                 <div class="span-item">
18 |                     <div class="span-header">
19 |                         <h4>{{ span.get('name', 'Unknown') }}</h4>
20 |                         <span class="span-id">ID: {{ span.get('span_id') }}</span>
21 |                     </div>
22 |                     <div class="span-details">
23 |                         <p><strong>Service:</strong> {{ span.get('service', 'N/A') }}</p>
24 |                         <p><strong>Resource:</strong> {{ span.get('resource', 'N/A') }}</p>
25 |                         <p><strong>Duration:</strong> {{ span.get('duration', 'N/A') }}</p>
26 |                     </div>
27 |                 </div>
28 |                 {% endfor %}
29 |             </div>
30 |         </div>
31 |     {% else %}
32 |         <div class="empty-state">
33 |             <p>No trace data available</p>
34 |         </div>
35 |     {% endif %}
36 | </div>
37 | {% endblock %}


--------------------------------------------------------------------------------
/LICENSE.BSD3:
--------------------------------------------------------------------------------
 1 | Copyright (c) 2016, Datadog <info@datadoghq.com>
 2 | All rights reserved.
 3 | 
 4 | Redistribution and use in source and binary forms, with or without
 5 | modification, are permitted provided that the following conditions are met:
 6 |     * Redistributions of source code must retain the above copyright
 7 |       notice, this list of conditions and the following disclaimer.
 8 |     * Redistributions in binary form must reproduce the above copyright
 9 |       notice, this list of conditions and the following disclaimer in the
10 |       documentation and/or other materials provided with the distribution.
11 |     * Neither the name of Datadog nor the
12 |       names of its contributors may be used to endorse or promote products
13 |       derived from this software without specific prior written permission.
14 | 
15 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
16 | ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 | WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 | DISCLAIMED. IN NO EVENT SHALL DATADOG BE LIABLE FOR ANY
19 | DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20 | (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21 | LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22 | ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24 | SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_multi_trace.json:
--------------------------------------------------------------------------------
 1 | [[
 2 |   {
 3 |     "name": "root0",
 4 |     "service": "",
 5 |     "resource": "root0",
 6 |     "trace_id": 0,
 7 |     "span_id": 1,
 8 |     "parent_id": 0,
 9 |     "type": "",
10 |     "error": 0,
11 |     "meta": {
12 |       "_dd.p.dm": "-0",
13 |       "_dd.p.tid": "68e68f4300000000",
14 |       "language": "python",
15 |       "runtime-id": "a17fe963cb6c4dfaaf0bbb73d29976e9"
16 |     },
17 |     "metrics": {
18 |       "_dd.top_level": 1,
19 |       "_dd.tracer_kr": 1.0,
20 |       "_sampling_priority_v1": 1,
21 |       "process_id": 44199
22 |     },
23 |     "duration": 89000,
24 |     "start": 1759940419298507000
25 |   },
26 |      {
27 |        "name": "child0",
28 |        "service": "",
29 |        "resource": "child0",
30 |        "trace_id": 0,
31 |        "span_id": 2,
32 |        "parent_id": 1,
33 |        "type": "",
34 |        "error": 0,
35 |        "duration": 14000,
36 |        "start": 1759940419298570000
37 |      }],
38 | [
39 |   {
40 |     "name": "root1",
41 |     "service": "",
42 |     "resource": "root1",
43 |     "trace_id": 1,
44 |     "span_id": 1,
45 |     "parent_id": 0,
46 |     "type": "",
47 |     "error": 0,
48 |     "meta": {
49 |       "_dd.p.dm": "-0",
50 |       "_dd.p.tid": "68e68f4300000000",
51 |       "language": "python",
52 |       "runtime-id": "a17fe963cb6c4dfaaf0bbb73d29976e9"
53 |     },
54 |     "metrics": {
55 |       "_dd.top_level": 1,
56 |       "_dd.tracer_kr": 1.0,
57 |       "_sampling_priority_v1": 1,
58 |       "process_id": 44199
59 |     },
60 |     "duration": 34000,
61 |     "start": 1759940419314834000
62 |   },
63 |      {
64 |        "name": "child1",
65 |        "service": "",
66 |        "resource": "child1",
67 |        "trace_id": 1,
68 |        "span_id": 2,
69 |        "parent_id": 1,
70 |        "type": "",
71 |        "error": 0,
72 |        "duration": 7000,
73 |        "start": 1759940419314856000
74 |      }]]
75 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_models_gpt-4_get_b13c5b23.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       User-Agent:
12 |       - OpenAI/NodeJS/3.0.0
13 |     method: GET
14 |     uri: https://api.openai.com/v1/models/gpt-4
15 |   response:
16 |     body:
17 |       string: "{\n  \"id\": \"gpt-4\",\n  \"object\": \"model\",\n  \"created\": 1687882411,\n
18 |         \ \"owned_by\": \"openai\"\n}"
19 |     headers:
20 |       CF-RAY:
21 |       - 951540766be938dd-IAD
22 |       Connection:
23 |       - keep-alive
24 |       Content-Encoding:
25 |       - gzip
26 |       Content-Type:
27 |       - application/json
28 |       Date:
29 |       - Tue, 17 Jun 2025 20:21:11 GMT
30 |       Server:
31 |       - cloudflare
32 |       Set-Cookie:
33 |       - __cf_bm=hTLEdkQKk_HOs8DnXPLfgXlnmK6u6HqZMwSkHwWcIDI-1750191671-1.0.1.1-E6GEEMQHZ4Fw89j8QE3jbYGBqNytQ7_P2tCX9KhC.ZxCI7DgYjNdbEePerNf20jWPsCBfSnFp3EJcbtOGNdV7.g55V1xcm1H6Pb2d0tmeTA;
34 |         path=/; expires=Tue, 17-Jun-25 20:51:11 GMT; domain=.api.openai.com; HttpOnly;
35 |         Secure; SameSite=None
36 |       - _cfuvid=4mKS2ZVVbbgOhcxLUtSnrVT50sE1QKhSr9XNcHEbzxs-1750191671971-0.0.1.1-604800000;
37 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
38 |       Transfer-Encoding:
39 |       - chunked
40 |       X-Content-Type-Options:
41 |       - nosniff
42 |       alt-svc:
43 |       - h3=":443"; ma=86400
44 |       cf-cache-status:
45 |       - DYNAMIC
46 |       openai-processing-ms:
47 |       - '71'
48 |       openai-version:
49 |       - '2020-10-01'
50 |       strict-transport-security:
51 |       - max-age=31536000; includeSubDomains; preload
52 |       x-request-id:
53 |       - 8d21e05f908d4a94606a18cda0956050
54 |     status:
55 |       code: 200
56 |       message: OK
57 | version: 1
58 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_files_file-RpTpuvRVtnKpdKZb7DDGto_delete_3ee48409.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '0'
13 |       User-Agent:
14 |       - OpenAI/NodeJS/3.0.0
15 |     method: DELETE
16 |     uri: https://api.openai.com/v1/files/file-RpTpuvRVtnKpdKZb7DDGto
17 |   response:
18 |     body:
19 |       string: "{\n  \"object\": \"file\",\n  \"deleted\": true,\n  \"id\": \"file-RpTpuvRVtnKpdKZb7DDGto\"\n}\n"
20 |     headers:
21 |       CF-RAY:
22 |       - 952d08ef7dc88fb5-BOS
23 |       Connection:
24 |       - keep-alive
25 |       Content-Encoding:
26 |       - gzip
27 |       Content-Type:
28 |       - application/json
29 |       Date:
30 |       - Fri, 20 Jun 2025 17:37:35 GMT
31 |       Server:
32 |       - cloudflare
33 |       Set-Cookie:
34 |       - __cf_bm=MCoPJ5omfnrx_gkAal1RafE3bxIY0oRvm3MFnLqZ6Rk-1750441055-1.0.1.1-rwYZM5vgXzvoLzO9sz74tXTKUI1e2.15eUEaDK1nDtgFxIIYs9VJM9EVkkG.NDMZHDCa1BTPrVFAk9tZaEXeM7eofW_BJala9n0I_h_welU;
35 |         path=/; expires=Fri, 20-Jun-25 18:07:35 GMT; domain=.api.openai.com; HttpOnly;
36 |         Secure; SameSite=None
37 |       - _cfuvid=DLn6ZZ1zb_07PDhRsO9LiZDU9FvXq6mnChI6qeIqM90-1750441055075-0.0.1.1-604800000;
38 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
39 |       Transfer-Encoding:
40 |       - chunked
41 |       X-Content-Type-Options:
42 |       - nosniff
43 |       access-control-allow-origin:
44 |       - '*'
45 |       alt-svc:
46 |       - h3=":443"; ma=86400
47 |       cf-cache-status:
48 |       - DYNAMIC
49 |       openai-organization:
50 |       - datadog-staging
51 |       openai-processing-ms:
52 |       - '150'
53 |       openai-version:
54 |       - '2020-10-01'
55 |       strict-transport-security:
56 |       - max-age=31536000; includeSubDomains; preload
57 |       x-request-id:
58 |       - req_81dd6e59809460874a67de5ab494362c
59 |     status:
60 |       code: 200
61 |       message: OK
62 | version: 1
63 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_trace_distributed_same_payload.json:
--------------------------------------------------------------------------------
 1 | [[
 2 |   {
 3 |     "name": "root0",
 4 |     "service": "",
 5 |     "resource": "root0",
 6 |     "trace_id": 0,
 7 |     "span_id": 1,
 8 |     "parent_id": 0,
 9 |     "type": "",
10 |     "error": 0,
11 |     "meta": {
12 |       "_dd.p.dm": "-0",
13 |       "_dd.p.tid": "68e68f4400000000",
14 |       "language": "python",
15 |       "runtime-id": "4462e49f0f9b432ab62de16a0798af9f"
16 |     },
17 |     "metrics": {
18 |       "_dd.top_level": 1,
19 |       "_dd.tracer_kr": 1.0,
20 |       "_sampling_priority_v1": 1,
21 |       "process_id": 44234
22 |     },
23 |     "duration": 65000,
24 |     "start": 1759940420028363000
25 |   },
26 |      {
27 |        "name": "child0",
28 |        "service": "",
29 |        "resource": "child0",
30 |        "trace_id": 0,
31 |        "span_id": 2,
32 |        "parent_id": 1,
33 |        "type": "",
34 |        "error": 0,
35 |        "duration": 12000,
36 |        "start": 1759940420028405000
37 |      },
38 |         {
39 |           "name": "root1",
40 |           "service": "",
41 |           "resource": "root1",
42 |           "trace_id": 0,
43 |           "span_id": 3,
44 |           "parent_id": 2,
45 |           "type": "",
46 |           "error": 0,
47 |           "meta": {
48 |             "_dd.p.dm": "-0",
49 |             "_dd.p.tid": "68e68f4400000000",
50 |             "language": "python",
51 |             "runtime-id": "4462e49f0f9b432ab62de16a0798af9f"
52 |           },
53 |           "metrics": {
54 |             "_dd.top_level": 1,
55 |             "_dd.tracer_kr": 1.0,
56 |             "_sampling_priority_v1": 1,
57 |             "process_id": 44234
58 |           },
59 |           "duration": 26000,
60 |           "start": 1759940420042497000
61 |         },
62 |            {
63 |              "name": "child1",
64 |              "service": "",
65 |              "resource": "child1",
66 |              "trace_id": 0,
67 |              "span_id": 4,
68 |              "parent_id": 3,
69 |              "type": "",
70 |              "error": 0,
71 |              "duration": 7000,
72 |              "start": 1759940420042512000
73 |            }]]
74 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_models_ft_gpt-4.1-mini-2025-04-14_datadog-staging__BkaILRSh_delete_c70f2992.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '0'
13 |       User-Agent:
14 |       - OpenAI/NodeJS/3.0.0
15 |     method: DELETE
16 |     uri: https://api.openai.com/v1/models/ft:gpt-4.1-mini-2025-04-14:datadog-staging::BkaILRSh
17 |   response:
18 |     body:
19 |       string: "{\n  \"id\": \"ft:gpt-4.1-mini-2025-04-14:datadog-staging::BkaILRSh\",\n  \"object\":
20 |         \"model\",\n  \"deleted\": true,\n  \"owned_by\": \"openai-internal\"\n}\n"
21 |     headers:
22 |       CF-RAY:
23 |       - 952d4b555b924cf8-BOS
24 |       Connection:
25 |       - keep-alive
26 |       Content-Encoding:
27 |       - gzip
28 |       Content-Type:
29 |       - application/json
30 |       Date:
31 |       - Fri, 20 Jun 2025 18:22:54 GMT
32 |       Server:
33 |       - cloudflare
34 |       Set-Cookie:
35 |       - __cf_bm=0rbg4_AraJjttMovkzOwzGxCJGxjNqj6hu5ztY9psU0-1750443774-1.0.1.1-clTbG2KCuJXnGVuozRV2XWtb52oYAOg2YjOeS04sUdvO4ULdq4UL78T9YfW53mr9fqZUOOgrXY4f3iHyW2kEaC6m9_J0tVX9GwXdrl3X0Rk;
36 |         path=/; expires=Fri, 20-Jun-25 18:52:54 GMT; domain=.api.openai.com; HttpOnly;
37 |         Secure; SameSite=None
38 |       - _cfuvid=OyrB3il.9uQDkze0RiSQ.14bQJjLxfkrJ_aWei3U5fs-1750443774617-0.0.1.1-604800000;
39 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
40 |       Transfer-Encoding:
41 |       - chunked
42 |       X-Content-Type-Options:
43 |       - nosniff
44 |       access-control-allow-origin:
45 |       - '*'
46 |       alt-svc:
47 |       - h3=":443"; ma=86400
48 |       cf-cache-status:
49 |       - DYNAMIC
50 |       openai-organization:
51 |       - datadog-staging
52 |       openai-processing-ms:
53 |       - '37'
54 |       openai-version:
55 |       - '2020-10-01'
56 |       strict-transport-security:
57 |       - max-age=31536000; includeSubDomains; preload
58 |       x-request-id:
59 |       - req_88ad71a5e16b94df838479a8b3e6f990
60 |     status:
61 |       code: 200
62 |       message: Forbidden
63 | version: 1
64 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_files_file-RpTpuvRVtnKpdKZb7DDGto_get_1afefb93.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       User-Agent:
12 |       - OpenAI/NodeJS/3.0.0
13 |     method: GET
14 |     uri: https://api.openai.com/v1/files/file-RpTpuvRVtnKpdKZb7DDGto
15 |   response:
16 |     body:
17 |       string: "{\n  \"object\": \"file\",\n  \"id\": \"file-RpTpuvRVtnKpdKZb7DDGto\",\n
18 |         \ \"purpose\": \"fine-tune\",\n  \"filename\": \"fine-tune.jsonl\",\n  \"bytes\":
19 |         1386,\n  \"created_at\": 1750424589,\n  \"expires_at\": null,\n  \"status\":
20 |         \"processed\",\n  \"status_details\": null\n}\n"
21 |     headers:
22 |       CF-RAY:
23 |       - 952b78e66f4e4ced-BOS
24 |       Connection:
25 |       - keep-alive
26 |       Content-Encoding:
27 |       - gzip
28 |       Content-Type:
29 |       - application/json
30 |       Date:
31 |       - Fri, 20 Jun 2025 13:04:29 GMT
32 |       Server:
33 |       - cloudflare
34 |       Set-Cookie:
35 |       - __cf_bm=iTIpJyjcSLMLc_1kT90oyUz6gvRRU_kmOwjkGOBpfCw-1750424669-1.0.1.1-kabbQH5a9vEByKMZagYWcu4tBeOiVLMgTFWtnUmC955HtYF1_0imQbBaxpbxJ_C4c_OEm0OiyuE6OMFKqUzx6Wzq_ckw55G5USwMoQ3IYjk;
36 |         path=/; expires=Fri, 20-Jun-25 13:34:29 GMT; domain=.api.openai.com; HttpOnly;
37 |         Secure; SameSite=None
38 |       - _cfuvid=3.EMyzsipqW41yeKaakMYINOMYwdNPyB2QCzyDkZIgo-1750424669466-0.0.1.1-604800000;
39 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
40 |       Transfer-Encoding:
41 |       - chunked
42 |       X-Content-Type-Options:
43 |       - nosniff
44 |       access-control-allow-origin:
45 |       - '*'
46 |       alt-svc:
47 |       - h3=":443"; ma=86400
48 |       cf-cache-status:
49 |       - DYNAMIC
50 |       openai-organization:
51 |       - datadog-staging
52 |       openai-processing-ms:
53 |       - '37'
54 |       openai-version:
55 |       - '2020-10-01'
56 |       strict-transport-security:
57 |       - max-age=31536000; includeSubDomains; preload
58 |       x-request-id:
59 |       - req_c524603ac776cac5708cdc1d9d047f61
60 |     status:
61 |       code: 200
62 |       message: OK
63 | version: 1
64 | 


--------------------------------------------------------------------------------
/riotfile.py:
--------------------------------------------------------------------------------
 1 | from riot import Venv
 2 | from riot import latest
 3 | 
 4 | 
 5 | with open("test_deps.txt") as f:
 6 |     testing_deps = [line.strip() for line in f.readlines()]
 7 | 
 8 | 
 9 | venv = Venv(
10 |     pys=["3"],
11 |     venvs=[
12 |         Venv(
13 |             name="test",
14 |             command="pytest {cmdargs}",
15 |             pys=["3.11", "3.12", "3.13"],
16 |             pkgs={pkg: latest for pkg in testing_deps},
17 |         ),
18 |         Venv(
19 |             pkgs={
20 |                 "black": latest,
21 |                 "isort": latest,
22 |                 "toml": latest,
23 |             },
24 |             pys=["3.11"],
25 |             venvs=[
26 |                 Venv(
27 |                     name="black",
28 |                     command="black {cmdargs}",
29 |                 ),
30 |                 Venv(
31 |                     name="fmt",
32 |                     command="isort . && black .",
33 |                 ),
34 |                 Venv(
35 |                     name="check_fmt",
36 |                     command="isort --check . && black --check .",
37 |                 ),
38 |             ],
39 |         ),
40 |         Venv(
41 |             name="flake8",
42 |             command="flake8 {cmdargs}",
43 |             pys=["3.11"],
44 |             pkgs={
45 |                 "flake8": latest,
46 |                 "flake8-blind-except": latest,
47 |                 "flake8-builtins": latest,
48 |                 "flake8-docstrings": latest,
49 |                 # "flake8-logging-format": latest,
50 |                 "flake8-rst-docstrings": latest,
51 |                 "pygments": latest,
52 |                 "toml": latest,
53 |             },
54 |         ),
55 |         Venv(
56 |             name="mypy",
57 |             create=True,
58 |             command="mypy {cmdargs}",
59 |             pys=["3.11"],
60 |             pkgs={
61 |                 "mypy": "==1.7.1",
62 |                 "pytest": latest,
63 |                 "types-protobuf": latest,
64 |                 "types-requests": latest,
65 |                 "types-setuptools": latest,
66 |                 "types-PyYAML": latest,
67 |             },
68 |         ),
69 |         Venv(
70 |             name="reno",
71 |             pkgs={
72 |                 "reno": latest,
73 |             },
74 |             command="reno {cmdargs}",
75 |         ),
76 |     ],
77 | )
78 | 


--------------------------------------------------------------------------------
/setup.py:
--------------------------------------------------------------------------------
 1 | from setuptools import find_packages
 2 | from setuptools import setup
 3 | 
 4 | 
 5 | with open("README.md", "r") as f:
 6 |     long_description = f.read()
 7 | 
 8 | with open("test_deps.txt") as f:
 9 |     testing_deps = [line.strip() for line in f.readlines()]
10 | 
11 | setup(
12 |     name="ddapm-test-agent",
13 |     description="Test agent for Datadog APM client libraries",
14 |     url="https://github.com/Datadog/dd-apm-test-agent",
15 |     author="Kyle Verhoog",
16 |     author_email="kyle@verhoog.ca",
17 |     classifiers=[
18 |         "Programming Language :: Python",
19 |         "Programming Language :: Python :: 3.11",
20 |         "Programming Language :: Python :: 3.12",
21 |         "Programming Language :: Python :: 3.13",
22 |     ],
23 |     long_description=long_description,
24 |     long_description_content_type="text/markdown",
25 |     license="BSD 3",
26 |     packages=find_packages(exclude=["tests*", "releasenotes", "scripts"]),
27 |     package_data={"ddapm_test_agent": ["py.typed", "templates/*", "static/*"]},
28 |     python_requires=">=3.8",
29 |     install_requires=[
30 |         "aiohttp",
31 |         "ddsketch[serialization]",
32 |         "msgpack",
33 |         "requests",
34 |         "typing_extensions",
35 |         "yarl",
36 |         "requests-aws4auth",
37 |         "jinja2>=3.0.0",
38 |         "pyyaml",
39 |         # ddtrace libraries officially support opentelemetry-proto 1.33.1
40 |         # which implements the v1.7.0 spec
41 |         "opentelemetry-proto>1.33.0,<1.37.0",
42 |         "protobuf>=3.19.0",
43 |         "grpcio>=1.66.2,<2.0",
44 |         "pywin32; sys_platform == 'win32'",
45 |     ],
46 |     tests_require=testing_deps,
47 |     setup_requires=["setuptools_scm"],
48 |     use_scm_version=True,
49 |     entry_points={
50 |         "console_scripts": [
51 |             "ddapm-test-agent=ddapm_test_agent.agent:main",
52 |             "ddapm-test-agent-fmt=ddapm_test_agent.fmt:main",
53 |             "ddapm-test-agent-session-start=ddapm_test_agent.cmd:main_session_start",
54 |             "ddapm-test-agent-snapshot=ddapm_test_agent.cmd:main_snapshot",
55 |         ]
56 |     },
57 |     extras_require={
58 |         "testing": testing_deps,
59 |     },
60 |     # Required for mypy compatibility, see
61 |     # https://mypy.readthedocs.io/en/stable/installed_packages.html#making-pep-561-compatible-packages
62 |     zip_safe=False,
63 | )
64 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/tracestats.py:
--------------------------------------------------------------------------------
 1 | from typing import List
 2 | from typing import TypedDict
 3 | 
 4 | from ddsketch import DDSketch
 5 | from ddsketch.pb.ddsketch_pb2 import DDSketch as DDSketchProto
 6 | import msgpack
 7 | 
 8 | 
 9 | # Note that class attributes are golang style to match the payload.
10 | 
11 | 
12 | class StatsAggr(TypedDict):
13 |     Name: str
14 |     Resource: str
15 |     Service: str
16 |     Type: str
17 |     HTTPStatusCode: int
18 |     Synthetics: bool
19 |     Hits: int
20 |     TopLevelHits: int
21 |     Duration: int
22 |     Errors: int
23 |     OkSummary: DDSketch
24 |     ErrorSummary: DDSketch
25 | 
26 | 
27 | class StatsBucket(TypedDict):
28 |     Start: int
29 |     Duration: int
30 |     Stats: List[StatsAggr]
31 | 
32 | 
33 | class v06StatsPayload(TypedDict):
34 |     Hostname: str
35 |     Env: str
36 |     Version: str
37 |     Stats: List[StatsBucket]
38 | 
39 | 
40 | def decode_v06(data: bytes) -> v06StatsPayload:
41 |     payload = msgpack.unpackb(data)
42 |     stats_buckets: List[StatsBucket] = []
43 |     for raw_bucket in payload["Stats"]:
44 |         stats: List[StatsAggr] = []
45 |         for raw_stats in raw_bucket["Stats"]:
46 |             stat = StatsAggr(
47 |                 Name=raw_stats["Name"],
48 |                 Resource=raw_stats["Resource"],
49 |                 Service=raw_stats.get("Service") or "",
50 |                 Type=raw_stats.get("Type") or "",
51 |                 HTTPStatusCode=raw_stats.get("HTTPStatusCode") or 0,
52 |                 Synthetics=raw_stats["Synthetics"],
53 |                 Hits=raw_stats["Hits"],
54 |                 TopLevelHits=raw_stats["TopLevelHits"],
55 |                 Duration=raw_stats["Duration"],
56 |                 Errors=raw_stats["Errors"],
57 |                 OkSummary=DDSketchProto().ParseFromString(raw_stats["OkSummary"]),
58 |                 ErrorSummary=DDSketchProto().ParseFromString(raw_stats["ErrorSummary"]),
59 |             )
60 |             stats.append(stat)
61 | 
62 |         bucket = StatsBucket(
63 |             Start=raw_bucket["Start"],
64 |             Duration=raw_bucket["Duration"],
65 |             Stats=stats,
66 |         )
67 |         stats_buckets.append(bucket)
68 | 
69 |     return v06StatsPayload(
70 |         Hostname=payload.get("Hostname", ""),
71 |         Env=payload.get("Env", ""),
72 |         Version=payload.get("Version", ""),
73 |         Stats=stats_buckets,
74 |     )
75 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_files_get_68efa25d.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       User-Agent:
12 |       - OpenAI/NodeJS/3.0.0
13 |     method: GET
14 |     uri: https://api.openai.com/v1/files
15 |   response:
16 |     body:
17 |       string: "{\n  \"object\": \"list\",\n  \"data\": [\n    {\n  \"object\": \"file\",\n  \"id\": \"file-VkHpbu69EdKZ3bbRtjeptc\",\n
18 |         \ \"purpose\": \"fine-tune\",\n  \"filename\": \"fine-tune.jsonl\",\n  \"bytes\":
19 |         62,\n  \"created_at\": 1750265397,\n  \"expires_at\": null,\n  \"status\":
20 |         \"processed\",\n  \"status_details\": null\n}\n  ],\n  \"has_more\":
21 |         false,\n  \"first_id\": \"file-E9JJ2o2RwQg8HPKS4VZ8T8\",\n  \"last_id\": \"file-VkHpbu69EdKZ3bbRtjeptc\"\n}\n"
22 |     headers:
23 |       CF-RAY:
24 |       - 951586261c9f379e-IAD
25 |       Connection:
26 |       - keep-alive
27 |       Content-Encoding:
28 |       - gzip
29 |       Content-Type:
30 |       - application/json
31 |       Date:
32 |       - Tue, 17 Jun 2025 21:08:46 GMT
33 |       Server:
34 |       - cloudflare
35 |       Set-Cookie:
36 |       - __cf_bm=ba.2HPdsF0Gehmqx24l.R1sKPCKntjMpA0J58ZfNYfE-1750194526-1.0.1.1-FlloGo5PDLgBGy2c_yub0tfKlyMNvLtzbeT5ZcDogJUsLx7eyphJg1inzsIdenkZeffjqhHt7Y3JzgChfz5sPdlW3pgKs6eqqSbGK7Y0Jzk;
37 |         path=/; expires=Tue, 17-Jun-25 21:38:46 GMT; domain=.api.openai.com; HttpOnly;
38 |         Secure; SameSite=None
39 |       - _cfuvid=rOL_2O1ii6SiV1YoOe1afD9HnWAh7c8D6gmrdAwGoRQ-1750194526617-0.0.1.1-604800000;
40 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
41 |       Transfer-Encoding:
42 |       - chunked
43 |       X-Content-Type-Options:
44 |       - nosniff
45 |       access-control-allow-origin:
46 |       - '*'
47 |       alt-svc:
48 |       - h3=":443"; ma=86400
49 |       cf-cache-status:
50 |       - DYNAMIC
51 |       openai-organization:
52 |       - datadog-staging
53 |       openai-processing-ms:
54 |       - '1274'
55 |       openai-version:
56 |       - '2020-10-01'
57 |       strict-transport-security:
58 |       - max-age=31536000; includeSubDomains; preload
59 |       x-request-id:
60 |       - req_7f4a10ea591871dd9baadf92dc9df583
61 |     status:
62 |       code: 200
63 |       message: OK
64 | version: 1
65 | 


--------------------------------------------------------------------------------
/tests/test_tracerflare.py:
--------------------------------------------------------------------------------
 1 | from aiohttp import FormData
 2 | 
 3 | 
 4 | async def test_tracerflare(agent):
 5 |     expected_output = {
 6 |         "source": "tracer_test",
 7 |         "case_id": "12345",
 8 |         "email": "its.me@datadoghq.com",
 9 |         "hostname": "my.hostname",
10 |         "flare_file": "UEsFBgAAAAAAAAAAAAAAAAAAAAAAAA==",
11 |     }
12 |     form = FormData()
13 |     form.add_field("source", "tracer_test")
14 |     form.add_field("case_id", "12345")
15 |     form.add_field("email", "its.me@datadoghq.com")
16 |     form.add_field("hostname", "my.hostname")
17 |     form.add_field(
18 |         "flare_file",
19 |         b"PK\x05\x06\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0",
20 |         filename="test-flare.zip",
21 |         content_type="application/octet-stream",
22 |     )
23 |     resp = await agent.post("/tracer_flare/v1", data=form)
24 |     assert resp.status == 200
25 |     flares = await agent.get("/test/session/tracerflares")
26 |     assert await flares.json() == [expected_output]
27 | 
28 | 
29 | async def test_tracerflare_missing_case_id(agent):
30 |     expected_output = {
31 |         "source": "tracer_test",
32 |         "email": "its.me@datadoghq.com",
33 |         "hostname": "my.hostname",
34 |         "flare_file": "UEsFBgAAAAAAAAAAAAAAAAAAAAAAAA==",
35 |     }
36 |     form = FormData()
37 |     form.add_field("source", "tracer_test")
38 |     form.add_field("email", "its.me@datadoghq.com")
39 |     form.add_field("hostname", "my.hostname")
40 |     form.add_field(
41 |         "flare_file",
42 |         b"PK\x05\x06\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0",
43 |         filename="test-flare.zip",
44 |         content_type="application/octet-stream",
45 |     )
46 |     resp = await agent.post("/tracer_flare/v1", data=form)
47 |     assert resp.status == 400
48 |     flares = await agent.get("/test/session/tracerflares")
49 |     assert await flares.json() == [expected_output]
50 | 
51 | 
52 | async def test_tracerflare_not_multipart(agent):
53 |     expected_output = {
54 |         "error": "multipart/* content type expected",
55 |     }
56 |     form = {
57 |         "source": "tracer_test",
58 |         "case_id": "12345",
59 |         "email": "its.me@datadoghq.com",
60 |         "hostname": "my.hostname",
61 |     }
62 |     resp = await agent.post("/tracer_flare/v1", data=form)
63 |     assert resp.status == 400
64 |     flares = await agent.get("/test/session/tracerflares")
65 |     assert await flares.json() == [expected_output]
66 | 


--------------------------------------------------------------------------------
/.github/workflows/publish_pypi.yml:
--------------------------------------------------------------------------------
 1 | name: PyPI
 2 | 
 3 | on:
 4 |   pull_request:
 5 |   release:
 6 |     types:
 7 |       - published
 8 | 
 9 | jobs:
10 |   build_wheel:
11 |     name: Build wheels
12 |     runs-on: ubuntu-latest
13 | 
14 |     steps:
15 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
16 |         # Include all history and tags
17 |         with:
18 |           fetch-depth: 0
19 | 
20 |       - uses: actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065 # v5.6.0
21 |         name: Install Python
22 |         with:
23 |           python-version: '3.12'
24 | 
25 |       - name: Build wheels
26 |         run: |
27 |           pip install wheel
28 |           pip wheel --no-deps -w dist .
29 | 
30 |       - uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
31 |         with:
32 |           name: wheel
33 |           path: dist/*.whl
34 | 
35 |   build_sdist:
36 |     name: Build source distribution
37 |     runs-on: ubuntu-latest
38 |     steps:
39 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
40 |         # Include all history and tags
41 |         with:
42 |           fetch-depth: 0
43 | 
44 |       - uses: actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065 # v5.6.0
45 |         name: Install Python
46 |         with:
47 |           python-version: '3.12'
48 | 
49 |       - name: Build sdist
50 |         run: |
51 |           python -m pip install --upgrade setuptools
52 |           python setup.py sdist
53 | 
54 |       - uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
55 |         with:
56 |           name: sdist
57 |           path: dist/*.tar.gz
58 | 
59 |   upload_pypi:
60 |     needs: [build_wheel, build_sdist]
61 |     runs-on: ubuntu-latest
62 |     if: github.event_name == 'release' && github.event.action == 'published'
63 |     steps:
64 |       - uses: actions/download-artifact@d3f86a106a0bac45b974a628896c90dbdf5c8093 # v4.3.0
65 |         with:
66 |           name: sdist
67 |           path: dist
68 | 
69 |       - uses: actions/download-artifact@d3f86a106a0bac45b974a628896c90dbdf5c8093 # v4.3.0
70 |         with:
71 |           name: wheel
72 |           path: dist
73 | 
74 |       - uses: pypa/gh-action-pypi-publish@76f52bc884231f62b9a034ebfe128415bbaabdfc # v1.12.4
75 |         with:
76 |           user: __token__
77 |           password: ${{ secrets.PYPI_TOKEN }}
78 |           # To test: repository_url: https://test.pypi.org/legacy/
79 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/cmd.py:
--------------------------------------------------------------------------------
 1 | import argparse
 2 | import os
 3 | import sys
 4 | 
 5 | import requests
 6 | import yarl
 7 | 
 8 | 
 9 | def _add_token_arg(parser: argparse.ArgumentParser) -> None:
10 |     """Add the test session token argument to the parser."""
11 |     parser.add_argument(
12 |         "--test-session-token",
13 |         type=str,
14 |         default=os.environ.get("TEST_SESSION_TOKEN"),
15 |         help="Test session token to query with.",
16 |     )
17 | 
18 | 
19 | def _add_agent_url_arg(parser: argparse.ArgumentParser) -> None:
20 |     """Add the agent url argument to the given parser."""
21 |     parser.add_argument(
22 |         "--agent-url",
23 |         type=str,
24 |         default=os.environ.get(
25 |             "DD_TRACE_AGENT_URL",
26 |             os.environ.get("DD_AGENT_URL", "http://localhost:8126"),
27 |         ),
28 |         help=("Test agent URL. Default is http://localhost:8126"),
29 |     )
30 | 
31 | 
32 | def main_session_start() -> None:
33 |     """Entrypoint for the start-session command"""
34 |     parser = argparse.ArgumentParser(
35 |         description=(
36 |             "Start a test agent session with a given token. "
37 |             "All data submitted with this token will be associated with this session."
38 |         ),
39 |         prog="ddapm-test-agent-start-session",
40 |     )
41 |     _add_agent_url_arg(parser)
42 |     _add_token_arg(parser)
43 |     parsed_args = parser.parse_args(sys.argv[1:])
44 |     url = yarl.URL(parsed_args.agent_url).with_path("/test/session/start")
45 |     resp = requests.get(str(url), params={"test_session_token": parsed_args.test_session_token})
46 |     if resp.status_code != 200:
47 |         print(resp.text)
48 |         sys.exit(1)
49 |     print(resp.text)
50 |     sys.exit(0)
51 | 
52 | 
53 | def main_snapshot() -> None:
54 |     """Entrypoint for the snapshot command"""
55 |     parser = argparse.ArgumentParser(
56 |         description=("Perform a snapshot test for the data received in the session."),
57 |         prog="ddapm-test-agent-snapshot",
58 |     )
59 |     _add_agent_url_arg(parser)
60 |     _add_token_arg(parser)
61 |     parsed_args = parser.parse_args(sys.argv[1:])
62 |     if not parsed_args.test_session_token:
63 |         print(
64 |             "Error: a test token is required! Please specify one with --test-session-token"
65 |             " command line argument or the TEST_SESSION_TOKEN environment variable."
66 |         )
67 |         sys.exit(1)
68 |     url = yarl.URL(parsed_args.agent_url).with_path("/test/session/snapshot")
69 |     resp = requests.get(str(url), params={"test_session_token": parsed_args.test_session_token})
70 |     if resp.status_code != 200:
71 |         print(resp.text)
72 |         sys.exit(1)
73 |     print(resp.text)
74 |     sys.exit(0)
75 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_completions_post_6224acff.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo","prompt":"Hello, OpenAI!","max_tokens":100,"temperature":0.5,"n":1,"stream":false}'
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '107'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/NodeJS/3.0.0
17 |     method: POST
18 |     uri: https://api.openai.com/v1/completions
19 |   response:
20 |     body:
21 |       string: "{\n  \"error\": {\n    \"message\": \"This is a chat model and not
22 |         supported in the v1/completions endpoint. Did you mean to use v1/chat/completions?\",\n
23 |         \   \"type\": \"invalid_request_error\",\n    \"param\": \"model\",\n    \"code\":
24 |         null\n  }\n}"
25 |     headers:
26 |       CF-RAY:
27 |       - 952d0d4dbddc4cd9-BOS
28 |       Connection:
29 |       - keep-alive
30 |       Content-Encoding:
31 |       - gzip
32 |       Content-Type:
33 |       - application/json
34 |       Date:
35 |       - Fri, 20 Jun 2025 17:40:33 GMT
36 |       Server:
37 |       - cloudflare
38 |       Set-Cookie:
39 |       - __cf_bm=4qRBBQ2tClOn.HUfMDi0I6kdldko0XllOhgwqCZSmVk-1750441233-1.0.1.1-nmAFxZAQIPn7TDmIq09ZzrN3KSC1_RYkuGq.1ecMgUIbp0I7NNlFyT54wr2G4cOA6tA2aKl3tUdUCi3Q5FmJ1lqgYXo3L9VdExXf3s4BlkU;
40 |         path=/; expires=Fri, 20-Jun-25 18:10:33 GMT; domain=.api.openai.com; HttpOnly;
41 |         Secure; SameSite=None
42 |       - _cfuvid=o.Qm9FQMaV0x_UdplrdzScU4H6ggFdobpAgWA.47qTg-1750441233686-0.0.1.1-604800000;
43 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
44 |       Transfer-Encoding:
45 |       - chunked
46 |       X-Content-Type-Options:
47 |       - nosniff
48 |       access-control-expose-headers:
49 |       - X-Request-ID
50 |       alt-svc:
51 |       - h3=":443"; ma=86400
52 |       cf-cache-status:
53 |       - DYNAMIC
54 |       openai-organization:
55 |       - datadog-staging
56 |       openai-processing-ms:
57 |       - '18'
58 |       openai-version:
59 |       - '2020-10-01'
60 |       strict-transport-security:
61 |       - max-age=31536000; includeSubDomains; preload
62 |       x-envoy-upstream-service-time:
63 |       - '22'
64 |       x-ratelimit-limit-requests:
65 |       - '10000'
66 |       x-ratelimit-limit-tokens:
67 |       - '50000000'
68 |       x-ratelimit-remaining-requests:
69 |       - '9999'
70 |       x-ratelimit-remaining-tokens:
71 |       - '49999995'
72 |       x-ratelimit-reset-requests:
73 |       - 6ms
74 |       x-ratelimit-reset-tokens:
75 |       - 0s
76 |       x-request-id:
77 |       - req_80b3ab0bd1abceca50643613ae87c9dd
78 |     status:
79 |       code: 404
80 |       message: Not Found
81 | version: 1
82 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_917478d0.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo-instruct","messages":[{"role":"system","content":"You
 4 |       are a helpful assistant."},{"role":"user","content":"Hello, OpenAI!"}],"temperature":0.5,"stream":false,"max_tokens":100,"n":1,"user":"dd-trace-test"}'
 5 |     headers:
 6 |       Accept:
 7 |       - application/json, text/plain, */*
 8 |       Accept-Encoding:
 9 |       - gzip, deflate
10 |       Connection:
11 |       - keep-alive
12 |       Content-Length:
13 |       - '228'
14 |       Content-Type:
15 |       - application/json
16 |       User-Agent:
17 |       - OpenAI/NodeJS/3.3.0
18 |     method: POST
19 |     uri: https://api.openai.com/v1/chat/completions
20 |   response:
21 |     body:
22 |       string: "{\n  \"error\": {\n    \"message\": \"This is not a chat model and
23 |         thus not supported in the v1/chat/completions endpoint. Did you mean to use
24 |         v1/completions?\",\n    \"type\": \"invalid_request_error\",\n    \"param\":
25 |         \"model\",\n    \"code\": null\n  }\n}"
26 |     headers:
27 |       CF-RAY:
28 |       - 952d64efa8ae0f4b-EWR
29 |       Connection:
30 |       - keep-alive
31 |       Content-Encoding:
32 |       - gzip
33 |       Content-Type:
34 |       - application/json
35 |       Date:
36 |       - Fri, 20 Jun 2025 18:40:23 GMT
37 |       Server:
38 |       - cloudflare
39 |       Set-Cookie:
40 |       - __cf_bm=fBNcGUfek2z2W4oYr_OiBzbWupBkMn6yJDTUEsVK83g-1750444823-1.0.1.1-er3MbB1om9ZTEex6ZgPA2aY_f4pw5ZfHZsBE7NArGzLIJiHEMCYHYLidsoWkdvtjvnygkxTi9qpppP50rgvXOF2iwlqjS.lBpyCuYdCdICE;
41 |         path=/; expires=Fri, 20-Jun-25 19:10:23 GMT; domain=.api.openai.com; HttpOnly;
42 |         Secure; SameSite=None
43 |       - _cfuvid=5HDHMd8Uhua8g6XUU8tLL86azHv_b5POVNA0rm7.nPY-1750444823058-0.0.1.1-604800000;
44 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
45 |       Transfer-Encoding:
46 |       - chunked
47 |       X-Content-Type-Options:
48 |       - nosniff
49 |       access-control-expose-headers:
50 |       - X-Request-ID
51 |       alt-svc:
52 |       - h3=":443"; ma=86400
53 |       cf-cache-status:
54 |       - DYNAMIC
55 |       openai-organization:
56 |       - datadog-staging
57 |       openai-processing-ms:
58 |       - '6'
59 |       openai-version:
60 |       - '2020-10-01'
61 |       strict-transport-security:
62 |       - max-age=31536000; includeSubDomains; preload
63 |       x-envoy-upstream-service-time:
64 |       - '8'
65 |       x-ratelimit-limit-requests:
66 |       - '3500'
67 |       x-ratelimit-limit-tokens:
68 |       - '90000'
69 |       x-ratelimit-remaining-requests:
70 |       - '3499'
71 |       x-ratelimit-remaining-tokens:
72 |       - '89985'
73 |       x-ratelimit-reset-requests:
74 |       - 17ms
75 |       x-ratelimit-reset-tokens:
76 |       - 9ms
77 |       x-request-id:
78 |       - req_1ec17d77b9aae56ba87a06300b83c29c
79 |     status:
80 |       code: 404
81 |       message: Not Found
82 | version: 1
83 | 


--------------------------------------------------------------------------------
/.github/workflows/main.yml:
--------------------------------------------------------------------------------
 1 | name: CI
 2 | on:
 3 |   pull_request:
 4 |   push:
 5 |     branches:
 6 |       - master
 7 | jobs:
 8 |   check:
 9 |     runs-on: ubuntu-latest
10 |     steps:
11 |       - uses: actions/setup-python@7f4fc3e22c37d6ff65e88745f38bd3157c663f7c # v4.9.1
12 |         with:
13 |           # Run typing/formatting/etc checks on the oldest version we support
14 |           python-version: '3.11'
15 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
16 |         with:
17 |           fetch-depth: 0
18 |       - run: pip install --upgrade 'pip<24.1'
19 |       - run: pip install .[testing]
20 |       - run: riot -v run -s flake8
21 |       - run: riot -v run mypy
22 |   test:
23 |     strategy:
24 |       matrix:
25 |         os: [ubuntu-latest, macos-latest, windows-latest]
26 |         python-version: ["3.11", "3.12", "3.13"]
27 |     runs-on: ${{ matrix.os }}
28 |     steps:
29 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
30 |       - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v6.0.0
31 |         with:
32 |           python-version: ${{ matrix.python-version }}
33 |       - name: Downgrade pip for riot compatibility
34 |         if: runner.os != 'Windows'
35 |         run: pip install --upgrade 'pip<24.1'
36 |       - name: install deps
37 |         run: pip install .[testing]
38 |       - name: install pip-tools
39 |         if: runner.os != 'Windows'
40 |         run: pip install 'pip-tools>=7.4.0'
41 |       # Windows: Use vanilla pytest (riot doesn't support Windows - see https://github.com/DataDog/riot/issues/74)
42 |       - name: Run tests (Windows - direct pytest)
43 |         if: runner.os == 'Windows'
44 |         run: |
45 |           echo "Using pytest directly due to riot Windows limitation (https://github.com/DataDog/riot/issues/74)"
46 |           python -m pytest -v --basetemp=$env:TEMP\pytest tests/ --ignore=tests/test_container.py
47 |       - name: Run tests (Unix - riot)
48 |         if: runner.os != 'Windows'
49 |         run: |
50 |           riot run --pass-env -p ${{ matrix.python-version}} test
51 |   cassettes-size:
52 |     runs-on: ubuntu-latest
53 |     steps:
54 |       - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
55 |       - name: Check VCR cassettes size
56 |         run: |
57 |           CASSETTES_SIZE=$(du -sb vcr-cassettes/ 2>/dev/null | cut -f1 || echo "0")
58 | 
59 |           MAX_SIZE=2097152
60 | 
61 |           echo "VCR cassettes size: ${CASSETTES_SIZE} bytes ($((CASSETTES_SIZE / 1024 / 1024))MB)"
62 |           echo "Maximum allowed size: ${MAX_SIZE} bytes (2MB)"
63 | 
64 |           if [ "$CASSETTES_SIZE" -gt "$MAX_SIZE" ]; then
65 |             echo "❌ VCR cassettes directory is larger than 2MB!"
66 |             echo "Current size: $((CASSETTES_SIZE / 1024 / 1024))MB"
67 |             echo "Please review and clean up large cassette files."
68 |             exit 1
69 |           else
70 |             echo "✅ VCR cassettes directory size is within limits"
71 |           fi
72 | 
73 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/logs.py:
--------------------------------------------------------------------------------
 1 | """OTLP Logs handling for the test agent."""
 2 | 
 3 | import json
 4 | import logging
 5 | from typing import Any
 6 | from typing import Dict
 7 | 
 8 | from aiohttp import ClientSession
 9 | from google.protobuf.json_format import MessageToDict
10 | from grpc import aio as grpc_aio
11 | from opentelemetry.proto.collector.logs.v1.logs_service_pb2 import ExportLogsServiceRequest
12 | from opentelemetry.proto.collector.logs.v1.logs_service_pb2 import ExportLogsServiceResponse
13 | from opentelemetry.proto.collector.logs.v1.logs_service_pb2_grpc import LogsServiceServicer
14 | 
15 | 
16 | LOGS_ENDPOINT = "/v1/logs"
17 | 
18 | 
19 | log = logging.getLogger(__name__)
20 | 
21 | 
22 | def decode_logs_request(request_body: bytes, content_type: str) -> Dict[str, Any]:
23 |     if content_type == "application/json":
24 |         parsed_json = json.loads(request_body)
25 |         if not isinstance(parsed_json, dict):
26 |             raise Exception("JSON payload must be an object")
27 |         return parsed_json
28 |     elif content_type == "application/x-protobuf":
29 |         export_request = ExportLogsServiceRequest()
30 |         export_request.ParseFromString(request_body)
31 |         return protobuf_to_dict(export_request)
32 |     else:
33 |         raise ValueError(f"Content-Type must be application/x-protobuf or application/json, got {content_type}")
34 | 
35 | 
36 | def protobuf_to_dict(pb_obj: Any) -> Dict[str, Any]:
37 |     return MessageToDict(pb_obj, preserving_proto_field_name=True)
38 | 
39 | 
40 | class OTLPLogsGRPCServicer(LogsServiceServicer):
41 | 
42 |     def __init__(self, http_port: int):
43 |         self.http_url = f"http://127.0.0.1:{http_port}"
44 | 
45 |     async def Export(
46 |         self, request: ExportLogsServiceRequest, context: grpc_aio.ServicerContext
47 |     ) -> ExportLogsServiceResponse:
48 |         try:
49 |             protobuf_data = request.SerializeToString()
50 |             headers = {"Content-Type": "application/x-protobuf"}
51 |             metadata = dict(context.invocation_metadata())
52 |             if "session-token" in metadata:
53 |                 headers["Session-Token"] = metadata["session-token"]
54 |             async with ClientSession(self.http_url) as session:
55 |                 async with session.post(LOGS_ENDPOINT, headers=headers, data=protobuf_data) as resp:
56 |                     context.set_trailing_metadata([("http-status", str(resp.status))])
57 |                     response = ExportLogsServiceResponse()
58 |                     if resp.status >= 400:
59 |                         response.partial_success.rejected_log_records = len(request.resource_logs)
60 |                         response.partial_success.error_message = f"HTTP {resp.status}: {await resp.text()}"
61 |                     return response
62 |         except Exception as e:
63 |             context.set_trailing_metadata([("http-status", "500"), ("error", str(e))])
64 |             response = ExportLogsServiceResponse()
65 |             response.partial_success.rejected_log_records = len(request.resource_logs)
66 |             response.partial_success.error_message = f"Forward failed: {str(e)}"
67 |             return response
68 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_images_generations_post_736070ea.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"prompt":"sleepy capybara with monkey on top","n":1,"size":"1024x1024","response_format":"url","model":"dall-e-3"}'
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '115'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/NodeJS/3.3.0
17 |     method: POST
18 |     uri: https://api.openai.com/v1/images/generations
19 |   response:
20 |     body:
21 |       string: "{\n  \"created\": 1750279401,\n  \"data\": [\n    {\n      \"revised_prompt\":
22 |         \"A capybara, the largest rodent native to South America, comfortably lounges
23 |         in its habitat, its eyes half-closed in a sleepy expression. Perched neatly
24 |         on top of the capybara is a small monkey, possibly curious and adventurous,
25 |         gripping onto the capybara's fur for balance and looking around vigilantly
26 |         at the surroundings. The overall dynamic suggests a surprising and heartwarming
27 |         camaraderie between these two different species sharing the same environment.\",\n
28 |         \     \"url\": \"https://oaidalleapiprodscus.blob.core.windows.net/private/org-GKgUkEpTs8iJbqUCqnL6JW5H/user-DoUONI1dg5wtyS7luFo7MncN/img-O27kpXrkfcYRYaX5oao0vKuL.png?st=2025-06-18T19%3A43%3A21Z&se=2025-06-18T21%3A43%3A21Z&sp=r&sv=2024-08-04&sr=b&rscd=inline&rsct=image/png&skoid=cc612491-d948-4d2e-9821-2683df3719f5&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2025-06-18T12%3A22%3A54Z&ske=2025-06-19T12%3A22%3A54Z&sks=b&skv=2024-08-04&sig=e30ayD5TJOJ44c1VInZBPw0Vml7UoIJW5gR3UMt9%2Bng%3D\"\n
29 |         \   }\n  ]\n}"
30 |     headers:
31 |       CF-RAY:
32 |       - 951d9dff28b742a1-EWR
33 |       Connection:
34 |       - keep-alive
35 |       Content-Encoding:
36 |       - gzip
37 |       Content-Type:
38 |       - application/json
39 |       Date:
40 |       - Wed, 18 Jun 2025 20:43:21 GMT
41 |       Server:
42 |       - cloudflare
43 |       Set-Cookie:
44 |       - __cf_bm=hq172MsyXT7Eeww2f7m1sPkPVvUrTF6Zv37HV08talY-1750279401-1.0.1.1-gMqrfMJGr8csWwS29kCjvwye4UhiqIGWmUVigBnsnF.rp9.Q_Q2EWrrqrxmM3gTmdEsmVDjL3PWloya9KFimtmK7X0U2UP7im0HMvuOmwlk;
45 |         path=/; expires=Wed, 18-Jun-25 21:13:21 GMT; domain=.api.openai.com; HttpOnly;
46 |         Secure; SameSite=None
47 |       - _cfuvid=9eoXPqQ9O5XvM4GjJqSJmhKmV2pSUIt.76TfatMT9UY-1750279401521-0.0.1.1-604800000;
48 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
49 |       Transfer-Encoding:
50 |       - chunked
51 |       X-Content-Type-Options:
52 |       - nosniff
53 |       alt-svc:
54 |       - h3=":443"; ma=86400
55 |       cf-cache-status:
56 |       - DYNAMIC
57 |       openai-organization:
58 |       - datadog-staging
59 |       openai-processing-ms:
60 |       - '13421'
61 |       openai-version:
62 |       - '2020-10-01'
63 |       strict-transport-security:
64 |       - max-age=31536000; includeSubDomains; preload
65 |       x-request-id:
66 |       - req_31669501eb43d2e6b4aad57c69c052ac
67 |     status:
68 |       code: 200
69 |       message: OK
70 | version: 1
71 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_completions_post_9bb49464.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo-instruct","prompt":"Hello, OpenAI!","max_tokens":100,"temperature":0.5,"n":1,"stream":false}'
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '116'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/NodeJS/3.0.0
17 |     method: POST
18 |     uri: https://api.openai.com/v1/completions
19 |   response:
20 |     body:
21 |       string: "{\n  \"id\": \"cmpl-BjWrtEGE0hTMNDEZxiiEK7j3Z8MVm\",\n  \"object\":
22 |         \"text_completion\",\n  \"created\": 1750190805,\n  \"model\": \"gpt-3.5-turbo-instruct:20230824-v2\",\n
23 |         \ \"choices\": [\n    {\n      \"text\": \"\\n\\nHello! How can I assist you?\",\n
24 |         \     \"index\": 0,\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n
25 |         \   }\n  ],\n  \"usage\": {\n    \"prompt_tokens\": 5,\n    \"completion_tokens\":
26 |         9,\n    \"total_tokens\": 14\n  }\n}\n"
27 |     headers:
28 |       CF-RAY:
29 |       - 95152b501b3a8062-IAD
30 |       Cache-Control:
31 |       - no-cache, must-revalidate
32 |       Connection:
33 |       - keep-alive
34 |       Content-Encoding:
35 |       - gzip
36 |       Content-Type:
37 |       - application/json
38 |       Date:
39 |       - Tue, 17 Jun 2025 20:06:45 GMT
40 |       Server:
41 |       - cloudflare
42 |       Set-Cookie:
43 |       - __cf_bm=0PJgEsRC7NjP7TNQXH1rQkRjbCafftESid.n97ne2VI-1750190805-1.0.1.1-tBasMJEfmpgg1zB8JHRgPbwMEauEFJfDwEjwFxnWzlun75_vPR_TjnGg5amVVam9ed78YBv2L4jCA.2Ycac_PrWQ9.Qsnh8Qy1BIZi4L.Tk;
44 |         path=/; expires=Tue, 17-Jun-25 20:36:45 GMT; domain=.api.openai.com; HttpOnly;
45 |         Secure; SameSite=None
46 |       - _cfuvid=gfCDjic8jHqy7eAUA9dx3KAgSe9r.7fh.UUFwkuAHT8-1750190805199-0.0.1.1-604800000;
47 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
48 |       Transfer-Encoding:
49 |       - chunked
50 |       X-Content-Type-Options:
51 |       - nosniff
52 |       access-control-allow-origin:
53 |       - '*'
54 |       access-control-expose-headers:
55 |       - X-Request-ID
56 |       alt-svc:
57 |       - h3=":443"; ma=86400
58 |       cf-cache-status:
59 |       - DYNAMIC
60 |       openai-model:
61 |       - gpt-3.5-turbo-instruct:20230824-v2
62 |       openai-organization:
63 |       - datadog-staging
64 |       openai-processing-ms:
65 |       - '510'
66 |       openai-version:
67 |       - '2020-10-01'
68 |       strict-transport-security:
69 |       - max-age=31536000; includeSubDomains; preload
70 |       via:
71 |       - envoy-router-8478bc989-k82xn
72 |       x-envoy-upstream-service-time:
73 |       - '517'
74 |       x-ratelimit-limit-requests:
75 |       - '3500'
76 |       x-ratelimit-limit-tokens:
77 |       - '90000'
78 |       x-ratelimit-remaining-requests:
79 |       - '3499'
80 |       x-ratelimit-remaining-tokens:
81 |       - '89995'
82 |       x-ratelimit-reset-requests:
83 |       - 17ms
84 |       x-ratelimit-reset-tokens:
85 |       - 2ms
86 |       x-request-id:
87 |       - req_17b612a8e208f5b7b89c95378f18f64a
88 |     status:
89 |       code: 200
90 |       message: OK
91 | version: 1
92 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_moderations_post_532b3315.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"input":"I want to harm the robots"}'
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '37'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/NodeJS/3.3.0
17 |     method: POST
18 |     uri: https://api.openai.com/v1/moderations
19 |   response:
20 |     body:
21 |       string: "{\n  \"id\": \"modr-BkWEObamZdptfnhFjPJtkDLfGU7BT\",\n  \"model\":
22 |         \"text-moderation-007\",\n  \"results\": [\n    {\n      \"flagged\": true,\n
23 |         \     \"categories\": {\n        \"sexual\": false,\n        \"hate\": false,\n
24 |         \       \"harassment\": false,\n        \"self-harm\": false,\n        \"sexual/minors\":
25 |         false,\n        \"hate/threatening\": false,\n        \"violence/graphic\":
26 |         false,\n        \"self-harm/intent\": false,\n        \"self-harm/instructions\":
27 |         false,\n        \"harassment/threatening\": false,\n        \"violence\":
28 |         true\n      },\n      \"category_scores\": {\n        \"sexual\": 0.00003649308200692758,\n
29 |         \       \"hate\": 0.000206793614779599,\n        \"harassment\": 0.01880560629069805,\n
30 |         \       \"self-harm\": 9.369699910166673e-6,\n        \"sexual/minors\": 5.352579250939016e-7,\n
31 |         \       \"hate/threatening\": 0.00012469793728087097,\n        \"violence/graphic\":
32 |         4.941105999023421e-6,\n        \"self-harm/intent\": 3.630466153481393e-6,\n
33 |         \       \"self-harm/instructions\": 8.705083764937172e-9,\n        \"harassment/threatening\":
34 |         0.025648830458521843,\n        \"violence\": 0.9218947291374207\n      }\n
35 |         \   }\n  ]\n}"
36 |     headers:
37 |       CF-RAY:
38 |       - 952baa1ae8978f73-BOS
39 |       Connection:
40 |       - keep-alive
41 |       Content-Encoding:
42 |       - gzip
43 |       Content-Type:
44 |       - application/json
45 |       Date:
46 |       - Fri, 20 Jun 2025 13:38:05 GMT
47 |       Server:
48 |       - cloudflare
49 |       Set-Cookie:
50 |       - __cf_bm=zv1b3ZAlaOyWQ9.7YxkxGUPXsNZia4BXMry7LJSq7JI-1750426685-1.0.1.1-UqwnPe_5zDkeOSh3IGg9yL2FvOfaOZ3KsKFfYTdO_cUlOCn4MjlWDU7WJS_ILElLjOqtVXy2wSG4UtOpWbiTyaFTsnTlWTbzTXl.3kaQkTc;
51 |         path=/; expires=Fri, 20-Jun-25 14:08:05 GMT; domain=.api.openai.com; HttpOnly;
52 |         Secure; SameSite=None
53 |       - _cfuvid=dfzU1nBKTl7eSXfrjOOdnGPyjVTDeU49FBjU35142dc-1750426685063-0.0.1.1-604800000;
54 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
55 |       Transfer-Encoding:
56 |       - chunked
57 |       X-Content-Type-Options:
58 |       - nosniff
59 |       alt-svc:
60 |       - h3=":443"; ma=86400
61 |       cf-cache-status:
62 |       - DYNAMIC
63 |       openai-organization:
64 |       - datadog-staging
65 |       openai-processing-ms:
66 |       - '198'
67 |       openai-version:
68 |       - '2020-10-01'
69 |       strict-transport-security:
70 |       - max-age=31536000; includeSubDomains; preload
71 |       x-request-id:
72 |       - req_606a53c1fbf99a2b8a48a8aa52c98810
73 |     status:
74 |       code: 200
75 |       message: OK
76 | version: 1
77 | 


--------------------------------------------------------------------------------
/.github/workflows/docker.yml:
--------------------------------------------------------------------------------
 1 | name: Publish latest Docker images
 2 | on:
 3 |   push:
 4 |   release:
 5 |     types:
 6 |       - published
 7 | jobs:
 8 |   build-and-publish:
 9 |     runs-on: ubuntu-latest
10 |     permissions:
11 |       packages: write
12 |     if: github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/tags/')
13 |     steps:
14 |     - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
15 | 
16 |     - name: Set up QEMU
17 |       uses: docker/setup-qemu-action@2b82ce82d56a2a04d2637cd93a637ae1b359c0a7 # v2.2.0
18 | 
19 |     - name: Set up Docker Buildx
20 |       id: buildx
21 |       uses: docker/setup-buildx-action@885d1462b80bc1c1c7f0b00334ad271f09369c55 # v2.10.0
22 | 
23 |     - name: Available platforms
24 |       run: echo ${{ steps.buildx.outputs.platforms }}
25 | 
26 |     - name: Set Docker Image Tag
27 |       id: set_names
28 |       run: |
29 |         if [ "${{ startsWith(github.ref, 'refs/tags/') }}" == "true" ]; then
30 |           DOCKER_IMAGE_TAG=$(echo ${GITHUB_REF} | sed -e "s/refs\/tags\///g")
31 |         else
32 |           DOCKER_IMAGE_TAG="latest"
33 |         fi
34 |         DOCKER_IMAGE_NAME=ddapm-test-agent
35 |         DOCKER_IMAGE_NAME=$(echo ghcr.io/${GITHUB_REPOSITORY}/${DOCKER_IMAGE_NAME} | tr '[:upper:]' '[:lower:]')
36 |         DOCKER_IMAGE_NAME_WITH_TAG=$(echo ${DOCKER_IMAGE_NAME}:${DOCKER_IMAGE_TAG} | tr '[:upper:]' '[:lower:]')
37 |         echo "Using image name '$DOCKER_IMAGE_NAME_WITH_TAG'"
38 |         echo "image_name=$DOCKER_IMAGE_NAME_WITH_TAG" >> $GITHUB_OUTPUT
39 | 
40 |     - name: Login to Docker
41 |       run: docker login -u publisher -p ${{ secrets.GITHUB_TOKEN }} ghcr.io
42 | 
43 |     - name: Docker Build
44 |       uses: docker/build-push-action@1104d471370f9806843c095c1db02b5a90c5f8b6 # v3.3.1
45 |       with:
46 |         push: true
47 |         tags: ${{ steps.set_names.outputs.image_name }}
48 |         platforms: "linux/amd64,linux/arm64/v8"
49 |         context: .
50 | 
51 |   build-and-publish-windows:
52 |     runs-on: windows-latest
53 |     permissions:
54 |       packages: write
55 |     if: github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/tags/')
56 |     steps:
57 |     - uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
58 | 
59 |     - name: Set Docker Image Tag
60 |       id: set_names
61 |       shell: pwsh
62 |       run: |
63 |         if ("${{ startsWith(github.ref, 'refs/tags/') }}" -eq "true") {
64 |           $DOCKER_IMAGE_TAG = "${{ github.ref }}".Replace("refs/tags/", "") + "-windows"
65 |         } else {
66 |           $DOCKER_IMAGE_TAG = "latest-windows"
67 |         }
68 |         $DOCKER_IMAGE_NAME = "ddapm-test-agent"
69 |         $DOCKER_IMAGE_NAME = "ghcr.io/${{ github.repository }}/${DOCKER_IMAGE_NAME}".ToLower()
70 |         $DOCKER_IMAGE_NAME_WITH_TAG = "${DOCKER_IMAGE_NAME}:${DOCKER_IMAGE_TAG}".ToLower()
71 |         Write-Host "Using image name '$DOCKER_IMAGE_NAME_WITH_TAG'"
72 |         "image_name=$DOCKER_IMAGE_NAME_WITH_TAG" >> $env:GITHUB_OUTPUT
73 | 
74 |     - name: Login to Docker
75 |       run: docker login -u publisher -p ${{ secrets.GITHUB_TOKEN }} ghcr.io
76 | 
77 |     - name: Docker Build
78 |       shell: pwsh
79 |       run: |
80 |         docker build -t ${{ steps.set_names.outputs.image_name }} -f ./Dockerfile.windows .
81 |         docker push ${{ steps.set_names.outputs.image_name }}
82 | 


--------------------------------------------------------------------------------
/vcr-cassettes/deepseek/deepseek_chat_completions_post_afb6ccab.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: "{\n  \"model\": \"deepseek-chat\",\n  \"messages\": [\n    {\n      \"role\":
 4 |       \"system\",\n      \"content\": \"You are a helpful assistant.\"\n    },\n    {\n
 5 |       \     \"role\": \"user\",\n      \"content\": \"Hello, OpenAI!\"\n    }\n  ],\n
 6 |       \ \"temperature\": 0.5,\n  \"stream\": false,\n  \"max_tokens\": 100,\n  \"n\":
 7 |       1,\n  \"user\": \"dd-trace-test\"\n}"
 8 |     headers:
 9 |       Accept:
10 |       - application/json
11 |       Accept-Encoding:
12 |       - gzip,deflate
13 |       Connection:
14 |       - keep-alive
15 |       Content-Length:
16 |       - '303'
17 |       Content-Type:
18 |       - application/json
19 |       User-Agent:
20 |       - OpenAI/JS 4.0.0
21 |       X-Stainless-Arch:
22 |       - arm64
23 |       X-Stainless-Lang:
24 |       - js
25 |       X-Stainless-Os:
26 |       - MacOS
27 |       X-Stainless-Package-Version:
28 |       - 4.0.0
29 |       X-Stainless-Runtime:
30 |       - node
31 |       X-Stainless-Runtime-Version:
32 |       - v22.13.1
33 |     method: POST
34 |     uri: https://api.deepseek.com/chat/completions
35 |   response:
36 |     body:
37 |       string: "{\n  \"id\": \"chatcmpl-BkZxQcJvaxbyldxLmrNffCobBnd84\",\n  \"object\":
38 |         \"chat.completion\",\n  \"created\": 1750441008,\n  \"model\": \"deepseek-chat\",\n
39 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
40 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
41 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
42 |         null,\n      \"finish_reason\": \"stop\"\n    }\n  ],\n  \"usage\": {\n    \"prompt_tokens\":
43 |         21,\n    \"completion_tokens\": 9,\n    \"total_tokens\": 30,\n    \"prompt_tokens_details\":
44 |         {\n      \"cached_tokens\": 0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
45 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
46 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
47 |         \"default\",\n  \"system_fingerprint\": \"fp_07871e2ad8\"\n}\n"
48 |     headers:
49 |       CF-RAY:
50 |       - 952d93a97f810f6f-EWR
51 |       Connection:
52 |       - keep-alive
53 |       Content-Length:
54 |       - '111'
55 |       Content-Type:
56 |       - application/json
57 |       Date:
58 |       - Fri, 20 Jun 2025 19:12:17 GMT
59 |       Server:
60 |       - cloudflare
61 |       Set-Cookie:
62 |       - __cf_bm=OVJpDC4uunjsiV8_fYewKdBcNb7Lh25RKaoIltEW3kI-1750446737-1.0.1.1-lRGs1fPN.cl7rWoNdw0jH0rMPmx3S8uA1jM71WtdVfi62cHgK9IXbg.AxCYHXFsIIHd01lpXSGIZV5p5pkHPhfKhPhwhXOOPtDvCY0bNVDg;
63 |         path=/; expires=Fri, 20-Jun-25 19:42:17 GMT; domain=.deepseek.com; HttpOnly;
64 |         Secure; SameSite=None
65 |       Strict-Transport-Security:
66 |       - max-age=31536000; includeSubDomains; preload
67 |       X-Content-Type-Options:
68 |       - nosniff
69 |       access-control-allow-credentials:
70 |       - 'true'
71 |       cf-cache-status:
72 |       - DYNAMIC
73 |       vary:
74 |       - origin, access-control-request-method, access-control-request-headers
75 |       x-ds-trace-id:
76 |       - 40ea0f5519c26ab95e6ffaf4badd4c3a
77 |     status:
78 |       code: 200
79 |       message: OK
80 | version: 1
81 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_get_265489e7.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json
 7 |       Accept-Encoding:
 8 |       - gzip,deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Type:
12 |       - application/json
13 |       User-Agent:
14 |       - OpenAI/JS 4.17.0
15 |       X-Stainless-Arch:
16 |       - arm64
17 |       X-Stainless-Lang:
18 |       - js
19 |       X-Stainless-Os:
20 |       - MacOS
21 |       X-Stainless-Package-Version:
22 |       - 4.17.0
23 |       X-Stainless-Runtime:
24 |       - node
25 |       X-Stainless-Runtime-Version:
26 |       - v22.13.1
27 |     method: GET
28 |     uri: https://api.openai.com/v1/fine_tuning/jobs/ftjob-q9CUUUsHJemGUVQ1Ecc01zcf
29 |   response:
30 |     body:
31 |       string: "{\n  \"object\": \"fine_tuning.job\",\n  \"id\": \"ftjob-q9CUUUsHJemGUVQ1Ecc01zcf\",\n
32 |         \ \"model\": \"gpt-4.1-mini-2025-04-14\",\n  \"created_at\": 1750424715,\n
33 |         \ \"finished_at\": null,\n  \"fine_tuned_model\": null,\n  \"organization_id\":
34 |         \"org-GKgUkEpTs8iJbqUCqnL6JW5H\",\n  \"result_files\": [],\n  \"status\":
35 |         \"validating_files\",\n  \"validation_file\": null,\n  \"training_file\":
36 |         \"file-RpTpuvRVtnKpdKZb7DDGto\",\n  \"hyperparameters\": {\n    \"n_epochs\":
37 |         10,\n    \"batch_size\": 1,\n    \"learning_rate_multiplier\": 2\n  },\n  \"trained_tokens\":
38 |         null,\n  \"error\": {},\n  \"user_provided_suffix\": null,\n  \"seed\": 836925998,\n
39 |         \ \"estimated_finish\": null,\n  \"integrations\": [],\n  \"metadata\": null,\n
40 |         \ \"usage_metrics\": null,\n  \"shared_with_openai\": false,\n  \"eval_id\":
41 |         null,\n  \"method\": {\n    \"type\": \"supervised\",\n    \"supervised\":
42 |         {\n      \"hyperparameters\": {\n        \"n_epochs\": 10,\n        \"batch_size\":
43 |         1,\n        \"learning_rate_multiplier\": 2.0\n      }\n    }\n  }\n}"
44 |     headers:
45 |       CF-RAY:
46 |       - 952b7ae12c714d0a-BOS
47 |       Connection:
48 |       - keep-alive
49 |       Content-Encoding:
50 |       - gzip
51 |       Content-Type:
52 |       - application/json
53 |       Date:
54 |       - Fri, 20 Jun 2025 13:05:50 GMT
55 |       Server:
56 |       - cloudflare
57 |       Set-Cookie:
58 |       - __cf_bm=4X8.PQnGnMdqYMJPFgpl3Xv9vBm0GyCdwalBWwxnnL4-1750424750-1.0.1.1-I1N77TVNdO55CkyiKXBBfOTfNEsl_oGqW_clVkpna2ZF4lUiBJsqHWuwfmNCln3heIhE4xqfrM23cW8FlGpmkGS8W1_puEpd2vOVmgatixA;
59 |         path=/; expires=Fri, 20-Jun-25 13:35:50 GMT; domain=.api.openai.com; HttpOnly;
60 |         Secure; SameSite=None
61 |       - _cfuvid=s0UCTHrD32bKDx5yR4WCeCwWq8b0w59ZRFMY8djKcOA-1750424750479-0.0.1.1-604800000;
62 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
63 |       Transfer-Encoding:
64 |       - chunked
65 |       X-Content-Type-Options:
66 |       - nosniff
67 |       alt-svc:
68 |       - h3=":443"; ma=86400
69 |       cf-cache-status:
70 |       - DYNAMIC
71 |       openai-organization:
72 |       - datadog-staging
73 |       openai-processing-ms:
74 |       - '94'
75 |       openai-version:
76 |       - '2020-10-01'
77 |       strict-transport-security:
78 |       - max-age=31536000; includeSubDomains; preload
79 |       x-request-id:
80 |       - req_ff237dd043d00b74115549f3765f0c8b
81 |     status:
82 |       code: 200
83 |       message: OK
84 | version: 1
85 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_fine_tuning_jobs_get_ebc57c6f.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json
 7 |       Accept-Encoding:
 8 |       - gzip,deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Type:
12 |       - application/json
13 |       User-Agent:
14 |       - OpenAI/JS 4.17.0
15 |       X-Stainless-Arch:
16 |       - arm64
17 |       X-Stainless-Lang:
18 |       - js
19 |       X-Stainless-Os:
20 |       - MacOS
21 |       X-Stainless-Package-Version:
22 |       - 4.17.0
23 |       X-Stainless-Runtime:
24 |       - node
25 |       X-Stainless-Runtime-Version:
26 |       - v22.13.1
27 |     method: GET
28 |     uri: https://api.openai.com/v1/fine_tuning/jobs
29 |   response:
30 |     body:
31 |       string: "{\n  \"object\": \"list\",\n  \"data\": [\n    {\n  \"object\": \"fine_tuning.job\",\n  \"id\": \"ftjob-q9CUUUsHJemGUVQ1Ecc01zcf\",\n
32 |         \ \"model\": \"gpt-4.1-mini-2025-04-14\",\n  \"created_at\": 1750424715,\n
33 |         \ \"finished_at\": null,\n  \"fine_tuned_model\": null,\n  \"organization_id\":
34 |         \"org-GKgUkEpTs8iJbqUCqnL6JW5H\",\n  \"result_files\": [],\n  \"status\":
35 |         \"validating_files\",\n  \"validation_file\": null,\n  \"training_file\":
36 |         \"file-RpTpuvRVtnKpdKZb7DDGto\",\n  \"hyperparameters\": {\n    \"n_epochs\":
37 |         10,\n    \"batch_size\": 1,\n    \"learning_rate_multiplier\": 2\n  },\n  \"trained_tokens\":
38 |         null,\n  \"error\": {},\n  \"user_provided_suffix\": null,\n  \"seed\": 836925998,\n
39 |         \ \"estimated_finish\": null,\n  \"integrations\": [],\n  \"metadata\": null,\n
40 |         \ \"usage_metrics\": null,\n  \"shared_with_openai\": false,\n  \"eval_id\":
41 |         null,\n  \"method\": {\n    \"type\": \"supervised\",\n    \"supervised\":
42 |         {\n      \"hyperparameters\": {\n        \"n_epochs\": 10,\n        \"batch_size\":
43 |         1,\n        \"learning_rate_multiplier\": 2.0\n      }\n    }\n  }\n}],\n  \"has_more\": false\n}"
44 |     headers:
45 |       CF-RAY:
46 |       - 952b85d04c908fe7-BOS
47 |       Connection:
48 |       - keep-alive
49 |       Content-Encoding:
50 |       - gzip
51 |       Content-Type:
52 |       - application/json
53 |       Date:
54 |       - Fri, 20 Jun 2025 13:13:18 GMT
55 |       Server:
56 |       - cloudflare
57 |       Set-Cookie:
58 |       - __cf_bm=j6BfbbK5vsC4Z9jwyfpejhuoHozgXJKQJ_9dgUpCE4E-1750425198-1.0.1.1-31113.Hy2aXpoWElaspQIyPiPzL_QZTWwzvSC.MZJ3OjDxwLtMz7NAF5CSfFmzqJayafGQpyi3YEPCEh5BDskL.V47SvGcgbPys9RUA2auQ;
59 |         path=/; expires=Fri, 20-Jun-25 13:43:18 GMT; domain=.api.openai.com; HttpOnly;
60 |         Secure; SameSite=None
61 |       - _cfuvid=1X2fHIQDYAXkTLek90WryeHpIYexljTKB9lmWwwgLjA-1750425198293-0.0.1.1-604800000;
62 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
63 |       Transfer-Encoding:
64 |       - chunked
65 |       X-Content-Type-Options:
66 |       - nosniff
67 |       alt-svc:
68 |       - h3=":443"; ma=86400
69 |       cf-cache-status:
70 |       - DYNAMIC
71 |       openai-organization:
72 |       - datadog-staging
73 |       openai-processing-ms:
74 |       - '63'
75 |       openai-version:
76 |       - '2020-10-01'
77 |       strict-transport-security:
78 |       - max-age=31536000; includeSubDomains; preload
79 |       x-request-id:
80 |       - req_d41eff0dc4e9afb4fb0c5fad1c35b1a8
81 |     status:
82 |       code: 200
83 |       message: OK
84 | version: 1
85 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_cancel_post_d7c16b38.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json
 7 |       Accept-Encoding:
 8 |       - gzip,deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '0'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/JS 4.17.0
17 |       X-Stainless-Arch:
18 |       - arm64
19 |       X-Stainless-Lang:
20 |       - js
21 |       X-Stainless-Os:
22 |       - MacOS
23 |       X-Stainless-Package-Version:
24 |       - 4.17.0
25 |       X-Stainless-Runtime:
26 |       - node
27 |       X-Stainless-Runtime-Version:
28 |       - v22.13.1
29 |     method: POST
30 |     uri: https://api.openai.com/v1/fine_tuning/jobs/ftjob-q9CUUUsHJemGUVQ1Ecc01zcf/cancel
31 |   response:
32 |     body:
33 |       string: "{\n  \"object\": \"fine_tuning.job\",\n  \"id\": \"ftjob-q9CUUUsHJemGUVQ1Ecc01zcf\",\n
34 |         \ \"model\": \"gpt-4.1-mini-2025-04-14\",\n  \"created_at\": 1750424715,\n
35 |         \ \"finished_at\": null,\n  \"fine_tuned_model\": null,\n  \"organization_id\":
36 |         \"org-GKgUkEpTs8iJbqUCqnL6JW5H\",\n  \"result_files\": [],\n  \"status\":
37 |         \"cancelled\",\n  \"validation_file\": null,\n  \"training_file\": \"file-RpTpuvRVtnKpdKZb7DDGto\",\n
38 |         \ \"hyperparameters\": {\n    \"n_epochs\": 10,\n    \"batch_size\": 1,\n
39 |         \   \"learning_rate_multiplier\": 2\n  },\n  \"trained_tokens\": null,\n  \"error\":
40 |         {},\n  \"user_provided_suffix\": null,\n  \"seed\": 836925998,\n  \"estimated_finish\":
41 |         null,\n  \"integrations\": [],\n  \"metadata\": null,\n  \"usage_metrics\":
42 |         null,\n  \"shared_with_openai\": false,\n  \"eval_id\": null,\n  \"method\":
43 |         {\n    \"type\": \"supervised\",\n    \"supervised\": {\n      \"hyperparameters\":
44 |         {\n        \"n_epochs\": 10,\n        \"batch_size\": 1,\n        \"learning_rate_multiplier\":
45 |         2.0\n      }\n    }\n  }\n}"
46 |     headers:
47 |       CF-RAY:
48 |       - 952b7d769ce24cf4-BOS
49 |       Connection:
50 |       - keep-alive
51 |       Content-Encoding:
52 |       - gzip
53 |       Content-Type:
54 |       - application/json
55 |       Date:
56 |       - Fri, 20 Jun 2025 13:07:36 GMT
57 |       Server:
58 |       - cloudflare
59 |       Set-Cookie:
60 |       - __cf_bm=Z93292EGLkRDIvTjg45KS0Fm1v5Jfl4gPQ9JZdxgZyE-1750424856-1.0.1.1-VNT9UerQLlKM_V3KspcgPLrI18DGiSqs19juc5iHFSwzJno.V.WOgWaHjkKWAJDPLqm.XTPYfat8rnCyQnVg42jK1U7vAStJb8TfSlUi0ko;
61 |         path=/; expires=Fri, 20-Jun-25 13:37:36 GMT; domain=.api.openai.com; HttpOnly;
62 |         Secure; SameSite=None
63 |       - _cfuvid=BA_TJGdKZrUT_kiryyrxq7tZqFLPb78N6FXkDf8eUK4-1750424856571-0.0.1.1-604800000;
64 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
65 |       Transfer-Encoding:
66 |       - chunked
67 |       X-Content-Type-Options:
68 |       - nosniff
69 |       alt-svc:
70 |       - h3=":443"; ma=86400
71 |       cf-cache-status:
72 |       - DYNAMIC
73 |       openai-organization:
74 |       - datadog-staging
75 |       openai-processing-ms:
76 |       - '90'
77 |       openai-version:
78 |       - '2020-10-01'
79 |       strict-transport-security:
80 |       - max-age=31536000; includeSubDomains; preload
81 |       x-request-id:
82 |       - req_1544829063eb9b601b7ee18420ec8237
83 |     status:
84 |       code: 200
85 |       message: OK
86 | version: 1
87 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_fine_tuning_jobs_post_c3b6e4a9.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: "{\n  \"training_file\": \"file-RpTpuvRVtnKpdKZb7DDGto\",\n  \"model\":
 4 |       \"gpt-4.1-mini-2025-04-14\"\n}"
 5 |     headers:
 6 |       Accept:
 7 |       - application/json
 8 |       Accept-Encoding:
 9 |       - gzip,deflate
10 |       Connection:
11 |       - keep-alive
12 |       Content-Length:
13 |       - '90'
14 |       Content-Type:
15 |       - application/json
16 |       User-Agent:
17 |       - OpenAI/JS 4.17.0
18 |       X-Stainless-Arch:
19 |       - arm64
20 |       X-Stainless-Lang:
21 |       - js
22 |       X-Stainless-Os:
23 |       - MacOS
24 |       X-Stainless-Package-Version:
25 |       - 4.17.0
26 |       X-Stainless-Runtime:
27 |       - node
28 |       X-Stainless-Runtime-Version:
29 |       - v22.13.1
30 |     method: POST
31 |     uri: https://api.openai.com/v1/fine_tuning/jobs
32 |   response:
33 |     body:
34 |       string: "{\n  \"object\": \"fine_tuning.job\",\n  \"id\": \"ftjob-q9CUUUsHJemGUVQ1Ecc01zcf\",\n
35 |         \ \"model\": \"gpt-4.1-mini-2025-04-14\",\n  \"created_at\": 1750424715,\n
36 |         \ \"finished_at\": null,\n  \"fine_tuned_model\": null,\n  \"organization_id\":
37 |         \"org-GKgUkEpTs8iJbqUCqnL6JW5H\",\n  \"result_files\": [],\n  \"status\":
38 |         \"validating_files\",\n  \"validation_file\": null,\n  \"training_file\":
39 |         \"file-RpTpuvRVtnKpdKZb7DDGto\",\n  \"hyperparameters\": {\n    \"n_epochs\":
40 |         \"auto\",\n    \"batch_size\": \"auto\",\n    \"learning_rate_multiplier\":
41 |         \"auto\"\n  },\n  \"trained_tokens\": null,\n  \"error\": {},\n  \"user_provided_suffix\":
42 |         null,\n  \"seed\": 836925998,\n  \"estimated_finish\": null,\n  \"integrations\":
43 |         [],\n  \"metadata\": null,\n  \"usage_metrics\": null,\n  \"shared_with_openai\":
44 |         false,\n  \"eval_id\": null,\n  \"method\": {\n    \"type\": \"supervised\",\n
45 |         \   \"supervised\": {\n      \"hyperparameters\": {\n        \"batch_size\":
46 |         \"auto\",\n        \"learning_rate_multiplier\": \"auto\",\n        \"n_epochs\":
47 |         \"auto\"\n      }\n    }\n  }\n}"
48 |     headers:
49 |       CF-RAY:
50 |       - 952b7a020bb78f7e-BOS
51 |       Connection:
52 |       - keep-alive
53 |       Content-Encoding:
54 |       - gzip
55 |       Content-Type:
56 |       - application/json
57 |       Date:
58 |       - Fri, 20 Jun 2025 13:05:15 GMT
59 |       Server:
60 |       - cloudflare
61 |       Set-Cookie:
62 |       - __cf_bm=lkj_sQRxOobZqR5KxE2EV3JZFdebX5CDsR683u1lIR4-1750424715-1.0.1.1-9JSD1vKH7JFGuAh47YWbELGWUE1rLg1f2j.tONU4QNcYgZ0URMm6Yac9fCvCeJEO_jLa0b6Ia2K7Z2zEWNZOUyPi594zWuwQmUfP0BXwdlw;
63 |         path=/; expires=Fri, 20-Jun-25 13:35:15 GMT; domain=.api.openai.com; HttpOnly;
64 |         Secure; SameSite=None
65 |       - _cfuvid=ANj22mHIgvhPlyjz.tKSjQdDyo3UGqAxbaNkCuoCIIU-1750424715244-0.0.1.1-604800000;
66 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
67 |       Transfer-Encoding:
68 |       - chunked
69 |       X-Content-Type-Options:
70 |       - nosniff
71 |       alt-svc:
72 |       - h3=":443"; ma=86400
73 |       cf-cache-status:
74 |       - DYNAMIC
75 |       openai-organization:
76 |       - datadog-staging
77 |       openai-processing-ms:
78 |       - '565'
79 |       openai-version:
80 |       - '2020-10-01'
81 |       strict-transport-security:
82 |       - max-age=31536000; includeSubDomains; preload
83 |       x-request-id:
84 |       - req_67e131ca9620e8d04c898aeacfc0656a
85 |     status:
86 |       code: 200
87 |       message: OK
88 | version: 1
89 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/snapshots.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | {% from "macros.html" import page_layout, empty_state, action_button, error_message %}
 3 | 
 4 | {% block content %}
 5 | {% set extra_header %}
 6 |     <div class="snapshot-info">
 7 |         <span class="snapshot-dir">Directory: {{ snapshot_dir }}</span>
 8 |     </div>
 9 | {% endset %}
10 | 
11 | {% call page_layout("Snapshots", extra_header=extra_header) %}
12 |     {% if error %}
13 |         {{ error_message(error) }}
14 |     {% endif %}
15 | 
16 |     {% if snapshots %}
17 |         <div class="snapshots-controls">
18 |             <div class="snapshots-stats">
19 |                 <p><strong><span id="total-count">{{ snapshots|length }}</span></strong> snapshot files found (<span id="filtered-count">{{ snapshots|length }}</span> shown)</p>
20 |             </div>
21 |             <div class="filter-controls">
22 |                 <input type="text" id="filename-filter" placeholder="Filter by filename..." class="filter-input">
23 |             </div>
24 |         </div>
25 |         <div class="snapshots-table">
26 |             <table>
27 |                 <thead>
28 |                     <tr>
29 |                         <th>Filename</th>
30 |                         <th>Size</th>
31 |                         <th>Modified</th>
32 |                     </tr>
33 |                 </thead>
34 |                 <tbody>
35 |                     {% for snapshot in snapshots %}
36 |                     <tr class="snapshot-row" onclick="window.location.href='/snapshots/{{ snapshot.filename }}'" style="cursor: pointer;">
37 |                         <td class="filename-cell">{{ snapshot.filename }}</td>
38 |                         <td class="size-cell">{{ "%.1f" | format(snapshot.size / 1024) }} KB</td>
39 |                         <td class="modified-cell">{{ snapshot.modified | timestamp_format }}</td>
40 |                     </tr>
41 |                     {% endfor %}
42 |                 </tbody>
43 |             </table>
44 |         </div>
45 |     {% else %}
46 |         {% if error %}
47 |             {{ empty_state("Unable to load snapshots") }}
48 |         {% else %}
49 |             {{ empty_state("No snapshot files found in " + snapshot_dir) }}
50 |         {% endif %}
51 |     {% endif %}
52 | {% endcall %}
53 | 
54 | <script>
55 | document.addEventListener('DOMContentLoaded', function() {
56 |     const filterInput = document.getElementById('filename-filter');
57 |     const filteredCountSpan = document.getElementById('filtered-count');
58 |     const table = document.querySelector('.snapshots-table table tbody');
59 | 
60 |     if (filterInput && table) {
61 |         filterInput.addEventListener('input', function() {
62 |             const filterValue = this.value.toLowerCase().trim();
63 |             const rows = table.getElementsByTagName('tr');
64 |             let visibleCount = 0;
65 | 
66 |             for (let i = 0; i < rows.length; i++) {
67 |                 const filenameCell = rows[i].querySelector('.filename-cell');
68 |                 if (filenameCell) {
69 |                     const filename = filenameCell.textContent.toLowerCase();
70 |                     if (filename.includes(filterValue)) {
71 |                         rows[i].style.display = '';
72 |                         visibleCount++;
73 |                     } else {
74 |                         rows[i].style.display = 'none';
75 |                     }
76 |                 }
77 |             }
78 | 
79 |             if (filteredCountSpan) {
80 |                 filteredCountSpan.textContent = visibleCount;
81 |             }
82 |         });
83 |     }
84 | });
85 | </script>
86 | {% endblock %}


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_completions_post_33681ea4.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo-instruct","prompt":"Hello, OpenAI!","max_tokens":100,"temperature":0.5,"n":3,"stream":false}'
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Length:
12 |       - '116'
13 |       Content-Type:
14 |       - application/json
15 |       User-Agent:
16 |       - OpenAI/NodeJS/3.0.0
17 |     method: POST
18 |     uri: https://api.openai.com/v1/completions
19 |   response:
20 |     body:
21 |       string: "{\n  \"id\": \"cmpl-BkZALGR9oEKDvmxHU9lT9Z97hfTit\",\n  \"object\":
22 |         \"text_completion\",\n  \"created\": 1750437965,\n  \"model\": \"gpt-3.5-turbo-instruct:20230824-v2\",\n
23 |         \ \"choices\": [\n    {\n      \"text\": \"\\n\\nHello! How are you?\",\n
24 |         \     \"index\": 0,\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n
25 |         \   },\n    {\n      \"text\": \"\\n\\nHello! How can I assist you?\",\n      \"index\":
26 |         1,\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n    },\n
27 |         \   {\n      \"text\": \"\\n\\nHello! How can I help you?\",\n      \"index\":
28 |         2,\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n    }\n
29 |         \ ],\n  \"usage\": {\n    \"prompt_tokens\": 5,\n    \"completion_tokens\":
30 |         25,\n    \"total_tokens\": 30\n  }\n}\n"
31 |     headers:
32 |       CF-RAY:
33 |       - 952cbd7fdb714ccc-BOS
34 |       Cache-Control:
35 |       - no-cache, must-revalidate
36 |       Connection:
37 |       - keep-alive
38 |       Content-Encoding:
39 |       - gzip
40 |       Content-Type:
41 |       - application/json
42 |       Date:
43 |       - Fri, 20 Jun 2025 16:46:05 GMT
44 |       Server:
45 |       - cloudflare
46 |       Set-Cookie:
47 |       - __cf_bm=j6S2mBNhMz7EyUcHP91.lw4c75jHfXfzqtTxGQBXlY4-1750437965-1.0.1.1-dvknSL73XTW886LF2Bj_0TZjLv7VJ5jo30vRINm1oGOFlZ_ck0tqXXBlDz9m1nbVKjKM3g_LX1ePfG8pQsHezOwSEjN4tZJoJzQGP61jDH4;
48 |         path=/; expires=Fri, 20-Jun-25 17:16:05 GMT; domain=.api.openai.com; HttpOnly;
49 |         Secure; SameSite=None
50 |       - _cfuvid=6Rd7vDg4fZ.0RQuw5Z9yTsbc.daHokuxDLbwfzsd_eY-1750437965227-0.0.1.1-604800000;
51 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
52 |       Transfer-Encoding:
53 |       - chunked
54 |       X-Content-Type-Options:
55 |       - nosniff
56 |       access-control-allow-origin:
57 |       - '*'
58 |       access-control-expose-headers:
59 |       - X-Request-ID
60 |       alt-svc:
61 |       - h3=":443"; ma=86400
62 |       cf-cache-status:
63 |       - DYNAMIC
64 |       openai-model:
65 |       - gpt-3.5-turbo-instruct:20230824-v2
66 |       openai-organization:
67 |       - datadog-staging
68 |       openai-processing-ms:
69 |       - '251'
70 |       openai-version:
71 |       - '2020-10-01'
72 |       strict-transport-security:
73 |       - max-age=31536000; includeSubDomains; preload
74 |       via:
75 |       - envoy-router-6655b8f567-qtdk4
76 |       x-envoy-upstream-service-time:
77 |       - '259'
78 |       x-ratelimit-limit-requests:
79 |       - '3500'
80 |       x-ratelimit-limit-tokens:
81 |       - '90000'
82 |       x-ratelimit-remaining-requests:
83 |       - '3499'
84 |       x-ratelimit-remaining-tokens:
85 |       - '89994'
86 |       x-ratelimit-reset-requests:
87 |       - 17ms
88 |       x-ratelimit-reset-tokens:
89 |       - 4ms
90 |       x-request-id:
91 |       - req_51fef3cc389c0b66be59910908e7fcac
92 |     status:
93 |       code: 200
94 |       message: OK
95 | version: 1
96 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_files_file-RpTpuvRVtnKpdKZb7DDGto_content_get_60bd10ef.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json, text/plain, */*
 7 |       Accept-Encoding:
 8 |       - gzip, deflate
 9 |       Connection:
10 |       - keep-alive
11 |       User-Agent:
12 |       - OpenAI/NodeJS/3.0.0
13 |     method: GET
14 |     uri: https://api.openai.com/v1/files/file-RpTpuvRVtnKpdKZb7DDGto/content
15 |   response:
16 |     body:
17 |       string: '{"messages":[{"role":"user","content":"What is the meaning of life?"},{"role":"assistant","content":"The
18 |         meaning of life is 42"}]}
19 | 
20 |         {"messages":[{"role":"user","content":"What is 5 + 5?"},{"role":"assistant","content":"5
21 |         + 5 equals 10"}]}
22 | 
23 |         {"messages":[{"role":"user","content":"What is the capital of France?"},{"role":"assistant","content":"The
24 |         capital of France is Paris"}]}
25 | 
26 |         {"messages":[{"role":"user","content":"What is the largest planet in our solar
27 |         system?"},{"role":"assistant","content":"Jupiter is the largest planet in
28 |         our solar system"}]}
29 | 
30 |         {"messages":[{"role":"user","content":"How many sides does a triangle have?"},{"role":"assistant","content":"A
31 |         triangle has three sides"}]}
32 | 
33 |         {"messages":[{"role":"user","content":"What is the chemical symbol for gold?"},{"role":"assistant","content":"The
34 |         chemical symbol for gold is Au"}]}
35 | 
36 |         {"messages":[{"role":"user","content":"What is the opposite of hot?"},{"role":"assistant","content":"The
37 |         opposite of hot is cold"}]}
38 | 
39 |         {"messages":[{"role":"user","content":"How many days are in a week?"},{"role":"assistant","content":"There
40 |         are seven days in a week"}]}
41 | 
42 |         {"messages":[{"role":"user","content":"What color is the sky on a clear day?"},{"role":"assistant","content":"The
43 |         sky is blue on a clear day"}]}
44 | 
45 |         {"messages":[{"role":"user","content":"What is the square root of 16?"},{"role":"assistant","content":"The
46 |         square root of 16 is 4"}]}'
47 |     headers:
48 |       CF-RAY:
49 |       - 952b793afbe34ce3-BOS
50 |       Connection:
51 |       - keep-alive
52 |       Content-Length:
53 |       - '1386'
54 |       Content-Type:
55 |       - application/octet-stream
56 |       Date:
57 |       - Fri, 20 Jun 2025 13:04:44 GMT
58 |       Server:
59 |       - cloudflare
60 |       Set-Cookie:
61 |       - __cf_bm=p4OUNBzFwXWKJBaO4EUDnxlwKu0MyNZBku0DhvWKzsc-1750424684-1.0.1.1-820WErCSEHFe.sEIIs0TFGg8CyXvfA026S6wnlnIUxir3m35MsRR_aZ9YY_5E03dOMPIRHI4Utl6c5j00lSeKFakRJ2w0N3y.M4CH10Ba8o;
62 |         path=/; expires=Fri, 20-Jun-25 13:34:44 GMT; domain=.api.openai.com; HttpOnly;
63 |         Secure; SameSite=None
64 |       - _cfuvid=PT6_Ee9hkmsz1ttcU97AqzHjDj1RI30Noh.Bgsl7KjM-1750424684171-0.0.1.1-604800000;
65 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
66 |       X-Content-Type-Options:
67 |       - nosniff
68 |       access-control-allow-origin:
69 |       - '*'
70 |       alt-svc:
71 |       - h3=":443"; ma=86400
72 |       cf-cache-status:
73 |       - DYNAMIC
74 |       content-disposition:
75 |       - attachment; filename="fine-tune.jsonl"
76 |       openai-organization:
77 |       - datadog-staging
78 |       openai-processing-ms:
79 |       - '1347'
80 |       openai-version:
81 |       - '2020-10-01'
82 |       strict-transport-security:
83 |       - max-age=31536000; includeSubDomains; preload
84 |       x-request-id:
85 |       - req_578fa30d1e795003fe291d980cf20a74
86 |     status:
87 |       code: 200
88 |       message: OK
89 | version: 1
90 | 


--------------------------------------------------------------------------------
/tests/integration_snapshots/test_trace_stats.json:
--------------------------------------------------------------------------------
  1 | [[
  2 |   {
  3 |     "name": "http.request",
  4 |     "service": "",
  5 |     "resource": "/users/view",
  6 |     "trace_id": 0,
  7 |     "span_id": 1,
  8 |     "parent_id": 0,
  9 |     "type": "",
 10 |     "error": 0,
 11 |     "meta": {
 12 |       "_dd.p.dm": "-3",
 13 |       "_dd.p.ksr": "0",
 14 |       "_dd.p.tid": "68e68ed400000000",
 15 |       "language": "python",
 16 |       "runtime-id": "6c0e6f7ea24a423bb099504dddd4f5e6"
 17 |     },
 18 |     "metrics": {
 19 |       "_dd.rule_psr": 0.0,
 20 |       "_dd.top_level": 1,
 21 |       "_dd.tracer_kr": 1.0,
 22 |       "_sampling_priority_v1": -1,
 23 |       "process_id": 43445
 24 |     },
 25 |     "duration": 23000,
 26 |     "start": 1759940308841385000
 27 |   }],
 28 | [
 29 |   {
 30 |     "name": "http.request",
 31 |     "service": "",
 32 |     "resource": "/users/view",
 33 |     "trace_id": 1,
 34 |     "span_id": 1,
 35 |     "parent_id": 0,
 36 |     "type": "",
 37 |     "error": 0,
 38 |     "meta": {
 39 |       "_dd.p.dm": "-3",
 40 |       "_dd.p.ksr": "0",
 41 |       "_dd.p.tid": "68e68ed400000000",
 42 |       "language": "python",
 43 |       "runtime-id": "6c0e6f7ea24a423bb099504dddd4f5e6"
 44 |     },
 45 |     "metrics": {
 46 |       "_dd.rule_psr": 0.0,
 47 |       "_dd.top_level": 1,
 48 |       "_dd.tracer_kr": 1.0,
 49 |       "_sampling_priority_v1": -1,
 50 |       "process_id": 43445
 51 |     },
 52 |     "duration": 7000,
 53 |     "start": 1759940308841551000
 54 |   }],
 55 | [
 56 |   {
 57 |     "name": "http.request",
 58 |     "service": "",
 59 |     "resource": "/users/view",
 60 |     "trace_id": 2,
 61 |     "span_id": 1,
 62 |     "parent_id": 0,
 63 |     "type": "",
 64 |     "error": 0,
 65 |     "meta": {
 66 |       "_dd.p.dm": "-3",
 67 |       "_dd.p.ksr": "0",
 68 |       "_dd.p.tid": "68e68ed400000000",
 69 |       "language": "python",
 70 |       "runtime-id": "6c0e6f7ea24a423bb099504dddd4f5e6"
 71 |     },
 72 |     "metrics": {
 73 |       "_dd.rule_psr": 0.0,
 74 |       "_dd.top_level": 1,
 75 |       "_dd.tracer_kr": 1.0,
 76 |       "_sampling_priority_v1": -1,
 77 |       "process_id": 43445
 78 |     },
 79 |     "duration": 5000,
 80 |     "start": 1759940308841584000
 81 |   }],
 82 | [
 83 |   {
 84 |     "name": "http.request",
 85 |     "service": "",
 86 |     "resource": "/users/view",
 87 |     "trace_id": 3,
 88 |     "span_id": 1,
 89 |     "parent_id": 0,
 90 |     "type": "",
 91 |     "error": 0,
 92 |     "meta": {
 93 |       "_dd.p.dm": "-3",
 94 |       "_dd.p.ksr": "0",
 95 |       "_dd.p.tid": "68e68ed400000000",
 96 |       "language": "python",
 97 |       "runtime-id": "6c0e6f7ea24a423bb099504dddd4f5e6"
 98 |     },
 99 |     "metrics": {
100 |       "_dd.rule_psr": 0.0,
101 |       "_dd.top_level": 1,
102 |       "_dd.tracer_kr": 1.0,
103 |       "_sampling_priority_v1": -1,
104 |       "process_id": 43445
105 |     },
106 |     "duration": 3000,
107 |     "start": 1759940308841610000
108 |   }],
109 | [
110 |   {
111 |     "name": "http.request",
112 |     "service": "",
113 |     "resource": "/users/view",
114 |     "trace_id": 4,
115 |     "span_id": 1,
116 |     "parent_id": 0,
117 |     "type": "",
118 |     "error": 1,
119 |     "meta": {
120 |       "_dd.p.dm": "-3",
121 |       "_dd.p.ksr": "0",
122 |       "_dd.p.tid": "68e68ed400000000",
123 |       "language": "python",
124 |       "runtime-id": "6c0e6f7ea24a423bb099504dddd4f5e6"
125 |     },
126 |     "metrics": {
127 |       "_dd.rule_psr": 0.0,
128 |       "_dd.top_level": 1,
129 |       "_dd.tracer_kr": 1.0,
130 |       "_sampling_priority_v1": -1,
131 |       "process_id": 43445
132 |     },
133 |     "duration": 4000,
134 |     "start": 1759940308841628000
135 |   }]]
136 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_57612363.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo","messages":[{"role":"system","content":"You are
 4 |       a helpful assistant."},{"role":"user","content":"Hello, OpenAI!"}],"temperature":0.5,"stream":false,"max_tokens":100,"n":1,"user":"dd-trace-test"}'
 5 |     headers:
 6 |       Accept:
 7 |       - application/json, text/plain, */*
 8 |       Accept-Encoding:
 9 |       - gzip, deflate
10 |       Connection:
11 |       - keep-alive
12 |       Content-Length:
13 |       - '219'
14 |       Content-Type:
15 |       - application/json
16 |       User-Agent:
17 |       - OpenAI/NodeJS/3.3.0
18 |     method: POST
19 |     uri: https://api.openai.com/v1/chat/completions
20 |   response:
21 |     body:
22 |       string: "{\n  \"id\": \"chatcmpl-BkXa0GDvXwjXmEh6LZMuw0iQFKt8d\",\n  \"object\":
23 |         \"chat.completion\",\n  \"created\": 1750431868,\n  \"model\": \"gpt-3.5-turbo-0125\",\n
24 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
25 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
26 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
27 |         null,\n      \"finish_reason\": \"stop\"\n    }\n  ],\n  \"usage\": {\n    \"prompt_tokens\":
28 |         22,\n    \"completion_tokens\": 9,\n    \"total_tokens\": 31,\n    \"prompt_tokens_details\":
29 |         {\n      \"cached_tokens\": 0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
30 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
31 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
32 |         \"default\",\n  \"system_fingerprint\": null\n}\n"
33 |     headers:
34 |       CF-RAY:
35 |       - 952c28a96f1ec351-EWR
36 |       Connection:
37 |       - keep-alive
38 |       Content-Encoding:
39 |       - gzip
40 |       Content-Type:
41 |       - application/json
42 |       Date:
43 |       - Fri, 20 Jun 2025 15:04:28 GMT
44 |       Server:
45 |       - cloudflare
46 |       Set-Cookie:
47 |       - __cf_bm=GuQFHiu.wMEyW0zJ6baqAJE9kyw93buvPmgLM.5klxE-1750431868-1.0.1.1-23veSeLqD7zNLyTTuMGByp8ut1oNUuU.nI.8YRxjzBuFIlGlX738TGr6uRkSKvlk1VOfDjTR0ZAuXXdu5R.vVUh0x9PNSW607dLCd9ty.q4;
48 |         path=/; expires=Fri, 20-Jun-25 15:34:28 GMT; domain=.api.openai.com; HttpOnly;
49 |         Secure; SameSite=None
50 |       - _cfuvid=lY7guA00Orgtvo39tD.tireurOTxrW9IDyM20I_voxM-1750431868854-0.0.1.1-604800000;
51 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
52 |       Transfer-Encoding:
53 |       - chunked
54 |       X-Content-Type-Options:
55 |       - nosniff
56 |       access-control-expose-headers:
57 |       - X-Request-ID
58 |       alt-svc:
59 |       - h3=":443"; ma=86400
60 |       cf-cache-status:
61 |       - DYNAMIC
62 |       openai-organization:
63 |       - datadog-staging
64 |       openai-processing-ms:
65 |       - '236'
66 |       openai-version:
67 |       - '2020-10-01'
68 |       strict-transport-security:
69 |       - max-age=31536000; includeSubDomains; preload
70 |       x-envoy-upstream-service-time:
71 |       - '240'
72 |       x-ratelimit-limit-requests:
73 |       - '10000'
74 |       x-ratelimit-limit-tokens:
75 |       - '50000000'
76 |       x-ratelimit-remaining-requests:
77 |       - '9999'
78 |       x-ratelimit-remaining-tokens:
79 |       - '49999985'
80 |       x-ratelimit-reset-requests:
81 |       - 6ms
82 |       x-ratelimit-reset-tokens:
83 |       - 0s
84 |       x-request-id:
85 |       - req_ee6face1aeb3c6085f9963c427425c04
86 |     status:
87 |       code: 200
88 |       message: OK
89 | version: 1
90 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/tracestats_snapshot.py:
--------------------------------------------------------------------------------
 1 | import json
 2 | from typing import List
 3 | 
 4 | from .checks import CheckTrace
 5 | from .tracestats import StatsBucket
 6 | 
 7 | 
 8 | def _normalize_statsbuckets(buckets: List[StatsBucket]) -> List[StatsBucket]:
 9 |     """Normalize the stats bucket by normalizing the time buckets."""
10 |     # Make a copy of the buckets, note that the sketches are not copied.
11 |     normed_buckets = []
12 |     for bucket in buckets:
13 |         bcopy = bucket.copy()
14 |         bcopy["Stats"] = [aggr.copy() for aggr in bucket["Stats"]]  # Copy the aggregations
15 |         normed_buckets.append(bcopy)
16 | 
17 |     # Order the buckets by time
18 |     normed_buckets = sorted(normed_buckets, key=lambda b: b["Start"])
19 | 
20 |     # Sort aggr for a bucket alphanumerically
21 |     for bucket in normed_buckets:
22 |         # Sort aggrs to get a consistent list of buckets, since order does not matter.
23 |         bucket["Stats"] = sorted(
24 |             bucket["Stats"],
25 |             key=lambda b: (
26 |                 b["Name"],
27 |                 b["Resource"],
28 |                 b["Type"],
29 |                 b.get("Synthetics"),
30 |                 b.get("HTTPStatusCode"),
31 |                 b["Hits"],
32 |             ),
33 |         )
34 | 
35 |     start = normed_buckets[0]["Start"]
36 |     for b in normed_buckets:
37 |         b["Start"] -= start
38 | 
39 |     return normed_buckets
40 | 
41 | 
42 | def snapshot(expected_stats: List[StatsBucket], received_stats: List[StatsBucket]) -> None:
43 |     # Normalize the stats buckets by making them independent of time. Only ordering matters.
44 | 
45 |     # Sort the buckets by start time.
46 |     normed_expected = _normalize_statsbuckets(expected_stats)
47 |     normed_received = _normalize_statsbuckets(received_stats)
48 | 
49 |     # TODO: do better matching and comparing to aid in debugging
50 |     assert len(normed_received) == len(
51 |         normed_expected
52 |     ), f"Number of stats buckets ({len(normed_received)}) doesn't match expected ({len(normed_expected)})."
53 | 
54 |     with CheckTrace.add_frame(f"snapshot compare of {len(normed_received)} stats buckets"):
55 |         # Do a really rough comparison.
56 |         for i, (exp_bucket, rec_bucket) in enumerate(zip(normed_expected, normed_received)):
57 |             with CheckTrace.add_frame(f"comparison of bucket {i}") as frame:
58 |                 exp_aggrs = exp_bucket["Stats"]
59 |                 rec_aggrs = rec_bucket["Stats"]
60 |                 frame.add_item(f"Expected aggregated stats: {exp_aggrs}")
61 |                 frame.add_item(f"Received aggregated stats: {rec_aggrs}")
62 |                 assert len(exp_aggrs) == len(
63 |                     rec_aggrs
64 |                 ), f"Number of aggregations ({len(rec_aggrs)}) in bucket {i} doesn't match expected: ({len(exp_aggrs)})."
65 | 
66 |                 for j, (exp_aggr, rec_aggr) in enumerate(zip(exp_aggrs, rec_aggrs)):
67 |                     # Omit duration and sketches for now
68 |                     # Duration and sketches will be noisy
69 |                     for attr in (
70 |                         "Name",
71 |                         "Resource",
72 |                         "Type",
73 |                         "Synthetics",
74 |                         "Hits",
75 |                         "TopLevelHits",
76 |                         "Errors",
77 |                         "HTTPStatusCode",
78 |                     ):
79 |                         exp_value, rec_value = exp_aggr[attr], rec_aggr[attr]  # type: ignore
80 |                         if exp_value != rec_value:
81 |                             raise AssertionError(
82 |                                 f"Expected value ('{exp_value}') for field '{attr}' does not match received value: '{rec_value}'."
83 |                             )
84 | 
85 | 
86 | def generate(received_stats: List[StatsBucket]) -> str:
87 |     return f"{json.dumps(_normalize_statsbuckets(received_stats), indent=2)}\n"
88 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/remoteconfig.py:
--------------------------------------------------------------------------------
 1 | import base64
 2 | import datetime
 3 | import hashlib
 4 | import json
 5 | from typing import Any
 6 | from typing import Dict
 7 | from typing import Optional
 8 | 
 9 | 
10 | class RemoteConfigServer:
11 |     _responses: Dict[Optional[str], Any] = {}
12 | 
13 |     def _update_response(self, token: Optional[str], data: Dict[str, Any]) -> None:
14 |         if self._responses.get(token):
15 |             self._responses[token].update(data)
16 |         else:
17 |             self._create_response(token, data)
18 | 
19 |     def _create_response(self, token: Optional[str], data: Dict[str, Any]) -> None:
20 |         self._responses[token] = data
21 | 
22 |     async def _get_response(self, token: Optional[str]) -> Any:
23 |         return self._responses.get(token, {})
24 | 
25 |     def update_config_response(self, token: Optional[str], data: Dict[str, Any]) -> None:
26 |         self._update_response(token, data)
27 | 
28 |     @staticmethod
29 |     def _build_config_path_response(path: str, msg: Any) -> Dict[str, Any]:
30 |         expires_date = datetime.datetime.strftime(
31 |             datetime.datetime.now() + datetime.timedelta(days=1), "%Y-%m-%dT%H:%M:%SZ"
32 |         )
33 |         msg_enc = bytes(json.dumps(msg), encoding="utf-8")
34 |         data = {
35 |             "signatures": [{"keyid": "", "sig": ""}],
36 |             "signed": {
37 |                 "_type": "targets",
38 |                 "custom": {"opaque_backend_state": ""},
39 |                 "expires": expires_date,
40 |                 "spec_version": "1.0.0",
41 |                 "targets": {
42 |                     path: {
43 |                         "custom": {"c": [""], "v": 0},
44 |                         "hashes": {"sha256": hashlib.sha256(msg_enc).hexdigest()},
45 |                         "length": len(msg_enc),
46 |                     }
47 |                 },
48 |                 "version": 0,
49 |             },
50 |         }
51 |         remote_config_payload = {
52 |             "roots": [
53 |                 str(
54 |                     base64.b64encode(
55 |                         bytes(
56 |                             json.dumps(
57 |                                 {
58 |                                     "signatures": [],
59 |                                     "signed": {
60 |                                         "_type": "root",
61 |                                         "consistent_snapshot": True,
62 |                                         "expires": "1986-12-11T00:00:00Z",
63 |                                         "keys": {},
64 |                                         "roles": {},
65 |                                         "spec_version": "1.0",
66 |                                         "version": 2,
67 |                                     },
68 |                                 }
69 |                             ),
70 |                             encoding="utf-8",
71 |                         )
72 |                     ),
73 |                     encoding="utf-8",
74 |                 )
75 |             ],
76 |             "targets": str(base64.b64encode(bytes(json.dumps(data), encoding="utf-8")), encoding="utf-8"),
77 |             "target_files": [
78 |                 {
79 |                     "path": path,
80 |                     "raw": str(base64.b64encode(msg_enc), encoding="utf-8"),
81 |                 }
82 |             ],
83 |             "client_configs": [path],
84 |         }
85 |         return remote_config_payload
86 | 
87 |     def create_config_path_response(self, token: Optional[str], path: str, msg: Any) -> None:
88 |         remote_config_payload = self._build_config_path_response(path, msg)
89 |         self.create_config_response(token, remote_config_payload)
90 | 
91 |     def create_config_response(self, token: Optional[str], data: Dict[str, Any]) -> None:
92 |         self._create_response(token, data)
93 | 
94 |     async def get_config_response(self, token: Optional[str]) -> Any:
95 |         return await self._get_response(token)
96 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_655a8ef2.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo","messages":[{"role":"user","content":"What is
 4 |       the weather in New York City?"}],"functions":[{"name":"get_weather","description":"Get
 5 |       the weather in a given city","parameters":{"type":"object","properties":{"city":{"type":"string","description":"The
 6 |       city to get the weather for"}}}}],"function_call":"auto","stream":false}'
 7 |     headers:
 8 |       Accept:
 9 |       - application/json, text/plain, */*
10 |       Accept-Encoding:
11 |       - gzip, deflate
12 |       Connection:
13 |       - keep-alive
14 |       Content-Length:
15 |       - '346'
16 |       Content-Type:
17 |       - application/json
18 |       User-Agent:
19 |       - OpenAI/NodeJS/3.3.0
20 |     method: POST
21 |     uri: https://api.openai.com/v1/chat/completions
22 |   response:
23 |     body:
24 |       string: "{\n  \"id\": \"chatcmpl-Bkav0WiivW3boplX6SfS25N5tZjIE\",\n  \"object\":
25 |         \"chat.completion\",\n  \"created\": 1750444702,\n  \"model\": \"gpt-3.5-turbo-0125\",\n
26 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
27 |         \"assistant\",\n        \"content\": null,\n        \"function_call\": {\n
28 |         \         \"name\": \"get_weather\",\n          \"arguments\": \"{\\\"city\\\":\\\"New
29 |         York City\\\"}\"\n        },\n        \"refusal\": null,\n        \"annotations\":
30 |         []\n      },\n      \"logprobs\": null,\n      \"finish_reason\": \"function_call\"\n
31 |         \   }\n  ],\n  \"usage\": {\n    \"prompt_tokens\": 65,\n    \"completion_tokens\":
32 |         16,\n    \"total_tokens\": 81,\n    \"prompt_tokens_details\": {\n      \"cached_tokens\":
33 |         0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
34 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
35 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
36 |         \"default\",\n  \"system_fingerprint\": null\n}\n"
37 |     headers:
38 |       CF-RAY:
39 |       - 952d61feebd19056-BOS
40 |       Connection:
41 |       - keep-alive
42 |       Content-Encoding:
43 |       - gzip
44 |       Content-Type:
45 |       - application/json
46 |       Date:
47 |       - Fri, 20 Jun 2025 18:38:23 GMT
48 |       Server:
49 |       - cloudflare
50 |       Set-Cookie:
51 |       - __cf_bm=mlrOwzBeyTEjWyNj2fciCZtivYipbkstmM6lDpaA9eQ-1750444703-1.0.1.1-apZ7VamzlWw01_hH0wBfmm6FGBPQThdlanEBUag8qo2gddOmQZXV3ntuFvVYIn1PZMmrfQbuSlK8leLksxQiLmsjCg7swCI2i9o7VpDt3c4;
52 |         path=/; expires=Fri, 20-Jun-25 19:08:23 GMT; domain=.api.openai.com; HttpOnly;
53 |         Secure; SameSite=None
54 |       - _cfuvid=TFlt5iAAE_VU.gJJLXNO6tjK8Etb2ZGIdql_0aIefnI-1750444703295-0.0.1.1-604800000;
55 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
56 |       Transfer-Encoding:
57 |       - chunked
58 |       X-Content-Type-Options:
59 |       - nosniff
60 |       access-control-expose-headers:
61 |       - X-Request-ID
62 |       alt-svc:
63 |       - h3=":443"; ma=86400
64 |       cf-cache-status:
65 |       - DYNAMIC
66 |       openai-organization:
67 |       - datadog-staging
68 |       openai-processing-ms:
69 |       - '314'
70 |       openai-version:
71 |       - '2020-10-01'
72 |       strict-transport-security:
73 |       - max-age=31536000; includeSubDomains; preload
74 |       x-envoy-upstream-service-time:
75 |       - '318'
76 |       x-ratelimit-limit-requests:
77 |       - '10000'
78 |       x-ratelimit-limit-tokens:
79 |       - '50000000'
80 |       x-ratelimit-remaining-requests:
81 |       - '9999'
82 |       x-ratelimit-remaining-tokens:
83 |       - '49999987'
84 |       x-ratelimit-reset-requests:
85 |       - 6ms
86 |       x-ratelimit-reset-tokens:
87 |       - 0s
88 |       x-request-id:
89 |       - req_b21e85708cf94d2591ae389d3a1c2d02
90 |     status:
91 |       code: 200
92 |       message: OK
93 | version: 1
94 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_649d8162.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-4o-mini","messages":[{"role":"user","content":[{"type":"text","text":"What
 4 |       is in this image?"},{"type":"image_url","image_url":{"url":"https://tinyurl.com/4mfz54bx"}}]}]}'
 5 |     headers:
 6 |       Accept:
 7 |       - application/json, text/plain, */*
 8 |       Accept-Encoding:
 9 |       - gzip, deflate
10 |       Connection:
11 |       - keep-alive
12 |       Content-Length:
13 |       - '184'
14 |       Content-Type:
15 |       - application/json
16 |       User-Agent:
17 |       - OpenAI/NodeJS/3.3.0
18 |     method: POST
19 |     uri: https://api.openai.com/v1/chat/completions
20 |   response:
21 |     body:
22 |       string: "{\n  \"id\": \"chatcmpl-BkXmjJeTCIKgy3PZ03ZYVnQFA5Sza\",\n  \"object\":
23 |         \"chat.completion\",\n  \"created\": 1750432657,\n  \"model\": \"gpt-4o-mini-2024-07-18\",\n
24 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
25 |         \"assistant\",\n        \"content\": \"The image features a cat with striking
26 |         blue eyes and a mix of light and dark fur. The background appears to be black,
27 |         emphasizing the cat's features.\",\n        \"refusal\": null,\n        \"annotations\":
28 |         []\n      },\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n
29 |         \   }\n  ],\n  \"usage\": {\n    \"prompt_tokens\": 36848,\n    \"completion_tokens\":
30 |         31,\n    \"total_tokens\": 36879,\n    \"prompt_tokens_details\": {\n      \"cached_tokens\":
31 |         0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
32 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
33 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
34 |         \"default\",\n  \"system_fingerprint\": \"fp_62a23a81ef\"\n}\n"
35 |     headers:
36 |       CF-RAY:
37 |       - 952c3be5c8f88f7b-BOS
38 |       Connection:
39 |       - keep-alive
40 |       Content-Encoding:
41 |       - gzip
42 |       Content-Type:
43 |       - application/json
44 |       Date:
45 |       - Fri, 20 Jun 2025 15:17:40 GMT
46 |       Server:
47 |       - cloudflare
48 |       Set-Cookie:
49 |       - __cf_bm=bTyTTi2RpTe5ENkmNk1gSZLHB6TaAOn4hP67KV5bz14-1750432660-1.0.1.1-gYEtvicfSp9m2wu8oCzWVor7rG.0Bxok_BKH9fdwSoKfWKYX.fP4MB44hJTV5JW8WZup9E8mm4PiTKIC8ZVyG6QTL_jQiIFUAxT.IMGC8Ps;
50 |         path=/; expires=Fri, 20-Jun-25 15:47:40 GMT; domain=.api.openai.com; HttpOnly;
51 |         Secure; SameSite=None
52 |       - _cfuvid=N8CxhlHAS9ekG3EkMsDg5WCr_fdo60E1cAQnK225Vso-1750432660355-0.0.1.1-604800000;
53 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
54 |       Transfer-Encoding:
55 |       - chunked
56 |       X-Content-Type-Options:
57 |       - nosniff
58 |       access-control-expose-headers:
59 |       - X-Request-ID
60 |       alt-svc:
61 |       - h3=":443"; ma=86400
62 |       cf-cache-status:
63 |       - DYNAMIC
64 |       openai-organization:
65 |       - datadog-staging
66 |       openai-processing-ms:
67 |       - '3954'
68 |       openai-version:
69 |       - '2020-10-01'
70 |       strict-transport-security:
71 |       - max-age=31536000; includeSubDomains; preload
72 |       x-envoy-upstream-service-time:
73 |       - '3259'
74 |       x-ratelimit-limit-input-images:
75 |       - '50000'
76 |       x-ratelimit-limit-requests:
77 |       - '30000'
78 |       x-ratelimit-limit-tokens:
79 |       - '150000000'
80 |       x-ratelimit-remaining-input-images:
81 |       - '49999'
82 |       x-ratelimit-remaining-requests:
83 |       - '29999'
84 |       x-ratelimit-remaining-tokens:
85 |       - '149999226'
86 |       x-ratelimit-reset-input-images:
87 |       - 1ms
88 |       x-ratelimit-reset-requests:
89 |       - 2ms
90 |       x-ratelimit-reset-tokens:
91 |       - 0s
92 |       x-request-id:
93 |       - req_9906374ee983c1241767608f21705b47
94 |     status:
95 |       code: 200
96 |       message: OK
97 | version: 1
98 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_fine_tuning_jobs_ftjob-q9CUUUsHJemGUVQ1Ecc01zcf_events_get_557d820e.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: null
 4 |     headers:
 5 |       Accept:
 6 |       - application/json
 7 |       Accept-Encoding:
 8 |       - gzip,deflate
 9 |       Connection:
10 |       - keep-alive
11 |       Content-Type:
12 |       - application/json
13 |       User-Agent:
14 |       - OpenAI/JS 4.17.0
15 |       X-Stainless-Arch:
16 |       - arm64
17 |       X-Stainless-Lang:
18 |       - js
19 |       X-Stainless-Os:
20 |       - MacOS
21 |       X-Stainless-Package-Version:
22 |       - 4.17.0
23 |       X-Stainless-Runtime:
24 |       - node
25 |       X-Stainless-Runtime-Version:
26 |       - v22.13.1
27 |     method: GET
28 |     uri: https://api.openai.com/v1/fine_tuning/jobs/ftjob-q9CUUUsHJemGUVQ1Ecc01zcf/events
29 |   response:
30 |     body:
31 |       string: "{\n  \"object\": \"list\",\n  \"data\": [\n    {\n      \"object\":
32 |         \"fine_tuning.job.event\",\n      \"id\": \"ftevent-Cx5ZK5J7luO94qrNem4VPvgC\",\n
33 |         \     \"created_at\": 1750424922,\n      \"level\": \"warn\",\n      \"message\":
34 |         \"The job was stopped due to a cancellation request\",\n      \"data\": null,\n
35 |         \     \"type\": \"message\"\n    },\n    {\n      \"object\": \"fine_tuning.job.event\",\n
36 |         \     \"id\": \"ftevent-5KrghIsIPhIjF04a6Dgge4fb\",\n      \"created_at\":
37 |         1750424800,\n      \"level\": \"info\",\n      \"message\": \"Fine-tuning
38 |         job started\",\n      \"data\": null,\n      \"type\": \"message\"\n    },\n
39 |         \   {\n      \"object\": \"fine_tuning.job.event\",\n      \"id\": \"ftevent-BPfhTk63npLjsE1sC61hMui2\",\n
40 |         \     \"created_at\": 1750424795,\n      \"level\": \"info\",\n      \"message\":
41 |         \"Files validated, moving job to queued state\",\n      \"data\": {},\n      \"type\":
42 |         \"message\"\n    },\n    {\n      \"object\": \"fine_tuning.job.event\",\n
43 |         \     \"id\": \"ftevent-azRJ5VTezaTOaP6g16NcHR17\",\n      \"created_at\":
44 |         1750424715,\n      \"level\": \"info\",\n      \"message\": \"Validating training
45 |         file: file-RpTpuvRVtnKpdKZb7DDGto\",\n      \"data\": {},\n      \"type\":
46 |         \"message\"\n    },\n    {\n      \"object\": \"fine_tuning.job.event\",\n
47 |         \     \"id\": \"ftevent-1046aVlktddmVhWzAxcC2z0X\",\n      \"created_at\":
48 |         1750424715,\n      \"level\": \"info\",\n      \"message\": \"Created fine-tuning
49 |         job: ftjob-q9CUUUsHJemGUVQ1Ecc01zcf\",\n      \"data\": {},\n      \"type\":
50 |         \"message\"\n    }\n  ],\n  \"has_more\": false\n}"
51 |     headers:
52 |       CF-RAY:
53 |       - 952b829358018f85-BOS
54 |       Connection:
55 |       - keep-alive
56 |       Content-Encoding:
57 |       - gzip
58 |       Content-Type:
59 |       - application/json
60 |       Date:
61 |       - Fri, 20 Jun 2025 13:11:05 GMT
62 |       Server:
63 |       - cloudflare
64 |       Set-Cookie:
65 |       - __cf_bm=IXZa8aHNVV8cz_Y4qAqMecqx7_.vyqYkoRg..En5Ry4-1750425065-1.0.1.1-VRH_gqQr2SpG.0gAD_dW3pZ97Fw5KnDQ1SPyHqAkbiT6vUV4t03tizPxD6PjiuiHDCewEOv6fKGN1pnnq05CQ5hQ8y_YCG1ZM8A4_pHdGPk;
66 |         path=/; expires=Fri, 20-Jun-25 13:41:05 GMT; domain=.api.openai.com; HttpOnly;
67 |         Secure; SameSite=None
68 |       - _cfuvid=jvGW6q5kCbZxJc_KMlWW.Suje7ZSAVedBKpd2Jc1M54-1750425065627-0.0.1.1-604800000;
69 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
70 |       Transfer-Encoding:
71 |       - chunked
72 |       X-Content-Type-Options:
73 |       - nosniff
74 |       alt-svc:
75 |       - h3=":443"; ma=86400
76 |       cf-cache-status:
77 |       - DYNAMIC
78 |       openai-organization:
79 |       - datadog-staging
80 |       openai-processing-ms:
81 |       - '54'
82 |       openai-version:
83 |       - '2020-10-01'
84 |       strict-transport-security:
85 |       - max-age=31536000; includeSubDomains; preload
86 |       x-request-id:
87 |       - req_9f0b6dc8bde4df07193394349e722e68
88 |     status:
89 |       code: 200
90 |       message: OK
91 | version: 1
92 | 


--------------------------------------------------------------------------------
/tests/test_container.py:
--------------------------------------------------------------------------------
  1 | import contextlib
  2 | import os
  3 | import platform
  4 | import shutil
  5 | import subprocess
  6 | import time
  7 | from typing import Dict
  8 | from typing import Generator
  9 | from typing import List
 10 | from typing import Tuple
 11 | 
 12 | import pytest
 13 | 
 14 | 
 15 | def is_docker_available():
 16 |     """Check if the Docker binary is available on the system."""
 17 |     return shutil.which("docker") is not None
 18 | 
 19 | 
 20 | pytestmark = [
 21 |     pytest.mark.skipif(os.getenv("SKIP_CONTAINER") is not None, reason="SKIP_CONTAINER set"),
 22 |     pytest.mark.skipif(not is_docker_available(), reason="Docker is not available"),
 23 | ]
 24 | 
 25 | 
 26 | class DockerContainer:
 27 |     def __init__(self, cid: str):
 28 |         self.id = cid
 29 | 
 30 |     def logs(self):
 31 |         p = subprocess.run(["docker", "logs", self.id], capture_output=True, check=True)
 32 |         return p.stdout.decode(), p.stderr.decode()
 33 | 
 34 | 
 35 | @contextlib.contextmanager
 36 | def docker_run(
 37 |     image: str,
 38 |     env: Dict[str, str],
 39 |     volumes: List[str],
 40 |     cmd: List[str] = [],
 41 |     ports: List[Tuple[str, str]] = [],
 42 | ) -> Generator[DockerContainer, None, None]:
 43 |     _cmd: List[str] = [
 44 |         "docker",
 45 |         "run",
 46 |         "-i",
 47 |         "--rm",
 48 |         "--detach",
 49 |     ]
 50 |     for k, v in env.items():
 51 |         _cmd.extend(["-e", "%s=%s" % (k, v)])
 52 |     for v in volumes:
 53 |         _cmd.extend(["-v", v])
 54 |     for k, v in ports:
 55 |         _cmd.extend(["-p", "%s:%s" % (k, v)])
 56 |     _cmd += [image]
 57 |     _cmd.extend(cmd)
 58 | 
 59 |     # Run the docker container
 60 |     p = subprocess.run(_cmd, capture_output=True)
 61 |     assert p.returncode == 0, p.stderr
 62 |     cid = p.stdout.decode().strip()
 63 |     assert cid
 64 |     yield DockerContainer(cid)
 65 |     # Kill the container
 66 |     subprocess.run(["docker", "kill", cid], capture_output=True, check=True)
 67 | 
 68 | 
 69 | @pytest.fixture(scope="session")
 70 | def build_image():
 71 |     subprocess.run(
 72 |         [
 73 |             "docker",
 74 |             "build",
 75 |             "-t",
 76 |             "ddapm-test-agent:test",
 77 |             "-f",
 78 |             "Dockerfile",
 79 |             ".",
 80 |         ],
 81 |         check=True,
 82 |     )
 83 |     yield
 84 |     subprocess.run(
 85 |         [
 86 |             "docker",
 87 |             "rmi",
 88 |             "-f",
 89 |             "ddapm-test-agent:test",
 90 |         ],
 91 |         check=True,
 92 |     )
 93 | 
 94 | 
 95 | @pytest.mark.skipif(platform.system() == "Linux", reason="No socket mounting issues on Linux")
 96 | def test_container_uds(build_image, tmp_path_factory):
 97 |     uds_dir = tmp_path_factory.mktemp("uds")
 98 | 
 99 |     with docker_run(
100 |         image="ddapm-test-agent:test",
101 |         volumes=[f"{str(uds_dir)}:/opt/datadog-agent/run"],
102 |         env={"DD_APM_RECEIVER_SOCKET": "/opt/datadog-agent/run/apm.socket"},
103 |     ) as c:
104 |         for i in range(50):
105 |             stdout, stderr = c.logs()
106 |             if "could not set permissions" in stderr:
107 |                 break
108 |             time.sleep(0.1)
109 |         else:
110 |             raise Exception("Test agent did not start in time: %s" % stderr)
111 | 
112 | 
113 | @pytest.mark.skipif(platform.system() != "Windows", reason="Named pipes are Windows-specific")
114 | def test_container_named_pipe(build_image, tmp_path_factory):
115 |     named_pipe_dir = tmp_path_factory.mktemp("named_pipe")
116 | 
117 |     with docker_run(
118 |         image="ddapm-test-agent:test",
119 |         volumes=[f"{str(named_pipe_dir)}:/opt/datadog-agent/run"],
120 |         env={"DD_APM_RECEIVER_NAMED_PIPE": "/opt/datadog-agent/run/apm.pipe"},
121 |     ) as c:
122 |         for i in range(50):
123 |             stdout, stderr = c.logs()
124 |             if "could not set permissions" in stderr or "could not create named pipe" in stderr:
125 |                 break
126 |             time.sleep(0.1)
127 |         else:
128 |             raise Exception("Test agent did not start in time: %s" % stderr)
129 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_4dce8db6.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo","messages":[{"role":"user","content":"What is
 4 |       the weather in New York City?"}],"tools":[{"type":"function","function":{"name":"get_weather","description":"Get
 5 |       the weather in a given city","parameters":{"type":"object","properties":{"city":{"type":"string","description":"The
 6 |       city to get the weather for"}}}}}],"tool_choice":"auto","stream":false}'
 7 |     headers:
 8 |       Accept:
 9 |       - application/json, text/plain, */*
10 |       Accept-Encoding:
11 |       - gzip, deflate
12 |       Connection:
13 |       - keep-alive
14 |       Content-Length:
15 |       - '371'
16 |       Content-Type:
17 |       - application/json
18 |       User-Agent:
19 |       - OpenAI/NodeJS/3.3.0
20 |     method: POST
21 |     uri: https://api.openai.com/v1/chat/completions
22 |   response:
23 |     body:
24 |       string: "{\n  \"id\": \"chatcmpl-BkXxzVZJidYwUN9O92szsgtTFdaDm\",\n  \"object\":
25 |         \"chat.completion\",\n  \"created\": 1750433355,\n  \"model\": \"gpt-3.5-turbo-0125\",\n
26 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
27 |         \"assistant\",\n        \"content\": null,\n        \"tool_calls\": [\n          {\n
28 |         \           \"id\": \"call_H1u2zgdjr7Ha5bQoYHq9i763\",\n            \"type\":
29 |         \"function\",\n            \"function\": {\n              \"name\": \"get_weather\",\n
30 |         \             \"arguments\": \"{\\\"city\\\":\\\"New York City\\\"}\"\n            }\n
31 |         \         }\n        ],\n        \"refusal\": null,\n        \"annotations\":
32 |         []\n      },\n      \"logprobs\": null,\n      \"finish_reason\": \"tool_calls\"\n
33 |         \   }\n  ],\n  \"usage\": {\n    \"prompt_tokens\": 65,\n    \"completion_tokens\":
34 |         16,\n    \"total_tokens\": 81,\n    \"prompt_tokens_details\": {\n      \"cached_tokens\":
35 |         0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
36 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
37 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
38 |         \"default\",\n  \"system_fingerprint\": null\n}\n"
39 |     headers:
40 |       CF-RAY:
41 |       - 952c4cf54a978fea-BOS
42 |       Connection:
43 |       - keep-alive
44 |       Content-Encoding:
45 |       - gzip
46 |       Content-Type:
47 |       - application/json
48 |       Date:
49 |       - Fri, 20 Jun 2025 15:29:15 GMT
50 |       Server:
51 |       - cloudflare
52 |       Set-Cookie:
53 |       - __cf_bm=lwbC2CHslMY0bfBlRd.abcvstLtZYwaJ39.n1eS9JG4-1750433355-1.0.1.1-BkuJdjTwLvog0CiJkN6Y9MUy6sUA33xdjkTxcZfaSheQ6ajKGrI1EQPThO7bcMyD6YH1EaPXh8et52eWmXp1RafMd8wafsmLR1G5JSOytaU;
54 |         path=/; expires=Fri, 20-Jun-25 15:59:15 GMT; domain=.api.openai.com; HttpOnly;
55 |         Secure; SameSite=None
56 |       - _cfuvid=MhSTF_Evj4D65btDYu2rj2seJvh_1FZymg0MU1lg60o-1750433355637-0.0.1.1-604800000;
57 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
58 |       Transfer-Encoding:
59 |       - chunked
60 |       X-Content-Type-Options:
61 |       - nosniff
62 |       access-control-expose-headers:
63 |       - X-Request-ID
64 |       alt-svc:
65 |       - h3=":443"; ma=86400
66 |       cf-cache-status:
67 |       - DYNAMIC
68 |       openai-organization:
69 |       - datadog-staging
70 |       openai-processing-ms:
71 |       - '338'
72 |       openai-version:
73 |       - '2020-10-01'
74 |       strict-transport-security:
75 |       - max-age=31536000; includeSubDomains; preload
76 |       x-envoy-upstream-service-time:
77 |       - '340'
78 |       x-ratelimit-limit-requests:
79 |       - '10000'
80 |       x-ratelimit-limit-tokens:
81 |       - '50000000'
82 |       x-ratelimit-remaining-requests:
83 |       - '9999'
84 |       x-ratelimit-remaining-tokens:
85 |       - '49999987'
86 |       x-ratelimit-reset-requests:
87 |       - 6ms
88 |       x-ratelimit-reset-tokens:
89 |       - 0s
90 |       x-request-id:
91 |       - req_0903f0cfdf26e3659bb349882a0395e7
92 |     status:
93 |       code: 200
94 |       message: OK
95 | version: 1
96 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/templates/dashboard.html:
--------------------------------------------------------------------------------
 1 | {% extends "base.html" %}
 2 | 
 3 | {% block content %}
 4 | <div class="dashboard">
 5 |     <div class="dashboard-header">
 6 |         <h2>Agent Configuration Dashboard</h2>
 7 |     </div>
 8 | 
 9 |     <div class="dashboard-grid">
10 |         <div class="dashboard-card">
11 |             <div class="card-header">
12 |                 <h3>Server Configuration</h3>
13 |             </div>
14 |             <div class="card-content">
15 |                 <div class="stat-item">
16 |                     <span class="stat-label">Web UI Port:</span>
17 |                     <span class="stat-value">{{ web_ui_port }}</span>
18 |                 </div>
19 |                 <div class="stat-item">
20 |                     <span class="stat-label">APM Server Port:</span>
21 |                     <span class="stat-value">{{ apm_port }}</span>
22 |                 </div>
23 |                 <div class="stat-item">
24 |                     <span class="stat-label">OTLP HTTP Port:</span>
25 |                     <span class="stat-value">{{ otlp_http_port }}</span>
26 |                 </div>
27 |                 <div class="stat-item">
28 |                     <span class="stat-label">OTLP GRPC Port:</span>
29 |                     <span class="stat-value">{{ otlp_grpc_port }}</span>
30 |                 </div>
31 |                 <div class="stat-item">
32 |                     <span class="stat-label">Proxying to Agent:</span>
33 |                     <span class="stat-value {{ 'proxy-enabled' if is_proxying else 'proxy-disabled' }}">
34 |                         {{ 'Yes' if is_proxying else 'No' }}
35 |                     </span>
36 |                 </div>
37 |                 {% if is_proxying %}
38 |                 <div class="stat-item">
39 |                     <span class="stat-label">Agent URL:</span>
40 |                     <span class="stat-value">{{ agent_url }}</span>
41 |                 </div>
42 |                 {% endif %}
43 |                 <div class="stat-item">
44 |                     <span class="stat-label">Snapshot Directory:</span>
45 |                     <span class="stat-value">{{ snapshot_dir }}</span>
46 |                 </div>
47 |                 <div class="stat-item">
48 |                     <span class="stat-label">VCR Enabled:</span>
49 |                     <span class="stat-value">{{ 'Yes' if vcr_enabled else 'No' }}</span>
50 |                 </div>
51 |                 <div class="stat-item">
52 |                     <span class="stat-label">Error Responses Disabled:</span>
53 |                     <span class="stat-value">{{ 'Yes' if error_responses_disabled else 'No' }}</span>
54 |                 </div>
55 |                 <div class="stat-item">
56 |                     <span class="stat-label">Max Requests in Memory:</span>
57 |                     <span class="stat-value">{{ max_requests }}</span>
58 |                 </div>
59 |             </div>
60 |         </div>
61 | 
62 | 
63 | 
64 |         <div class="dashboard-card">
65 |             <div class="card-header">
66 |                 <h3>Enabled Servers</h3>
67 |             </div>
68 |             <div class="card-content">
69 |                 <div class="stat-item">
70 |                     <span class="stat-label">Web UI:</span>
71 |                     <span class="stat-value">{{ 'Enabled' if enabled_servers.web_ui else 'Disabled' }}</span>
72 |                 </div>
73 |                 <div class="stat-item">
74 |                     <span class="stat-label">APM Server:</span>
75 |                     <span class="stat-value">{{ 'Enabled' if enabled_servers.apm_server else 'Disabled' }}</span>
76 |                 </div>
77 |                 <div class="stat-item">
78 |                     <span class="stat-label">OTLP HTTP Server:</span>
79 |                     <span class="stat-value">{{ 'Enabled' if enabled_servers.otlp_http else 'Disabled' }}</span>
80 |                 </div>
81 |                 <div class="stat-item">
82 |                     <span class="stat-label">OTLP GRPC Server:</span>
83 |                     <span class="stat-value">{{ 'Enabled' if enabled_servers.otlp_grpc else 'Disabled' }}</span>
84 |                 </div>
85 |             </div>
86 |         </div>
87 | 
88 |     </div>
89 | </div>
90 | {% endblock %}


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_23558eed.yaml:
--------------------------------------------------------------------------------
  1 | interactions:
  2 | - request:
  3 |     body: '{"model":"gpt-3.5-turbo","max_tokens":100,"temperature":0.5,"messages":[{"role":"system","content":"You
  4 |       are a helpful assistant"},{"role":"user","content":"Hello, OpenAI!"}]}'
  5 |     headers:
  6 |       ? !!python/object/apply:multidict._multidict.istr
  7 |       - Accept
  8 |       : - '*/*'
  9 |       ? !!python/object/apply:multidict._multidict.istr
 10 |       - Accept-Encoding
 11 |       : - gzip, deflate
 12 |       ? !!python/object/apply:multidict._multidict.istr
 13 |       - Accept-Language
 14 |       : - '*'
 15 |       ? !!python/object/apply:multidict._multidict.istr
 16 |       - Connection
 17 |       : - keep-alive
 18 |       Content-Length:
 19 |       - '174'
 20 |       ? !!python/object/apply:multidict._multidict.istr
 21 |       - Content-Type
 22 |       : - application/json
 23 |       ? !!python/object/apply:multidict._multidict.istr
 24 |       - User-Agent
 25 |       : - node
 26 |       ? !!python/object/apply:multidict._multidict.istr
 27 |       - sec-fetch-mode
 28 |       : - cors
 29 |     method: POST
 30 |     uri: https://api.openai.com/v1/chat/completions
 31 |   response:
 32 |     body:
 33 |       string: "{\n  \"id\": \"chatcmpl-BmQAy5YZmY2VqBd67WEjAf7PL8pMR\",\n  \"object\":
 34 |         \"chat.completion\",\n  \"created\": 1750880064,\n  \"model\": \"gpt-3.5-turbo-0125\",\n
 35 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
 36 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
 37 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
 38 |         null,\n      \"finish_reason\": \"stop\"\n    }\n  ],\n  \"usage\": {\n    \"prompt_tokens\":
 39 |         21,\n    \"completion_tokens\": 9,\n    \"total_tokens\": 30,\n    \"prompt_tokens_details\":
 40 |         {\n      \"cached_tokens\": 0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
 41 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
 42 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
 43 |         \"default\",\n  \"system_fingerprint\": null\n}\n"
 44 |     headers:
 45 |       CF-RAY:
 46 |       - 9556e6f3febe13fa-IAD
 47 |       Connection:
 48 |       - keep-alive
 49 |       Content-Encoding:
 50 |       - gzip
 51 |       Content-Type:
 52 |       - application/json
 53 |       Date:
 54 |       - Wed, 25 Jun 2025 19:34:25 GMT
 55 |       Server:
 56 |       - cloudflare
 57 |       Set-Cookie:
 58 |       - __cf_bm=PoMWCL_iwiPZatxbVmgOIHz_yEW8xd5hjuNvRD5rfWA-1750880065-1.0.1.1-itEymv55j7cHFlHe_6ERngKYZtdalXjWdESFrQddtIZlYxDyV0j87rJuuLz10AeGqlquF4cGNxEi_HwwSsyZPsk8khlDU1UTGA30xD.9Nf4;
 59 |         path=/; expires=Wed, 25-Jun-25 20:04:25 GMT; domain=.api.openai.com; HttpOnly;
 60 |         Secure; SameSite=None
 61 |       - _cfuvid=KBEZcQ1NQPGarw46tNOzt.KftcQpOXra9vHtamVWxXE-1750880065108-0.0.1.1-604800000;
 62 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
 63 |       Transfer-Encoding:
 64 |       - chunked
 65 |       X-Content-Type-Options:
 66 |       - nosniff
 67 |       access-control-expose-headers:
 68 |       - X-Request-ID
 69 |       alt-svc:
 70 |       - h3=":443"; ma=86400
 71 |       cf-cache-status:
 72 |       - DYNAMIC
 73 |       openai-organization:
 74 |       - datadog-staging
 75 |       openai-processing-ms:
 76 |       - '250'
 77 |       openai-version:
 78 |       - '2020-10-01'
 79 |       strict-transport-security:
 80 |       - max-age=31536000; includeSubDomains; preload
 81 |       x-envoy-upstream-service-time:
 82 |       - '259'
 83 |       x-ratelimit-limit-requests:
 84 |       - '10000'
 85 |       x-ratelimit-limit-tokens:
 86 |       - '50000000'
 87 |       x-ratelimit-remaining-requests:
 88 |       - '9999'
 89 |       x-ratelimit-remaining-tokens:
 90 |       - '49999985'
 91 |       x-ratelimit-reset-requests:
 92 |       - 6ms
 93 |       x-ratelimit-reset-tokens:
 94 |       - 0s
 95 |       x-request-id:
 96 |       - req_b5d28a41a22c127bf77ef18ffae70449
 97 |     status:
 98 |       code: 200
 99 |       message: OK
100 | version: 1
101 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/fmt.py:
--------------------------------------------------------------------------------
  1 | import argparse
  2 | import glob
  3 | import json
  4 | import logging
  5 | import os
  6 | import sys
  7 | from typing import List
  8 | from typing import Optional
  9 | 
 10 | from . import _get_version
 11 | from .trace_snapshot import generate_snapshot
 12 | 
 13 | 
 14 | log = logging.getLogger(__name__)
 15 | 
 16 | 
 17 | def _resolve_files(files: List[str]) -> List[str]:
 18 |     """Return a list of json files resolved from the provided list of directories or files"""
 19 |     resolved = []
 20 |     for fname in files:
 21 |         if os.path.isdir(fname):
 22 |             curdir = os.getcwd()
 23 |             try:
 24 |                 os.chdir(fname)
 25 |                 resolved.extend([os.path.join(fname, f) for f in glob.glob("**/*.json", recursive=True)])
 26 |             finally:
 27 |                 os.chdir(curdir)
 28 |         else:
 29 |             resolved.append(fname)
 30 |     return resolved
 31 | 
 32 | 
 33 | def main(args: Optional[List[str]] = None) -> None:
 34 |     if args is None:
 35 |         args = sys.argv[1:]
 36 |     parser = argparse.ArgumentParser(
 37 |         description="Datadog APM Test Agent Snapshot Formatter",
 38 |         prog="ddapm-test-agent-fmt",
 39 |     )
 40 |     parser.add_argument(
 41 |         "-v",
 42 |         "--version",
 43 |         action="store_true",
 44 |         dest="version",
 45 |         help="Print version info and exit.",
 46 |     )
 47 |     parser.add_argument(
 48 |         "-c",
 49 |         "--check",
 50 |         action="store_true",
 51 |         dest="check",
 52 |         help="Do not rewrite files, error if any files would be changed.",
 53 |     )
 54 |     parser.add_argument(
 55 |         "--log-level",
 56 |         type=str,
 57 |         default=os.environ.get("LOG_LEVEL", "INFO"),
 58 |         help="Set the log level. DEBUG, INFO, WARNING, ERROR, CRITICAL.",
 59 |     )
 60 |     parser.add_argument(
 61 |         "files",
 62 |         metavar="FILE",
 63 |         type=str,
 64 |         nargs="+",
 65 |         help="Specific snapshot files or directories to format.",
 66 |     )
 67 | 
 68 |     parsed_args = parser.parse_args(args=args)
 69 |     logging.basicConfig(level=parsed_args.log_level)
 70 | 
 71 |     if parsed_args.version:
 72 |         print(_get_version())
 73 |         sys.exit(0)
 74 | 
 75 |     # Find all json files
 76 |     resolved_files = _resolve_files(parsed_args.files)
 77 |     log.info("Found %d snapshot files to process", len(resolved_files))
 78 | 
 79 |     trace_files = []
 80 |     trace_stats_files = []
 81 |     for f in resolved_files:
 82 |         if f.endswith("_tracestats.json"):
 83 |             trace_stats_files.append(f)
 84 |         else:
 85 |             trace_files.append(f)
 86 |     log.info("Found %d trace snapshot files to process", len(trace_files))
 87 |     log.info("Found %d trace stats snapshot files to process", len(trace_stats_files))
 88 | 
 89 |     has_errors = False
 90 |     for fname in trace_files:
 91 |         log.debug("Checking snapshot file %r", fname)
 92 |         try:
 93 |             # Read the original file data
 94 |             with open(fname, "r") as fp:
 95 |                 original = fp.read()
 96 | 
 97 |             # Parse and re-format
 98 |             traces = json.loads(original)
 99 |             formatted = generate_snapshot(traces)
100 | 
101 |             # Only do anything if something changed
102 |             if formatted != original:
103 |                 log.debug("Snapshot file %r has changes", fname)
104 |                 if parsed_args.check:
105 |                     # If we are in check mode and we changed the content, error
106 |                     log.error("Snapshot file %r would be reformatted!", fname)
107 |                     has_errors = True
108 |                 else:
109 |                     # Rewrite the original file with the new formatted version
110 |                     with open(fname, "w") as fp:
111 |                         fp.write(formatted)
112 |                     log.info("Snapshot file %r was formatted", fname)
113 | 
114 |         except Exception:
115 |             log.exception("Error processing file %r", fname)
116 |             has_errors = True
117 | 
118 |     if has_errors:
119 |         sys.exit(-1)
120 | 
121 | 
122 | if __name__ == "__main__":
123 |     main()
124 | 


--------------------------------------------------------------------------------
/ddapm_test_agent/metrics.py:
--------------------------------------------------------------------------------
 1 | """OTLP Metrics handling for the test agent."""
 2 | 
 3 | import json
 4 | import logging
 5 | from typing import Any
 6 | from typing import Dict
 7 | 
 8 | from aiohttp import ClientSession
 9 | from google.protobuf.json_format import MessageToDict
10 | from grpc import aio as grpc_aio
11 | from opentelemetry.proto.collector.metrics.v1.metrics_service_pb2 import ExportMetricsServiceRequest
12 | from opentelemetry.proto.collector.metrics.v1.metrics_service_pb2 import ExportMetricsServiceResponse
13 | from opentelemetry.proto.collector.metrics.v1.metrics_service_pb2_grpc import MetricsServiceServicer
14 | 
15 | 
16 | METRICS_ENDPOINT = "/v1/metrics"
17 | 
18 | 
19 | log = logging.getLogger(__name__)
20 | 
21 | 
22 | def decode_metrics_request(request_body: bytes, content_type: str) -> Dict[str, Any]:
23 |     if content_type == "application/json":
24 |         parsed_json = json.loads(request_body)
25 |         if not isinstance(parsed_json, dict):
26 |             raise Exception("JSON payload must be an object")
27 |         return parsed_json
28 |     elif content_type == "application/x-protobuf":
29 |         export_request = ExportMetricsServiceRequest()
30 |         export_request.ParseFromString(request_body)
31 |         return protobuf_to_dict(export_request)
32 |     else:
33 |         raise ValueError(f"Content-Type must be application/x-protobuf or application/json, got {content_type}")
34 | 
35 | 
36 | def protobuf_to_dict(pb_obj: Any) -> Dict[str, Any]:
37 |     return MessageToDict(pb_obj, preserving_proto_field_name=True)
38 | 
39 | 
40 | class OTLPMetricsGRPCServicer(MetricsServiceServicer):
41 | 
42 |     def __init__(self, http_port: int):
43 |         self.http_url = f"http://127.0.0.1:{http_port}"
44 | 
45 |     def _count_data_points(self, request: ExportMetricsServiceRequest) -> int:
46 |         return len(
47 |             [
48 |                 dp
49 |                 for rm in request.resource_metrics
50 |                 for sm in rm.scope_metrics
51 |                 for m in sm.metrics
52 |                 for dp in (
53 |                     m.gauge.data_points
54 |                     if m.HasField("gauge")
55 |                     else (
56 |                         m.sum.data_points
57 |                         if m.HasField("sum")
58 |                         else (
59 |                             m.histogram.data_points
60 |                             if m.HasField("histogram")
61 |                             else (
62 |                                 m.exponential_histogram.data_points
63 |                                 if m.HasField("exponential_histogram")
64 |                                 else m.summary.data_points if m.HasField("summary") else []
65 |                             )
66 |                         )
67 |                     )
68 |                 )
69 |             ]
70 |         )
71 | 
72 |     async def Export(
73 |         self, request: ExportMetricsServiceRequest, context: grpc_aio.ServicerContext
74 |     ) -> ExportMetricsServiceResponse:
75 |         try:
76 |             protobuf_data = request.SerializeToString()
77 |             headers = {"Content-Type": "application/x-protobuf"}
78 |             metadata = dict(context.invocation_metadata())
79 |             if "session-token" in metadata:
80 |                 headers["Session-Token"] = metadata["session-token"]
81 |             async with ClientSession(self.http_url) as session:
82 |                 async with session.post(METRICS_ENDPOINT, headers=headers, data=protobuf_data) as resp:
83 |                     context.set_trailing_metadata([("http-status", str(resp.status))])
84 |                     response = ExportMetricsServiceResponse()
85 |                     if resp.status >= 400:
86 |                         response.partial_success.rejected_data_points = self._count_data_points(request)
87 |                         response.partial_success.error_message = f"HTTP {resp.status}: {await resp.text()}"
88 |                     return response
89 |         except Exception as e:
90 |             context.set_trailing_metadata([("http-status", "500"), ("error", str(e))])
91 |             response = ExportMetricsServiceResponse()
92 |             response.partial_success.rejected_data_points = self._count_data_points(request)
93 |             response.partial_success.error_message = f"Forward failed: {str(e)}"
94 |             return response
95 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_d8fb132a.yaml:
--------------------------------------------------------------------------------
 1 | interactions:
 2 | - request:
 3 |     body: '{"model":"gpt-3.5-turbo","messages":[{"role":"system","content":"You are
 4 |       a helpful assistant."},{"role":"user","content":"Hello, OpenAI!"}],"temperature":0.5,"stream":false,"max_tokens":100,"n":3,"user":"dd-trace-test"}'
 5 |     headers:
 6 |       Accept:
 7 |       - application/json, text/plain, */*
 8 |       Accept-Encoding:
 9 |       - gzip, deflate
10 |       Connection:
11 |       - keep-alive
12 |       Content-Length:
13 |       - '219'
14 |       Content-Type:
15 |       - application/json
16 |       User-Agent:
17 |       - OpenAI/NodeJS/3.3.0
18 |     method: POST
19 |     uri: https://api.openai.com/v1/chat/completions
20 |   response:
21 |     body:
22 |       string: "{\n  \"id\": \"chatcmpl-BkXeAXECQrVxTEnQakC9YnODFxTQ8\",\n  \"object\":
23 |         \"chat.completion\",\n  \"created\": 1750432126,\n  \"model\": \"gpt-3.5-turbo-0125\",\n
24 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
25 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
26 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
27 |         null,\n      \"finish_reason\": \"stop\"\n    },\n    {\n      \"index\":
28 |         1,\n      \"message\": {\n        \"role\": \"assistant\",\n        \"content\":
29 |         \"Hello! How can I assist you today?\",\n        \"refusal\": null,\n        \"annotations\":
30 |         []\n      },\n      \"logprobs\": null,\n      \"finish_reason\": \"stop\"\n
31 |         \   },\n    {\n      \"index\": 2,\n      \"message\": {\n        \"role\":
32 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
33 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
34 |         null,\n      \"finish_reason\": \"stop\"\n    }\n  ],\n  \"usage\": {\n    \"prompt_tokens\":
35 |         22,\n    \"completion_tokens\": 27,\n    \"total_tokens\": 49,\n    \"prompt_tokens_details\":
36 |         {\n      \"cached_tokens\": 0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
37 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
38 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
39 |         \"default\",\n  \"system_fingerprint\": null\n}\n"
40 |     headers:
41 |       CF-RAY:
42 |       - 952c2ef79f3f8fe4-BOS
43 |       Connection:
44 |       - keep-alive
45 |       Content-Encoding:
46 |       - gzip
47 |       Content-Type:
48 |       - application/json
49 |       Date:
50 |       - Fri, 20 Jun 2025 15:08:47 GMT
51 |       Server:
52 |       - cloudflare
53 |       Set-Cookie:
54 |       - __cf_bm=AMpK7hfILT8guhJ5Kmc.Jf9pLksJfgPBZ21R2.yC5G4-1750432127-1.0.1.1-caH8SeKwKgtF8XFq50c620mQW1Prmp3DbO6u_WGdIlAQXjsSvslXxsMhAVcMx_emvxdMvShNjGrD.cOzR5JXR82cIxDTWekDBkjwjnEXX6s;
55 |         path=/; expires=Fri, 20-Jun-25 15:38:47 GMT; domain=.api.openai.com; HttpOnly;
56 |         Secure; SameSite=None
57 |       - _cfuvid=_8QCtYppGK_NVUgSga2iPPsqy0JegC5ach.lvF.Ii90-1750432127261-0.0.1.1-604800000;
58 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
59 |       Transfer-Encoding:
60 |       - chunked
61 |       X-Content-Type-Options:
62 |       - nosniff
63 |       access-control-expose-headers:
64 |       - X-Request-ID
65 |       alt-svc:
66 |       - h3=":443"; ma=86400
67 |       cf-cache-status:
68 |       - DYNAMIC
69 |       openai-organization:
70 |       - datadog-staging
71 |       openai-processing-ms:
72 |       - '332'
73 |       openai-version:
74 |       - '2020-10-01'
75 |       strict-transport-security:
76 |       - max-age=31536000; includeSubDomains; preload
77 |       x-envoy-upstream-service-time:
78 |       - '342'
79 |       x-ratelimit-limit-requests:
80 |       - '10000'
81 |       x-ratelimit-limit-tokens:
82 |       - '50000000'
83 |       x-ratelimit-remaining-requests:
84 |       - '9999'
85 |       x-ratelimit-remaining-tokens:
86 |       - '49999983'
87 |       x-ratelimit-reset-requests:
88 |       - 6ms
89 |       x-ratelimit-reset-tokens:
90 |       - 0s
91 |       x-request-id:
92 |       - req_b923f8efc6f0c912be18936344d4e3c2
93 |     status:
94 |       code: 200
95 |       message: OK
96 | version: 1
97 | 


--------------------------------------------------------------------------------
/vcr-cassettes/openai/openai_chat_completions_post_ce69ed91.yaml:
--------------------------------------------------------------------------------
  1 | interactions:
  2 | - request:
  3 |     body: "{\n  \"model\": \"gpt-4o\",\n  \"messages\": [\n    {\n      \"role\":
  4 |       \"system\",\n      \"content\": \"You are a helpful assistant\"\n    },\n    {\n
  5 |       \     \"role\": \"user\",\n      \"content\": \"Hello, OpenAI!\"\n    }\n  ],\n
  6 |       \ \"temperature\": 0.5,\n  \"max_tokens\": 100,\n  \"stream\": false,\n  \"n\":
  7 |       1,\n  \"user\": \"dd-trace-test\"\n}"
  8 |     headers:
  9 |       Accept:
 10 |       - application/json
 11 |       Accept-Encoding:
 12 |       - gzip,deflate
 13 |       Connection:
 14 |       - keep-alive
 15 |       Content-Length:
 16 |       - '295'
 17 |       Content-Type:
 18 |       - application/json
 19 |       User-Agent:
 20 |       - OpenAI/JS 4.85.0
 21 |       X-Stainless-Arch:
 22 |       - arm64
 23 |       X-Stainless-Helper-Method:
 24 |       - beta.chat.completions.parse
 25 |       X-Stainless-Lang:
 26 |       - js
 27 |       X-Stainless-Os:
 28 |       - MacOS
 29 |       X-Stainless-Package-Version:
 30 |       - 4.85.0
 31 |       X-Stainless-Retry-Count:
 32 |       - '0'
 33 |       X-Stainless-Runtime:
 34 |       - node
 35 |       X-Stainless-Runtime-Version:
 36 |       - v22.13.1
 37 |       X-Stainless-Timeout:
 38 |       - '600000'
 39 |     method: POST
 40 |     uri: https://api.openai.com/v1/chat/completions
 41 |   response:
 42 |     body:
 43 |       string: "{\n  \"id\": \"chatcmpl-BkZxQcJvaxbyldxLmrNffCobBnd84\",\n  \"object\":
 44 |         \"chat.completion\",\n  \"created\": 1750441008,\n  \"model\": \"gpt-4o-2024-08-06\",\n
 45 |         \ \"choices\": [\n    {\n      \"index\": 0,\n      \"message\": {\n        \"role\":
 46 |         \"assistant\",\n        \"content\": \"Hello! How can I assist you today?\",\n
 47 |         \       \"refusal\": null,\n        \"annotations\": []\n      },\n      \"logprobs\":
 48 |         null,\n      \"finish_reason\": \"stop\"\n    }\n  ],\n  \"usage\": {\n    \"prompt_tokens\":
 49 |         21,\n    \"completion_tokens\": 9,\n    \"total_tokens\": 30,\n    \"prompt_tokens_details\":
 50 |         {\n      \"cached_tokens\": 0,\n      \"audio_tokens\": 0\n    },\n    \"completion_tokens_details\":
 51 |         {\n      \"reasoning_tokens\": 0,\n      \"audio_tokens\": 0,\n      \"accepted_prediction_tokens\":
 52 |         0,\n      \"rejected_prediction_tokens\": 0\n    }\n  },\n  \"service_tier\":
 53 |         \"default\",\n  \"system_fingerprint\": \"fp_07871e2ad8\"\n}\n"
 54 |     headers:
 55 |       CF-RAY:
 56 |       - 952d07cdcd548f9c-BOS
 57 |       Connection:
 58 |       - keep-alive
 59 |       Content-Encoding:
 60 |       - gzip
 61 |       Content-Type:
 62 |       - application/json
 63 |       Date:
 64 |       - Fri, 20 Jun 2025 17:36:48 GMT
 65 |       Server:
 66 |       - cloudflare
 67 |       Set-Cookie:
 68 |       - __cf_bm=ByDNjXQAlPfUE4O7Y5ISxcbw5kE.mI7kAGV4_cTSxCU-1750441008-1.0.1.1-4JaLZ4u09.dEdgsKlgZZTGC58kw7YkSetEc1iAl.pgSahVQW.ebMAYgILdUMfN.yUjLihgERIXpfNtX7PtU9y.w6UieglhDfbnXIpPebK08;
 69 |         path=/; expires=Fri, 20-Jun-25 18:06:48 GMT; domain=.api.openai.com; HttpOnly;
 70 |         Secure; SameSite=None
 71 |       - _cfuvid=83X1cyA6eaLH6NGBf0udLBJcLZ4bq1U6f97bo8qqv9k-1750441008744-0.0.1.1-604800000;
 72 |         path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
 73 |       Transfer-Encoding:
 74 |       - chunked
 75 |       X-Content-Type-Options:
 76 |       - nosniff
 77 |       access-control-expose-headers:
 78 |       - X-Request-ID
 79 |       alt-svc:
 80 |       - h3=":443"; ma=86400
 81 |       cf-cache-status:
 82 |       - DYNAMIC
 83 |       openai-organization:
 84 |       - datadog-staging
 85 |       openai-processing-ms:
 86 |       - '223'
 87 |       openai-version:
 88 |       - '2020-10-01'
 89 |       strict-transport-security:
 90 |       - max-age=31536000; includeSubDomains; preload
 91 |       x-envoy-upstream-service-time:
 92 |       - '226'
 93 |       x-ratelimit-limit-requests:
 94 |       - '10000'
 95 |       x-ratelimit-limit-tokens:
 96 |       - '30000000'
 97 |       x-ratelimit-remaining-requests:
 98 |       - '9999'
 99 |       x-ratelimit-remaining-tokens:
100 |       - '29999985'
101 |       x-ratelimit-reset-requests:
102 |       - 6ms
103 |       x-ratelimit-reset-tokens:
104 |       - 0s
105 |       x-request-id:
106 |       - req_1a669521263f5170b72834dc20255396
107 |     status:
108 |       code: 200
109 |       message: OK
110 | version: 1
111 | 


--------------------------------------------------------------------------------
/tests/trace_utils.py:
--------------------------------------------------------------------------------
  1 | from random import Random
  2 | from typing import Any
  3 | from typing import Dict
  4 | from typing import List
  5 | from typing import Optional
  6 | 
  7 | from ddapm_test_agent.trace import Span
  8 | from ddapm_test_agent.trace import Trace
  9 | from ddapm_test_agent.trace import dfs_order
 10 | from ddapm_test_agent.trace import root_span
 11 | from ddapm_test_agent.trace import verify_span
 12 | 
 13 | 
 14 | # Fix the seed for deterministic results between runs
 15 | _random = Random(1234)
 16 | 
 17 | 
 18 | _TYPES = {
 19 |     "web": {
 20 |         "name": ["django.request", "flask.request", "rack.request"],
 21 |         "service": ["django", "flask", "rack"],
 22 |         "resource": [
 23 |             "/users",
 24 |             "/feed",
 25 |             "/posts",
 26 |             "/home",
 27 |         ],
 28 |     },
 29 |     "cache": {
 30 |         "name": ["memcached.command"],
 31 |         "service": ["memcached"],
 32 |         "resource": [
 33 |             "get",
 34 |             "get_many",
 35 |             "set",
 36 |         ],
 37 |     },
 38 |     "sql": {
 39 |         "name": ["mysql.query", "postgres.query"],
 40 |         "service": ["mysql", "postgres"],
 41 |         "resource": [
 42 |             "SELECT * from users",
 43 |         ],
 44 |     },
 45 | }
 46 | 
 47 | 
 48 | def span(rnd: Random = _random, **kwargs: Any) -> Span:
 49 |     if "type" not in kwargs:
 50 |         kwargs["type"] = rnd.choice(list(_TYPES.keys()))
 51 | 
 52 |     data = _TYPES[kwargs["type"]]
 53 |     for k in ["name", "service"]:
 54 |         i = rnd.choice(list(range(len(data["name"]))))
 55 |         if k not in kwargs:
 56 |             kwargs[k] = data[k][i]
 57 | 
 58 |     if "resource" not in kwargs:
 59 |         kwargs["resource"] = rnd.choice(data["resource"])
 60 | 
 61 |     for k in ["trace_id", "span_id"]:
 62 |         if k not in kwargs:
 63 |             kwargs[k] = random_id(rnd)
 64 | 
 65 |     # Don't assign a parent id by default
 66 |     if "parent_id" not in kwargs:
 67 |         kwargs["parent_id"] = None
 68 | 
 69 |     if "start" not in kwargs:
 70 |         kwargs["start"] = rnd.randint(0, 2**48)
 71 | 
 72 |     if "duration" not in kwargs:
 73 |         kwargs["duration"] = rnd.randint(100, 10**10)
 74 | 
 75 |     if "meta" not in kwargs:
 76 |         kwargs["meta"] = {}
 77 | 
 78 |     if "metrics" not in kwargs:
 79 |         kwargs["metrics"] = {}
 80 |     return verify_span(kwargs)
 81 | 
 82 | 
 83 | def _prufers_trace(n: int, rnd: Random = _random) -> Trace:
 84 |     """Return a randomly generated trace tree with `n` spans.
 85 | 
 86 |     https://en.wikipedia.org/wiki/Pr%C3%BCfer_sequence
 87 |     """
 88 |     a = [rnd.randint(1, n) for _ in range(n - 2)]
 89 |     spans = [span(span_id=i) for i in range(1, n + 1)]
 90 |     degree: Dict[int, int] = {}
 91 |     for s in spans:
 92 |         degree[s["span_id"]] = 1
 93 |     for i in a:
 94 |         degree[i] += 1
 95 | 
 96 |     for i in a:
 97 |         for s in spans:
 98 |             if degree[s["span_id"]] == 1:
 99 |                 s["parent_id"] = i
100 |                 degree[i] -= 1
101 |                 degree[s["span_id"]] -= 1
102 |                 break
103 | 
104 |     u = 0
105 |     for s in spans:
106 |         if degree[s["span_id"]] == 1:
107 |             if u == 0:
108 |                 u = s["span_id"]
109 |             else:
110 |                 s["parent_id"] = u
111 |                 break
112 | 
113 |     # TODO: randomize the span ids?
114 |     return list(dfs_order(spans))
115 | 
116 | 
117 | def random_trace(
118 |     nspans: int,
119 |     rng: Random = _random,
120 |     trace_id: Optional[int] = None,
121 |     parent_id: Optional[int] = None,
122 |     remove_keys: Optional[List[str]] = None,
123 | ) -> Trace:
124 |     # TODO:
125 |     #   represent arbitrary random services (subtrees in spans)
126 |     #   resource names (should only be on service entry)
127 |     #   smarter type (should only be on service entry)
128 |     #   sampling decisions
129 |     #   dd_origin?
130 |     assert nspans > 0
131 |     if not trace_id:
132 |         trace_id = random_id(rng)
133 |     t = _prufers_trace(nspans, rng)
134 |     root = root_span(t)
135 |     if parent_id:
136 |         root["parent_id"] = parent_id
137 |     for s in t:
138 |         if s is not root:
139 |             del s["type"]
140 |             del s["resource"]
141 |         s["trace_id"] = trace_id
142 |         if remove_keys:
143 |             for k in remove_keys:
144 |                 del s[k]  # type: ignore
145 | 
146 |     return t
147 | 
148 | 
149 | def random_id(rng: Random = _random) -> int:
150 |     return rng.randint(0, 2**64)
151 | 


--------------------------------------------------------------------------------