Skip to main content

Connect agent SDKs to context-graph components (actions-graph, skills-graph, etc.)

Project description

Agent Context Graph

Connect any agent runtime to any context-graph component.

Agent Context Graph is a lightweight adapter layer that decouples runtime-specific hooks from graph storage. It routes a common event protocol from runtime adapters to graph connectors, so you can mix and match SDKs and graph components.

Runtime Adapter  ->  Event Protocol  ->  Graph Connector(s)
(Claude,         (ToolStart,         (SkillGraphConnector,
 OpenAI)          ToolEnd, ...)       custom connectors, ...)

Runtime plugins are the distribution layer for host-specific hook wiring. They install hooks, skills, and setup helpers for a runtime, then call Agent Context Graph. They are not graph components and should not encode graph-specific meaning.

Installation

pip install agent-context-graph

With runtime adapters:

pip install agent-context-graph[claude]
pip install agent-context-graph[openai]

Graph connectors live in the graph packages that persist the data. For the skills graph connector:

pip install skills-graph[agent-context-graph]

Quick Start

Claude Agent SDK

from agent_context_graph import AgentLink
from agent_context_graph.adapters.claude import ClaudeAdapter
from claude_agent_sdk import ClaudeAgentOptions, query
from skills_graph import SkillGraph
from skills_graph.connector import SkillGraphConnector

# 1. Set up graph storage
skills = SkillGraph()
skills.setup()

# 2. Wire up the link
link = AgentLink()
link.add_connector(SkillGraphConnector(skills))

# 3. Create adapter
adapter = ClaudeAdapter(
    link,
    session_id="my-session",
    session_kwargs={"model": "claude-sonnet-4-20250514", "tags": ["review"]},
)

# 4. Use with Claude Agent SDK
async for message in query(
    prompt="Review the available skills",
    options=ClaudeAgentOptions(hooks=adapter.get_runtime_hooks()),
):
    print(message)

OpenAI Agents SDK

from agent_context_graph import AgentLink
from agent_context_graph.adapters.openai import OpenAIAdapter
from agents import Agent, Runner, function_tool
from skills_graph import SkillGraph
from skills_graph.connector import SkillGraphConnector

# 1. Set up graph storage
skills = SkillGraph()
skills.setup()

# 2. Define a tool whose name matches the SkillGraphConnector defaults
@function_tool
def get_skill(name: str) -> str:
    skill = skills.get_skill(name)
    if skill is None:
        return f"Skill '{name}' not found."
    return f"{skill.name}: {skill.description}\n{skill.content}"

# 3. Wire up the link
link = AgentLink()
link.add_connector(SkillGraphConnector(skills))

# 4. Create adapter
adapter = OpenAIAdapter(
    link,
    session_id="my-session",
    session_kwargs={"model": "gpt-4o-mini"},
)

# 5. Run with hooks
agent = Agent(
    name="Skill Assistant",
    instructions="Use get_skill when the user asks for a named skill.",
    tools=[get_skill],
    model="gpt-4o-mini",
)
result = await Runner.run(
    agent,
    "Get the skill called 'cypher-basics'",
    hooks=adapter.get_runtime_hooks(),
)

# 6. Signal end (OpenAI SDK doesn't have a stop hook)
adapter.end_session()

Command Hook Runtimes

Some agent applications run hooks as external commands instead of in-process SDK callbacks. Runtime adapters should keep the product-specific JSON mapping at the edge, emit the shared Event protocol, and leave graph persistence in connectors such as SkillGraphConnector.

The installed command is runtime-dispatched:

agent-context-graph hook <command> [options]

Implemented:

Runtime Adapter Hook Shape
OpenAI Codex CodexHooksAdapter Command receives one JSON object on stdin
Claude Code ClaudeCodeHooksAdapter Command receives one JSON object on stdin

OpenAI Codex Hooks

Codex hook configuration can be installed either as local environment wiring or as a user-level Codex plugin.

The runtime-plugin flow is:

Codex Plugin -> Codex Runtime Adapter -> Event Protocol -> Graph Connector -> Memgraph

The plugin installs Codex hook wiring. The Codex runtime adapter normalizes the hook payload. Graph connectors such as SkillGraphConnector decide what those events mean in their graph.

Prerequisites:

  • Memgraph running and reachable over Bolt. Defaults are bolt://localhost:7687, empty user/password, and database memgraph.
  • A Python environment that contains agent-context-graph and skills-graph[agent-context-graph].
  • Codex CLI or IDE extension running in a project that trusts the project-local .codex/ layer.

