coco-framework 0.1.0__tar.gz

coco_framework-0.1.0/.gitignore

@@ -0,0 +1,14 @@
+.venv/
+__pycache__/
+*.pyc
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+.coverage
+.coverage.*
+htmlcov/
+*.egg-info/
+*.whl
+*.tar.gz
+dist/
+build/

coco_framework-0.1.0/LICENSE

@@ -0,0 +1,24 @@
+MIT License
+
+Copyright (c) LangChain, Inc.
+Copyright (c) 2024 LangChain, Inc.
+Copyright (c) 2026 Coco AI Pty Ltd
+
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR 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.

coco_framework-0.1.0/NOTICE

@@ -0,0 +1,15 @@
+coco-framework
+
+Everything under libs/ is derived from two MIT-licensed projects and renamed by
+scripts/sync_upstream.py:
+
+  langchain-ai/langchain   Copyright (c) LangChain, Inc.
+  langchain-ai/langgraph   Copyright (c) 2024 LangChain, Inc.
+
+libs/core/coco_core/_protocol.py is vendored from langchain-protocol 0.0.19,
+Copyright (c) 2026 LangChain, Inc., MIT.
+
+Each library under libs/ keeps its upstream LICENSE file unchanged. UPSTREAM.md
+records the commit each library was taken from and every patch applied after
+the rename. Changes are Copyright (c) 2026 Coco AI Pty Ltd, under the same MIT
+licence.

coco_framework-0.1.0/PKG-INFO

