komodor-agentops 0.1.0__tar.gz

komodor_agentops-0.1.0/LICENSE

@@ -0,0 +1,53 @@
+Komodor AgentOps License
+
+The Komodor AgentOps SDK, together with its associated libraries, documentation,
+and files (the "Software"), is provided by Komodor Ltd. ("Komodor") for integration
+with the Komodor service (the "Service"). By installing, copying, or using the
+Software, you agree to these terms.
+
+© 2020 - Komodor Ltd.
+
+Grant. Subject to your compliance with these terms, Komodor grants you a
+non-exclusive, non-transferable, non-sublicensable, worldwide, revocable, and
+limited license to (a) install and use the Software to develop, test, and operate
+integrations of your applications and services with the Service, or to use
+applications and services that embed the Software, and (b) reproduce
+and distribute verbatim copies of the Software, but only as embedded within your own
+applications, services, or container images, and not on a standalone basis. No
+separate agreement with Komodor is required; anyone who obtains the Software
+receives this license. If you have a separate written agreement with Komodor
+covering the Software, that agreement controls to the extent of any conflict.
+
+You may develop and operate your own software that imports, calls, subclasses,
+configures, or otherwise depends on the Software through its published interfaces;
+doing so is permitted use and does not, by itself, create a modified version or
+derivative work of the Software.
+
+Restrictions. You shall not: (a) distribute the Software on a standalone basis, or
+resell, lease, or sublicense it; (b) offer the Software itself, or substantially its
+standalone functionality, to third parties as a hosted or managed service — this
+does not restrict you from operating or offering your own applications or services
+that embed the Software; (c) modify the Software or create a derivative work of its
+files or code; (d) remove, alter, or obscure any copyright, license, or proprietary
+notice, all of which you must retain in every copy you distribute; or (e) claim any
+proprietary interest in the Software, or contest or infringe Komodor's intellectual
+property rights in it.
+
+Ownership. The Software is licensed, not sold. Komodor retains all right, title, and
+interest in and to the Software and all intellectual property rights in it. No
+rights are granted except as expressly stated here, and no patent rights are
+granted. This license does not permit use of Komodor's names or trademarks, except
+for reasonable and customary use in describing the origin of the Software.
+
+Termination. This license terminates automatically if you breach it and do not cure
+the breach within thirty (30) days of Komodor's written notice. On termination you
+must stop installing, using, and distributing the Software, except that copies
+already embedded in products you distributed to end users before termination may
+remain in use. The Restrictions and this "No warranty; no liability" section survive termination.
+
+No warranty; no liability. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
+KIND, EXPRESS OR IMPLIED, INCLUDING THE IMPLIED WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE, TITLE, AND NON-INFRINGEMENT. IN NO EVENT SHALL
+KOMODOR OR THE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES, OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT, OR OTHERWISE, ARISING FROM, OUT
+OF, OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

komodor_agentops-0.1.0/PKG-INFO

