auto-coder 0.1.76__tar.gz → 0.1.77__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (76) hide show
  1. {auto-coder-0.1.76 → auto-coder-0.1.77}/PKG-INFO +4 -4
  2. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/PKG-INFO +4 -4
  3. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/requires.txt +3 -3
  4. auto-coder-0.1.77/src/autocoder/command_args.py +298 -0
  5. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/index/index.py +2 -2
  6. auto-coder-0.1.77/src/autocoder/version.py +1 -0
  7. auto-coder-0.1.76/src/autocoder/command_args.py +0 -179
  8. auto-coder-0.1.76/src/autocoder/version.py +0 -1
  9. {auto-coder-0.1.76 → auto-coder-0.1.77}/README.md +0 -0
  10. {auto-coder-0.1.76 → auto-coder-0.1.77}/setup.cfg +0 -0
  11. {auto-coder-0.1.76 → auto-coder-0.1.77}/setup.py +0 -0
  12. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/SOURCES.txt +0 -0
  13. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/dependency_links.txt +0 -0
  14. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/entry_points.txt +0 -0
  15. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/auto_coder.egg-info/top_level.txt +0 -0
  16. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/__init__.py +0 -0
  17. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/agent/__init__.py +0 -0
  18. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/agent/coder.py +0 -0
  19. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/auto_coder.py +0 -0
  20. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/JupyterClient.py +0 -0
  21. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/ShellClient.py +0 -0
  22. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/__init__.py +0 -0
  23. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/anything2images.py +0 -0
  24. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/audio.py +0 -0
  25. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/cleaner.py +0 -0
  26. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_execute.py +0 -0
  27. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_generate.py +0 -0
  28. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_generate_diff.py +0 -0
  29. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_generate_editblock.py +0 -0
  30. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_generate_strict_diff.py +0 -0
  31. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_merge.py +0 -0
  32. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_merge_diff.py +0 -0
  33. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_merge_editblock.py +0 -0
  34. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/code_auto_merge_strict_diff.py +0 -0
  35. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/command_templates.py +0 -0
  36. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/const.py +0 -0
  37. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/git_utils.py +0 -0
  38. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/image_to_page.py +0 -0
  39. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/llm_rerank.py +0 -0
  40. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/screenshots.py +0 -0
  41. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/search.py +0 -0
  42. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/search_replace.py +0 -0
  43. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/common/types.py +0 -0
  44. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/db/__init__.py +0 -0
  45. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/db/store.py +0 -0
  46. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/__init__.py +0 -0
  47. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/__init__.py +0 -0
  48. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/action.py +0 -0
  49. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/copilot.py +0 -0
  50. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/plugins/__init__.py +0 -0
  51. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/plugins/action_regex_project.py +0 -0
  52. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/dispacher/actions/plugins/action_translate.py +0 -0
  53. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/index/__init__.py +0 -0
  54. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/index/for_command.py +0 -0
  55. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/lang.py +0 -0
  56. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/pyproject/__init__.py +0 -0
  57. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/rag/__init__.py +0 -0
  58. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/rag/api_server.py +0 -0
  59. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/rag/llm_wrapper.py +0 -0
  60. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/rag/simple_rag.py +0 -0
  61. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/rag/types.py +0 -0
  62. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/regexproject/__init__.py +0 -0
  63. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/suffixproject/__init__.py +0 -0
  64. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/tsproject/__init__.py +0 -0
  65. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/utils/__init__.py +0 -0
  66. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/utils/llm_client_interceptors.py +0 -0
  67. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/utils/multi_turn.py +0 -0
  68. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/utils/print_table.py +0 -0
  69. {auto-coder-0.1.76 → auto-coder-0.1.77}/src/autocoder/utils/rest.py +0 -0
  70. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_code_auto_merge.py +0 -0
  71. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_image_to_page.py +0 -0
  72. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_init_command.py +0 -0
  73. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_llm_rerank.py +0 -0
  74. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_shell_client.py +0 -0
  75. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_simple_rag.py +0 -0
  76. {auto-coder-0.1.76 → auto-coder-0.1.77}/tests/test_tsproject.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: auto-coder
3
- Version: 0.1.76
3
+ Version: 0.1.77
4
4
  Summary: AutoCoder: AutoCoder
5
5
  Author: allwefantasy
6
6
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -12,7 +12,7 @@ Requires-Dist: ninja
12
12
  Requires-Dist: accelerate