@@ -0,0 +1,184 @@
+Metadata-Version: 2.5
+Name: coco-framework
+Version: 0.1.0
+Summary: The agent runtime for Coco: core messages, tools, runnables, stateful agent graphs, checkpointers, and model integrations.
+Project-URL: Homepage, https://trustcoco.ai
+Project-URL: Repository, https://github.com/coco-ai-au/coco-framework
+License: MIT License
+        
+        Copyright (c) LangChain, Inc.
+        Copyright (c) 2024 LangChain, Inc.
+        Copyright (c) 2026 Coco AI Pty Ltd
+        
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR 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.
+License-File: LICENSE
+License-File: NOTICE
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: <4.0.0,>=3.10.0
+Requires-Dist: aiosqlite>=0.20.0
+Requires-Dist: anthropic<2.0.0,>=0.120.0
+Requires-Dist: jsonpatch<2.0.0,>=1.33.0
+Requires-Dist: openai<4.0.0,>=2.26.0
+Requires-Dist: ormsgpack>=1.12.0
+Requires-Dist: packaging<27.0.0,>=23.2.0
+Requires-Dist: pydantic<3.0.0,>=2.7.4
+Requires-Dist: pyyaml<7.0.0,>=5.3.0
+Requires-Dist: sqlite-vec>=0.1.6
+Requires-Dist: tenacity!=8.4.0,<10.0.0,>=8.1.0
+Requires-Dist: tiktoken<1.0.0,>=0.7.0
+Requires-Dist: typing-extensions<5.0.0,>=4.7.0
+Requires-Dist: uuid-utils<1.0,>=0.12.0
+Requires-Dist: xxhash>=3.5.0
+Provides-Extra: all
+Requires-Dist: aiosqlite>=0.20.0; extra == 'all'
+Requires-Dist: anthropic<2.0.0,>=0.120.0; extra == 'all'
+Requires-Dist: azure-identity<2.0.0,>=1.15.0; extra == 'all'
+Requires-Dist: fastmcp<4.0.0,>=3.1.1; extra == 'all'
+Requires-Dist: openai<4.0.0,>=2.26.0; extra == 'all'
+Requires-Dist: ormsgpack>=1.12.0; extra == 'all'
+Requires-Dist: psycopg-pool>=3.2.0; extra == 'all'
+Requires-Dist: psycopg>=3.2.0; extra == 'all'
+Requires-Dist: sqlite-vec>=0.1.6; extra == 'all'
+Requires-Dist: tiktoken<1.0.0,>=0.7.0; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic<2.0.0,>=0.120.0; extra == 'anthropic'
+Provides-Extra: azure-identity
+Requires-Dist: azure-identity<2.0.0,>=1.15.0; extra == 'azure-identity'
+Provides-Extra: mcp
+Requires-Dist: fastmcp<4.0.0,>=3.1.1; extra == 'mcp'
+Provides-Extra: openai
+Requires-Dist: openai<4.0.0,>=2.26.0; extra == 'openai'
+Requires-Dist: tiktoken<1.0.0,>=0.7.0; extra == 'openai'
+Provides-Extra: postgres
+Requires-Dist: ormsgpack>=1.12.0; extra == 'postgres'
+Requires-Dist: psycopg-pool>=3.2.0; extra == 'postgres'
+Requires-Dist: psycopg>=3.2.0; extra == 'postgres'
+Provides-Extra: sqlite
+Requires-Dist: aiosqlite>=0.20.0; extra == 'sqlite'
+Requires-Dist: sqlite-vec>=0.1.6; extra == 'sqlite'
+Description-Content-Type: text/markdown
+
+# coco-framework
+
+The agent runtime for Coco: LangChain and LangGraph, forked at pinned commits and
+renamed to Coco by one script. Agents built on it get Coco's check on every tool
+call from `../coco-sdk`.
+
+```python
+from coco.agents import create_agent
+from coco_core.tools import tool
+from coco_graph.checkpoint.memory import InMemorySaver
+from coco_openai import ChatOpenAI
+```
+
+## Install
+
+Install the full framework directly with `pip`:
+
+```bash
+pip install .                                                   # from a local checkout
+pip install -e .                                                # editable development install
+pip install "git+https://github.com/coco-ai-au/coco-framework.git"
+```
+
+Or use the coco-sdk environment, which installs every library here:
+
+```bash
+cd ../coco-sdk && uv sync && source .venv/bin/activate
+```
+
+To work on the fork itself or build and publish the distributions:
+
+```bash
+python3 scripts/sync_upstream.py --report   # rebuild libs/ from the pins, list leftover upstream names
+scripts/dev.sh setup                        # .venv with every library and its test/build tools
+scripts/dev.sh test                         # each library's upstream unit tests, renamed
+python3 scripts/build.py --clean            # build and verify all 9 trustcoco-* wheels + sdists in dist/
+python3 scripts/build.py --include-root     # also build the bundled coco-framework distribution
+python3 scripts/build.py --publish          # upload built distributions with twine/uv
+```
+
+`sync_upstream.py` needs the upstream clones at the pinned commits in
+`build_reference/` above this folder (or `COCO_UPSTREAM_DIR`).
+
+## What is here
+
+| Layer | Distribution | Import | From upstream |
+|---|---|---|---|
+| Core | `trustcoco-core` | `coco_core` | `langchain-core` 1.6.5 |
+| Graph | `trustcoco-graph`, `-prebuilt`, `-checkpoint`, `-checkpoint-sqlite`, `-checkpoint-postgres` | `coco_graph` | `langgraph` 1.2.12 and its libraries |
+| Agents | `trustcoco` | `coco` | `langchain` 1.4.2 |
+| Models | `trustcoco-openai`, `trustcoco-anthropic` | `coco_openai`, `coco_anthropic` | the two partner packages |
+
+Messages, tools, runnables and prompts; `StateGraph`, the runtime, `interrupt()`
+and checkpointers; `create_agent` and its middleware; an OpenAI-compatible and an
+Anthropic chat model.
+
+Left out: upstream's hosted-platform clients (`langgraph-sdk`, the CLI,
+`RemoteGraph`), its tracing client (`langsmith`, replaced by a local `CocoTracer`
+that sends nothing), text splitters, community integrations and JavaScript.
+
+## How the fork is made
+
+`scripts/sync_upstream.py` copies the pinned upstream libraries into `libs/`,
+renames modules, distributions, classes and environment variables, removes the
+hosted-platform paths, copies Coco's own files from `overlay/`, and applies
+`patches/`. It runs again on every upstream bump, and a bump lands only when the
+renamed upstream tests pass. Nobody edits `libs/` by hand.
+
+[`UPSTREAM.md`](UPSTREAM.md) lists the pins, every name mapping, every removed
+path, overlay file and patch with its reason, and the test results.
+
+## Where Coco attaches
+
+`create_agent` runs every tool call through `wrap_tool_call`. `CocoGuard` from
+coco-sdk is that middleware: it runs the call, denies it with a reason, or pauses
+the graph with `interrupt()` until a person decides. The session a contract reads
+is the graph's thread id. The framework carries no Coco enforcement code, which
+keeps the fork mechanical.
+
+## Decisions
+
+| Question | Decided |
+|---|---|
+| Compatibility with upstream integrations | A clean rename. Upstream partner packages import `langchain_core` and do not load here; we fork a provider when a design partner needs one |
+| Package names | `trustcoco-*` distributions, `coco*` imports. PyPI on the day we open source |
+| Guard on by default | Yes, once the SDK is stable: `create_agent` installs `CocoGuard` in Monitor, with a logged opt out. Not built yet |
+| Upstream cadence | Pinned. Bump for a fix we need; review the gap each quarter |
+
+Publishes as wheels to Coco's private package index by `scripts/build.py --publish`
+(or `coco-sdk/scripts/publish.sh`); the install is in coco-sdk's README. PyPI on the
+day we open source.
+
+## Licence
+
+MIT. `LICENSE` keeps "Copyright (c) LangChain, Inc." beside ours, and `NOTICE`
+names what derives from upstream. The Coco name replaces theirs everywhere else.