@@ -0,0 +1,218 @@
+Metadata-Version: 2.4
+Name: komodor-agentops
+Version: 0.1.0
+Summary: Python SDK for AgentOps observability.
+Author: Komodor Ltd.
+License-Expression: LicenseRef-Komodor-Proprietary
+Project-URL: Homepage, https://agentops.komodor.com
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: agentops-otel~=0.1.0
+Requires-Dist: agentops-rpc~=0.1.0
+Requires-Dist: a2a-sdk[http-server]<2,>=1.0
+Requires-Dist: croniter>=2.0
+Requires-Dist: fastapi>=0.136.1
+Requires-Dist: httpx
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pydantic-settings>=2.0
+Requires-Dist: python-frontmatter>=1.2.0
+Requires-Dist: pyyaml>=6.0.3
+Requires-Dist: uvicorn[standard]>=0.46.0
+Provides-Extra: server
+Provides-Extra: langchain
+Requires-Dist: langchain-core; extra == "langchain"
+Provides-Extra: claude-code
+Requires-Dist: claude-agent-sdk>=0.1.81; extra == "claude-code"
+Provides-Extra: adk
+Requires-Dist: google-adk[extensions]>=2.2.0; extra == "adk"
+Provides-Extra: agno
+Requires-Dist: agno<3,>=2.6; extra == "agno"
+Requires-Dist: anthropic>=0.40; extra == "agno"
+Provides-Extra: all
+Requires-Dist: komodor-agentops[adk,agno,claude-code,langchain]; extra == "all"
+Dynamic: license-file
+
+# komodor-agentops
+
+Python SDK for AgentOps observability, including the worker runtime, with
+optional extras for framework integrations (LangChain, Claude Agent SDK, ADK,
+Agno).
+
+## Install
+
+```bash
+pip install komodor-agentops
+
+# With framework adapters
+pip install komodor-agentops[langchain]
+pip install komodor-agentops[claude-code]
+pip install komodor-agentops[adk]
+pip install komodor-agentops[agno]
+pip install komodor-agentops[all]           # Everything
+```
+
+## Quick Start
+
+### @observe() decorator
+
+Wrap functions to emit span events automatically:
+
+```python
+from komodor_agentops import observe
+
+@observe(name="summarize", as_type="llm")
+async def summarize(text: str) -> str:
+    ...
+```
+
+### AgentOps client (lightweight event buffer)
+
+```python
+from komodor_agentops import AgentOps
+
+client = AgentOps(agent_id="my-agent", endpoint="http://localhost:8000")
+await client.log("Processing started", run_id="run_1")
+await client.flush()
+```
+
+### TransportClient (full controlplane client)
+
+For direct controlplane interaction (heartbeat, run lifecycle, event ingest):
+
+```python
+from komodor_agentops import TransportClient, AgentSpec
+
+agent = AgentSpec(agent_id="my-agent", agent_card={"name": "My Agent"})
+async with TransportClient(endpoint="http://localhost:8000", worker_id="wrk_1", agent=agent) as ops:
+    await ops.heartbeat()
+    run = await ops.start_run_direct("run_1", input_payload={"prompt": "hello"})
+    await ops.emit_span_start(run_id="run_1", span_id="s1", name="tool", span_kind="tool")
+    await ops.emit_span_end(run_id="run_1", span_id="s1", name="tool", span_kind="tool")
+    await ops.complete_run("run_1", output={"result": "done"})
+```
+
+### AgentOpsWorker (unified server)
+
+Run a worker with A2A protocol + AgentOps streaming:
+
+```python
+from pathlib import Path
+
+from komodor_agentops import AgentOpsWorker, AgentSpec, A2AMessage, A2ATask
+
+spec = AgentSpec.from_dir(
+    Path(__file__).parent,
+    agent_card={"skills": [{"id": "search", "name": "search"}]},
+)
+
+async def handler(message: A2AMessage) -> A2ATask:
+    return A2ATask(id=message.task_id, status={"state": "completed"})
+
+worker = AgentOpsWorker(agent=spec, handler=handler)
+await worker.serve(host="0.0.0.0", port=8010)
+```
+
+The standard AgentOps agent layout is:
+
+```text
+my_agent/
+  agent-spec.yaml
+  agent.md
+  worker.py
+  skills/
+    triage.md
+    rca/SKILL.md
+```
+
+`AgentSpec.from_dir(Path(__file__).parent, ...)` requires `agent-spec.yaml`, then loads `agent.md` and `skills/` automatically. The worker registers loaded skills during heartbeat, and SDK-owned LLM adapters can prepend the agent context at invocation time.
+
+```yaml
+schema_version: 1
+agent_id: my-agent
+name: My Agent
+description: Does useful work.
+owner: AgentOps
+repo: https://github.com/komodorio/agentops
+source_path: packages/workers/my_agent
+labels:
+  category: example
+```
+
+To make a worker available in the AgentOps Chat UI, advertise chat capability on its agent card:
+
+```python
+spec = AgentSpec(
+    agent_id="my-chat-agent",
+    agent_card={
+        "name": "My Chat Agent",
+        "capabilities": {"chat": True, "ask": True, "streaming": True},
+    },
+)
+```
+
+Control plane chat sessions invoke the selected worker over A2A with input data that includes `messages` (conversation history), `model`, and `prompt` (latest user text). A2A message parts use the proto-style shape accepted by `AgentOpsWorker`, such as `{"text": "..."}` and `{"data": {...}}`.
+
+## Framework Adapters
+
+### LangChain
+
+```python
+from komodor_agentops.langchain import KomodorCallbackHandler
+
+handler = KomodorCallbackHandler(ops=transport_client, controlplane_run_id="run_1")
+chain.invoke(input, config={"callbacks": [handler]})
+```
+
+### Claude Code
+
+Install hooks that forward Claude Code events to the controlplane:
+
+```bash
+install-cc-hooks --endpoint http://localhost:8000
+```
+
+SDK hooks for `claude-agent-sdk`:
+
+```python
+from komodor_agentops.claude_code.sdk_hooks import agent_ops_hooks_for_run
+
+hooks = agent_ops_hooks_for_run("run_1")
+```
+
+## Architecture
+
+```
+komodor-agentops
+  agentops-rpc, agentops-otel (wire types + OTel bootstrap)
+  fastapi, uvicorn, a2a-sdk (worker/server runtime)
+  httpx, pydantic, pydantic-settings, croniter, pyyaml, python-frontmatter
+    |
+    +-- [langchain]   -> langchain-core
+    +-- [claude-code] -> claude-agent-sdk
+    +-- [adk]         -> google-adk
+    +-- [agno]        -> agno, anthropic
+    +-- [server]      -> no-op alias (kept so existing [server] refs resolve)
+```
+
+## Package Structure
+
+```
+src/komodor_agentops/
+  __init__.py       # Public API
+  _client.py        # Event buffer + flush
+  _transport.py     # Full controlplane HTTP client
+  _context.py       # TraceContext + ContextVar
+  _events.py        # AgentOpsEvent dataclass
+  _flush.py         # Async event batching
+  _observe.py       # @observe() decorator
+  _rpc.py           # JSON-RPC helpers
+  _types.py         # Internal SDK types (AgentSpec, RunJob, etc.)
+  worker.py         # AgentOpsWorker facade
+  py.typed          # PEP 561 marker
+  a2a/              # A2A protocol bridge
+  stream/           # SSE/JSON-RPC streaming
+  hooks/            # Reporter + heartbeat
+  langchain/        # LangChain callback handler
+  claude_code/      # Claude Code integration
+```