13
13
  Requires-Dist: bitsandbytes==0.39.0
14
14
  Requires-Dist: transformers>=4.35.0
15
- Requires-Dist: torch==2.1.2
15
+ Requires-Dist: torch<2.3,>=2.1.2
16
16
  Requires-Dist: ray[default]>=2.9.1
17
17
  Requires-Dist: sentencepiece
18
18
  Requires-Dist: pyarrow==12.0.1
@@ -23,7 +23,7 @@ Requires-Dist: paramiko
23
23
  Requires-Dist: einops
24
24
  Requires-Dist: tqdm
25
25
  Requires-Dist: loguru
26
- Requires-Dist: pyjava>=0.6.20
26
+ Requires-Dist: pyjava>=0.6.21
27
27
  Requires-Dist: tiktoken
28
28
  Requires-Dist: transformers_stream_generator
29
29
  Requires-Dist: optimum
@@ -41,7 +41,7 @@ Requires-Dist: websocket-client
41
41
  Requires-Dist: sqlmodel
42
42
  Requires-Dist: wudao
43
43
  Requires-Dist: jieba
44
- Requires-Dist: byzerllm>=0.1.92
44
+ Requires-Dist: byzerllm>=0.1.93
45
45
  Requires-Dist: patch
46
46
  Requires-Dist: diff_match_patch
47
47
  Requires-Dist: GitPython
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: auto-coder
3
- Version: 0.1.76
3
+ Version: 0.1.77
4
4
  Summary: AutoCoder: AutoCoder
5
5
  Author: allwefantasy
6
6
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -12,7 +12,7 @@ Requires-Dist: ninja
12
12
  Requires-Dist: accelerate
13
13
  Requires-Dist: bitsandbytes==0.39.0
14
14
  Requires-Dist: transformers>=4.35.0
15
- Requires-Dist: torch==2.1.2
15
+ Requires-Dist: torch<2.3,>=2.1.2
16
16
  Requires-Dist: ray[default]>=2.9.1
17
17
  Requires-Dist: sentencepiece
18
18
  Requires-Dist: pyarrow==12.0.1
@@ -23,7 +23,7 @@ Requires-Dist: paramiko
23
23
  Requires-Dist: einops
24
24
  Requires-Dist: tqdm
25
25
  Requires-Dist: loguru
26
- Requires-Dist: pyjava>=0.6.20
26
+ Requires-Dist: pyjava>=0.6.21
27
27
  Requires-Dist: tiktoken
28
28
  Requires-Dist: transformers_stream_generator
29
29
  Requires-Dist: optimum
@@ -41,7 +41,7 @@ Requires-Dist: websocket-client
41
41
  Requires-Dist: sqlmodel
42
42
  Requires-Dist: wudao
43
43
  Requires-Dist: jieba
44
- Requires-Dist: byzerllm>=0.1.92
44
+ Requires-Dist: byzerllm>=0.1.93
45
45
  Requires-Dist: patch
46
46
  Requires-Dist: diff_match_patch
47
47
  Requires-Dist: GitPython
@@ -3,7 +3,7 @@ ninja
3
3
  accelerate
4
4
  bitsandbytes==0.39.0
5
5
  transformers>=4.35.0
6
- torch==2.1.2
6
+ torch<2.3,>=2.1.2
7
7
  ray[default]>=2.9.1
8
8
  sentencepiece
9
9
  pyarrow==12.0.1
@@ -14,7 +14,7 @@ paramiko
14
14
  einops
15
15
  tqdm
16
16
  loguru
17
- pyjava>=0.6.20
17
+ pyjava>=0.6.21
18
18
  tiktoken
19
19
  transformers_stream_generator
20
20
  optimum
@@ -32,7 +32,7 @@ websocket-client
32
32
  sqlmodel
33
33
  wudao
34
34
  jieba
35
- byzerllm>=0.1.92
35
+ byzerllm>=0.1.93
36
36
  patch
37
37
  diff_match_patch
38
38
  GitPython