coco_framework-0.1.0/README.md

@@ -0,0 +1,96 @@
+# coco-framework
+
+The agent runtime for Coco: LangChain and LangGraph, forked at pinned commits and
+renamed to Coco by one script. Agents built on it get Coco's check on every tool
+call from `../coco-sdk`.
+
+```python
+from coco.agents import create_agent
+from coco_core.tools import tool
+from coco_graph.checkpoint.memory import InMemorySaver
+from coco_openai import ChatOpenAI
+```
+
+## Install
+
+Install the full framework directly with `pip`:
+
+```bash
+pip install .                                                   # from a local checkout
+pip install -e .                                                # editable development install
+pip install "git+https://github.com/coco-ai-au/coco-framework.git"
+```
+
+Or use the coco-sdk environment, which installs every library here:
+
+```bash
+cd ../coco-sdk && uv sync && source .venv/bin/activate
+```
+
+To work on the fork itself or build and publish the distributions:
+
+```bash
+python3 scripts/sync_upstream.py --report   # rebuild libs/ from the pins, list leftover upstream names
+scripts/dev.sh setup                        # .venv with every library and its test/build tools
+scripts/dev.sh test                         # each library's upstream unit tests, renamed
+python3 scripts/build.py --clean            # build and verify all 9 trustcoco-* wheels + sdists in dist/
+python3 scripts/build.py --include-root     # also build the bundled coco-framework distribution
+python3 scripts/build.py --publish          # upload built distributions with twine/uv
+```
+
+`sync_upstream.py` needs the upstream clones at the pinned commits in
+`build_reference/` above this folder (or `COCO_UPSTREAM_DIR`).
+
+## What is here
+
+| Layer | Distribution | Import | From upstream |
+|---|---|---|---|
+| Core | `trustcoco-core` | `coco_core` | `langchain-core` 1.6.5 |
+| Graph | `trustcoco-graph`, `-prebuilt`, `-checkpoint`, `-checkpoint-sqlite`, `-checkpoint-postgres` | `coco_graph` | `langgraph` 1.2.12 and its libraries |
+| Agents | `trustcoco` | `coco` | `langchain` 1.4.2 |
+| Models | `trustcoco-openai`, `trustcoco-anthropic` | `coco_openai`, `coco_anthropic` | the two partner packages |
+
+Messages, tools, runnables and prompts; `StateGraph`, the runtime, `interrupt()`
+and checkpointers; `create_agent` and its middleware; an OpenAI-compatible and an
+Anthropic chat model.
+
+Left out: upstream's hosted-platform clients (`langgraph-sdk`, the CLI,
+`RemoteGraph`), its tracing client (`langsmith`, replaced by a local `CocoTracer`
+that sends nothing), text splitters, community integrations and JavaScript.
+
+## How the fork is made
+
+`scripts/sync_upstream.py` copies the pinned upstream libraries into `libs/`,
+renames modules, distributions, classes and environment variables, removes the
+hosted-platform paths, copies Coco's own files from `overlay/`, and applies
+`patches/`. It runs again on every upstream bump, and a bump lands only when the
+renamed upstream tests pass. Nobody edits `libs/` by hand.
+
+[`UPSTREAM.md`](UPSTREAM.md) lists the pins, every name mapping, every removed
+path, overlay file and patch with its reason, and the test results.
+
+## Where Coco attaches
+
+`create_agent` runs every tool call through `wrap_tool_call`. `CocoGuard` from
+coco-sdk is that middleware: it runs the call, denies it with a reason, or pauses
+the graph with `interrupt()` until a person decides. The session a contract reads
+is the graph's thread id. The framework carries no Coco enforcement code, which
+keeps the fork mechanical.
+
+## Decisions
+
+| Question | Decided |
+|---|---|
+| Compatibility with upstream integrations | A clean rename. Upstream partner packages import `langchain_core` and do not load here; we fork a provider when a design partner needs one |
+| Package names | `trustcoco-*` distributions, `coco*` imports. PyPI on the day we open source |
+| Guard on by default | Yes, once the SDK is stable: `create_agent` installs `CocoGuard` in Monitor, with a logged opt out. Not built yet |
+| Upstream cadence | Pinned. Bump for a fix we need; review the gap each quarter |
+
+Publishes as wheels to Coco's private package index by `scripts/build.py --publish`
+(or `coco-sdk/scripts/publish.sh`); the install is in coco-sdk's README. PyPI on the
+day we open source.
+
+## Licence
+
+MIT. `LICENSE` keeps "Copyright (c) LangChain, Inc." beside ours, and `NOTICE`
+names what derives from upstream. The Coco name replaces theirs everywhere else.

