doc-store-client-dev 0.1.185__py3-none-any.whl

doc_store_client/__init__.py

@@ -0,0 +1,194 @@
+"""Public API for the transport-neutral doc-store client package."""
+
+from importlib.metadata import PackageNotFoundError as _PackageNotFoundError
+from importlib.metadata import version as _distribution_version
+from pathlib import Path as _Path
+import tomllib as _tomllib
+
+from .client import (
+    DOC_STORE_COMMANDS,
+    DocStoreClient,
+    DocStoreClientError,
+    DocStoreNotFoundError,
+    FileAlreadyLockedError,
+    LockServiceUnavailableError,
+    SessionNotFoundError,
+)
+from .owned_document_cleanup import OwnedDocumentCleanup, OwnedDocumentCleanupError
+from .text_verification import text_verification_phrase
+from .models import (
+    BackupEmergencyResult,
+    BackupInspectFile,
+    BackupInspectResult,
+    BackupPoint,
+    BackupPointsResult,
+    BackupRecoverCatalogResult,
+    BackupRestoreResult,
+    BackupRunResult,
+    BackupStatusResult,
+    ChapterGetRequest,
+    ChapterGetResult,
+    ChapterTextGetRequest,
+    ChunkHistoryRequest,
+    ChunkHistoryResult,
+    ChunkVersionAddRequest,
+    ChunkVersionAddResult,
+    ChunkVersionDeleteRequest,
+    ChunkVersionDeleteResult,
+    ChunkVersionDiffRequest,
+    ChunkVersionDiffResult,
+    ChunkVersionGetRequest,
+    ChunkVersionGetResult,
+    ChunkVersionItem,
+    ChunkVersionListRequest,
+    ChunkVersionListResult,
+    ChunkVersionRestoreRequest,
+    ChunkVersionRestoreResult,
+    ChunkVersionRetireRequest,
+    ChunkVersionRetireResult,
+    ChunkVersionSetCurrentRequest,
+    ChunkVersionSetCurrentResult,
+    ChunkVersionTextMutationRequest,
+    ChunkVersionUpdateRequest,
+    ChunkVersionUpdateResult,
+    DocumentChunkRequest,
+    DocumentChunkResult,
+    DocumentCreateRequest,
+    DocumentCreateResult,
+    DocumentDeleteRequest,
+    DocumentDeleteResult,
+    DocumentGetRequest,
+    DocumentGetResult,
+    DocumentRebindRequest,
+    DocumentRebindResult,
+    DocumentUpdateRequest,
+    DocumentUpdateResult,
+    DocumentWriteRequest,
+    DocumentWriteResult,
+    EntityGetRequest,
+    EntityGetResult,
+    EntityTextGetRequest,
+    EntityTextGetResult,
+    EntityIdsRequest,
+    EntityLifecycleResult,
+    EntityListRequest,
+    EntityListResult,
+    EntityOwnerTreeRequest,
+    EntityOwnerTreeResult,
+    EntityReferencesRequest,
+    EntityReferencesResult,
+    OperationState,
+    ParagraphGetByNumberRequest,
+    ParagraphGetByNumberResult,
+    ParagraphGetRequest,
+    ParagraphGetResult,
+    ProcessingStatusRequest,
+    ProcessingStatusResult,
+    RankedSearchHit,
+    RetrievalRequest,
+    RetrievalResult,
+    SearchResult,
+    ServerError,
+    SemanticChunkMetadataUpdateRequest,
+    SemanticChunkMetadataUpdateResult,
+    SourceFileReconstructRequest,
+    TextReconstructionRequest,
+    TextReconstructionResult,
+)
+
+try:
+    __version__ = _distribution_version("doc-store-client")
+except _PackageNotFoundError:
+    _pyproject = _Path(__file__).resolve().parents[2] / "pyproject.toml"
+    __version__ = _tomllib.loads(_pyproject.read_text(encoding="utf-8"))["project"]["version"]
+
+# The package includes ``py.typed`` in its distribution metadata.
+__all__ = [
+    "BackupEmergencyResult",
+    "BackupInspectFile",
+    "BackupInspectResult",
+    "BackupPoint",
+    "BackupPointsResult",
+    "BackupRecoverCatalogResult",
+    "BackupRestoreResult",
+    "BackupRunResult",
+    "BackupStatusResult",
+    "ChapterGetRequest",
+    "ChapterGetResult",
+    "ChapterTextGetRequest",
+    "ChunkHistoryRequest",
+    "ChunkHistoryResult",
+    "ChunkVersionAddRequest",
+    "ChunkVersionAddResult",
+    "ChunkVersionDeleteRequest",
+    "ChunkVersionDeleteResult",
+    "ChunkVersionDiffRequest",
+    "ChunkVersionDiffResult",
+    "ChunkVersionGetRequest",
+    "ChunkVersionGetResult",
+    "ChunkVersionItem",
+    "ChunkVersionListRequest",
+    "ChunkVersionListResult",
+    "ChunkVersionRestoreRequest",
+    "ChunkVersionRestoreResult",
+    "ChunkVersionRetireRequest",
+    "ChunkVersionRetireResult",
+    "ChunkVersionSetCurrentRequest",
+    "ChunkVersionSetCurrentResult",
+    "ChunkVersionTextMutationRequest",
+    "ChunkVersionUpdateRequest",
+    "ChunkVersionUpdateResult",
+    "DOC_STORE_COMMANDS",
+    "DocStoreClient",
+    "DocStoreClientError",
+    "DocStoreNotFoundError",
+    "FileAlreadyLockedError",
+    "LockServiceUnavailableError",
+    "SessionNotFoundError",
+    "OwnedDocumentCleanup",
+    "OwnedDocumentCleanupError",
+    "DocumentCreateRequest",
+    "DocumentCreateResult",
+    "DocumentChunkRequest",
+    "DocumentChunkResult",
+    "DocumentDeleteRequest",
+    "DocumentDeleteResult",
+    "DocumentGetRequest",
+    "DocumentGetResult",
+    "DocumentRebindRequest",
+    "DocumentRebindResult",
+    "DocumentUpdateRequest",
+    "DocumentUpdateResult",
+    "DocumentWriteRequest",
+    "DocumentWriteResult",
+    "EntityGetRequest",
+    "EntityGetResult",
+    "EntityTextGetRequest",
+    "EntityTextGetResult",
+    "EntityIdsRequest",
+    "EntityLifecycleResult",
+    "EntityListRequest",
+    "EntityListResult",
+    "EntityOwnerTreeRequest",
+    "EntityOwnerTreeResult",
+    "EntityReferencesRequest",
+    "EntityReferencesResult",
+    "OperationState",
+    "ParagraphGetByNumberRequest",
+    "ParagraphGetByNumberResult",
+    "ParagraphGetRequest",
+    "ParagraphGetResult",
+    "ProcessingStatusRequest",
+    "ProcessingStatusResult",
+    "RankedSearchHit",
+    "RetrievalRequest",
+    "RetrievalResult",
+    "SearchResult",
+    "ServerError",
+    "SemanticChunkMetadataUpdateRequest",
+    "SemanticChunkMetadataUpdateResult",
+    "SourceFileReconstructRequest",
+    "TextReconstructionRequest",
+    "TextReconstructionResult",
+    "text_verification_phrase",
+]

