server.py

from __future__ import annotations

import os

from mcp.server.fastmcp import FastMCP

from deepfetch import __version__
from deepfetch.logging import configure_logging
from deepfetch.search.internet_search import (
    INTERNET_SEARCH_TOOL_DESCRIPTION,
    internet_search,
)
from deepfetch.search.pdf_utils import (
    PDF_EXTRACT_TEXT_TOOL_DESCRIPTION,
    pdf_extract_text,
)

_log_path = os.getenv("DEEPFETCH_LOG_PATH")
_log_dir = os.getenv("DEEPFETCH_LOG_DIR")
if _log_path or _log_dir:
    configure_logging(
        os.getenv("DEEPFETCH_LOG_LEVEL", "INFO"),
        log_path=_log_path,
        run_id=os.getenv("DEEPFETCH_LOG_RUN_ID"),
    )

mcp = FastMCP("deepfetch")
mcp._mcp_server.version = __version__

mcp.tool(description=INTERNET_SEARCH_TOOL_DESCRIPTION)(internet_search)
mcp.tool(description=PDF_EXTRACT_TEXT_TOOL_DESCRIPTION)(pdf_extract_text)


def run() -> None:
    transport = str(os.getenv("DEEPFETCH_TRANSPORT", "stdio") or "stdio").strip()
    if not transport:
        transport = "stdio"
    mcp.run(transport=transport)


if __name__ == "__main__":
    run()