For a global plugin proof of concept, see:

context-graph/plugins/agent-context-graph-codex

That plugin expects agent-context-graph to be available on PATH and Memgraph to be reachable from the Codex process. A global install can use uv tool:

uv tool install agent-context-graph --with "skills-graph[agent-context-graph]"

Then install the Codex plugin through a user plugin marketplace. Keep graph credentials in the process environment, not in the plugin hook file.

Check the installed hook environment with:

agent-context-graph doctor --runtime codex --connector skills-graph

For a public Git-backed marketplace install:

codex plugin marketplace add memgraph/ai-toolkit --sparse .agents/plugins

Local .codex/ files remain useful for source development and per-project experiments. This repository ignores .codex/.

Claude Code Hooks

Claude Code hook configuration can be installed as a Claude Code plugin.

The runtime-plugin flow is:

Claude Code Plugin -> Claude Code Runtime Adapter -> Event Protocol -> Graph Connector -> Memgraph

For a public Git-backed marketplace install, add the marketplace inside Claude Code:

/plugin marketplace add memgraph/ai-toolkit

Then install:

/plugin install agent-context-graph-claude@context-graph-plugins

The streamlined setup only needs two pieces of local information:

  • where to write the Codex project config, usually your repo root
  • where Memgraph is, plus optional auth/database values

If Memgraph is running locally with defaults:

agent-context-graph setup codex --project-dir "$PWD" --setup-schema

--setup-schema connects to Memgraph immediately and runs SkillGraph().setup().

If you need non-default Memgraph connection values:

agent-context-graph setup codex \
  --project-dir /path/to/your/repo \
  --memgraph-url bolt://localhost:7687 \
  --memgraph-user "" \
  --memgraph-password "" \
  --memgraph-database memgraph \
  --setup-schema

The --memgraph-* options are used for --setup-schema, but they are not written into .codex/hooks.json.

For source development in this workspace:

uv run --package skills-graph --extra agent-context-graph \
  python -m agent_context_graph.cli setup codex \
  --project-dir /path/to/your/repo \
  --memgraph-url bolt://localhost:7687 \
  --setup-schema

The command writes local, ignored files:

.codex/config.toml
.codex/hooks.json

It refuses to overwrite existing generated files unless you pass --force.

The generated hook command does not embed any Memgraph connection values. At runtime, Codex must run with the needed MEMGRAPH_* variables in its process environment, or the hooks will use memgraph-toolbox defaults.

If Memgraph requires a password, provide MEMGRAPH_PASSWORD to the Codex process environment. .codex/hooks.json should not contain Memgraph credentials.

Keep the Python environment used by the generated hook command around. Codex will run that absolute command path for every hook event.

To smoke test the generated command, copy the "command" value from .codex/hooks.json and run:

printf '{"hook_event_name":"Stop","session_id":"test"}' | COMMAND

The expected output is:

{"continue": true}

If you prefer manual setup:

  1. Make skills-graph able to reach Memgraph, then initialize and seed your skill graph once:
export MEMGRAPH_URL="bolt://localhost:7687"
export MEMGRAPH_USER=""
export MEMGRAPH_PASSWORD=""
from skills_graph import SkillGraph

skills = SkillGraph()
skills.setup()
  1. Install the hook command and the graph connector in the same Python environment:
python -m venv ~/.venvs/agent-context-graph-hooks
~/.venvs/agent-context-graph-hooks/bin/python -m pip install \
  "agent-context-graph" \
  "skills-graph[agent-context-graph]"

For source development in this workspace, use this command instead of the venv binary:

cd /path/to/ai-toolkit
uv run --package skills-graph --extra agent-context-graph \
  python -m agent_context_graph.cli hook run codex --connector skills-graph
  1. Generate private Codex hook config in the workspace:
agent-context-graph hook init codex --connector skills-graph

For source development in this workspace:

uv run --package skills-graph --extra agent-context-graph \
  python -m agent_context_graph.cli hook init codex --connector skills-graph

The wizard writes local, ignored files:

.codex/config.toml
.codex/hooks.json

It refuses to overwrite existing generated files unless you pass --force.

The generated config enables Codex hooks and points all supported Codex hook events at a command like:

agent-context-graph hook run codex --connector skills-graph

The resulting .codex/hooks.json has this shape:

