hindsight-api 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

hindsight_api/cli.py

@@ -1,127 +0,0 @@
-"""
-Command-line interface for Hindsight API.
-
-Run the server with:
-    hindsight-api
-
-Stop with Ctrl+C.
-"""
-import argparse
-import asyncio
-import atexit
-import os
-import signal
-import sys
-from typing import Optional
-
-import uvicorn
-
-from . import MemoryEngine
-from .api import create_app
-
-
-# Disable tokenizers parallelism to avoid warnings
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-# Global reference for cleanup
-_memory: Optional[MemoryEngine] = None
-
-
-def _cleanup():
-    """Synchronous cleanup function to stop resources on exit."""
-    global _memory
-    if _memory is not None and _memory._pg0 is not None:
-        try:
-            loop = asyncio.new_event_loop()
-            loop.run_until_complete(_memory._pg0.stop())
-            loop.close()
-            print("\npg0 stopped.")
-        except Exception as e:
-            print(f"\nError stopping pg0: {e}")
-
-
-def _signal_handler(signum, frame):
-    """Handle SIGINT/SIGTERM to ensure cleanup."""
-    print(f"\nReceived signal {signum}, shutting down...")
-    _cleanup()
-    sys.exit(0)
-
-
-def main():
-    """Main entry point for the CLI."""
-    global _memory
-
-    parser = argparse.ArgumentParser(
-        prog="hindsight-api",
-        description="Hindsight API Server",
-    )
-    parser.add_argument(
-        "--host", default="0.0.0.0",
-        help="Host to bind to (default: 0.0.0.0)"
-    )
-    parser.add_argument(
-        "--port", type=int, default=8888,
-        help="Port to bind to (default: 8888)"
-    )
-    parser.add_argument(
-        "--log-level", default="info",
-        choices=["critical", "error", "warning", "info", "debug", "trace"],
-        help="Log level (default: info)"
-    )
-    parser.add_argument(
-        "--access-log", action="store_true",
-        help="Enable access log"
-    )
-
-    args = parser.parse_args()
-
-    # Register cleanup handlers
-    atexit.register(_cleanup)
-    signal.signal(signal.SIGINT, _signal_handler)
-    signal.signal(signal.SIGTERM, _signal_handler)
-
-    # Get configuration from environment variables
-    db_url = os.getenv("HINDSIGHT_API_DATABASE_URL", "pg0")
-    llm_provider = os.getenv("HINDSIGHT_API_LLM_PROVIDER", "groq")
-    llm_api_key = os.getenv("HINDSIGHT_API_LLM_API_KEY", "")
-    llm_model = os.getenv("HINDSIGHT_API_LLM_MODEL", "openai/gpt-oss-20b")
-    llm_base_url = os.getenv("HINDSIGHT_API_LLM_BASE_URL") or None
-
-    # Create MemoryEngine
-    _memory = MemoryEngine(
-        db_url=db_url,
-        memory_llm_provider=llm_provider,
-        memory_llm_api_key=llm_api_key,
-        memory_llm_model=llm_model,
-        memory_llm_base_url=llm_base_url,
-    )
-
-    # Create FastAPI app
-    app = create_app(
-        memory=_memory,
-        http_api_enabled=True,
-        mcp_api_enabled=True,
-        mcp_mount_path="/mcp",
-        initialize_memory=True,
-    )
-
-    # Prepare uvicorn config
-    uvicorn_config = {
-        "app": app,
-        "host": args.host,
-        "port": args.port,
-        "log_level": args.log_level,
-        "access_log": args.access_log,
-    }
-
-    print(f"\nStarting Hindsight API...")
-    print(f"  URL: http://{args.host}:{args.port}")
-    print(f"  Database: {db_url}")
-    print(f"  LLM Provider: {llm_provider}")
-    print()
-
-    uvicorn.run(**uvicorn_config)
-
-
-if __name__ == "__main__":
-    main()

hindsight_api/web/__init__.py

@@ -1,12 +0,0 @@
-"""
-Web interface for memory system.
-
-Provides FastAPI app and visualization interface.
-"""
-from hindsight_api.api import create_app
-
-# Note: Don't import app from .server here to avoid circular import warnings
-# when running with `python -m hindsight_api.web.server`
-# If you need the app, import it directly: from hindsight_api.web.server import app
-
-__all__ = ["create_app"]

hindsight_api/web/server.py