komodor_agentops-0.1.0/README.md

@@ -0,0 +1,183 @@
+# komodor-agentops
+
+Python SDK for AgentOps observability, including the worker runtime, with
+optional extras for framework integrations (LangChain, Claude Agent SDK, ADK,
+Agno).
+
+## Install
+
+```bash
+pip install komodor-agentops
+
+# With framework adapters
+pip install komodor-agentops[langchain]
+pip install komodor-agentops[claude-code]
+pip install komodor-agentops[adk]
+pip install komodor-agentops[agno]
+pip install komodor-agentops[all]           # Everything
+```
+
+## Quick Start
+
+### @observe() decorator
+
+Wrap functions to emit span events automatically:
+
+```python
+from komodor_agentops import observe
+
+@observe(name="summarize", as_type="llm")
+async def summarize(text: str) -> str:
+    ...
+```
+
+### AgentOps client (lightweight event buffer)
+
+```python
+from komodor_agentops import AgentOps
+
+client = AgentOps(agent_id="my-agent", endpoint="http://localhost:8000")
+await client.log("Processing started", run_id="run_1")
+await client.flush()
+```
+
+### TransportClient (full controlplane client)
+
+For direct controlplane interaction (heartbeat, run lifecycle, event ingest):
+
+```python
+from komodor_agentops import TransportClient, AgentSpec
+
+agent = AgentSpec(agent_id="my-agent", agent_card={"name": "My Agent"})
+async with TransportClient(endpoint="http://localhost:8000", worker_id="wrk_1", agent=agent) as ops:
+    await ops.heartbeat()
+    run = await ops.start_run_direct("run_1", input_payload={"prompt": "hello"})
+    await ops.emit_span_start(run_id="run_1", span_id="s1", name="tool", span_kind="tool")
+    await ops.emit_span_end(run_id="run_1", span_id="s1", name="tool", span_kind="tool")
+    await ops.complete_run("run_1", output={"result": "done"})
+```
+
+### AgentOpsWorker (unified server)
+
+Run a worker with A2A protocol + AgentOps streaming:
+
+```python
+from pathlib import Path
+
+from komodor_agentops import AgentOpsWorker, AgentSpec, A2AMessage, A2ATask
+
+spec = AgentSpec.from_dir(
+    Path(__file__).parent,
+    agent_card={"skills": [{"id": "search", "name": "search"}]},
+)
+
+async def handler(message: A2AMessage) -> A2ATask:
+    return A2ATask(id=message.task_id, status={"state": "completed"})
+
+worker = AgentOpsWorker(agent=spec, handler=handler)
+await worker.serve(host="0.0.0.0", port=8010)
+```
+
+The standard AgentOps agent layout is:
+
+```text
+my_agent/
+  agent-spec.yaml
+  agent.md
+  worker.py
+  skills/
+    triage.md
+    rca/SKILL.md
+```
+
+`AgentSpec.from_dir(Path(__file__).parent, ...)` requires `agent-spec.yaml`, then loads `agent.md` and `skills/` automatically. The worker registers loaded skills during heartbeat, and SDK-owned LLM adapters can prepend the agent context at invocation time.
+
+```yaml
+schema_version: 1
+agent_id: my-agent
+name: My Agent
+description: Does useful work.
+owner: AgentOps
+repo: https://github.com/komodorio/agentops
+source_path: packages/workers/my_agent
+labels:
+  category: example
+```
+
+To make a worker available in the AgentOps Chat UI, advertise chat capability on its agent card:
+
+```python
+spec = AgentSpec(
+    agent_id="my-chat-agent",
+    agent_card={
+        "name": "My Chat Agent",
+        "capabilities": {"chat": True, "ask": True, "streaming": True},
+    },
+)
+```
+
+Control plane chat sessions invoke the selected worker over A2A with input data that includes `messages` (conversation history), `model`, and `prompt` (latest user text). A2A message parts use the proto-style shape accepted by `AgentOpsWorker`, such as `{"text": "..."}` and `{"data": {...}}`.
+
+## Framework Adapters
+
+### LangChain
+
+```python
+from komodor_agentops.langchain import KomodorCallbackHandler
+
+handler = KomodorCallbackHandler(ops=transport_client, controlplane_run_id="run_1")
+chain.invoke(input, config={"callbacks": [handler]})
+```
+
+### Claude Code
+
+Install hooks that forward Claude Code events to the controlplane:
+
+```bash
+install-cc-hooks --endpoint http://localhost:8000
+```
+
+SDK hooks for `claude-agent-sdk`:
+
+```python
+from komodor_agentops.claude_code.sdk_hooks import agent_ops_hooks_for_run
+
+hooks = agent_ops_hooks_for_run("run_1")
+```
+
+## Architecture
+
+```
+komodor-agentops
+  agentops-rpc, agentops-otel (wire types + OTel bootstrap)
+  fastapi, uvicorn, a2a-sdk (worker/server runtime)
+  httpx, pydantic, pydantic-settings, croniter, pyyaml, python-frontmatter
+    |
+    +-- [langchain]   -> langchain-core
+    +-- [claude-code] -> claude-agent-sdk
+    +-- [adk]         -> google-adk
+    +-- [agno]        -> agno, anthropic
+    +-- [server]      -> no-op alias (kept so existing [server] refs resolve)
+```
+
+## Package Structure
+
+```
+src/komodor_agentops/
+  __init__.py       # Public API
+  _client.py        # Event buffer + flush
+  _transport.py     # Full controlplane HTTP client
+  _context.py       # TraceContext + ContextVar
+  _events.py        # AgentOpsEvent dataclass
+  _flush.py         # Async event batching
+  _observe.py       # @observe() decorator
+  _rpc.py           # JSON-RPC helpers
+  _types.py         # Internal SDK types (AgentSpec, RunJob, etc.)
+  worker.py         # AgentOpsWorker facade
+  py.typed          # PEP 561 marker
+  a2a/              # A2A protocol bridge
+  stream/           # SSE/JSON-RPC streaming
+  hooks/            # Reporter + heartbeat
+  langchain/        # LangChain callback handler
+  claude_code/      # Claude Code integration
+```