coco_framework-0.1.0/UPSTREAM.md

@@ -0,0 +1,116 @@
+# Upstream
+
+`scripts/sync_upstream.py` builds `libs/` from these pins. To move a pin, check the
+clone out at the new commit, change the commit in the script and in this file, run
+the script, and land the change only when the renamed tests pass.
+
+| Upstream | Commit | Date | Versions |
+|---|---|---|---|
+| langchain-ai/langchain | `d750819` | 27 Sep 2026 | langchain-core 1.6.5, langchain-tests 1.1.9, langchain 1.4.2, langchain-openai 1.6.6, langchain-anthropic 1.7.4 |
+| langchain-ai/langgraph | `07b3318` | 27 Sep 2026 | langgraph 1.2.12, langgraph-prebuilt 1.1.0, langgraph-checkpoint 4.2.0 |
+
+| Folder | Distribution | Import | From |
+|---|---|---|---|
+| `libs/core` | `trustcoco-core` | `coco_core` | `langchain/libs/core` |
+| `libs/standard-tests` | `trustcoco-tests`, test only | `coco_tests` | `langchain/libs/standard-tests` |
+| `libs/coco_graph` | `trustcoco-graph` | `coco_graph` | `langgraph/libs/langgraph` |
+| `libs/checkpoint` | `trustcoco-graph-checkpoint` | `coco_graph.checkpoint` | `langgraph/libs/checkpoint` |
+| `libs/checkpoint-sqlite` | `trustcoco-graph-checkpoint-sqlite` | `coco_graph.checkpoint.sqlite` | `langgraph/libs/checkpoint-sqlite` |
+| `libs/checkpoint-postgres` | `trustcoco-graph-checkpoint-postgres` | `coco_graph.checkpoint.postgres` | `langgraph/libs/checkpoint-postgres` |
+| `libs/checkpoint-conformance` | `trustcoco-graph-checkpoint-conformance`, test only | `coco_graph.checkpoint.conformance` | `langgraph/libs/checkpoint-conformance` |
+| `libs/prebuilt` | `trustcoco-graph-prebuilt` | `coco_graph.prebuilt` | `langgraph/libs/prebuilt` |
+| `libs/coco_v1` | `trustcoco` | `coco` | `langchain/libs/langchain_v1` |
+| `libs/partners/openai` | `trustcoco-openai` | `coco_openai` | `langchain/libs/partners/openai` |
+| `libs/partners/anthropic` | `trustcoco-anthropic` | `coco_anthropic` | `langchain/libs/partners/anthropic` |
+
+Folder names are upstream's after the rename, so the relative paths upstream writes
+between its libraries still resolve. Versions follow upstream's until we cut our own,
+so the ranges between libraries keep resolving.
+
+## What the rename does
+
+- Module names: `langchain_core` to `coco_core`, `langgraph` to `coco_graph`,
+  `langchain` to `coco`, `langchain_openai` to `coco_openai` and so on.
+- Distribution names: `langchain-*` to `trustcoco-*`, `langgraph-*` to
+  `trustcoco-graph-*`, `langchain` to `trustcoco`, `langgraph` to `trustcoco-graph`.
+- Class and constant names: `LangChain` to `Coco`, `LangGraph` to `CocoGraph`,
+  `LANGCHAIN_` to `COCO_`, `LANGGRAPH_` to `COCO_GRAPH_`.
+- Kept as written: every URL, the names of external packages (`langsmith`,
+  `langchain-protocol`), the upstream platform's API key variables, and the tracing
+  variables `langsmith` reads.
+- Dropped: dependencies and sources for libraries we do not fork, and extras naming
+  a `trustcoco-*` package we do not produce.
+- Our pyproject links and a generated README replace upstream's. `LICENSE` files are
+  never rewritten.
+
+## Removed
+
+| Path | Why |
+|---|---|
+| `coco_graph/pregel/remote.py`, `_remote_run_stream.py` and their two test files | `RemoteGraph` is the client for upstream's hosted platform. Its `langgraph-sdk` dependency installs upstream's `langchain-core` beside ours |
+| `coco_core/tracers/evaluation.py`, `coco_core/document_loaders/langsmith.py` | The evaluation tracer and the dataset loader call the upstream vendor's hosted service |
+| `coco_tests/_langsmith_plugin.py` | A pytest plugin that reports test runs to the same service |
+| Tests of the above, and the tracing interop tests in core and the graph | They exercise the vendor's tracing client |
+
+## Coco's own files
+
+`overlay/<folder>/` is copied over the renamed library before the patches run.
+
+| File | What it is |
+|---|---|
+| `core/coco_core/_protocol.py` | The agent streaming protocol types, vendored from `langchain-protocol` 0.0.19 (MIT). The rename points every import of that package here |
+| `core/coco_core/tracers/_context.py` | The tracing context, `traceable`, the environment helpers and the `RunTree` run record, with the names and shapes upstream imported from its vendor's tracing client. Nothing leaves the process. The script points those imports here (`TRACING_IMPORTS`) |
+| `core/coco_core/tracers/coco.py` | `CocoTracer`, replacing upstream's tracer at that path. Same interface, keeps runs in memory, sends nothing |
+
+## Patches
+
+Applied in order after the rename, from `patches/<folder>/`.
+
+| Patch | Why |
+|---|---|
+| `core/001-guard-import-names-trustcoco.patch` | `guard_import` told users to `pip install coco-core`. Coco modules ship in `trustcoco-*` distributions |
+| `coco_graph/001-base-user-without-platform-client.patch` | `runtime.py` imported `BaseUser` from `langgraph-sdk`. Defined locally as a runtime-checkable protocol, the same shape |
+| `coco_v1/001-test-distribution-name.patch` | A test lists dependency names, and the rename read its bare `"langgraph"` as a module |
+| `core/002-tracing-without-vendor-client.patch` | Drops the removed evaluation tracer and loader from the package exports, and one test of the vendor's `@traceable` |
+| `core/003-sys-info-distribution-names.patch` | Maps `coco` and `coco_*` module names to `trustcoco` and `trustcoco-*` distributions when `print_sys_info` queries `importlib.metadata` |
+| `coco_graph/002-tests-without-vendor-tracing.patch` | Drops two tests that trace a graph through the vendor's client |
+| `coco_graph/003-version-distribution-name.patch` | Looks up `trustcoco-graph` (with fallback to `coco-framework` and `__package__`) in `coco_graph/version.py` so `__version__` resolves when installed |
+| `coco_v1/002-provider-package-names.patch` | Maps `coco_*` provider module names to `trustcoco-*` in `init_chat_model` and `init_embeddings` install hints, and isolates the missing-`coco_anthropic` test when `trustcoco-anthropic` is installed |
+| `checkpoint/001-version-distribution-name.patch` | Queries the `trustcoco` distribution in `ensure_embeddings` version checks and install hints |
+| `checkpoint-postgres/001-version-distribution-name.patch` | Queries `trustcoco-graph` in the `checkpoint-postgres` compatibility version check |
+
+To make a patch: sync, copy the file aside, edit it under `libs/`, then
+`diff -u --label a/<path> --label b/<path> <copy> libs/<folder>/<path> > patches/<folder>/NNN-name.patch`
+with `<path>` relative to the library folder, and sync again to check it applies.
+
+## Upstream packages no longer needed
+
+Removed 28 Sep 2026. `import langsmith` and `import langchain_protocol` fail in the
+framework venv, and every library imports without them.
+
+- `langsmith`, the upstream vendor's tracing client, replaced by
+  `coco_core.tracers._context`.
+- `langchain-protocol`, the streaming message types, vendored as `coco_core._protocol`.
+
+## Tests
+
+`scripts/dev.sh setup` creates `.venv` and installs every library editable.
+`scripts/dev.sh test` runs each library's unit tests. Run on 28 Sep 2026 at the pins
+above, after the vendor client came out:
+
+| Library | Result | Notes |
+|---|---|---|
+| core | 2197 passed | 96 fewer than before the vendor client came out: the removed tracer, loader and interop tests |
+| standard-tests | 148 passed | the removed pytest plugin's tests account for the difference |
+| coco_graph | 1977 passed | upstream's four failures under the first rename are fixed by `coco_graph/001`; two tracing tests removed by `002`. One timing-dependent test failed once under `-n 16` on a loaded machine and passed on three reruns |
+| prebuilt | 228 passed | |
+| checkpoint | 159 passed, 16 skipped | the skipped tests need Redis |
+| checkpoint-sqlite | 128 passed | a few tests retry once, as upstream configures |
+| checkpoint-conformance | 1 passed | |
+| checkpoint-postgres | not run | needs a Postgres server; upstream starts one with Docker |
+| coco_v1 | 1219 passed | missing-`coco_anthropic` test isolated with `monkeypatch` in `coco_v1/002` so all 20 tests in `test_tool_emulator.py` pass when `trustcoco-anthropic` is installed |
+| partners/openai | 669 passed | token counting downloads tiktoken encodings |
+| partners/anthropic | 410 passed | |
+
+Before the fixes, the same failing tests were run against untouched upstream in the
+same environment to separate rename bugs from environment gaps.

