coco-framework 0.1.0__py3-none-any.whl

coco_anthropic/data/__init__.py

@@ -0,0 +1 @@
+"""Model profile data. All edits should be made in profile_augmentations.toml."""

coco_anthropic/data/_profiles.py

@@ -0,0 +1,491 @@
+"""Auto-generated model profiles.
+
+DO NOT EDIT THIS FILE MANUALLY.
+This file is generated by the trustcoco-profiles CLI tool.
+
+It contains data derived from the models.dev project.
+
+Source: https://github.com/sst/models.dev
+License: MIT License
+
+To update these data, refer to the instructions here:
+
+https://docs.langchain.com/oss/python/langchain/models#updating-or-overwriting-profile-data
+"""
+
+from typing import Any
+
+_PROFILES: dict[str, dict[str, Any]] = {
+    "claude-fable-5": {
+        "name": "Claude Fable 5",
+        "release_date": "2026-06-07",
+        "last_updated": "2026-06-09",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-fable-5-1": {
+        "name": "Claude Fable 5.1",
+        "release_date": "2026-09-01",
+        "last_updated": "2026-09-01",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-haiku-4-5": {
+        "name": "Claude Haiku 4.5 (latest)",
+        "release_date": "2025-10-15",
+        "last_updated": "2025-10-15",
+        "open_weights": False,
+        "max_input_tokens": 200000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+    },
+    "claude-haiku-4-5-20251001": {
+        "name": "Claude Haiku 4.5",
+        "release_date": "2025-10-15",
+        "last_updated": "2025-10-15",
+        "open_weights": False,
+        "max_input_tokens": 200000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": False,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+    },
+    "claude-opus-4-5": {
+        "name": "Claude Opus 4.5 (latest)",
+        "release_date": "2025-11-24",
+        "last_updated": "2025-11-24",
+        "open_weights": False,
+        "max_input_tokens": 200000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-4-5-20251101": {
+        "name": "Claude Opus 4.5",
+        "release_date": "2025-11-24",
+        "last_updated": "2025-11-01",
+        "open_weights": False,
+        "max_input_tokens": 200000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": False,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-4-6": {
+        "name": "Claude Opus 4.6",
+        "release_date": "2026-02-04",
+        "last_updated": "2026-03-13",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-4-7": {
+        "name": "Claude Opus 4.7",
+        "release_date": "2026-04-14",
+        "last_updated": "2026-04-16",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-4-8": {
+        "name": "Claude Opus 4.8",
+        "release_date": "2026-05-28",
+        "last_updated": "2026-05-28",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-5": {
+        "name": "Claude Opus 5",
+        "release_date": "2026-07-24",
+        "last_updated": "2026-07-24",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-opus-5-5": {
+        "name": "Claude Opus 5.5",
+        "release_date": "2026-09-22",
+        "last_updated": "2026-09-22",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "medium",
+    },
+    "claude-sonnet-4-5": {
+        "name": "Claude Sonnet 4.5 (latest)",
+        "release_date": "2025-09-29",
+        "last_updated": "2025-09-29",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+    },
+    "claude-sonnet-4-5-20250929": {
+        "name": "Claude Sonnet 4.5",
+        "release_date": "2025-09-29",
+        "last_updated": "2025-09-29",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 64000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": False,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+    },
+    "claude-sonnet-4-6": {
+        "name": "Claude Sonnet 4.6",
+        "release_date": "2026-02-17",
+        "last_updated": "2026-03-13",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": True,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+    "claude-sonnet-5": {
+        "name": "Claude Sonnet 5",
+        "release_date": "2026-06-29",
+        "last_updated": "2026-06-30",
+        "open_weights": False,
+        "max_input_tokens": 1000000,
+        "max_output_tokens": 128000,
+        "text_inputs": True,
+        "image_inputs": True,
+        "audio_inputs": False,
+        "pdf_inputs": True,
+        "video_inputs": False,
+        "text_outputs": True,
+        "image_outputs": False,
+        "audio_outputs": False,
+        "video_outputs": False,
+        "reasoning_output": True,
+        "tool_calling": True,
+        "structured_output": True,
+        "attachment": True,
+        "temperature": False,
+        "image_url_inputs": True,
+        "pdf_tool_message": True,
+        "image_tool_message": True,
+        "tool_call_streaming": True,
+        "reasoning_effort_levels": [
+            "low",
+            "medium",
+            "high",
+            "xhigh",
+            "max",
+        ],
+        "reasoning_effort_default": "high",
+    },
+}

coco_anthropic/data/profile_augmentations.toml

@@ -0,0 +1,69 @@
+provider = "anthropic"
+
+[overrides]
+image_url_inputs = true
+pdf_inputs = true
+pdf_tool_message = true
+image_tool_message = true
+structured_output = false
+tool_call_streaming = true
+
+[overrides."claude-haiku-4-5"]
+structured_output = true
+
+[overrides."claude-sonnet-4-5"]
+structured_output = true
+
+[overrides."claude-sonnet-4-6"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-4-5"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-4-5-20251101"]
+reasoning_effort_levels = ["low", "medium", "high"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-4-6"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-4-7"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-4-8"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-5"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-opus-5-5"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "medium"
+
+[overrides."claude-sonnet-5"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-fable-5"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"
+
+[overrides."claude-fable-5-1"]
+structured_output = true
+reasoning_effort_levels = ["low", "medium", "high", "xhigh", "max"]
+reasoning_effort_default = "high"

coco_anthropic/experimental.py

@@ -0,0 +1,140 @@
+"""Experimental tool-calling support for Anthropic chat models."""
+
+from __future__ import annotations
+
+import json
+from typing import (
+    Any,
+)
+
+SYSTEM_PROMPT_FORMAT = """In this environment you have access to a set of tools you can use to answer the user's question.
+
+You may call them like this:
+<function_calls>
+<invoke>
+<tool_name>$TOOL_NAME</tool_name>
+<parameters>
+<$PARAMETER_NAME>$PARAMETER_VALUE</$PARAMETER_NAME>
+...
+</parameters>
+</invoke>
+</function_calls>
+
+Here are the tools available:
+<tools>
+{formatted_tools}
+</tools>"""  # noqa: E501
+
+TOOL_FORMAT = """<tool_description>
+<tool_name>{tool_name}</tool_name>
+<description>{tool_description}</description>
+<parameters>
+{formatted_parameters}
+</parameters>
+</tool_description>"""
+
+TOOL_PARAMETER_FORMAT = """<parameter>
+<name>{parameter_name}</name>
+<type>{parameter_type}</type>
+<description>{parameter_description}</description>
+</parameter>"""
+
+
+def _get_type(parameter: dict[str, Any]) -> str:
+    if "type" in parameter:
+        return parameter["type"]
+    if "anyOf" in parameter:
+        return json.dumps({"anyOf": parameter["anyOf"]})
+    if "allOf" in parameter:
+        return json.dumps({"allOf": parameter["allOf"]})
+    return json.dumps(parameter)
+
+
+def get_system_message(tools: list[dict]) -> str:
+    """Generate a system message that describes the available tools."""
+    tools_data: list[dict] = [
+        {
+            "tool_name": tool["name"],
+            "tool_description": tool["description"],
+            "formatted_parameters": "\n".join(
+                [
+                    TOOL_PARAMETER_FORMAT.format(
+                        parameter_name=name,
+                        parameter_type=_get_type(parameter),
+                        parameter_description=parameter.get("description"),
+                    )
+                    for name, parameter in tool["parameters"]["properties"].items()
+                ],
+            ),
+        }
+        for tool in tools
+    ]
+    tools_formatted = "\n".join(
+        [
+            TOOL_FORMAT.format(
+                tool_name=tool["tool_name"],
+                tool_description=tool["tool_description"],
+                formatted_parameters=tool["formatted_parameters"],
+            )
+            for tool in tools_data
+        ],
+    )
+    return SYSTEM_PROMPT_FORMAT.format(formatted_tools=tools_formatted)
+
+
+def _xml_to_dict(t: Any) -> str | dict[str, Any]:
+    # Base case: If the element has no children, return its text or an empty string.
+    if len(t) == 0:
+        return t.text or ""
+
+    # Recursive case: The element has children. Convert them into a dictionary.
+    d: dict[str, Any] = {}
+    for child in t:
+        if child.tag not in d:
+            d[child.tag] = _xml_to_dict(child)
+        else:
+            # Handle multiple children with the same tag
+            if not isinstance(d[child.tag], list):
+                d[child.tag] = [d[child.tag]]  # Convert existing entry into a list
+            d[child.tag].append(_xml_to_dict(child))
+    return d
+
+
+def _xml_to_function_call(invoke: Any, tools: list[dict]) -> dict[str, Any]:
+    name = invoke.find("tool_name").text
+    arguments = _xml_to_dict(invoke.find("parameters"))
+
+    # make list elements in arguments actually lists
+    filtered_tools = [tool for tool in tools if tool["name"] == name]
+    if len(filtered_tools) > 0 and not isinstance(arguments, str):
+        tool = filtered_tools[0]
+        for key, value in arguments.items():
+            if (
+                key in tool["parameters"]["properties"]
+                and "type" in tool["parameters"]["properties"][key]
+            ):
+                if tool["parameters"]["properties"][key][
+                    "type"
+                ] == "array" and not isinstance(value, list):
+                    arguments[key] = [value]
+                if (
+                    tool["parameters"]["properties"][key]["type"] != "object"
+                    and isinstance(value, dict)
+                    and len(value.keys()) == 1
+                ):
+                    arguments[key] = next(iter(value.values()))
+
+    return {
+        "function": {
+            "name": name,
+            "arguments": json.dumps(arguments),
+        },
+        "type": "function",
+    }
+
+
+def _xml_to_tool_calls(elem: Any, tools: list[dict]) -> list[dict[str, Any]]:
+    """Convert an XML element and its children into a dictionary of dictionaries."""
+    invokes = elem.findall("invoke")
+
+    return [_xml_to_function_call(invoke, tools) for invoke in invokes]