komodor_agentops-0.1.0/pyproject.toml

@@ -0,0 +1,71 @@
+[project]
+name = "komodor-agentops"
+version = "0.1.0"
+description = "Python SDK for AgentOps observability."
+readme = "README.md"
+license = "LicenseRef-Komodor-Proprietary"
+license-files = ["LICENSE"]
+authors = [{ name = "Komodor Ltd." }]
+requires-python = ">=3.11"
+dependencies = [
+  "agentops-otel~=0.1.0",
+  "agentops-rpc~=0.1.0",
+  "a2a-sdk[http-server]>=1.0,<2",
+  "croniter>=2.0",
+  "fastapi>=0.136.1",
+  "httpx",
+  "pydantic>=2.0",
+  "pydantic-settings>=2.0",
+  "python-frontmatter>=1.2.0",
+  "pyyaml>=6.0.3",
+  "uvicorn[standard]>=0.46.0",
+]
+
+[project.urls]
+Homepage = "https://agentops.komodor.com"
+
+[project.optional-dependencies]
+server = []  # kept as no-op alias so [server] refs resolve
+langchain = [
+  "langchain-core",
+]
+claude-code = [
+  "claude-agent-sdk>=0.1.81",
+]
+adk = [
+  "google-adk[extensions]>=2.2.0",
+]
+agno = [
+  "agno>=2.6,<3",
+  "anthropic>=0.40",
+]
+all = [
+  "komodor-agentops[langchain,claude-code,adk,agno]",
+]
+
+[project.scripts]
+install-cc-hooks = "komodor_agentops.adapters.claude_code.install:cli"
+komodor-cc-hook = "komodor_agentops.adapters.claude_code.hook:main"
+agentops-run-worker = "komodor_agentops.worker.local:cli"
+
+[build-system]
+requires = ["setuptools>=77"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.package-dir]
+"" = "src"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+"komodor_agentops" = ["schemas/*.json"]
+
+[tool.uv.sources]
+agentops-otel = { workspace = true }
+agentops-rpc = { workspace = true }
+
+[dependency-groups]
+dev = [
+    "respx>=0.23.1",
+]