coco_framework-0.1.0/libs/checkpoint/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 LangChain, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR 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.

coco_framework-0.1.0/libs/checkpoint/README.md

@@ -0,0 +1,5 @@
+# trustcoco-graph-checkpoint
+
+Checkpoint interfaces and the in-memory checkpointer for Coco graphs.
+
+Part of coco-framework. Generated from upstream by `scripts/sync_upstream.py`; see `UPSTREAM.md` for the pinned source and `NOTICE` for attribution. Do not edit files here by hand.

coco_framework-0.1.0/libs/checkpoint/coco_graph/cache/base/__init__.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from collections.abc import Mapping, Sequence
+from typing import Generic, TypeVar
+
+from coco_graph.checkpoint.serde.base import SerializerProtocol
+from coco_graph.checkpoint.serde.jsonplus import JsonPlusSerializer
+
+ValueT = TypeVar("ValueT")
+Namespace = tuple[str, ...]
+FullKey = tuple[Namespace, str]
+
+
+class BaseCache(ABC, Generic[ValueT]):
+    """Base class for a cache."""
+
+    serde: SerializerProtocol = JsonPlusSerializer(pickle_fallback=False)
+
+    def __init__(self, *, serde: SerializerProtocol | None = None) -> None:
+        """Initialize the cache with a serializer."""
+        self.serde = serde or self.serde
+
+    @abstractmethod
+    def get(self, keys: Sequence[FullKey]) -> dict[FullKey, ValueT]:
+        """Get the cached values for the given keys."""
+
+    @abstractmethod
+    async def aget(self, keys: Sequence[FullKey]) -> dict[FullKey, ValueT]:
+        """Asynchronously get the cached values for the given keys."""
+
+    @abstractmethod
+    def set(self, pairs: Mapping[FullKey, tuple[ValueT, int | None]]) -> None:
+        """Set the cached values for the given keys and TTLs."""
+
+    @abstractmethod
+    async def aset(self, pairs: Mapping[FullKey, tuple[ValueT, int | None]]) -> None:
+        """Asynchronously set the cached values for the given keys and TTLs."""
+
+    @abstractmethod
+    def clear(self, namespaces: Sequence[Namespace] | None = None) -> None:
+        """Delete the cached values for the given namespaces.
+        If no namespaces are provided, clear all cached values."""
+
+    @abstractmethod
+    async def aclear(self, namespaces: Sequence[Namespace] | None = None) -> None:
+        """Asynchronously delete the cached values for the given namespaces.
+        If no namespaces are provided, clear all cached values."""

