trustgraph/docs/apis/api-librarian.md
cybermaggedon 89be656990
Release/v1.2 (#457)
* Bump setup.py versions for 1.1

* PoC MCP server (#419)

* Very initial MCP server PoC for TrustGraph

* Put service on port 8000

* Add MCP container and packages to buildout

* Update docs for API/CLI changes in 1.0 (#421)

* Update some API basics for the 0.23/1.0 API change

* Add MCP container push (#425)

* Add command args to the MCP server (#426)

* Host and port parameters

* Added websocket arg

* More docs

* MCP client support (#427)

- MCP client service
- Tool request/response schema
- API gateway support for mcp-tool
- Message translation for tool request & response
- Make mcp-tool using configuration service for information
  about where the MCP services are.

* Feature/react call mcp (#428)

Key Features

  - MCP Tool Integration: Added core MCP tool support with ToolClientSpec and ToolClient classes
  - API Enhancement: New mcp_tool method for flow-specific tool invocation
  - CLI Tooling: New tg-invoke-mcp-tool command for testing MCP integration
  - React Agent Enhancement: Fixed and improved multi-tool invocation capabilities
  - Tool Management: Enhanced CLI for tool configuration and management

Changes

  - Added MCP tool invocation to API with flow-specific integration
  - Implemented ToolClientSpec and ToolClient for tool call handling
  - Updated agent-manager-react to invoke MCP tools with configurable types
  - Enhanced CLI with new commands and improved help text
  - Added comprehensive documentation for new CLI commands
  - Improved tool configuration management

Testing

  - Added tg-invoke-mcp-tool CLI command for isolated MCP integration testing
  - Enhanced agent capability to invoke multiple tools simultaneously

* Test suite executed from CI pipeline (#433)

* Test strategy & test cases

* Unit tests

* Integration tests

* Extending test coverage (#434)

* Contract tests

* Testing embeedings

* Agent unit tests

* Knowledge pipeline tests

* Turn on contract tests

* Increase storage test coverage (#435)

* Fixing storage and adding tests

* PR pipeline only runs quick tests

* Empty configuration is returned as empty list, previously was not in response (#436)

* Update config util to take files as well as command-line text (#437)

* Updated CLI invocation and config model for tools and mcp (#438)

* Updated CLI invocation and config model for tools and mcp

* CLI anomalies

* Tweaked the MCP tool implementation for new model

* Update agent implementation to match the new model

* Fix agent tools, now all tested

* Fixed integration tests

* Fix MCP delete tool params

* Update Python deps to 1.2

* Update to enable knowledge extraction using the agent framework (#439)

* Implement KG extraction agent (kg-extract-agent)

* Using ReAct framework (agent-manager-react)
 
* ReAct manager had an issue when emitting JSON, which conflicts which ReAct manager's own JSON messages, so refactored ReAct manager to use traditional ReAct messages, non-JSON structure.
 
* Minor refactor to take the prompt template client out of prompt-template so it can be more readily used by other modules. kg-extract-agent uses this framework.

* Migrate from setup.py to pyproject.toml (#440)

* Converted setup.py to pyproject.toml

* Modern package infrastructure as recommended by py docs

* Install missing build deps (#441)

* Install missing build deps (#442)

* Implement logging strategy (#444)

* Logging strategy and convert all prints() to logging invocations

* Fix/startup failure (#445)

* Fix loggin startup problems

* Fix logging startup problems (#446)

* Fix logging startup problems (#447)

* Fixed Mistral OCR to use current API (#448)

* Fixed Mistral OCR to use current API

* Added PDF decoder tests

* Fix Mistral OCR ident to be standard pdf-decoder (#450)

* Fix Mistral OCR ident to be standard pdf-decoder

* Correct test

* Schema structure refactor (#451)

* Write schema refactor spec

* Implemented schema refactor spec

* Structure data mvp (#452)

* Structured data tech spec

* Architecture principles

* New schemas

* Updated schemas and specs

* Object extractor

* Add .coveragerc

* New tests

* Cassandra object storage

* Trying to object extraction working, issues exist

* Validate librarian collection (#453)

* Fix token chunker, broken API invocation (#454)

* Fix token chunker, broken API invocation (#455)

* Knowledge load utility CLI (#456)

* Knowledge loader

* More tests
2025-08-18 20:56:09 +01:00

9.1 KiB

TrustGraph Librarian API

This API provides document library management for TrustGraph. It handles document storage, metadata management, and processing orchestration using hybrid storage (MinIO for content, Cassandra for metadata) with multi-user support.

Request/response

Request

The request contains the following fields:

  • operation: The operation to perform (see operations below)
  • document_id: Document identifier (for document operations)
  • document_metadata: Document metadata object (for add/update operations)
    • id: Document identifier (required)
    • time: Unix timestamp in seconds as a float (required for add operations)
    • kind: MIME type of document (required, e.g., "text/plain", "application/pdf")
    • title: Document title (optional)
    • comments: Document comments (optional)
    • user: Document owner (required)
    • tags: Array of tags (optional)
    • metadata: Array of RDF triples (optional) - each triple has:
      • s: Subject with v (value) and e (is_uri boolean)
      • p: Predicate with v (value) and e (is_uri boolean)
      • o: Object with v (value) and e (is_uri boolean)
  • content: Document content as base64-encoded bytes (for add operations)
  • processing_id: Processing job identifier (for processing operations)
  • processing_metadata: Processing metadata object (for add-processing)
  • user: User identifier (required for most operations)
  • collection: Collection filter (optional for list operations)
  • criteria: Query criteria array (for filtering operations)

Response

The response contains the following fields:

  • error: Error information if operation fails
  • document_metadata: Single document metadata (for get operations)
  • content: Document content as base64-encoded bytes (for get-content)
  • document_metadatas: Array of document metadata (for list operations)
  • processing_metadatas: Array of processing metadata (for list-processing)

Document Operations

ADD-DOCUMENT - Add Document to Library

Request:

{
    "operation": "add-document",
    "document_metadata": {
        "id": "doc-123",
        "time": 1640995200.0,
        "kind": "application/pdf",
        "title": "Research Paper",
        "comments": "Important research findings",
        "user": "alice",
        "tags": ["research", "ai", "machine-learning"],
        "metadata": [
            {
                "s": {
                    "v": "http://example.com/doc-123",
                    "e": true
                },
                "p": {
                    "v": "http://purl.org/dc/elements/1.1/creator",
                    "e": true
                },
                "o": {
                    "v": "Dr. Smith",
                    "e": false
                }
            }
        ]
    },
    "content": "JVBERi0xLjQKMSAwIG9iago8PAovVHlwZSAvQ2F0YWxvZwovUGFnZXMgMiAwIFIKPj4KZW5kb2JqCg=="
}

Response:

{}

GET-DOCUMENT-METADATA - Get Document Metadata

Request:

{
    "operation": "get-document-metadata",
    "document_id": "doc-123",
    "user": "alice"
}

Response:

{
    "document_metadata": {
        "id": "doc-123",
        "time": 1640995200.0,
        "kind": "application/pdf",
        "title": "Research Paper",
        "comments": "Important research findings",
        "user": "alice",
        "tags": ["research", "ai", "machine-learning"],
        "metadata": [
            {
                "s": {
                    "v": "http://example.com/doc-123",
                    "e": true
                },
                "p": {
                    "v": "http://purl.org/dc/elements/1.1/creator",
                    "e": true
                },
                "o": {
                    "v": "Dr. Smith",
                    "e": false
                }
            }
        ]
    }
}

GET-DOCUMENT-CONTENT - Get Document Content

Request:

{
    "operation": "get-document-content",
    "document_id": "doc-123",
    "user": "alice"
}

Response:

{
    "content": "JVBERi0xLjQKMSAwIG9iago8PAovVHlwZSAvQ2F0YWxvZwovUGFnZXMgMiAwIFIKPj4KZW5kb2JqCg=="
}

LIST-DOCUMENTS - List User's Documents

Request:

{
    "operation": "list-documents",
    "user": "alice",
    "collection": "research"
}

Response:

{
    "document_metadatas": [
        {
            "id": "doc-123",
            "time": 1640995200.0,
            "kind": "application/pdf",
            "title": "Research Paper",
            "comments": "Important research findings",
            "user": "alice",
            "tags": ["research", "ai"]
        },
        {
            "id": "doc-124",
            "time": 1640995300.0,
            "kind": "text/plain",
            "title": "Meeting Notes",
            "comments": "Team meeting discussion",
            "user": "alice",
            "tags": ["meeting", "notes"]
        }
    ]
}

UPDATE-DOCUMENT - Update Document Metadata

Request:

{
    "operation": "update-document",
    "document_metadata": {
        "id": "doc-123",
        "time": 1640995500.0,
        "title": "Updated Research Paper",
        "comments": "Updated findings and conclusions",
        "user": "alice",
        "tags": ["research", "ai", "machine-learning", "updated"],
        "metadata": []
    }
}

Response:

{}

REMOVE-DOCUMENT - Remove Document

Request:

{
    "operation": "remove-document",
    "document_id": "doc-123",
    "user": "alice"
}

Response:

{}

Processing Operations

ADD-PROCESSING - Start Document Processing

Request:

{
    "operation": "add-processing",
    "processing_metadata": {
        "id": "proc-456",
        "document_id": "doc-123",
        "time": 1640995400.0,
        "flow": "pdf-extraction",
        "user": "alice",
        "collection": "research",
        "tags": ["extraction", "nlp"]
    }
}

Response:

{}

LIST-PROCESSING - List Processing Jobs

Request:

{
    "operation": "list-processing",
    "user": "alice",
    "collection": "research"
}

Response:

{
    "processing_metadatas": [
        {
            "id": "proc-456",
            "document_id": "doc-123",
            "time": 1640995400.0,
            "flow": "pdf-extraction",
            "user": "alice",
            "collection": "research",
            "tags": ["extraction", "nlp"]
        }
    ]
}

REMOVE-PROCESSING - Stop Processing Job

Request:

{
    "operation": "remove-processing",
    "processing_id": "proc-456",
    "user": "alice"
}

Response:

{}

REST service

The REST service is available at /api/v1/librarian and accepts the above request formats.

Websocket

Requests have a request object containing the operation fields. Responses have a response object containing the response fields.

Request:

{
    "id": "unique-request-id",
    "service": "librarian",
    "request": {
        "operation": "list-documents",
        "user": "alice"
    }
}

Response:

{
    "id": "unique-request-id",
    "response": {
        "document_metadatas": [...]
    },
    "complete": true
}

Pulsar

The Pulsar schema for the Librarian API is defined in Python code here:

https://github.com/trustgraph-ai/trustgraph/blob/master/trustgraph-base/trustgraph/schema/library.py

Default request queue: non-persistent://tg/request/librarian

Default response queue: non-persistent://tg/response/librarian

Request schema: trustgraph.schema.LibrarianRequest

Response schema: trustgraph.schema.LibrarianResponse

Python SDK

The Python SDK provides convenient access to the Librarian API:

from trustgraph.api.library import LibrarianClient

client = LibrarianClient()

# Add a document
with open("document.pdf", "rb") as f:
    content = f.read()
    
await client.add_document(
    doc_id="doc-123",
    title="Research Paper",
    content=content,
    user="alice",
    tags=["research", "ai"]
)

# Get document metadata
metadata = await client.get_document_metadata("doc-123", "alice")

# List documents
documents = await client.list_documents("alice", collection="research")

# Start processing
await client.add_processing(
    processing_id="proc-456",
    document_id="doc-123",
    flow="pdf-extraction",
    user="alice"
)

Features

  • Hybrid Storage: MinIO for content, Cassandra for metadata
  • Multi-user Support: User-based document ownership and access control
  • Rich Metadata: RDF-style metadata triples and tagging system
  • Processing Integration: Automatic triggering of document processing workflows
  • Content Types: Support for multiple document formats (PDF, text, etc.)
  • Collection Management: Optional document grouping by collection
  • Metadata Search: Query documents by metadata criteria

Use Cases

  • Document Management: Store and organize documents with rich metadata
  • Knowledge Extraction: Process documents to extract structured knowledge
  • Research Libraries: Manage collections of research papers and documents
  • Content Processing: Orchestrate document processing workflows
  • Multi-tenant Systems: Support multiple users with isolated document libraries