{
  "hooks": {
    "SessionStart": [
      {
        "matcher": "startup|resume|clear",
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ],
    "UserPromptSubmit": [
      {
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ],
    "PreToolUse": [
      {
        "matcher": "*",
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ],
    "PostToolUse": [
      {
        "matcher": "*",
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ],
    "PermissionRequest": [
      {
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ],
    "Stop": [
      {
        "hooks": [{ "type": "command", "command": "COMMAND", "timeout": 30 }]
      }
    ]
  }
}

The adapter records Codex SessionStart, UserPromptSubmit, PreToolUse, PostToolUse, PermissionRequest, and Stop payloads. MCP tool names such as mcp__skills__get_skill are normalized by skills-graph to the underlying get_skill operation.

Multiple Graph Components

from agent_context_graph import AgentLink
from agent_context_graph.adapters.claude import ClaudeAdapter
from skills_graph import SkillGraph
from skills_graph.connector import SkillGraphConnector

skills = SkillGraph()

link = AgentLink()
link.add_connector(SkillGraphConnector(skills))
link.add_connector(MyGraphConnector(...))

adapter = ClaudeAdapter(link, session_id="s-1")
hooks = adapter.get_runtime_hooks()

Connectors are owned by the graph packages because each graph package knows its own schema and persistence rules.

Architecture

Event Protocol

All runtime adapters emit runtime-agnostic Event dataclasses:

Event When
SessionStartEvent Agent session begins
SessionEndEvent Agent session ends
ToolStartEvent Before tool/function call
ToolEndEvent After tool/function returns
AgentStartEvent Agent/subagent begins
AgentEndEvent Agent/subagent finishes
LLMStartEvent Before LLM call
LLMEndEvent After LLM response
HandoffEvent Agent hands off to another
MessageEvent User/assistant/system message
ErrorOccurredEvent Error during execution

Runtime Adapters

Adapter Runtime Source Hook Mechanism
ClaudeAdapter Claude Agent SDK Dict of HookMatcher callbacks
OpenAIAdapter OpenAI Agents SDK RunHooksBase subclass
CodexHooksAdapter OpenAI Codex Command hooks reading JSON from stdin

Graph Connectors

Connector Graph Component Events Handled
SkillGraphConnector skills-graph Tool events matching skill access/search operations

Additional graph connectors should live in the packages that own those graph schemas.

Adding a New Runtime Adapter

Implement RuntimeAdapter:

from agent_context_graph import AgentLink, ToolStartEvent
from agent_context_graph.protocols import RuntimeAdapter

class MyRuntimeAdapter(RuntimeAdapter):
    def __init__(self, link: AgentLink, session_id: str):
        self._link = link
        self._session_id = session_id

    def get_runtime_hooks(self):
        # Return whatever your runtime expects.
        ...

    def _on_tool_call(self, name, args):
        self._link.emit(
            ToolStartEvent(
                session_id=self._session_id,
                tool_name=name,
                tool_input=args,
            )
        )

Adding a New Graph Component

Implement GraphConnector in the graph package:

from agent_context_graph import EventType
from agent_context_graph.protocols import GraphConnector

class MyGraphConnector(GraphConnector):
    def supports(self, event):
        return event.event_type in {EventType.TOOL_START, EventType.TOOL_END}

    def on_event(self, event):
        # Write to your graph component.
        ...

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

agent_context_graph-0.1.1.tar.gz (23.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

agent_context_graph-0.1.1-py3-none-any.whl (25.9 kB view details)

Uploaded Python 3

File details

Details for the file agent_context_graph-0.1.1.tar.gz.

File metadata

  • Download URL: agent_context_graph-0.1.1.tar.gz
  • Upload date:
  • Size: 23.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.11 {"installer":{"name":"uv","version":"0.11.11","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for agent_context_graph-0.1.1.tar.gz
Algorithm Hash digest
SHA256 39b388231d0565b47c303a00e90751881d429f079d0a319cb519ba2c86dde4d9
MD5 f5c84adc56fda232f74a7a2ab2396330
BLAKE2b-256 4c49d180b6f4a8d06f42c1df49d804d3bcce95dae554a7981248134bfbf0efe9

See more details on using hashes here.

File details

Details for the file agent_context_graph-0.1.1-py3-none-any.whl.

File metadata

  • Download URL: agent_context_graph-0.1.1-py3-none-any.whl
  • Upload date:
  • Size: 25.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.11 {"installer":{"name":"uv","version":"0.11.11","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for agent_context_graph-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 0a1c0b8fd1db9d376de4111da65f0c508e714b3b3daceb200f695d7964c01d1f
MD5 9972efd50efb10df2b904ffbfd132f2d
BLAKE2b-256 c73686d0ce460ea9df409304ec2858ff7d5526097663e02b26b274e37533dc98

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page