Obsidian Elite RAG MCP Server

""" Main entry point for just-prompt. """ import argparse import asyncio import logging import sys from dotenv import load_dotenv from .server import serve from .atoms.shared.utils import DEFAULT_MODEL, get_dynamic_default_models from .atoms.shared.validator import print_provider_availability # Load environment variables load_dotenv() # Configure logging logging.basicConfig( level=logging.INFO, format='%(asctime)s [%(levelname)s] %(message)s', datefmt='%Y-%m-%d %H:%M:%S' ) logger = logging.getLogger(__name__) def main(): """ Main entry point for just-prompt. """ parser = argparse.ArgumentParser(description="just-prompt - A lightweight MCP server for various LLM providers") parser.add_argument( "--default-models", default=None, help="Comma-separated list of default models to use for prompts and model name correction, in format provider:model. If not specified, will auto-detect best free OpenRouter models." ) parser.add_argument( "--log-level", choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], default="INFO", help="Logging level" ) parser.add_argument( "--show-providers", action="store_true", help="Show available providers and exit" ) args = parser.parse_args() # Set logging level logging.getLogger().setLevel(getattr(logging, args.log_level)) # Show provider availability and optionally exit if args.show_providers: print_provider_availability() sys.exit(0) try: # Get default models - use dynamic detection if not specified default_models = args.default_models if default_models is None: logger.info("No default models specified, auto-detecting best free OpenRouter models...") default_models = get_dynamic_default_models() logger.info(f"Using auto-detected models: {default_models}") # Start server (asyncio) asyncio.run(serve(default_models)) except Exception as e: logger.error(f"Error starting server: {e}") sys.exit(1) if __name__ == "__main__": main()