def parse_args()

in projects/dynasor/openai_client.py [0:0]

• 28 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser(description="OpenAI Chat Client")
    parser.add_argument("--api-key", default="EMPTY", help="OpenAI API key")
    parser.add_argument(
        "--base-url", default="http://localhost:8000/v1", help="OpenAI API base URL"
    )
    parser.add_argument(
        "--max-tokens", type=int, default=2048, help="Maximum tokens for completion"
    )
    parser.add_argument(
        "--disable-dynasor",
        action="store_true",
        help="Disable Dynasor"
    )
    parser.add_argument(
        "--probe-interval",
        type=int,
        default=32,
        help="Probe interval for adaptive compute"
    )
    parser.add_argument(
        "--certainty-window",
        type=int,
        default=2,
        help="Certainty window for adaptive compute"
    )
    parser.add_argument("--prompt", default="2+2=", help="User prompt")
    # parser.add_argument("--no-stream", action="store_true", help="Do not stream the response")
    return parser.parse_args()