doc_store_client/cli.py

@@ -0,0 +1,192 @@
+"""Command line entrypoint for the installed doc-store client."""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+import os
+from pathlib import Path
+from typing import Any
+from urllib.parse import urlsplit
+
+
+class _AdapterBridge:
+    def __init__(self, client: Any) -> None:
+        self._client = client
+
+    async def execute_command_unified(
+        self, command: str, params: dict[str, Any], **kwargs: Any
+    ) -> Any:
+        dispatch = getattr(self._client, "execute_command_unified", None)
+        if dispatch is None:
+            dispatch = getattr(self._client, "execute_command", None)
+        if dispatch is None:
+            raise TypeError("configured adapter exposes no supported command dispatch method")
+        result = await dispatch(command, params, **kwargs)
+        if isinstance(result, dict) and "error" in result:
+            return {
+                "success": False,
+                "code": "SERVER_ERROR",
+                "message": str(result["error"]),
+                "details": {},
+            }
+        return result
+
+    async def wait_for_job(self, job_id: str, **kwargs: Any) -> Any:
+        return await self._client.wait_for_job(job_id, **kwargs)
+
+    async def fetch_command_session_result(self, job_id: str) -> Any:
+        return await self._client.fetch_command_session_result(job_id)
+
+
+def _config_path(explicit: str | None) -> Path | None:
+    if explicit:
+        return Path(explicit).expanduser()
+    configured = os.environ.get("DOC_STORE_CLIENT_CONFIG")
+    if configured:
+        return Path(configured).expanduser()
+    current = Path.cwd().resolve()
+    for directory in (current, *current.parents):
+        candidate = directory / "doc-store-client.json"
+        if candidate.is_file():
+            return candidate
+    return None
+
+
+def _load_config(args: argparse.Namespace) -> dict[str, Any]:
+    path = _config_path(args.config)
+    config: dict[str, Any] = {}
+    if path is not None:
+        try:
+            value = json.loads(path.read_text(encoding="utf-8"))
+        except (OSError, json.JSONDecodeError) as exc:
+            raise ValueError(f"Unable to load configuration: {exc}") from exc
+        if not isinstance(value, dict):
+            raise ValueError("Configuration must be a JSON object")
+        config = value
+    if args.server_url:
+        config["server_url"] = args.server_url
+    elif os.environ.get("DOC_STORE_SERVER_URL"):
+        config.setdefault("server_url", os.environ["DOC_STORE_SERVER_URL"])
+    if not config.get("server_url"):
+        raise ValueError(
+            "No server URL configured; use --config, --server-url, "
+            "DOC_STORE_CLIENT_CONFIG, or DOC_STORE_SERVER_URL"
+        )
+    return config
+
+
+def _params(text: str | None) -> dict[str, Any]:
+    if text is None:
+        return {}
+    try:
+        value = json.loads(text)
+    except json.JSONDecodeError as exc:
+        raise ValueError(f"--params must be a JSON object: {exc.msg}") from exc
+    if not isinstance(value, dict):
+        raise ValueError("--params must be a JSON object")
+    return value
+
+
+async def _execute(args: argparse.Namespace) -> Any:
+    from mcp_proxy_adapter.client.jsonrpc_client import JsonRpcClient
+
+    from doc_store_client.client import DocStoreClient
+
+    config = _load_config(args)
+    parsed = urlsplit(str(config["server_url"]))
+    if not parsed.hostname:
+        raise ValueError("server_url must include a host")
+    protocol = str(config.get("protocol") or parsed.scheme or "http")
+    port = int(
+        config.get("port")
+        or parsed.port
+        or (443 if protocol in {"https", "mtls"} else 80)
+    )
+    transport = JsonRpcClient(
+        protocol=protocol,
+        host=str(config.get("host") or parsed.hostname),
+        port=port,
+        token_header=config.get("token_header"),
+        token=config.get("token"),
+        cert=config.get("cert"),
+        key=config.get("key"),
+        ca=config.get("ca"),
+        check_hostname=config.get("check_hostname"),
+        timeout=float(config["timeout"]) if config.get("timeout") is not None else None,
+    )
+
+    async def emit_event(event: dict[str, Any]) -> None:
+        print(
+            json.dumps(event, ensure_ascii=False, separators=(",", ":")),
+            file=os.sys.stderr,
+            flush=True,
+        )
+
+    client = DocStoreClient(_AdapterBridge(transport), event_callback=emit_event)
+    try:
+        if args.action == "call":
+            return await client.call(args.command, _params(args.params))
+        params = _params(args.params)
+        if args.file:
+            path = Path(args.file)
+            try:
+                payload = path.read_text(encoding="utf-8")
+            except OSError as exc:
+                raise ValueError(f"Unable to read --file: {exc}") from exc
+            params["raw_text"] = payload
+        return await client.call(args.action, params)
+    finally:
+        await transport.close()
+
+
+def _parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="doc-store-client",
+        description="Execute doc-store server commands through the installed client.",
+    )
+    parser.add_argument("--config", help="JSON config path; defaults to nearest doc-store-client.json")
+    parser.add_argument("--server-url", help="Override the configured server URL")
+    sub = parser.add_subparsers(dest="action", required=True)
+    call = sub.add_parser("call", help="Call any server command with a JSON object")
+    call.add_argument("command", help="Server command name")
+    call.add_argument("--params", help="JSON object passed to the command")
+    for name in ("document-create", "document-update", "import"):
+        command = sub.add_parser(name, help=f"Run {name} through the client")
+        command.set_defaults(action={"document-create": "document_create", "document-update": "document_update", "import": "document_import"}[name])
+        command.add_argument("--params", help="JSON object passed to the command")
+        command.add_argument("--file", help="UTF-8 text file to use as the document source")
+    return parser
+
+
+def main() -> int:
+    args = _parser().parse_args()
+    try:
+        result = asyncio.run(_execute(args))
+    except Exception as exc:
+        message = str(exc)
+        lowered = message.lower()
+        code = "CLI_ERROR"
+        details: Any = {}
+        server_error = getattr(exc, "error", None)
+        if server_error is not None:
+            code = str(getattr(server_error, "code", "SERVER_ERROR"))
+            message = str(getattr(server_error, "message", message))
+            details = getattr(server_error, "details", {}) or {}
+        if "no module named" in lowered:
+            code = "CLIENT_RUNTIME_INCOMPATIBLE"
+        print(
+            json.dumps(
+                {"code": code, "message": message, "details": details},
+                ensure_ascii=False,
+            ),
+            file=os.sys.stderr,
+        )
+        return 2
+    print(json.dumps(result, ensure_ascii=False, indent=2, default=str))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())