coco_framework-0.1.0/libs/checkpoint/coco_graph/cache/memory/__init__.py

@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+import datetime
+import threading
+from collections.abc import Mapping, Sequence
+
+from coco_graph.cache.base import BaseCache, FullKey, Namespace, ValueT
+from coco_graph.checkpoint.serde.base import SerializerProtocol
+
+
+class InMemoryCache(BaseCache[ValueT]):
+    def __init__(self, *, serde: SerializerProtocol | None = None):
+        super().__init__(serde=serde)
+        self._cache: dict[Namespace, dict[str, tuple[str, bytes, float | None]]] = {}
+        self._lock = threading.RLock()
+
+    def get(self, keys: Sequence[FullKey]) -> dict[FullKey, ValueT]:
+        """Get the cached values for the given keys."""
+        with self._lock:
+            if not keys:
+                return {}
+            now = datetime.datetime.now(datetime.timezone.utc).timestamp()
+            values: dict[FullKey, ValueT] = {}
+            for ns_tuple, key in keys:
+                ns = Namespace(ns_tuple)
+                if ns in self._cache and key in self._cache[ns]:
+                    enc, val, expiry = self._cache[ns][key]
+                    if expiry is None or now < expiry:
+                        values[(ns, key)] = self.serde.loads_typed((enc, val))
+                    else:
+                        del self._cache[ns][key]
+            return values
+
+    async def aget(self, keys: Sequence[FullKey]) -> dict[FullKey, ValueT]:
+        """Asynchronously get the cached values for the given keys."""
+        return self.get(keys)
+
+    def set(self, keys: Mapping[FullKey, tuple[ValueT, int | None]]) -> None:
+        """Set the cached values for the given keys."""
+        with self._lock:
+            now = datetime.datetime.now(datetime.timezone.utc)
+            for (ns, key), (value, ttl) in keys.items():
+                if ttl is not None:
+                    delta = datetime.timedelta(seconds=ttl)
+                    expiry: float | None = (now + delta).timestamp()
+                else:
+                    expiry = None
+                if ns not in self._cache:
+                    self._cache[ns] = {}
+                self._cache[ns][key] = (
+                    *self.serde.dumps_typed(value),
+                    expiry,
+                )
+
+    async def aset(self, keys: Mapping[FullKey, tuple[ValueT, int | None]]) -> None:
+        """Asynchronously set the cached values for the given keys."""
+        self.set(keys)
+
+    def clear(self, namespaces: Sequence[Namespace] | None = None) -> None:
+        """Delete the cached values for the given namespaces.
+        If no namespaces are provided, clear all cached values."""
+        with self._lock:
+            if namespaces is None:
+                self._cache.clear()
+            else:
+                for ns in namespaces:
+                    if ns in self._cache:
+                        del self._cache[ns]
+
+    async def aclear(self, namespaces: Sequence[Namespace] | None = None) -> None:
+        """Asynchronously delete the cached values for the given namespaces.
+        If no namespaces are provided, clear all cached values."""
+        self.clear(namespaces)