komodor_agentops-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

komodor_agentops-0.1.0/src/komodor_agentops/__init__.py

@@ -0,0 +1,92 @@
+from agentops_rpc import (
+    GRADER_CAPABILITY,
+    REVIEW_CAPABILITY,
+    USE_CASE_INPUT_KEY,
+    A2AMessage,
+    A2AMessagePart,
+    A2ATask,
+    GradeContext,
+    GradeResult,
+    RunJob,
+    UseCase,
+)
+from komodor_agentops.artifacts import ArtifactClient, get_artifact_client
+from komodor_agentops.client import AgentOpsClient as AgentOps
+from komodor_agentops.client import get_client
+from komodor_agentops.core.context import TraceContext
+from komodor_agentops.core.events import AgentOpsEvent
+from komodor_agentops.core.observe import observe
+from komodor_agentops.core.secrets import apply_secret_to_env, get_run_secrets, get_secret
+from komodor_agentops.core.skills import AgentMarkdown
+from komodor_agentops.core.transport import AgentOpsClient as TransportClient
+from komodor_agentops.core.types import AgentSpec
+from komodor_agentops.core.usage import drain_usage, record_usage, reset_usage, usage_from_anthropic_response
+from komodor_agentops.github import (
+    GitHubIntegrationError,
+    github_api_get,
+    github_api_patch,
+    github_api_post,
+    github_graphql,
+    github_installation_token,
+)
+from komodor_agentops.grade_context import fetch_grade_context
+from komodor_agentops.worker import (
+    AgentOpsWorker,
+    collect_env_secrets,
+    fetch_controlplane_secrets,
+    load_env_files,
+    run_worker_once,
+    run_worker_once_sync,
+)
+from komodor_agentops.worker.a2a.streaming import stream_status_text
+from komodor_agentops.worker.result import DESCRIPTIVE_OUTPUT_KEY, result_task
+from komodor_agentops.worker.server import create_worker_app, event_sink_from_client, register_worker_endpoint
+
+__all__ = [
+    "ArtifactClient",
+    "get_artifact_client",
+    "AgentOps",
+    "AgentMarkdown",
+    "AgentOpsEvent",
+    "AgentOpsWorker",
+    "A2AMessage",
+    "A2AMessagePart",
+    "A2ATask",
+    "AgentSpec",
+    "DESCRIPTIVE_OUTPUT_KEY",
+    "GRADER_CAPABILITY",
+    "REVIEW_CAPABILITY",
+    "USE_CASE_INPUT_KEY",
+    "GitHubIntegrationError",
+    "GradeContext",
+    "GradeResult",
+    "fetch_grade_context",
+    "RunJob",
+    "UseCase",
+    "apply_secret_to_env",
+    "collect_env_secrets",
+    "fetch_controlplane_secrets",
+    "TraceContext",
+    "TransportClient",
+    "create_worker_app",
+    "event_sink_from_client",
+    "get_client",
+    "get_run_secrets",
+    "github_api_get",
+    "github_api_patch",
+    "github_api_post",
+    "github_graphql",
+    "github_installation_token",
+    "get_secret",
+    "load_env_files",
+    "observe",
+    "record_usage",
+    "register_worker_endpoint",
+    "result_task",
+    "reset_usage",
+    "run_worker_once",
+    "run_worker_once_sync",
+    "drain_usage",
+    "stream_status_text",
+    "usage_from_anthropic_response",
+]

komodor_agentops-0.1.0/src/komodor_agentops/adapters/__init__.py

@@ -0,0 +1 @@
+"""Adapter modules for third-party agent frameworks."""

komodor_agentops-0.1.0/src/komodor_agentops/adapters/adk/__init__.py

@@ -0,0 +1,4 @@
+from komodor_agentops.adapters.adk.a2a_adapter import make_agentops_adk_interceptor, run_adk_pipeline
+from komodor_agentops.adapters.adk.adapter import instrument_adk_run
+
+__all__ = ["instrument_adk_run", "make_agentops_adk_interceptor", "run_adk_pipeline"]