wehub-resource-sync
c889a57b6b
Test Suites / Build CI Environment (push) Has been cancelled
Test Suites / Basic Tests (push) Has been cancelled
Test Suites / End-to-End Tests (push) Has been cancelled
Test Suites / CLI Tests (push) Has been cancelled
Test Suites / Slow End-to-End Tests (push) Has been cancelled
Test Suites / Graph Database Tests (push) Has been cancelled
Test Suites / Vector DB Tests (push) Has been cancelled
Test Suites / Temporal Graph Test (push) Has been cancelled
Test Suites / Search Test on Different DBs (push) Has been cancelled
Test Suites / Example Tests (push) Has been cancelled
Test Suites / Notebook Tests (push) Has been cancelled
Test Suites / OS and Python Tests Ubuntu (push) Has been cancelled
Test Suites / OS and Python Tests Extended (push) Has been cancelled
Test Suites / LLM Test Suite (push) Has been cancelled
Test Suites / S3 File Storage Test (push) Has been cancelled
Test Suites / Run Integration Tests (push) Has been cancelled
Test Suites / MCP Tests (push) Has been cancelled
Test Suites / Docker Compose Test (push) Has been cancelled
Test Suites / Docker CI test (push) Has been cancelled
Test Suites / Relational DB Migration Tests (push) Has been cancelled
Test Suites / Distributed Cognee Test (push) Has been cancelled
Test Suites / DB Examples Tests (push) Has been cancelled
Test Suites / Test Completion Status (push) Has been cancelled
Test Suites / Claude Code Review (push) Has been cancelled
Test Suites / basic checks (push) Has been cancelled
build | Build and Push Cognee MCP Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
build | Build and Push Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.11) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.12) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (kuzu, kuzu) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (neo4j, neo4j) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Examples (push) Has been cancelled
Weighted Edges Tests / Code Quality for Weighted Edges (push) Has been cancelled
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
"""Progressive-disclosure primitive: return a skill's procedure body by name.
|
|
|
|
The AgenticRetriever loads only Skill descriptions into the system prompt; the
|
|
full procedure is retrieved by the LLM calling this tool when it decides a
|
|
skill is relevant. Active skills for the current turn live in a ContextVar so
|
|
the handler does not need another graph round-trip.
|
|
"""
|
|
|
|
from typing import Any, Dict
|
|
|
|
from cognee.modules.engine.models import Tool
|
|
from cognee.modules.tools.context import active_skills_var, opened_skills_var
|
|
from cognee.modules.tools.errors import ToolInvocationError
|
|
from cognee.modules.tools.registry import register_builtin_tool
|
|
|
|
|
|
MAX_SKILL_BODY_CHARS = 12_000
|
|
|
|
TOOL = Tool(
|
|
name="load_skill",
|
|
description=(
|
|
"Load the full procedure body of a named skill. Use this when the skill "
|
|
"description suggests the skill applies to the current task."
|
|
),
|
|
input_schema={
|
|
"type": "object",
|
|
"properties": {"name": {"type": "string", "description": "Name of the skill to load."}},
|
|
"required": ["name"],
|
|
},
|
|
handler_ref="cognee.modules.tools.builtin.load_skill.handler",
|
|
permission_required="read",
|
|
readonly_hint=True,
|
|
)
|
|
|
|
|
|
async def handler(args: Dict[str, Any], **_) -> str:
|
|
name = args.get("name")
|
|
if not name:
|
|
raise ToolInvocationError("load_skill requires a 'name' argument")
|
|
|
|
skills = active_skills_var.get()
|
|
skill = skills.get(name)
|
|
if skill is None:
|
|
available = ", ".join(sorted(skills)) or "(none in scope)"
|
|
raise ToolInvocationError(
|
|
f"Skill {name!r} is not in the active skill set. Available: {available}"
|
|
)
|
|
|
|
opened = opened_skills_var.get()
|
|
if opened is not None:
|
|
opened.add(skill.name)
|
|
|
|
body = skill.procedure or "(this skill has no procedure body)"
|
|
if len(body) > MAX_SKILL_BODY_CHARS:
|
|
body = body[:MAX_SKILL_BODY_CHARS] + "\n... [truncated]"
|
|
return f"# Skill: {skill.name}\n{body}"
|
|
|
|
|
|
register_builtin_tool(TOOL)
|