@@ -0,0 +1,298 @@
1
+ import argparse
2
+ from autocoder.common import AutoCoderArgs
3
+ from autocoder.lang import lang_desc
4
+ import locale
5
+
6
+
7
+ def parse_args() -> AutoCoderArgs:
8
+ system_lang, _ = locale.getdefaultlocale()
9
+ lang = "zh" if system_lang and system_lang.startswith("zh") else "en"
10
+ desc = lang_desc[lang]
11
+
12
+ parser = argparse.ArgumentParser(description=desc["parser_desc"])
13
+ subparsers = parser.add_subparsers(dest="command")
14
+
15
+ parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
16
+ parser.add_argument("--git_url", help=desc["git_url"])
17
+ parser.add_argument("--target_file", required=False, help=desc["target_file"])
18
+ parser.add_argument("--query", help=desc["query"])
19
+ parser.add_argument("--template", default="common", help=desc["template"])
20
+ parser.add_argument("--project_type", default="py", help=desc["project_type"])
21
+ parser.add_argument("--execute", action="store_true", help=desc["execute"])
22
+ parser.add_argument("--package_name", default="", help=desc["package_name"])
23
+ parser.add_argument("--script_path", default="", help=desc["script_path"])
24
+
25
+ parser.add_argument("--model", default="", help=desc["model"])
26
+ parser.add_argument(
27
+ "--model_max_length", type=int, default=2000, help=desc["model_max_length"]
28
+ )
29
+ parser.add_argument(
30
+ "--model_max_input_length",
31
+ type=int,
32
+ default=6000,
33
+ help=desc["model_max_input_length"],
34
+ )
35
+
36
+ parser.add_argument("--vl_model", default="", help=desc["vl_model"])
37
+ parser.add_argument("--sd_model", default="", help=desc["sd_model"])
38
+ parser.add_argument("--emb_model", default="", help=desc["emb_model"])
39
+
40
+ parser.add_argument("--index_model", default="", help=desc["index_model"])
41
+ parser.add_argument(
42
+ "--index_model_max_length",
43
+ type=int,
44
+ default=0,
45
+ help=desc["index_model_max_length"],
46
+ )
47
+ parser.add_argument(
48
+ "--index_model_max_input_length",
49
+ type=int,
50
+ default=0,
51
+ help=desc["index_model_max_input_length"],
52
+ )
53
+ parser.add_argument(
54
+ "--index_model_anti_quota_limit",
55
+ type=int,
56
+ default=0,
57
+ help=desc["index_model_anti_quota_limit"],
58
+ )
59
+ parser.add_argument(
60
+ "--index_filter_level", type=int, default=0, help=desc["index_filter_level"]
61
+ )
62
+ parser.add_argument(
63
+ "--index_filter_workers", type=int, default=1, help=desc["index_filter_workers"]
64
+ )
65
+ parser.add_argument(
66
+ "--index_build_workers", type=int, default=1, help=desc["index_build_workers"]
67
+ )
68
+
69
+ parser.add_argument("--file", default=None, required=False, help=desc["file"])
70
+ parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
71
+ parser.add_argument(
72
+ "--anti_quota_limit", type=int, default=1, help=desc["anti_quota_limit"]
73
+ )
74
+ parser.add_argument(
75
+ "--skip_build_index", action="store_false", help=desc["skip_build_index"]
76
+ )
77
+ parser.add_argument(
78
+ "--print_request", action="store_true", help=desc["print_request"]
79
+ )
80
+ parser.add_argument("--code_model", default="", help=desc["code_model"])
81
+ parser.add_argument(
82
+ "--py_packages", required=False, default="", help=desc["py_packages"]
83
+ )
84
+ parser.add_argument(
85
+ "--human_as_model", action="store_true", help=desc["human_as_model"]
86
+ )
87
+ parser.add_argument("--urls", default="", help=desc["urls"])
88
+ parser.add_argument(
89
+ "--urls_use_model", action="store_true", help=desc["urls_use_model"]
90
+ )
91
+ parser.add_argument("--exclude_files", default="", help="")
92
+
93
+ parser.add_argument("--search", default="", help="")
94
+ parser.add_argument("--search_engine", default="", help=desc["search_engine"])
95
+ parser.add_argument(
96
+ "--search_engine_token", default="", help=desc["search_engine_token"]
97
+ )
98
+
99
+ parser.add_argument(
100
+ "--enable_rag_search",
101
+ nargs="?",
102
+ const=True,
103
+ default=False,
104
+ help=desc["enable_rag_search"],
105
+ )
106
+ parser.add_argument(
107
+ "--enable_rag_context",
108
+ nargs="?",
109
+ const=True,
110
+ default=False,
111
+ help=desc["enable_rag_context"],
112
+ )
113
+
114
+ parser.add_argument(
115
+ "--auto_merge", nargs="?", const=True, default=False, help=desc["auto_merge"]
116
+ )
117
+
118
+ parser.add_argument("--image_file", default="", help=desc["image_file"])
119
+ parser.add_argument("--image_mode", default="direct", help=desc["image_mode"])
120
+ parser.add_argument(
121
+ "--image_max_iter", type=int, default=1, help=desc["image_max_iter"]
122
+ )
123
+
124
+ parser.add_argument(
125
+ "--enable_multi_round_generate",
126
+ action="store_true",
127
+ help=desc["enable_multi_round_generate"],
128
+ )
129
+ parser.add_argument(
130
+ "--skip_confirm", action="store_true", help=desc["skip_confirm"]
131
+ )
132
+
133
+ revert_parser = subparsers.add_parser("revert", help=desc["revert_desc"])
134
+ revert_parser.add_argument("--file", help=desc["revert_desc"])
135
+
136
+ store_parser = subparsers.add_parser("store", help=desc["store_desc"])
137
+ store_parser.add_argument("--source_dir", default=".", help=desc["source_dir"])
138
+ store_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
139
+
140
+ index_parser = subparsers.add_parser("index", help=desc["index_desc"])
141
+ index_parser.add_argument("--file", help=desc["file"])
142
+ index_parser.add_argument("--model", default="", help=desc["model"])
143
+ index_parser.add_argument("--index_model", default="", help=desc["index_model"])
144
+ index_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
145
+ index_parser.add_argument("--project_type", default="py", help=desc["project_type"])
146
+ index_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
147
+
148
+ index_query_parser = subparsers.add_parser(
149
+ "index-query", help=desc["index_query_desc"]
150
+ ) # New subcommand
151
+ index_query_parser.add_argument("--file", help=desc["file"])
152
+ index_query_parser.add_argument("--model", default="", help=desc["model"])
153
+ index_query_parser.add_argument(
154
+ "--index_model", default="", help=desc["index_model"]
155
+ )
156
+ index_query_parser.add_argument(
157
+ "--source_dir", required=False, help=desc["source_dir"]
158
+ )
159
+ index_query_parser.add_argument("--query", help=desc["query"])
160
+ index_query_parser.add_argument(
161
+ "--index_filter_level", type=int, default=2, help=desc["index_filter_level"]
162
+ )
163
+ index_query_parser.add_argument(
164
+ "--ray_address", default="auto", help=desc["ray_address"]
165
+ )
166
+
167
+ doc_parser = subparsers.add_parser("doc", help=desc["doc_desc"])
168
+ doc_parser.add_argument("--urls", default="", help=desc["urls"])
169
+ doc_parser.add_argument("--model", default="", help=desc["model"])
170
+ doc_parser.add_argument("--target_file", default="", help=desc["target_file"])
171
+ doc_parser.add_argument("--file", default="", help=desc["file"])
172
+ doc_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
173
+ doc_parser.add_argument(
174
+ "--human_as_model", action="store_true", help=desc["human_as_model"]
175
+ )
176
+ doc_parser.add_argument(
177
+ "--urls_use_model", action="store_true", help=desc["urls_use_model"]
178
+ )
179
+ doc_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
180
+
181
+ doc_subparsers = doc_parser.add_subparsers(dest="doc_command")
182
+ doc_build_parse = doc_subparsers.add_parser("build", help="")
183
+ doc_build_parse.add_argument("--source_dir", default="", help="")
184
+ doc_build_parse.add_argument("--model", default="", help=desc["model"])
185
+ doc_build_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
186
+ doc_build_parse.add_argument("--file", default="", help=desc["file"])
187
+ doc_build_parse.add_argument(
188
+ "--ray_address", default="auto", help=desc["ray_address"]
189
+ )
190
+ doc_build_parse.add_argument(
191
+ "--required_exts", default="", help=desc["doc_build_parse_required_exts"]
192
+ )
193
+ doc_build_parse.add_argument(
194
+ "--collection", default="default", help="Collection name for indexing"
195
+ )
196
+ doc_build_parse.add_argument(
197
+ "--description", default="", help="Description of the indexed content"
198
+ )
199
+
200
+ doc_query_parse = doc_subparsers.add_parser("query", help="")
201
+ doc_query_parse.add_argument("--query", default="", help="")
202
+ doc_query_parse.add_argument("--source_dir", default=".", help="")
203
+ doc_query_parse.add_argument("--model", default="", help=desc["model"])
204
+ doc_query_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
205
+ doc_query_parse.add_argument("--file", default="", help=desc["file"])
206
+ doc_query_parse.add_argument(
207
+ "--ray_address", default="auto", help=desc["ray_address"]
208
+ )
209
+ doc_query_parse.add_argument("--execute", action="store_true", help=desc["execute"])
210
+ doc_query_parse.add_argument(
211
+ "--collections",
212
+ default="default",
213
+ help="Comma-separated list of collections to search",
214
+ )
215
+ doc_query_parse.add_argument(
216
+ "--description", default="", help="Description to route the query"
217
+ )
218
+
219
+ doc_chat_parse = doc_subparsers.add_parser("chat", help="")
220
+ doc_chat_parse.add_argument("--file", default="", help=desc["file"])
221
+ doc_chat_parse.add_argument("--model", default="", help=desc["model"])
222
+ doc_chat_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
223
+ doc_chat_parse.add_argument(
224
+ "--ray_address", default="auto", help=desc["ray_address"]
225
+ )
226
+ doc_chat_parse.add_argument("--source_dir", default=".", help="")
227
+ doc_chat_parse.add_argument(
228
+ "--collections",
229
+ default="default",
230
+ help="Comma-separated list of collections to search",
231
+ )
232
+ doc_chat_parse.add_argument(
233
+ "--description", default="", help="Description to route the query"
234
+ )
235
+
236
+ doc_serve_parse = doc_subparsers.add_parser("serve", help="")
237
+ doc_serve_parse.add_argument("--file", default="", help=desc["file"])
238
+ doc_serve_parse.add_argument("--model", default="", help=desc["model"])
239
+ doc_serve_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
240
+ doc_serve_parse.add_argument(
241
+ "--ray_address", default="auto", help=desc["ray_address"]
242
+ )
243
+ doc_serve_parse.add_argument("--source_dir", default=".", help="")
244
+ doc_serve_parse.add_argument("--host", default="", help="")
245
+ doc_serve_parse.add_argument("--port", type=int, default=8000, help="")
246
+ doc_serve_parse.add_argument("--uvicorn_log_level", default="info", help="")
247
+ doc_serve_parse.add_argument("--allow_credentials", action="store_true", help="")
248
+ doc_serve_parse.add_argument("--allowed_origins", default=["*"], help="")
249
+ doc_serve_parse.add_argument("--allowed_methods", default=["*"], help="")
250
+ doc_serve_parse.add_argument("--allowed_headers", default=["*"], help="")
251
+ doc_serve_parse.add_argument("--api_key", default="", help="")
252
+ doc_serve_parse.add_argument("--served_model_name", default="", help="")
253
+ doc_serve_parse.add_argument("--prompt_template", default="", help="")
254
+ doc_serve_parse.add_argument("--ssl_keyfile", default="", help="")
255
+ doc_serve_parse.add_argument("--ssl_certfile", default="", help="")
256
+ doc_serve_parse.add_argument("--response_role", default="assistant", help="")
257
+ doc_serve_parse.add_argument(
258
+ "--collections", default="default", help="Collection name for indexing"
259
+ )
260
+
261
+ init_parser = subparsers.add_parser("init", help=desc["init_desc"])
262
+ init_parser.add_argument("--source_dir", required=True, help=desc["init_dir"])
263
+
264
+ screenshot_parser = subparsers.add_parser(
265
+ "screenshot", help=desc["screenshot_desc"]
266
+ )
267
+ screenshot_parser.add_argument("--urls", required=True, help=desc["screenshot_url"])
268
+ screenshot_parser.add_argument(
269
+ "--output", required=True, help=desc["screenshot_output"]
270
+ )
271
+ screenshot_parser.add_argument("--source_dir", default=".", help="")
272
+
273
+ next_parser = subparsers.add_parser(
274
+ "next", help="Create a new action file based on the previous one"
275
+ )
276
+ next_parser.add_argument("name", help="Name for the new action file")
277
+ next_parser.add_argument("--source_dir", default=".", help="")
278
+
279
+ doc2html_parser = subparsers.add_parser(
280
+ "doc2html", help="Convert word/pdf document to html"
281
+ )
282
+ doc2html_parser.add_argument("--file", default="", help=desc["file"])
283
+ doc2html_parser.add_argument("--model", default="", help=desc["model"])
284
+ doc2html_parser.add_argument("--vl_model", default="", help=desc["vl_model"])
285
+ doc2html_parser.add_argument(
286
+ "--ray_address", default="auto", help=desc["ray_address"]
287
+ )
288
+ doc2html_parser.add_argument("--source_dir", default=".", help="")
289
+ doc2html_parser.add_argument(
290
+ "--urls", help="URL or path of the word/pdf document to convert"
291
+ )
292
+ doc2html_parser.add_argument(
293
+ "--output", help="Output directory to save the converted html file"
294
+ )
295
+
296
+ args = parser.parse_args()
297
+
298
+ return AutoCoderArgs(**vars(args)), args
@@ -201,14 +201,14 @@ class IndexManager:
201
201
  index_data = json.load(file)
