@@ -20,39 +20,31 @@ def main():
2020 parser = argparse .ArgumentParser (
2121 description = "Scan a directory or GitHub repository and optionally specify an LLM provider."
2222 )
23- parser .add_argument ("-d" , "--directory" , type = str , help = "The directory to scan" )
24- parser .add_argument (
25- "-u" , "--url" , type = str , help = "The GitHub repository URL to clone and scan"
26- )
27- parser .add_argument (
28- "-l" ,
29- "--llm_provider" ,
30- type = str ,
31- default = MODEL ,
32- choices = ["claude" , "sonnet" , "3.5" , "gpt4o" , "gpt4" , "gpt3.5" ],
33- help = "The LLM provider to use (optional)" ,
34- )
35- parser .add_argument (
36- "-p" ,
37- "--proper_nouns" ,
38- type = str ,
39- default = "* Llama3.1-70B \n * Cerebras \n * Cohere \n * OpenAI \n * AgentOps \n * Spellcaster" ,
40- help = "A string of proper nouns to include in the prompt (optional)" ,
41- )
42- parser .add_argument (
43- "-f" ,
44- "--file_types" ,
45- nargs = "+" ,
46- default = FILE_TYPES ,
47- help = "File types to scan (default: %(default)s)" ,
48- )
49- parser .add_argument (
50- "-m" ,
51- "--max_files" ,
52- type = int ,
53- default = MAX_FILES ,
54- help = "Maximum number of files to scan (default: %(default)s)" ,
55- )
23+ parser .add_argument ("-d" , "--directory" ,
24+ type = str , help = "The directory to scan" )
25+ parser .add_argument ("-u" , "--url" ,
26+ type = str , help = "The GitHub repository URL to clone and scan" )
27+ parser .add_argument ("-l" , "--llm_provider" ,
28+ type = str ,
29+ default = MODEL ,
30+ choices = ["claude" , "sonnet" , "3.5" , "gpt4o" , "gpt4" , "gpt3.5" ],
31+ help = "The LLM provider to use (optional)" ,
32+ )
33+ parser .add_argument ("-p" , "--proper_nouns" ,
34+ type = str ,
35+ default = "* Llama3.1-70B \n * Cerebras \n * Cohere \n * OpenAI \n * AgentOps \n * Spellcaster" ,
36+ help = "A string of proper nouns to include in the prompt (optional)" ,
37+ )
38+ parser .add_argument ("-f" , "--file_types" ,
39+ nargs = "+" ,
40+ default = FILE_TYPES ,
41+ help = "File types to scan (default: %(default)s)" ,
42+ )
43+ parser .add_argument ("-m" , "--max_files" ,
44+ type = int ,
45+ default = MAX_FILES ,
46+ help = "Maximum number of files to scan (default: %(default)s)" ,
47+ )
5648
5749 args = parser .parse_args ()
5850
0 commit comments