alix-doc-vault-tool 0.1.0__tar.gz

alix_doc_vault_tool-0.1.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: alix-doc-vault-tool
+Version: 0.1.0
+Summary: Documents Vault MCP tool for Alix. Fetches files, files per estate and downloads them into temporary folder for next processing
+Requires-Python: >=3.11
+Requires-Dist: mcp>=1.9.0
+Requires-Dist: httpx>=0.27
+Provides-Extra: dev
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"

alix_doc_vault_tool-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["setuptools>=61", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "alix-doc-vault-tool"
+version = "0.1.0"
+description = "Documents Vault MCP tool for Alix. Fetches files, files per estate and downloads them into temporary folder for next processing"
+requires-python = ">=3.11"
+dependencies = [
+  "mcp>=1.9.0",
+  "httpx>=0.27",
+]
+
+[project.optional-dependencies]
+dev = ["build", "twine"]
+
+[project.scripts]
+alix-doc-vault-tool = "alix_doc_vault_tool.mcp_server:main"
+
+[tool.setuptools.packages.find]
+where = ["src"]

alix_doc_vault_tool-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool/__init__.py

@@ -0,0 +1,12 @@
+__version__ = "0.1.0"
+
+from alix_doc_vault_tool.mcp_server import (
+    run_get_estate_documents,
+    run_get_file,
+)
+
+__all__ = [
+    "__version__",
+    "run_get_estate_documents",
+    "run_get_file",
+]

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool/__main__.py

@@ -0,0 +1,6 @@
+"""python -m alix_doc_vault_tool -> stdio MCP."""
+
+from alix_doc_vault_tool.mcp_server import main
+
+if __name__ == "__main__":
+    main()

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool/mcp_server.py

@@ -0,0 +1,348 @@
+"""Doc Vault MCP server for retrieving documents from the Alix Doc Vault.
+
+Env vars:
+    ALIX_API_URL          required. Base URL (e.g. http://localhost:8080) or full
+                          /graphql URL.
+    ALIX_API_KEY          required. Master token sent as the Authorization header.
+    ALIX_GRAPHQL_TIMEOUT  optional. Request timeout in seconds (default 30).
+    FILE_DOWNLOAD_TIMEOUT  optional. Request timeout in seconds for downloading files (default 30).
+"""
+
+from __future__ import annotations
+
+import json
+import os
+from typing import Any, Literal, TypedDict
+
+import httpx
+from mcp.server.fastmcp import FastMCP
+
+SERVER_NAME: str = "alix-doc-vault-tool"
+
+EstateDocumentType = Literal[
+    "DecedentDeathCertificate",
+    "PreDeceasedContactDeathCertificate",
+    "Trust",
+    "Will",
+    "AmendmentToTrust",
+    "Codicil",
+    "CustomerID",
+    "DecedentID",
+    "FuneralBillReceipt",
+    "FamilyMemberDeathCertificate",
+    "NAICReport",
+    "SocialSecurityEarningsReport",
+    "BondCertificate",
+    "AuthorityDocument",
+    "BankStatement",
+    "Other",
+]
+
+FileSource = Literal[
+    "Api",
+    "CustomerAppUpload",
+    "SpecialistEmUpload",
+    "ScanBox",
+    "SystemGenerated",
+    "EmailIntake",
+]
+
+_FILE_SELECTION = """
+id
+key
+fileName
+fileSource
+mime
+size
+public
+downloadUrl
+folder {
+  id
+  key
+  materializedPath
+  name
+}
+"""
+
+_GET_ESTATE_DOCUMENTS_QUERY = f"""
+query GetEstateDocuments($where: EstateWhereUniqueInput!) {{
+  estate(where: $where) {{
+    id
+    documents {{
+      id
+      documentType
+      files {{
+        file {{
+          {_FILE_SELECTION}
+        }}
+      }}
+    }}
+  }}
+}}
+"""
+
+_GET_FILE_QUERY = f"""
+query GetFile($where: FileWhereUniqueInput!) {{
+  file(where: $where) {{
+    id
+    {_FILE_SELECTION}
+  }}
+}}
+"""
+
+class Folder(TypedDict):
+    id: str
+    key: str | None
+    materializedPath: str | None
+    name: str
+
+class File(TypedDict):
+    id: str
+    key: str
+    fileName: str
+    fileSource: FileSource | None
+    mime: str
+    size: int
+    public: bool
+    downloadUrl: str | None
+    folder: Folder
+
+def  _resolve_endpoint(raw: str) -> str:
+    url = raw.rstrip("/")
+    if url.endswith("/graphql"):
+        return url
+    return f"{url}/graphql"
+
+
+def _read_alix_api_graphql_timeout() -> float:
+    raw = os.environ.get("ALIX_GRAPHQL_TIMEOUT")
+    if not raw:
+        return 30.0
+    try:
+        return float(raw)
+    except ValueError:
+        return 30.0
+
+def _read_file_download_timeout() -> float:
+    raw = os.environ.get("FILE_DOWNLOAD_TIMEOUT")
+    if not raw:
+        return 30.0
+    try:
+        return float(raw)
+    except ValueError:
+        return 30.0
+
+
+def _graphql_request(
+    *,
+    query: str,
+    variables: dict[str, Any],
+    operation_name: str,
+) -> dict[str, Any]:
+    api_url_raw = os.environ.get("ALIX_API_URL")
+    api_key = os.environ.get("ALIX_API_KEY")
+    if not api_url_raw:
+        return {"errors": [{"message": "ALIX_API_URL is not set."}]}
+    if not api_key:
+        return {"errors": [{"message": "ALIX_API_KEY is not set."}]}
+
+    endpoint = _resolve_endpoint(api_url_raw)
+    timeout = _read_alix_api_graphql_timeout()
+
+    payload: dict[str, Any] = {
+        "query": query,
+        "variables": variables,
+        "operationName": operation_name,
+    }
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": api_key,
+    }
+
+    try:
+        with httpx.Client(timeout=timeout) as client:
+            response = client.post(endpoint, json=payload, headers=headers)
+    except httpx.HTTPError as exc:
+        return {"errors": [{"message": f"Transport error: {exc.__class__.__name__}: {exc}"}]}
+
+    text = response.text
+    try:
+        body: Any = response.json()
+    except ValueError:
+        body = None
+
+    if response.is_error:
+        if isinstance(body, dict) and ("errors" in body or "data" in body):
+            return {
+                "data": body.get("data"),
+                "errors": body.get("errors")
+                or [{"message": f"HTTP {response.status_code}: {text or response.reason_phrase}"}],
+                "http_status": response.status_code,
+            }
+        return {
+            "data": None,
+            "errors": [{"message": f"HTTP {response.status_code}: {text or response.reason_phrase}"}],
+            "http_status": response.status_code,
+        }
+
+    if not isinstance(body, dict):
+        return {
+            "data": None,
+            "errors": [{"message": f"Non-JSON response from GraphQL endpoint: {text[:500]}"}],
+            "http_status": response.status_code,
+        }
+
+    return {
+        "data": body.get("data"),
+        "errors": body.get("errors"),
+        "extensions": body.get("extensions"),
+        "http_status": response.status_code,
+    }
+
+
+def run_get_estate_documents(estateId: str) -> str:
+    if not estateId or not isinstance(estateId, str):
+        return json.dumps({"errors": [{"message": "estateId is required."}]})
+
+    result = _graphql_request(
+        query=_GET_ESTATE_DOCUMENTS_QUERY,
+        variables={"where": {"id": estateId}},
+        operation_name="GetEstateDocuments",
+    )
+
+    if result.get("errors"):
+        return json.dumps(result, default=str)
+
+    data = result.get("data") if isinstance(result.get("data"), dict) else {}
+    estate = data.get("estate") if isinstance(data, dict) else None
+
+    if not estate:
+        return json.dumps(
+            {
+                **result,
+                "data": {"documents": None},
+            },
+            default=str,
+        )
+    documents = estate.get("documents") or []
+
+    return json.dumps(
+        {
+            **result,
+            "data": {"documents": documents},
+        },
+        default=str,
+    )
+
+
+def run_get_file(
+    fileId: str,
+) -> str:
+    if not fileId or not isinstance(fileId, str):
+        return json.dumps({"errors": [{"message": "fileId is required."}]})
+
+    where: dict[str, Any] = {
+        "id": fileId,
+    }
+
+    result = _graphql_request(
+        query=_GET_FILE_QUERY,
+        variables={"where": where},
+        operation_name="GetFile",
+    )
+
+    if result.get("errors"):
+        return json.dumps(result, default=str)
+
+    data = result.get("data") if isinstance(result.get("data"), dict) else {}
+    file = data.get("file") if isinstance(data, dict) else None
+
+    if not isinstance(file, dict):
+        return json.dumps(
+            {
+                **result,
+                "errors": [{"message": "Unexpected GraphQL response: `file` is missing."}],
+            },
+            default=str,
+        )
+
+    if not file:
+        return json.dumps(
+            {
+                **result,
+                "data": {"file": None},
+            },
+            default=str,
+        )
+
+    return json.dumps(
+        {
+            **result,
+            "data": {"file": file},
+        },
+        default=str,
+    )
+
+def download_file_to_temp_folder(file: File, temp_folder_path: str) -> str:
+    if not file or not isinstance(file, dict):
+        return json.dumps({"errors": [{"message": "file is required."}]})
+
+    download_url = file.get("downloadUrl")
+    if not download_url or not isinstance(download_url, str):
+        return json.dumps({"errors": [{"message": "downloadUrl is required in file."}]})
+
+    if not temp_folder_path or not isinstance(temp_folder_path, str):
+        return json.dumps({"errors": [{"message": "temp_folder_path is required."}]})
+
+    file_name = file.get("fileName")
+    if not file_name or not isinstance(file_name, str):
+        return json.dumps({"errors": [{"message": "fileName is required in file."}]})
+
+    os.makedirs(temp_folder_path, exist_ok=True)
+    file_path = os.path.join(temp_folder_path, file_name)
+
+    timeout = _read_file_download_timeout()
+    try:
+        with httpx.stream("GET", download_url, timeout=timeout) as response:
+            response.raise_for_status()
+            with open(file_path, "wb") as output_file:
+                for chunk in response.iter_bytes():
+                    output_file.write(chunk)
+    except httpx.HTTPError as exc:
+        return json.dumps({"errors": [{"message": f"Failed to download file: {exc}"}]})
+    except OSError as exc:
+        return json.dumps({"errors": [{"message": f"Failed to save file: {exc}"}]})
+
+    return json.dumps({"data": {"filePath": file_path}})
+
+def main() -> None:
+    mcp = FastMCP(SERVER_NAME)
+
+    mcp.tool(
+        name="getEstateDocuments",
+        title="Get estate documents",
+        description=(
+            "Fetch all documents for an estate"
+            "Returns raw JSON containing `{data, errors, extensions, http_status}`."
+        ),
+    )(run_get_estate_documents)
+
+    mcp.tool(
+        name="getFile",
+        title="Get file",
+        description=(
+            "Fetch a file by id"
+            "Returns raw JSON containing `{data, errors, extensions, http_status}`."
+        ),
+    )(run_get_file)
+
+    mcp.tool(
+        name="downloadFileToTempFolder",
+        title="Download file to temporary folder",
+        description=(
+            "Download a file to a temporary folder and return the file path. Get file info via other tools first, for example 'getEstateDocuments'."
+            "Returns raw JSON containing `{data, errors, extensions, http_status}`."
+        ),
+    )(download_file_to_temp_folder)
+
+    mcp.run()

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: alix-doc-vault-tool
+Version: 0.1.0
+Summary: Documents Vault MCP tool for Alix. Fetches files, files per estate and downloads them into temporary folder for next processing
+Requires-Python: >=3.11
+Requires-Dist: mcp>=1.9.0
+Requires-Dist: httpx>=0.27
+Provides-Extra: dev
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+pyproject.toml
+src/alix_doc_vault_tool/__init__.py
+src/alix_doc_vault_tool/__main__.py
+src/alix_doc_vault_tool/mcp_server.py
+src/alix_doc_vault_tool.egg-info/PKG-INFO
+src/alix_doc_vault_tool.egg-info/SOURCES.txt
+src/alix_doc_vault_tool.egg-info/dependency_links.txt
+src/alix_doc_vault_tool.egg-info/entry_points.txt
+src/alix_doc_vault_tool.egg-info/requires.txt
+src/alix_doc_vault_tool.egg-info/top_level.txt

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+alix-doc-vault-tool = alix_doc_vault_tool.mcp_server:main

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/requires.txt

@@ -0,0 +1,6 @@
+mcp>=1.9.0
+httpx>=0.27
+
+[dev]
+build
+twine

alix_doc_vault_tool-0.1.0/src/alix_doc_vault_tool.egg-info/top_level.txt

@@ -0,0 +1 @@
+alix_doc_vault_tool