cli.py

import argparse
from claude import Chat, Text, Vision

def main():
    class CustomFormatter(argparse.ArgumentDefaultsHelpFormatter,
                      argparse.RawDescriptionHelpFormatter):
        pass
    parser = argparse.ArgumentParser(
        description="""
    ------------------------------------------------------------------
                            Claude AI Toolkit                          
                   API Wrapper & Command-line Interface               
                          [v1.0.2] by @rmncldyo                      
    ------------------------------------------------------------------

    Claude AI toolit is an API wrapper and command-line interface for Anthropic's latest Claude 3.5 large-language models.

    | **Description**               	| **CLI Flag(s)**           	| **CLI Usage**                                         	|
    |-------------------------------	|---------------------------	|-------------------------------------------------------	|
    | Enable chat mode              	| `-c`,  `--chat`           	| --chat                                                	|
    | Enable text mode              	| `-t`,  `--text`           	| --text                                                	|
    | Enable vision mode            	| `-v`,  `--vision`         	| --vision                                              	|
    | User prompt                   	| `-p`,  `--prompt`         	| --prompt "Craft in depth lesson plans on any subject" 	|
    | Image file path or url        	| `-i`,  `--image`          	| --image "image_path_or_url"                           	|
    | API key for authentication    	| `-a`,  `--api_key`        	| --api_key "your_api_key"                              	|
    | Model to use                  	| `-m`,  `--model`          	| --model "claude-3-5-sonnet-20241022"                      |
    | Enable streaming mode         	| `-s`,  `--stream`         	| --stream                                              	|
    | System prompt (instructions)  	| `-sp,  '--system_prompt`  	| --system_prompt "You are an advanced AI assistant"    	|
    | Maximum tokens to generate    	| `-mt`, `--max_tokens`     	| --max_tokens 1024                                     	|
    | Sampling temperature          	| `-tm`, `--temperature`    	| --temperature 0.7                                     	|
    | Nucleus sampling threshold    	| `-tp`, `--top_p`          	| --top_p 0.9                                           	|
    | Top-k sampling threshold      	| `-tk`, `--top_k`          	| --top_k 40                                            	|
    | Stop sequences for completion 	| `-ss`, `--stop_sequences` 	| --stop_sequences ["\n", "."]                          	|
    """,
        formatter_class=CustomFormatter,
        epilog="For detailed usage information, visit our ReadMe here: github.com/RMNCLDYO/claude-ai-toolkit"
    )
    parser.add_argument('-c', '--chat', action='store_true', help='Enable chat mode')
    parser.add_argument('-t', '--text', action='store_true', help='Enable text mode')
    parser.add_argument('-v', '--vision', action='store_true', help='Enable vision mode')
    parser.add_argument('-p', '--prompt', type=str, help='Text or Vision prompt', metavar='')
    parser.add_argument('-i', '--image', type=str, help='Image file path or url', metavar='')
    parser.add_argument('-a', '--api_key', type=str, help='Claude API key for authentication', metavar='')
    parser.add_argument('-m', '--model', type=str, default='claude-3-5-sonnet-20241022', help='The model you would like to use', metavar='')
    parser.add_argument('-s', '--stream', action='store_true', help='Enable streaming mode for responses')
    parser.add_argument('-sp', '--system_prompt', type=str, help='Initial system prompt (instructions)', metavar='')
    parser.add_argument('-mt', '--max_tokens', type=int, help='Maximum number of tokens to generate', metavar='')
    parser.add_argument('-tm', '--temperature', type=float, help='Sampling temperature', metavar='')
    parser.add_argument('-tp', '--top_p', type=float, help='Nucleus sampling threshold', metavar='')
    parser.add_argument('-tk', '--top_k', type=int, help='Top-k sampling threshold', metavar='')
    parser.add_argument('-ss', '--stop_sequences', type=str, nargs='+', help='Stop sequences for completion', metavar='')

    args = parser.parse_args()
    
    if args.chat:
        Chat().run(args.api_key, args.model,  args.prompt, args.system_prompt, args.max_tokens, args.stop_sequences, args.stream, args.temperature, args.top_p, args.top_k)
    elif args.text:
        Text().run(args.api_key, args.model, args.prompt, args.system_prompt, args.max_tokens, args.stop_sequences, args.stream, args.temperature, args.top_p, args.top_k)
    elif args.vision:
        Vision().run(args.api_key, args.model, args.prompt, args.image, args.system_prompt, args.max_tokens, args.stop_sequences, args.stream, args.temperature, args.top_p, args.top_k)
    else:
        print("Error: Please specify a mode to use. Use --help for more information.")
        exit()

if __name__ == "__main__":
    main()