202
202
  else:
203
203
  index_data = {}
204
-
204
+
205
205
  @byzerllm.prompt()
206
206
  def error_message(source_dir: str, file_path: str):
207
207
  """
208
208
  The source_dir is different from the path in index file (e.g. file_path:{{ file_path }} source_dir:{{ source_dir }}).
209
209
  You may need to replace the prefix with the source_dir in the index file or Just delete the index file to rebuild it.
210
210
  """
211
-
211
+
212
212
  for item in index_data.keys():
213
213
  if not item.startswith(self.source_dir):
214
214
  logger.warning(
@@ -0,0 +1 @@
1
+ __version__ = "0.1.77"
@@ -1,179 +0,0 @@
1
- import argparse
2
- from autocoder.common import AutoCoderArgs
3
- from autocoder.lang import lang_desc
4
- import locale
5
-
6
-
7
- def parse_args() -> AutoCoderArgs:
8
- system_lang, _ = locale.getdefaultlocale()
9
- lang = "zh" if system_lang and system_lang.startswith("zh") else "en"
10
- desc = lang_desc[lang]
11
-
12
- parser = argparse.ArgumentParser(description=desc["parser_desc"])
13
- subparsers = parser.add_subparsers(dest="command")
14
-
15
- parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
16
- parser.add_argument("--git_url", help=desc["git_url"])
17
- parser.add_argument("--target_file", required=False, help=desc["target_file"])
18
- parser.add_argument("--query", help=desc["query"])
19
- parser.add_argument("--template", default="common", help=desc["template"])
20
- parser.add_argument("--project_type", default="py", help=desc["project_type"])
21
- parser.add_argument("--execute", action='store_true', help=desc["execute"])
22
- parser.add_argument("--package_name", default="", help=desc["package_name"])
23
- parser.add_argument("--script_path", default="", help=desc["script_path"])
24
-
25
- parser.add_argument("--model", default="", help=desc["model"])
26
- parser.add_argument("--model_max_length", type=int, default=2000, help=desc["model_max_length"])
27
- parser.add_argument("--model_max_input_length", type=int, default=6000, help=desc["model_max_input_length"])
28
-
29
- parser.add_argument("--vl_model", default="", help=desc["vl_model"])
30
- parser.add_argument("--sd_model", default="", help=desc["sd_model"])
31
- parser.add_argument("--emb_model", default="", help=desc["emb_model"])
32
-
33
- parser.add_argument("--index_model", default="", help=desc["index_model"])
34
- parser.add_argument("--index_model_max_length", type=int, default=0, help=desc["index_model_max_length"])
35
- parser.add_argument("--index_model_max_input_length", type=int, default=0, help=desc["index_model_max_input_length"])
36
- parser.add_argument("--index_model_anti_quota_limit", type=int, default=0, help=desc["index_model_anti_quota_limit"])
37
- parser.add_argument("--index_filter_level",type=int, default=0, help=desc["index_filter_level"])
38
- parser.add_argument("--index_filter_workers",type=int, default=1, help=desc["index_filter_workers"])
39
- parser.add_argument("--index_build_workers",type=int, default=1, help=desc["index_build_workers"])
40
-
41
- parser.add_argument("--file", default=None, required=False, help=desc["file"])
42
- parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
43
- parser.add_argument("--anti_quota_limit", type=int, default=1, help=desc["anti_quota_limit"])
44
- parser.add_argument("--skip_build_index", action='store_false', help=desc["skip_build_index"])
45
- parser.add_argument("--print_request", action='store_true', help=desc["print_request"])
46
- parser.add_argument("--code_model", default="", help=desc["code_model"])
47
- parser.add_argument("--py_packages", required=False, default="", help=desc["py_packages"])
48
- parser.add_argument("--human_as_model", action='store_true', help=desc["human_as_model"])
49
- parser.add_argument("--urls", default="", help=desc["urls"])
50
- parser.add_argument("--urls_use_model", action='store_true', help=desc["urls_use_model"])
51
- parser.add_argument("--exclude_files", default="", help="")
52
-
53
- parser.add_argument("--search", default="", help="")
54
- parser.add_argument("--search_engine", default="", help=desc["search_engine"])
55
- parser.add_argument("--search_engine_token", default="",help=desc["search_engine_token"])
56
-
57
- parser.add_argument("--enable_rag_search", nargs='?', const=True, default=False, help=desc["enable_rag_search"])
58
- parser.add_argument("--enable_rag_context", nargs='?',const=True, default=False, help=desc["enable_rag_context"])
59
-
60
- parser.add_argument("--auto_merge", nargs='?', const=True, default=False, help=desc["auto_merge"])
61
-
62
- parser.add_argument("--image_file", default="", help=desc["image_file"])
63
- parser.add_argument("--image_mode", default="direct", help=desc["image_mode"])
64
- parser.add_argument("--image_max_iter",type=int, default=1, help=desc["image_max_iter"])
65
-
66
- parser.add_argument("--enable_multi_round_generate", action='store_true',help=desc["enable_multi_round_generate"])
67
- parser.add_argument("--skip_confirm", action='store_true', help=desc["skip_confirm"])
68
-
69
- revert_parser = subparsers.add_parser("revert", help=desc["revert_desc"])
70
- revert_parser.add_argument("--file", help=desc["revert_desc"])
71
-
72
- store_parser = subparsers.add_parser("store", help=desc["store_desc"])
73
- store_parser.add_argument("--source_dir", help=desc["source_dir"])
74
- store_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
75
-
76
- index_parser = subparsers.add_parser("index", help=desc["index_desc"]) # New subcommand
77
- index_parser.add_argument("--file", help=desc["file"])
78
- index_parser.add_argument("--model", default="", help=desc["model"])
79
- index_parser.add_argument("--index_model", default="", help=desc["index_model"])
80
- index_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
81
- index_parser.add_argument("--project_type", default="py", help=desc["project_type"])
82
- index_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
83
-
84
- index_query_parser = subparsers.add_parser("index-query", help=desc["index_query_desc"]) # New subcommand
85
- index_query_parser.add_argument("--file", help=desc["file"])
86
- index_query_parser.add_argument("--model", default="", help=desc["model"])
87
- index_query_parser.add_argument("--index_model", default="", help=desc["index_model"])
88
- index_query_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
89
- index_query_parser.add_argument("--query", help=desc["query"])
90
- index_query_parser.add_argument("--index_filter_level",type=int, default=2, help=desc["index_filter_level"])
91
- index_query_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
92
-
93
- doc_parser = subparsers.add_parser("doc", help=desc["doc_desc"])
94
- doc_parser.add_argument("--urls", default="", help=desc["urls"])
95
- doc_parser.add_argument("--model", default="", help=desc["model"])
96
- doc_parser.add_argument("--target_file", default="", help=desc["target_file"])
97
- doc_parser.add_argument("--file",default="", help=desc["file"])
98
- doc_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
99
- doc_parser.add_argument("--human_as_model", action='store_true', help=desc["human_as_model"])
100
- doc_parser.add_argument("--urls_use_model", action='store_true', help=desc["urls_use_model"])
101
- doc_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
102
-
103
- doc_subparsers = doc_parser.add_subparsers(dest="doc_command")
104
- doc_build_parse = doc_subparsers.add_parser("build",help="")
105
- doc_build_parse.add_argument("--source_dir", default="", help="")
106
- doc_build_parse.add_argument("--model", default="", help=desc["model"])
107
- doc_build_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
108
- doc_build_parse.add_argument("--file",default="", help=desc["file"])
109
- doc_build_parse.add_argument("--ray_address", default="auto", help=desc["ray_address"])
110
- doc_build_parse.add_argument("--required_exts", default="", help=desc["doc_build_parse_required_exts"])
111
- doc_build_parse.add_argument("--collection", default="default", help="Collection name for indexing")
112
- doc_build_parse.add_argument("--description", default="", help="Description of the indexed content")
113
-
114
- doc_query_parse = doc_subparsers.add_parser("query",help="")
115
- doc_query_parse.add_argument("--query", default="", help="")
116
- doc_query_parse.add_argument("--source_dir", default=".", help="")
117
- doc_query_parse.add_argument("--model", default="", help=desc["model"])
118
- doc_query_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
119
- doc_query_parse.add_argument("--file",default="", help=desc["file"])
120
- doc_query_parse.add_argument("--ray_address", default="auto", help=desc["ray_address"])
121
- doc_query_parse.add_argument("--execute", action='store_true', help=desc["execute"])
122
- doc_query_parse.add_argument("--collections", default="default", help="Comma-separated list of collections to search")
123
- doc_query_parse.add_argument("--description", default="", help="Description to route the query")
124
-
125
- doc_chat_parse = doc_subparsers.add_parser("chat", help="")
126
- doc_chat_parse.add_argument("--file", default="", help=desc["file"])
127
- doc_chat_parse.add_argument("--model", default="", help=desc["model"])
128
- doc_chat_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
129
- doc_chat_parse.add_argument("--ray_address", default="auto", help=desc["ray_address"])
130
- doc_chat_parse.add_argument("--source_dir", default=".", help="")
131
- doc_chat_parse.add_argument("--collections", default="default", help="Comma-separated list of collections to search")
132
- doc_chat_parse.add_argument("--description", default="", help="Description to route the query")
133
-
134
- doc_serve_parse = doc_subparsers.add_parser("serve", help="")
135
- doc_serve_parse.add_argument("--file", default="", help=desc["file"])
136
- doc_serve_parse.add_argument("--model", default="", help=desc["model"])
137
- doc_serve_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
138
- doc_serve_parse.add_argument("--ray_address", default="auto", help=desc["ray_address"])
139
- doc_serve_parse.add_argument("--source_dir", default=".", help="")
140
- doc_serve_parse.add_argument("--host", default="", help="")
141
- doc_serve_parse.add_argument("--port", type=int, default=8000, help="")
142
- doc_serve_parse.add_argument("--uvicorn_log_level", default="info", help="")
143
- doc_serve_parse.add_argument("--allow_credentials", action='store_true', help="")
144
- doc_serve_parse.add_argument("--allowed_origins", default=["*"], help="")
145
- doc_serve_parse.add_argument("--allowed_methods", default=["*"], help="")
146
- doc_serve_parse.add_argument("--allowed_headers", default=["*"], help="")
147
- doc_serve_parse.add_argument("--api_key", default="", help="")
148
- doc_serve_parse.add_argument("--served_model_name", default="", help="")
149
- doc_serve_parse.add_argument("--prompt_template", default="", help="")
150
- doc_serve_parse.add_argument("--ssl_keyfile", default="", help="")
151
- doc_serve_parse.add_argument("--ssl_certfile", default="", help="")
152
- doc_serve_parse.add_argument("--response_role", default="assistant", help="")
153
- doc_serve_parse.add_argument("--collections", default="default", help="Collection name for indexing")
154
-
155
- init_parser = subparsers.add_parser("init", help=desc["init_desc"])
156
- init_parser.add_argument("--source_dir", required=True, help=desc["init_dir"])
157
-
158
- screenshot_parser = subparsers.add_parser("screenshot", help=desc["screenshot_desc"])
159
- screenshot_parser.add_argument("--urls", required=True, help=desc["screenshot_url"])
160
- screenshot_parser.add_argument("--output", required=True, help=desc["screenshot_output"])
161
- screenshot_parser.add_argument("--source_dir", default=".", help="")
162
-
163
- next_parser = subparsers.add_parser("next", help="Create a new action file based on the previous one")
164
- next_parser.add_argument("name", help="Name for the new action file")
165
- next_parser.add_argument("--source_dir", default=".", help="")
166
-
167
- doc2html_parser = subparsers.add_parser("doc2html", help="Convert word/pdf document to html")
168
- doc2html_parser.add_argument("--file", default="", help=desc["file"])
169
- doc2html_parser.add_argument("--model", default="", help=desc["model"])
170
- doc2html_parser.add_argument("--vl_model", default="", help=desc["vl_model"])
171
- doc2html_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
172
- doc2html_parser.add_argument("--source_dir", default=".", help="")
173
- doc2html_parser.add_argument("--urls", help="URL or path of the word/pdf document to convert")
174
- doc2html_parser.add_argument("--output",help="Output directory to save the converted html file")
175
-
176
-
177
- args = parser.parse_args()
178
-
179
- return AutoCoderArgs(**vars(args)),args
@@ -1 +0,0 @@
1
- __version__ = "0.1.76"
File without changes
File without changes
File without changes