@@ -1,109 +0,0 @@
-"""
-FastAPI server for memory graph visualization and API.
-
-Provides REST API endpoints for memory operations and serves
-the interactive visualization interface.
-"""
-import warnings
-
-# Filter deprecation warnings from third-party libraries
-warnings.filterwarnings("ignore", message="websockets.legacy is deprecated")
-warnings.filterwarnings("ignore", message="websockets.server.WebSocketServerProtocol is deprecated")
-
-import logging
-import os
-import argparse
-
-from hindsight_api import MemoryEngine
-from hindsight_api.api import create_app
-
-# Disable tokenizers parallelism to avoid warnings
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-
-# Create app at module level (required for uvicorn import string)
-_memory = MemoryEngine(
-    db_url=os.getenv("HINDSIGHT_API_DATABASE_URL", "pg0"),
-    memory_llm_provider=os.getenv("HINDSIGHT_API_LLM_PROVIDER", "groq"),
-    memory_llm_api_key=os.getenv("HINDSIGHT_API_LLM_API_KEY"),
-    memory_llm_model=os.getenv("HINDSIGHT_API_LLM_MODEL", "openai/gpt-oss-120b"),
-    memory_llm_base_url=os.getenv("HINDSIGHT_API_LLM_BASE_URL") or None,
-)
-
-# Check if MCP should be enabled
-mcp_enabled = os.getenv("HINDSIGHT_API_MCP_ENABLED", "true").lower() == "true"
-
-# Create unified app with both HTTP and optionally MCP
-app = create_app(
-    memory=_memory,
-    http_api_enabled=True,
-    mcp_api_enabled=mcp_enabled,
-    mcp_mount_path="/mcp"
-)
-
-
-if __name__ == "__main__":
-    import uvicorn
-
-    # Get log level from environment variable (default: info)
-    env_log_level = os.environ.get("HINDSIGHT_API_LOG_LEVEL", "info").lower()
-    if env_log_level not in ["critical", "error", "warning", "info", "debug", "trace"]:
-        env_log_level = "info"
-
-    # Parse CLI arguments
-    parser = argparse.ArgumentParser(description="Hindsight API Server")
-    parser.add_argument("--host", default="0.0.0.0", help="Host to bind to (default: 0.0.0.0)")
-    parser.add_argument("--port", type=int, default=8888, help="Port to bind to (default: 8888)")
-    parser.add_argument("--reload", action="store_true", help="Enable auto-reload on code changes")
-    parser.add_argument("--workers", type=int, default=1, help="Number of worker processes (default: 1)")
-    parser.add_argument("--log-level", default=env_log_level, choices=["critical", "error", "warning", "info", "debug", "trace"],
-                        help=f"Log level (default: {env_log_level}, from HINDSIGHT_API_LOG_LEVEL)")
-    parser.add_argument("--access-log", action="store_true", help="Enable access log")
-    parser.add_argument("--no-access-log", dest="access_log", action="store_false", help="Disable access log")
-    parser.add_argument("--proxy-headers", action="store_true", help="Enable X-Forwarded-Proto, X-Forwarded-For headers")
-    parser.add_argument("--forwarded-allow-ips", default=None, help="Comma separated list of IPs to trust with proxy headers")
-    parser.add_argument("--ssl-keyfile", default=None, help="SSL key file")
-    parser.add_argument("--ssl-certfile", default=None, help="SSL certificate file")
-    parser.set_defaults(access_log=False)
-
-    args = parser.parse_args()
-
-    # Configure Python logging based on log level
-    log_level_map = {
-        "critical": logging.CRITICAL,
-        "error": logging.ERROR,
-        "warning": logging.WARNING,
-        "info": logging.INFO,
-        "debug": logging.DEBUG,
-        "trace": logging.DEBUG,  # Python doesn't have TRACE, use DEBUG
-    }
-    logging.basicConfig(
-        level=log_level_map.get(args.log_level, logging.INFO),
-        format="%(asctime)s - %(levelname)s - %(name)s - %(message)s"
-    )
-    logging.info(f"Starting Hindsight API on {args.host}:{args.port}")
-
-    app_ref = "hindsight_api.web.server:app"
-
-    # Prepare uvicorn config
-    uvicorn_config = {
-        "app": app_ref,
-        "host": args.host,
-        "port": args.port,
-        "reload": args.reload,
-        "workers": args.workers,
-        "log_level": args.log_level,
-        "access_log": args.access_log,
-        "proxy_headers": args.proxy_headers,
-        "ws": "wsproto",  # Use wsproto instead of websockets to avoid deprecation warnings
-    }
-
-    # Add optional parameters if provided
-    if args.forwarded_allow_ips:
-        uvicorn_config["forwarded_allow_ips"] = args.forwarded_allow_ips
-    if args.ssl_keyfile:
-        uvicorn_config["ssl_keyfile"] = args.ssl_keyfile
-    if args.ssl_certfile:
-        uvicorn_config["ssl_certfile"] = args.ssl_certfile
-
-    uvicorn.run(**uvicorn_config)

hindsight_api-0.1.0.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-hindsight-api = hindsight_api.cli:main