serviette 0.1.0__py3-none-any.whl

serviette/__init__.py

@@ -0,0 +1,10 @@
+"""serviette — a universal, no-code RAG server built on the Pathway Live Data Framework.
+
+The project name is intentionally kept in a single constant so a rename only
+requires a find-and-replace across files plus updating ``APP_NAME`` here.
+"""
+
+APP_NAME = "serviette"
+__version__ = "0.1.0"
+
+__all__ = ["APP_NAME", "__version__"]

serviette/cli.py

@@ -0,0 +1,93 @@
+"""Single ``serviette`` entrypoint dispatching all subcommands.
+
+Subcommands:
+  serviette indexer    --config FILE   Build & run the Pathway indexing graph
+  serviette server     --config FILE   Start the FastAPI retrieval/RAG server
+  serviette up         --config FILE   Run indexer + server together (dev/demo)
+  serviette quickstart                 Interactive config-generation wizard
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+
+from serviette import APP_NAME, __version__
+
+_OLD_PATHWAY_HINT = (
+    "Your installed pathway ({version}) predates the vector-store connectors "
+    "serviette needs (pathway >= 0.32.1). Upgrade it in this environment:\n\n"
+    "  pip install -U 'pathway>=0.32.1'\n"
+)
+
+
+def _ensure_pathway_connectors() -> None:
+    """Fail with instructions, not an AttributeError, on an old pathway.
+
+    Engine-facing commands (indexer/up/demo) need the vector-store output
+    connectors that shipped in pathway 0.32.1; on an older install the first
+    symptom used to be an opaque crash deep in the stack.
+    """
+
+    import pathway as pw
+
+    if not hasattr(pw.io, "duckdb"):
+        raise SystemExit(_OLD_PATHWAY_HINT.format(version=pw.__version__))
+
+
+def main(argv: list[str] | None = None) -> None:
+    argv = list(sys.argv[1:] if argv is None else argv)
+
+    parser = argparse.ArgumentParser(prog=APP_NAME, description=__doc__)
+    parser.add_argument("--version", action="version", version=f"{APP_NAME} {__version__}")
+    sub = parser.add_subparsers(dest="command", required=True)
+    sub.add_parser("indexer", add_help=False, help="Run the indexer")
+    sub.add_parser("server", add_help=False, help="Run the server")
+    sub.add_parser("up", add_help=False, help="Run indexer + server together")
+    sub.add_parser("demo", add_help=False, help="Zero-to-chat demo on a bundled corpus")
+    sub.add_parser("frontend", add_help=False, help="Run the web chat frontend")
+    sub.add_parser("quickstart", add_help=False, help="Interactive config wizard")
+
+    # Parse only the first token so each subcommand owns the rest of argv.
+    args, rest = parser.parse_known_args(argv)
+
+    if args.command in ("indexer", "up", "demo"):
+        _ensure_pathway_connectors()
+    if args.command == "indexer":
+        from serviette.indexer.main import main as indexer_main
+
+        indexer_main(rest)
+    elif args.command == "server":
+        from serviette.server.config import load_server_config
+        from serviette.server.main import run
+
+        server_args = _parse_config_arg("server", rest)
+        run(load_server_config(server_args.config))
+    elif args.command == "up":
+        from serviette.up import main as up_main
+
+        up_main(rest)
+    elif args.command == "demo":
+        from serviette.demo import main as demo_main
+
+        demo_main(rest)
+    elif args.command == "frontend":
+        from serviette.frontend.config import load_frontend_config
+        from serviette.frontend.main import run
+
+        frontend_args = _parse_config_arg("frontend", rest)
+        run(load_frontend_config(frontend_args.config))
+    elif args.command == "quickstart":
+        from serviette.quickstart.wizard import main as quickstart_main
+
+        quickstart_main(rest)
+
+
+def _parse_config_arg(prog: str, rest: list[str]) -> argparse.Namespace:
+    p = argparse.ArgumentParser(prog=f"{APP_NAME} {prog}")
+    p.add_argument("--config", required=True, help="Path to the YAML config file")
+    return p.parse_args(rest)
+
+
+if __name__ == "__main__":
+    main()

serviette/config/__init__.py

@@ -0,0 +1,49 @@
+"""Shared configuration package for serviette."""
+
+from serviette.config.schema import (
+    ChromaConfig,
+    DuckDbConfig,
+    EmbedderConfig,
+    FrontendConfig,
+    FsSource,
+    GDriveSource,
+    LLMConfig,
+    MilvusConfig,
+    MongoDbConfig,
+    PersistenceConfig,
+    PgVectorConfig,
+    PineconeConfig,
+    QdrantConfig,
+    S3Source,
+    ServerConfig,
+    ServietteConfig,
+    SharePointSource,
+    Source,
+    SplitterConfig,
+    WeaviateConfig,
+    load_config,
+)
+
+__all__ = [
+    "ChromaConfig",
+    "DuckDbConfig",
+    "EmbedderConfig",
+    "FrontendConfig",
+    "FsSource",
+    "GDriveSource",
+    "LLMConfig",
+    "MilvusConfig",
+    "MongoDbConfig",
+    "PersistenceConfig",
+    "PgVectorConfig",
+    "PineconeConfig",
+    "QdrantConfig",
+    "S3Source",
+    "ServerConfig",
+    "ServietteConfig",
+    "SharePointSource",
+    "Source",
+    "SplitterConfig",
+    "WeaviateConfig",
+    "load_config",
+]