-
Notifications
You must be signed in to change notification settings - Fork 4
docs: update agents md #220
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Changes from all commits
Commits
Show all changes
3 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,118 +1,207 @@ | ||
| ## Project Overview | ||
| # Splunk Agent Observability Python SDK Agent Guide | ||
|
|
||
| Splunk Agent Observability Python SDK (`splunk-ao` on PyPI) — the official Python client for Splunk Agent Observability. Instrument LLM/agent apps, send traces and metrics, manage projects, datasets, experiments, and prompts. | ||
| This guide covers the entire repository. All paths are repository-root relative. Read `README.md` for supported user | ||
| workflows and `ARCHITECTURE.md` before changing telemetry, configuration, lifecycle, or integration behavior. | ||
|
|
||
| Successor to [`galileo-python`](https://github.com/rungalileo/galileo-python); migration notes in `splunk-ao-migration-tool/README.md`. | ||
| ## Permission Model | ||
|
|
||
| SDK code lives under `src/splunk_ao/`. Do not edit `galileo-core` or `src/splunk_ao/resources/` (auto-generated). | ||
| - Read-only discovery is allowed without approval: file reads, `rg`, `git status`, and `git diff`. | ||
| - Before running any project command, show the exact command, explain its scope, and ask. This includes tests, lint, | ||
| formatting, type checks, builds, installs, lock updates, scripts, code generation, and documentation generation. | ||
| - A requested implementation authorizes scoped file edits, not unrelated cleanup or expansion. | ||
| - Never stage or commit changes unless explicitly requested. Never push, publish, release, or bump a version without an | ||
| explicit request and separate confirmation of the exact action. | ||
| - Do not dispatch, rerun, cancel, or otherwise operate a GitHub Actions workflow unless the task explicitly requests it. | ||
| You may identify a relevant workflow as an optional next step, but do not seek approval to run it unless the user asks | ||
| to proceed. When execution is requested, show the exact workflow, ref, inputs, and command and wait for approval. | ||
| - Never expose credentials, tokens, `.env` contents, customer payloads, or private/internal planning material. | ||
|
|
||
| ### Deployment Modes | ||
| ## Subagents | ||
|
|
||
| | Mode | Auth | Notes | | ||
| |------|------|-------| | ||
| | **O11y Cloud** | `SPLUNK_AO_REALM` + `SPLUNK_AO_SF_TOKEN` | Do not set `SPLUNK_AO_CONSOLE_URL` / `SPLUNK_AO_API_URL` | | ||
| | **Standalone** | `SPLUNK_AO_API_KEY` + `SPLUNK_AO_CONSOLE_URL` | Self-hosted or legacy AO | | ||
| - Use subagents only for concrete, bounded, independent work where parallelism materially improves speed or quality. | ||
| Prefer read-heavy exploration, review, triage, and independent package analysis. | ||
| - Do not delegate trivial, tightly coupled, or sequential work. Avoid concurrent edits to the same files. | ||
| - The main agent owns scope, architectural decisions, integration, and final review. It must read task-defining | ||
| instructions itself rather than outsourcing its understanding. | ||
| - Give each subagent an explicit scope, relevant paths, constraints, and expected output. Require a concise, | ||
| evidence-backed handoff. | ||
| - Subagents inherit every permission boundary in this guide. Delegation must never bypass approval for project commands, | ||
| GitHub workflows, staging, commits, releases, or publishing. | ||
| - When subagents edit files, assign non-overlapping ownership and review the combined diff before completion. | ||
|
|
||
| Detection: `src/splunk_ao/deployment.py::resolve_deployment()`. Never mix o11y and standalone env vars. | ||
| ## Commands (Reference Only—Ask Before Running) | ||
|
|
||
| Optional defaults: `SPLUNK_AO_PROJECT`, `SPLUNK_AO_AGENT_STREAM` (deprecated alias: `SPLUNK_AO_LOG_STREAM`). | ||
|
|
||
| `SplunkAOConfig` bridges `SPLUNK_AO_*` → `GALILEO_*` for `galileo-core` (see `config.py::_BRIDGE`). | ||
|
|
||
| ## Build & Development | ||
| Root SDK (`splunk-ao`, Poetry 2.4.1): | ||
|
|
||
| ```bash | ||
| poetry install --all-extras --no-root # or: inv setup | ||
| poetry run pytest # single file: poetry run pytest tests/test_foo.py | ||
| inv test # with coverage | ||
| inv type-check # mypy | ||
| poetry run ruff check --fix src/ # lint + format | ||
| poetry install --all-extras --no-root | ||
| poetry run pytest tests/test_deployment.py -n 0 # targeted, deterministic | ||
| poetry run pytest # full unit suite | ||
| poetry run invoke test # full suite with terminal coverage | ||
| poetry run invoke type-check # configured mypy run | ||
| poetry run ruff check --no-fix src tests # lint without source rewrites | ||
| poetry run ruff format --check src tests # format check without source rewrites | ||
| poetry build | ||
| ``` | ||
|
|
||
| CI: mypy + pytest on Python 3.11–3.14 × Linux/macOS/Windows. Pre-commit: ruff + mypy. | ||
| A2A and ADK packages use uv/Hatch. Each block starts independently from the repository root. | ||
|
|
||
| ## Architecture | ||
| A2A: | ||
|
|
||
| ``` | ||
| src/splunk_ao/ | ||
| ├── project.py, dataset.py, experiment.py, prompt.py # Object-centric API (import via splunk_ao.__future__) | ||
| ├── logger/ # SplunkAOLogger — trace/span management | ||
| ├── handlers/ # LangChain, CrewAI, OpenAI Agents integrations | ||
| ├── openai/ # Drop-in OpenAI client wrapper | ||
| ├── resources/ # Auto-generated API client — DO NOT EDIT | ||
| ├── decorator.py # @log, splunk_ao_context | ||
| ├── config.py # SplunkAOConfig | ||
| └── deployment.py # O11y vs standalone detection | ||
| ``` | ||
|
|
||
| **Regenerate API client:** | ||
| ```bash | ||
| ./scripts/import-openapi-yaml.sh https://api.galileo.ai/client | ||
| ./scripts/auto-generate-api-client.sh | ||
| cd splunk-ao-a2a | ||
| uv sync --dev | ||
| uv run pytest | ||
| uv run mypy src/ | ||
| uv run ruff check --no-fix src tests # lint without source rewrites | ||
| uv build | ||
| ``` | ||
|
|
||
| Uses OpenAPI **Client API** (`/client`), not the main API (`/docs`). | ||
| ADK: | ||
|
|
||
| Depends on `galileo-core` for shared schemas and helpers; ongoing work to reduce this. | ||
|
|
||
| ## Key Patterns | ||
|
|
||
| **Object-centric API** (`__future__`): | ||
| ```python | ||
| from splunk_ao.__future__ import Project | ||
| project = Project.get(name="my-project") # retrieve | ||
| project = Project(name="new").create() # create | ||
| agent_streams = project.list_agent_streams() | ||
| ```bash | ||
| cd splunk-ao-adk | ||
| uv sync --dev | ||
| uv run pytest | ||
| uv run mypy src/ | ||
| uv run ruff check --no-fix src tests # lint without source rewrites | ||
| uv build | ||
| ``` | ||
|
|
||
| **Service layer** (procedural): | ||
| ```python | ||
| from splunk_ao.datasets import create_dataset | ||
| from splunk_ao.experiments import run_experiment | ||
| ``` | ||
| Potentially mutating commands require approval and a clean diff first: | ||
|
|
||
| **Logging:** | ||
| ```python | ||
| from splunk_ao import log, splunk_ao_context | ||
|
|
||
| @log | ||
| def my_workflow(): ... | ||
|
|
||
| with splunk_ao_context(project="my-project", agent_stream="prod"): | ||
| my_workflow() | ||
| ```bash | ||
| poetry run ruff check --fix <changed-paths> | ||
| poetry run ruff format <changed-paths> | ||
| poetry run pre-commit run --files <changed-files> | ||
| poetry run python scripts/create_docs.py | ||
| ``` | ||
|
|
||
| **Handlers:** `splunk_ao.handlers.langchain` (`SplunkAOCallback`), `splunk_ao.handlers.crewai` (`CrewAIEventListener`), `splunk_ao.openai` (drop-in wrapper). | ||
|
|
||
| ## Testing | ||
| Regenerate the low-level API client only when the task explicitly changes the Client OpenAPI contract, and still ask: | ||
|
|
||
| Fixtures in `tests/conftest.py`: `mock_request`, `mock_healthcheck`, `mock_login_api_key`. Tests use `--disable-socket`; env vars set in conftest for pytest-xdist. | ||
|
|
||
| CrewAI wraps stdout/stderr at import time. In tests, patch `_crewai_imports_resolved` / `CREWAI_AVAILABLE`, mock `AgentStreams`/`Projects`/`Traces`, and pass a mock `SplunkAOLogger` (see `tests/test_crewai_handler.py`). | ||
| ```bash | ||
| ./scripts/import-openapi-yaml.sh https://api.galileo.ai/client | ||
| ./scripts/auto-generate-api-client.sh | ||
| ``` | ||
|
|
||
| Use Given/When/Then comments in tests (`# Given: …`, `# When: …`, `# Then: …`). | ||
| The generator replaces `src/splunk_ao/resources/`; review the complete generated diff. It uses the Client API (`/client`), | ||
| not the main API documentation contract. | ||
|
|
||
| ## Stack and Package Map | ||
|
|
||
| | Area | Stack and responsibility | | ||
| |---|---| | ||
| | `src/splunk_ao/` | Python 3.11–3.14, Pydantic v2, OpenTelemetry 1.38, `galileo-core` 4.x; public SDK | | ||
| | `tests/` | pytest 9, xdist, respx, socket blocking, timeout and coverage plugins | | ||
| | `splunk-ao-a2a/` | Independently released native-OTel A2A instrumentation; uv/Hatch | | ||
| | `splunk-ao-adk/` | Independently released Google ADK handler integration; uv/Hatch | | ||
| | `splunk-ao-migration-tool/` | Migration documentation and examples, not a buildable package | | ||
| | `src/splunk_ao/resources/` | OpenAPI-generated transport client; never hand-edit | | ||
|
|
||
| The three buildable packages have independent versions, CI, and release workflows. Only the root `poetry.lock` is | ||
| tracked; A2A/ADK `uv.lock` files are ignored and may be created locally by uv. Validate every package a change touches. | ||
| CI supports Python 3.11–3.14; root CI also spans Linux, macOS, and Windows. | ||
|
|
||
| ## Architecture and Public Surfaces | ||
|
|
||
| - `src/splunk_ao/__init__.py` defines supported root imports. `src/splunk_ao/__future__/` is a compatibility/re-export | ||
| surface; do not assume object APIs are available only there. | ||
| - Singular modules (`project.py`, `dataset.py`, `experiment.py`, and peers) implement stateful object APIs. Plural modules | ||
| (`projects.py`, `datasets.py`, `experiments.py`, and peers) implement procedural/service APIs. Preserve both. | ||
| - `logger/`, `decorator.py`, `handlers/`, and `openai/` instrument applications. `otel.py` is the native OTel entry point. | ||
| - `exporter/` owns deployment-aware OTLP export, span normalization, lifecycle, and diagnostics. | ||
| - `config.py` bridges selected `SPLUNK_AO_*` variables to legacy `GALILEO_*` inputs used by `galileo-core`. | ||
| - `galileo-core` is an external dependency. Do not edit or vendor it here; adapt at this repository's boundary. | ||
| - See `ARCHITECTURE.md` for telemetry paths, ownership rules, and a change-impact map. | ||
|
|
||
| ## Configuration and Routing Invariants | ||
|
|
||
| | Deployment | Required authentication | | ||
| |---|---| | ||
| | O11y Cloud | `SPLUNK_AO_REALM` plus at least one token: `SPLUNK_AO_O11Y_TOKEN` for telemetry and, when permitted, CRUD; and/or `SPLUNK_AO_O11Y_API_TOKEN` for CRUD-only or dedicated CRUD authentication | | ||
| | Standalone | `SPLUNK_AO_API_KEY` plus `SPLUNK_AO_CONSOLE_URL`; `SPLUNK_AO_API_URL` is optional | | ||
|
|
||
| - Detection lives in `deployment.py::resolve_deployment()`. Never mix O11y and standalone variable sets. | ||
| - Project/Agent Stream selection is name XOR ID. Precedence is explicit argument, active context, environment, then | ||
| deployment defaults. Routing must agree in OTLP headers and Resource attributes. | ||
| - `OTEL_RESOURCE_ATTRIBUTES` is not an SDK routing override. Remove reserved routing keys before merging it. | ||
| - Configuration is stateful across `Configuration`, environment variables, and `SplunkAOConfig`; tests that change it | ||
| must reset all affected state and singleton instances. | ||
| - Never log auth headers, tokens, raw prompts, completions, embeddings, or large payloads. | ||
|
|
||
| ## Telemetry and Error Boundaries | ||
|
|
||
| - Handler/decorator/OpenAI/ADK telemetry uses the internal logged-step path and converts completed steps to immutable | ||
| OTel spans. The internal trace envelope is never exported as a span. | ||
| - `start_splunk_ao_span()` is SDK-native OTel. `add_splunk_ao_span_processor()` and A2A instrument caller-owned OTel. | ||
| - Never replace the process-global tracer provider. Register processors on the provided provider; respect ownership. | ||
| - Treat ended `ReadableSpan` objects as immutable. Normalize by copying at export, never by mutating private fields. | ||
| - Completed spans enqueue immediately. `flush()` drains completed work without ending active work; `terminate()` drains, | ||
| shuts down SDK-owned resources, and discards unfinished state. Caller-owned providers use `shutdown()`. | ||
| - CRUD/resource operations raise useful failures. Telemetry infrastructure failures must not break instrumented business | ||
| code; sanitize and rate-limit diagnostics. | ||
| - Preserve standard `gen_ai.*` attributes. New SDK-owned attributes use `splunk_ao.*`; do not introduce new proprietary | ||
| `galileo.*` wire attributes. | ||
| - Changes to propagation, IDs, parents, content schemas, or routing need coverage across every affected telemetry path. | ||
|
|
||
| ## Code Style | ||
|
|
||
| - Line length 120; ruff + mypy; numpy docstrings | ||
| - Conventional commits: `type(scope): description` | ||
| - Imports at module level (exception: lazy imports for optional deps like crewai) | ||
| - Duration vars need units: `timeout_seconds`, `delay_ms` | ||
| - Use `logging.getLogger(__name__)`; never log secrets or large payloads | ||
|
|
||
| **Error handling:** Resource ops (`create_project`, `get_dataset`, …) raise on failure. Telemetry/ingestion (`ingest_traces`, `flush`, `@log`) swallows infra errors — observability should not break user code. | ||
| - Line length 120; Ruff for lint/format; mypy for typing; NumPy-style public docstrings. | ||
| - Keep imports at module scope except intentional lazy imports for optional integrations. | ||
| - Use `logging.getLogger(__name__)`, typed signatures, and unit-bearing names such as `timeout_seconds` or `delay_ms`. | ||
| - Prefer the smallest compatible change. Do not combine feature work with drive-by formatting or generated diffs. | ||
| - Maintain sync, async, generator, and async-generator semantics where an API supports them. | ||
|
|
||
| ## Known Issues | ||
| Tests should show intent explicitly: | ||
|
|
||
| 1. **galileo-core dependency** — private package, contributor friction | ||
| 2. **Config state** — split across `Configuration`, `os.environ`, `SplunkAOConfig`; `connect()` must be called explicitly | ||
| 3. **Dataset versions** — API is 1-based, not 0-based | ||
| 4. **Experiment vs Playground** — SDK `Experiment` conflates two API concepts | ||
| 5. **Metadata** — SDK stringifies values in handlers; Trace vs Dataset APIs behave differently | ||
|
|
||
| ## References | ||
| ```python | ||
| def test_flush_does_not_end_active_trace(mock_request) -> None: | ||
| # Given: an active trace with one completed child span | ||
| # When: completed telemetry is flushed | ||
| # Then: the child is exported and the active trace remains open | ||
| ... | ||
| ``` | ||
|
|
||
| - PyPI: https://pypi.org/project/splunk-ao/ | ||
| - GitHub: https://github.com/splunk/splunk-ao-python | ||
| - Migration: `splunk-ao-migration-tool/README.md` | ||
| - Contributing: `CONTRIBUTING.md` | ||
| ## Testing Rules | ||
|
|
||
| - Add the closest focused regression test first; ask before running it. Run broader suites only after targeted confidence. | ||
| - Root tests inherit `-n auto`, network blocking, a 120-second timeout, and fake standalone credentials from pytest config. | ||
| Use `-n 0` for deterministic focused debugging. | ||
| - Set test environment variables before importing `splunk_ao`; xdist workers and Python 3.14 expose import-order leaks. | ||
| - Reuse `tests/conftest.py` fixtures such as `mock_request`, `mock_healthcheck`, and `mock_login_api_key`. Mock all network. | ||
| - Reset global OTel context, providers/processors, SDK configuration, loggers, and background resources after tests. | ||
| - Exercise success, exceptions, cancellation/early generator close, and cleanup for lifecycle-sensitive instrumentation. | ||
| - CrewAI is optional and excluded on Python 3.14. Follow `tests/test_crewai_handler.py`: patch | ||
| `_crewai_imports_resolved`, `CREWAI_AVAILABLE`, and `LITE_LLM_AVAILABLE`; mock logger-layer `AgentStreams`, `Projects`, | ||
| and `Traces`; inject a mock or hook-backed `SplunkAOLogger`; test both installed and unavailable behavior. | ||
| - Dataset version numbers are API-facing and 1-based. | ||
|
|
||
| ## Change Workflow and Git | ||
|
|
||
| 1. Read the public API, implementation, adjacent tests, and relevant architecture section before editing. | ||
| 2. Identify ownership: public wrapper, integration, converter, exporter, generated client, or external dependency. | ||
| 3. Preserve compatibility unless the task explicitly authorizes a breaking change. Update exports, docstrings, README usage, | ||
| tests, and `CHANGELOG.md` when public behavior changes. | ||
| 4. When changing package layout, authentication variables, telemetry paths, or lifecycle invariants documented here, | ||
| update `AGENTS.md` and `ARCHITECTURE.md` in the same change. | ||
| 5. Ask before running the exact validation commands. Report what ran, what did not run, and why. | ||
| 6. Review `git diff` for secrets, unrelated rewrites, generated churn, and platform-specific assumptions. | ||
|
|
||
| Use conventional commit subjects (`type(scope): description`) only when a commit is explicitly requested. Do not edit | ||
| versions, release workflows, or lockfiles as incidental cleanup. | ||
|
|
||
| ## Hard Boundaries | ||
|
|
||
| - Do not hand-edit `src/splunk_ao/resources/` or generated reference docs. | ||
| - Do not change release/publish configuration, dependency pins, or public compatibility aliases without task scope. | ||
| - Do not silently add network calls, global state, import-time side effects, unbounded queues, or non-daemon threads. | ||
| - Do not make tests depend on real credentials, live services, ordering, timing luck, or another test's state. | ||
| - Do not document unavailable internal context. Repository documentation must stand alone for public contributors. | ||
|
|
||
| ## Progressive References | ||
|
|
||
| - `README.md`: installation, authentication, supported APIs, and integration examples. | ||
| - `ARCHITECTURE.md`: package boundaries, telemetry data flow, lifecycle, and change-impact routing. | ||
| - `CONTRIBUTING.md`: contribution setup and generated-client workflow. | ||
| - `src/splunk_ao/README_API_CLIENT.md`: generated client's capabilities and limitations. | ||
| - `splunk-ao-migration-tool/README.md`: migration guidance from `galileo-python`. |
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.