auto-coder 1.0.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

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.

Potentially problematic release.


This version of auto-coder might be problematic. Click here for more details.

Files changed (574) hide show
  1. auto_coder-2.0.0.dist-info/LICENSE +158 -0
  2. auto_coder-2.0.0.dist-info/METADATA +558 -0
  3. auto_coder-2.0.0.dist-info/RECORD +795 -0
  4. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/WHEEL +1 -1
  5. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/entry_points.txt +3 -3
  6. autocoder/__init__.py +31 -0
  7. autocoder/agent/auto_filegroup.py +32 -13
  8. autocoder/agent/auto_learn_from_commit.py +9 -1
  9. autocoder/agent/base_agentic/__init__.py +3 -0
  10. autocoder/agent/base_agentic/agent_hub.py +1 -1
  11. autocoder/agent/base_agentic/base_agent.py +235 -136
  12. autocoder/agent/base_agentic/default_tools.py +119 -118
  13. autocoder/agent/base_agentic/test_base_agent.py +1 -1
  14. autocoder/agent/base_agentic/tool_registry.py +32 -20
  15. autocoder/agent/base_agentic/tools/read_file_tool_resolver.py +24 -3
  16. autocoder/agent/base_agentic/tools/write_to_file_tool_resolver.py +24 -11
  17. autocoder/agent/base_agentic/types.py +42 -0
  18. autocoder/agent/entry_command_agent/chat.py +73 -59
  19. autocoder/auto_coder.py +31 -40
  20. autocoder/auto_coder_rag.py +11 -1084
  21. autocoder/auto_coder_runner.py +970 -2345
  22. autocoder/auto_coder_terminal.py +26 -0
  23. autocoder/auto_coder_terminal_v3.py +190 -0
  24. autocoder/chat/conf_command.py +224 -124
  25. autocoder/chat/models_command.py +361 -299
  26. autocoder/chat/rules_command.py +79 -31
  27. autocoder/chat_auto_coder.py +988 -398
  28. autocoder/chat_auto_coder_lang.py +23 -732
  29. autocoder/commands/auto_command.py +25 -8
  30. autocoder/commands/auto_web.py +1 -1
  31. autocoder/commands/tools.py +44 -44
  32. autocoder/common/__init__.py +150 -128
  33. autocoder/common/ac_style_command_parser/__init__.py +39 -2
  34. autocoder/common/ac_style_command_parser/config.py +422 -0
  35. autocoder/common/ac_style_command_parser/parser.py +292 -78
  36. autocoder/common/ac_style_command_parser/test_parser.py +241 -16
  37. autocoder/common/ac_style_command_parser/test_typed_parser.py +342 -0
  38. autocoder/common/ac_style_command_parser/typed_parser.py +653 -0
  39. autocoder/common/action_yml_file_manager.py +25 -13
  40. autocoder/common/agent_events/__init__.py +52 -0
  41. autocoder/common/agent_events/agent_event_emitter.py +193 -0
  42. autocoder/common/agent_events/event_factory.py +177 -0
  43. autocoder/common/agent_events/examples.py +307 -0
  44. autocoder/common/agent_events/types.py +113 -0
  45. autocoder/common/agent_events/utils.py +68 -0
  46. autocoder/common/agent_hooks/__init__.py +44 -0
  47. autocoder/common/agent_hooks/examples.py +582 -0
  48. autocoder/common/agent_hooks/hook_executor.py +217 -0
  49. autocoder/common/agent_hooks/hook_manager.py +288 -0
  50. autocoder/common/agent_hooks/types.py +133 -0
  51. autocoder/common/agent_hooks/utils.py +99 -0
  52. autocoder/common/agent_query_queue/queue_executor.py +324 -0
  53. autocoder/common/agent_query_queue/queue_manager.py +325 -0
  54. autocoder/common/agents/__init__.py +11 -0
  55. autocoder/common/agents/agent_manager.py +323 -0
  56. autocoder/common/agents/agent_parser.py +189 -0
  57. autocoder/common/agents/example_usage.py +344 -0
  58. autocoder/common/agents/integration_example.py +330 -0
  59. autocoder/common/agents/test_agent_parser.py +545 -0
  60. autocoder/common/async_utils.py +101 -0
  61. autocoder/common/auto_coder_lang.py +23 -972
  62. autocoder/common/autocoderargs_parser/__init__.py +14 -0
  63. autocoder/common/autocoderargs_parser/parser.py +184 -0
  64. autocoder/common/autocoderargs_parser/tests/__init__.py +1 -0
  65. autocoder/common/autocoderargs_parser/tests/test_args_parser.py +235 -0
  66. autocoder/common/autocoderargs_parser/tests/test_token_parser.py +195 -0
  67. autocoder/common/autocoderargs_parser/token_parser.py +290 -0
  68. autocoder/common/buildin_tokenizer.py +2 -4
  69. autocoder/common/code_auto_generate.py +149 -74
  70. autocoder/common/code_auto_generate_diff.py +163 -70
  71. autocoder/common/code_auto_generate_editblock.py +179 -89
  72. autocoder/common/code_auto_generate_strict_diff.py +167 -72
  73. autocoder/common/code_auto_merge_editblock.py +13 -6
  74. autocoder/common/code_modification_ranker.py +1 -1
  75. autocoder/common/command_completer.py +3 -3
  76. autocoder/common/command_file_manager/manager.py +183 -47
  77. autocoder/common/command_file_manager/test_command_file_manager.py +507 -0
  78. autocoder/common/command_templates.py +1 -1
  79. autocoder/common/conf_utils.py +2 -4
  80. autocoder/common/conversations/config.py +11 -3
  81. autocoder/common/conversations/get_conversation_manager.py +100 -2
  82. autocoder/common/conversations/llm_stats_models.py +264 -0
  83. autocoder/common/conversations/manager.py +112 -28
  84. autocoder/common/conversations/models.py +16 -2
  85. autocoder/common/conversations/storage/index_manager.py +134 -10
  86. autocoder/common/core_config/__init__.py +63 -0
  87. autocoder/common/core_config/agentic_mode_manager.py +109 -0
  88. autocoder/common/core_config/base_manager.py +123 -0
  89. autocoder/common/core_config/compatibility.py +151 -0
  90. autocoder/common/core_config/config_manager.py +156 -0
  91. autocoder/common/core_config/conversation_manager.py +31 -0
  92. autocoder/common/core_config/exclude_manager.py +72 -0
  93. autocoder/common/core_config/file_manager.py +177 -0
  94. autocoder/common/core_config/human_as_model_manager.py +129 -0
  95. autocoder/common/core_config/lib_manager.py +54 -0
  96. autocoder/common/core_config/main_manager.py +81 -0
  97. autocoder/common/core_config/mode_manager.py +126 -0
  98. autocoder/common/core_config/models.py +70 -0
  99. autocoder/common/core_config/test_memory_manager.py +1056 -0
  100. autocoder/common/env_manager.py +282 -0
  101. autocoder/common/env_manager_usage_example.py +211 -0
  102. autocoder/common/file_checkpoint/conversation_checkpoint.py +19 -19
  103. autocoder/common/file_checkpoint/manager.py +264 -48
  104. autocoder/common/file_checkpoint/test_backup.py +1 -18
  105. autocoder/common/file_checkpoint/test_manager.py +270 -1
  106. autocoder/common/file_checkpoint/test_store.py +1 -17
  107. autocoder/common/file_handler/__init__.py +23 -0
  108. autocoder/common/file_handler/active_context_handler.py +159 -0
  109. autocoder/common/file_handler/add_files_handler.py +409 -0
  110. autocoder/common/file_handler/chat_handler.py +180 -0
  111. autocoder/common/file_handler/coding_handler.py +401 -0
  112. autocoder/common/file_handler/commit_handler.py +200 -0
  113. autocoder/common/file_handler/lib_handler.py +156 -0
  114. autocoder/common/file_handler/list_files_handler.py +111 -0
  115. autocoder/common/file_handler/mcp_handler.py +268 -0
  116. autocoder/common/file_handler/models_handler.py +493 -0
  117. autocoder/common/file_handler/remove_files_handler.py +172 -0
  118. autocoder/common/git_utils.py +44 -8
  119. autocoder/common/global_cancel.py +15 -6
  120. autocoder/common/ignorefiles/test_ignore_file_utils.py +1 -1
  121. autocoder/common/international/__init__.py +31 -0
  122. autocoder/common/international/demo_international.py +92 -0
  123. autocoder/common/international/message_manager.py +157 -0
  124. autocoder/common/international/messages/__init__.py +56 -0
  125. autocoder/common/international/messages/async_command_messages.py +507 -0
  126. autocoder/common/international/messages/auto_coder_messages.py +2208 -0
  127. autocoder/common/international/messages/chat_auto_coder_messages.py +1547 -0
  128. autocoder/common/international/messages/command_help_messages.py +986 -0
  129. autocoder/common/international/messages/conversation_command_messages.py +191 -0
  130. autocoder/common/international/messages/git_helper_plugin_messages.py +159 -0
  131. autocoder/common/international/messages/queue_command_messages.py +751 -0
  132. autocoder/common/international/messages/rules_command_messages.py +77 -0
  133. autocoder/common/international/messages/sdk_messages.py +1707 -0
  134. autocoder/common/international/messages/token_helper_plugin_messages.py +361 -0
  135. autocoder/common/international/messages/tool_display_messages.py +1212 -0
  136. autocoder/common/international/messages/workflow_exception_messages.py +473 -0
  137. autocoder/common/international/test_international.py +612 -0
  138. autocoder/common/linter_core/__init__.py +28 -0
  139. autocoder/common/linter_core/base_linter.py +61 -0
  140. autocoder/common/linter_core/config_loader.py +271 -0
  141. autocoder/common/linter_core/formatters/__init__.py +0 -0
  142. autocoder/common/linter_core/formatters/base_formatter.py +38 -0
  143. autocoder/common/linter_core/formatters/raw_formatter.py +17 -0
  144. autocoder/common/linter_core/linter.py +166 -0
  145. autocoder/common/linter_core/linter_factory.py +216 -0
  146. autocoder/common/linter_core/linter_manager.py +333 -0
  147. autocoder/common/linter_core/linters/__init__.py +9 -0
  148. autocoder/common/linter_core/linters/java_linter.py +342 -0
  149. autocoder/common/linter_core/linters/python_linter.py +115 -0
  150. autocoder/common/linter_core/linters/typescript_linter.py +119 -0
  151. autocoder/common/linter_core/models/__init__.py +7 -0
  152. autocoder/common/linter_core/models/lint_result.py +91 -0
  153. autocoder/common/linter_core/models.py +33 -0
  154. autocoder/common/linter_core/tests/__init__.py +3 -0
  155. autocoder/common/linter_core/tests/test_config_loader.py +323 -0
  156. autocoder/common/linter_core/tests/test_config_loading.py +308 -0
  157. autocoder/common/linter_core/tests/test_factory_manager.py +234 -0
  158. autocoder/common/linter_core/tests/test_formatters.py +147 -0
  159. autocoder/common/linter_core/tests/test_integration.py +317 -0
  160. autocoder/common/linter_core/tests/test_java_linter.py +496 -0
  161. autocoder/common/linter_core/tests/test_linters.py +265 -0
  162. autocoder/common/linter_core/tests/test_models.py +81 -0
  163. autocoder/common/linter_core/tests/verify_config_loading.py +296 -0
  164. autocoder/common/linter_core/tests/verify_fixes.py +183 -0
  165. autocoder/common/llm_friendly_package/__init__.py +31 -0
  166. autocoder/common/llm_friendly_package/base_manager.py +102 -0
  167. autocoder/common/llm_friendly_package/docs_manager.py +121 -0
  168. autocoder/common/llm_friendly_package/library_manager.py +171 -0
  169. autocoder/common/{llm_friendly_package.py → llm_friendly_package/main_manager.py} +204 -231
  170. autocoder/common/llm_friendly_package/models.py +40 -0
  171. autocoder/common/llm_friendly_package/test_llm_friendly_package.py +536 -0
  172. autocoder/common/llms/__init__.py +15 -0
  173. autocoder/common/llms/demo_error_handling.py +85 -0
  174. autocoder/common/llms/factory.py +142 -0
  175. autocoder/common/llms/manager.py +264 -0
  176. autocoder/common/llms/pricing.py +121 -0
  177. autocoder/common/llms/registry.py +288 -0
  178. autocoder/common/llms/schema.py +77 -0
  179. autocoder/common/llms/simple_demo.py +45 -0
  180. autocoder/common/llms/test_quick_model.py +116 -0
  181. autocoder/common/llms/test_remove_functionality.py +182 -0
  182. autocoder/common/llms/tests/__init__.py +1 -0
  183. autocoder/common/llms/tests/test_manager.py +330 -0
  184. autocoder/common/llms/tests/test_registry.py +364 -0
  185. autocoder/common/mcp_tools/__init__.py +62 -0
  186. autocoder/common/{mcp_tools.py → mcp_tools/executor.py} +49 -40
  187. autocoder/common/{mcp_hub.py → mcp_tools/hub.py} +42 -68
  188. autocoder/common/{mcp_server_install.py → mcp_tools/installer.py} +16 -28
  189. autocoder/common/{mcp_server.py → mcp_tools/server.py} +176 -48
  190. autocoder/common/mcp_tools/test_keyboard_interrupt.py +93 -0
  191. autocoder/common/mcp_tools/test_mcp_tools.py +391 -0
  192. autocoder/common/{mcp_server_types.py → mcp_tools/types.py} +121 -48
  193. autocoder/common/mcp_tools/verify_functionality.py +202 -0
  194. autocoder/common/model_speed_tester.py +32 -26
  195. autocoder/common/priority_directory_finder/__init__.py +142 -0
  196. autocoder/common/priority_directory_finder/examples.py +230 -0
  197. autocoder/common/priority_directory_finder/finder.py +283 -0
  198. autocoder/common/priority_directory_finder/models.py +236 -0
  199. autocoder/common/priority_directory_finder/test_priority_directory_finder.py +431 -0
  200. autocoder/common/project_scanner/__init__.py +18 -0
  201. autocoder/common/project_scanner/compat.py +77 -0
  202. autocoder/common/project_scanner/scanner.py +436 -0
  203. autocoder/common/project_tracker/__init__.py +27 -0
  204. autocoder/common/project_tracker/api.py +228 -0
  205. autocoder/common/project_tracker/demo.py +272 -0
  206. autocoder/common/project_tracker/tracker.py +487 -0
  207. autocoder/common/project_tracker/types.py +53 -0
  208. autocoder/common/pruner/__init__.py +67 -0
  209. autocoder/common/pruner/agentic_conversation_pruner.py +651 -102
  210. autocoder/common/pruner/conversation_message_ids_api.py +386 -0
  211. autocoder/common/pruner/conversation_message_ids_manager.py +347 -0
  212. autocoder/common/pruner/conversation_message_ids_pruner.py +473 -0
  213. autocoder/common/pruner/conversation_normalizer.py +347 -0
  214. autocoder/common/pruner/conversation_pruner.py +26 -6
  215. autocoder/common/pruner/test_agentic_conversation_pruner.py +554 -112
  216. autocoder/common/pruner/test_conversation_normalizer.py +502 -0
  217. autocoder/common/pruner/test_tool_content_detector.py +324 -0
  218. autocoder/common/pruner/tool_content_detector.py +227 -0
  219. autocoder/common/pruner/tools/__init__.py +18 -0
  220. autocoder/common/pruner/tools/query_message_ids.py +264 -0
  221. autocoder/common/pruner/tools/test_agentic_pruning_logic.py +432 -0
  222. autocoder/common/pruner/tools/test_message_ids_pruning_only.py +192 -0
  223. autocoder/common/pull_requests/__init__.py +9 -1
  224. autocoder/common/pull_requests/utils.py +122 -1
  225. autocoder/common/rag_manager/rag_manager.py +36 -40
  226. autocoder/common/rulefiles/__init__.py +53 -1
  227. autocoder/common/rulefiles/api.py +250 -0
  228. autocoder/common/rulefiles/core/__init__.py +14 -0
  229. autocoder/common/rulefiles/core/manager.py +241 -0
  230. autocoder/common/rulefiles/core/selector.py +805 -0
  231. autocoder/common/rulefiles/models/__init__.py +20 -0
  232. autocoder/common/rulefiles/models/index.py +16 -0
  233. autocoder/common/rulefiles/models/init_rule.py +18 -0
  234. autocoder/common/rulefiles/models/rule_file.py +18 -0
  235. autocoder/common/rulefiles/models/rule_relevance.py +14 -0
  236. autocoder/common/rulefiles/models/summary.py +16 -0
  237. autocoder/common/rulefiles/test_rulefiles.py +776 -0
  238. autocoder/common/rulefiles/utils/__init__.py +34 -0
  239. autocoder/common/rulefiles/utils/monitor.py +86 -0
  240. autocoder/common/rulefiles/utils/parser.py +230 -0
  241. autocoder/common/save_formatted_log.py +67 -10
  242. autocoder/common/search_replace.py +8 -1
  243. autocoder/common/search_replace_patch/__init__.py +24 -0
  244. autocoder/common/search_replace_patch/base.py +115 -0
  245. autocoder/common/search_replace_patch/manager.py +248 -0
  246. autocoder/common/search_replace_patch/patch_replacer.py +304 -0
  247. autocoder/common/search_replace_patch/similarity_replacer.py +306 -0
  248. autocoder/common/search_replace_patch/string_replacer.py +181 -0
  249. autocoder/common/search_replace_patch/tests/__init__.py +3 -0
  250. autocoder/common/search_replace_patch/tests/run_tests.py +126 -0
  251. autocoder/common/search_replace_patch/tests/test_base.py +188 -0
  252. autocoder/common/search_replace_patch/tests/test_empty_line_insert.py +233 -0
  253. autocoder/common/search_replace_patch/tests/test_integration.py +389 -0
  254. autocoder/common/search_replace_patch/tests/test_manager.py +351 -0
  255. autocoder/common/search_replace_patch/tests/test_patch_replacer.py +316 -0
  256. autocoder/common/search_replace_patch/tests/test_regex_replacer.py +306 -0
  257. autocoder/common/search_replace_patch/tests/test_similarity_replacer.py +384 -0
  258. autocoder/common/shell_commands/__init__.py +197 -0
  259. autocoder/common/shell_commands/background_process_notifier.py +346 -0
  260. autocoder/common/shell_commands/command_executor.py +1127 -0
  261. autocoder/common/shell_commands/error_recovery.py +541 -0
  262. autocoder/common/shell_commands/exceptions.py +120 -0
  263. autocoder/common/shell_commands/interactive_executor.py +476 -0
  264. autocoder/common/shell_commands/interactive_pexpect_process.py +623 -0
  265. autocoder/common/shell_commands/interactive_process.py +744 -0
  266. autocoder/common/shell_commands/interactive_session_manager.py +1014 -0
  267. autocoder/common/shell_commands/monitoring.py +529 -0
  268. autocoder/common/shell_commands/process_cleanup.py +386 -0
  269. autocoder/common/shell_commands/process_manager.py +606 -0
  270. autocoder/common/shell_commands/test_interactive_pexpect_process.py +281 -0
  271. autocoder/common/shell_commands/tests/__init__.py +6 -0
  272. autocoder/common/shell_commands/tests/conftest.py +118 -0
  273. autocoder/common/shell_commands/tests/test_background_process_notifier.py +703 -0
  274. autocoder/common/shell_commands/tests/test_command_executor.py +448 -0
  275. autocoder/common/shell_commands/tests/test_error_recovery.py +305 -0
  276. autocoder/common/shell_commands/tests/test_exceptions.py +299 -0
  277. autocoder/common/shell_commands/tests/test_execute_batch.py +588 -0
  278. autocoder/common/shell_commands/tests/test_indented_batch_commands.py +244 -0
  279. autocoder/common/shell_commands/tests/test_integration.py +664 -0
  280. autocoder/common/shell_commands/tests/test_monitoring.py +546 -0
  281. autocoder/common/shell_commands/tests/test_performance.py +632 -0
  282. autocoder/common/shell_commands/tests/test_process_cleanup.py +397 -0
  283. autocoder/common/shell_commands/tests/test_process_manager.py +606 -0
  284. autocoder/common/shell_commands/tests/test_timeout_config.py +343 -0
  285. autocoder/common/shell_commands/tests/test_timeout_manager.py +520 -0
  286. autocoder/common/shell_commands/timeout_config.py +315 -0
  287. autocoder/common/shell_commands/timeout_manager.py +352 -0
  288. autocoder/common/terminal_paste/__init__.py +14 -0
  289. autocoder/common/terminal_paste/demo.py +145 -0
  290. autocoder/common/terminal_paste/demo_paste_functionality.py +95 -0
  291. autocoder/common/terminal_paste/paste_handler.py +200 -0
  292. autocoder/common/terminal_paste/paste_manager.py +118 -0
  293. autocoder/common/terminal_paste/tests/__init__.py +1 -0
  294. autocoder/common/terminal_paste/tests/test_paste_handler.py +182 -0
  295. autocoder/common/terminal_paste/tests/test_paste_manager.py +126 -0
  296. autocoder/common/terminal_paste/utils.py +163 -0
  297. autocoder/common/test_autocoder_args.py +232 -0
  298. autocoder/common/test_env_manager.py +173 -0
  299. autocoder/common/test_env_manager_integration.py +159 -0
  300. autocoder/common/text_similarity/__init__.py +9 -0
  301. autocoder/common/text_similarity/demo.py +216 -0
  302. autocoder/common/text_similarity/examples.py +266 -0
  303. autocoder/common/text_similarity/test_text_similarity.py +306 -0
  304. autocoder/common/text_similarity/text_similarity.py +194 -0
  305. autocoder/common/text_similarity/utils.py +125 -0
  306. autocoder/common/todos/__init__.py +61 -0
  307. autocoder/common/todos/cache/__init__.py +16 -0
  308. autocoder/common/todos/cache/base_cache.py +89 -0
  309. autocoder/common/todos/cache/cache_manager.py +228 -0
  310. autocoder/common/todos/cache/memory_cache.py +225 -0
  311. autocoder/common/todos/config.py +155 -0
  312. autocoder/common/todos/exceptions.py +35 -0
  313. autocoder/common/todos/get_todo_manager.py +161 -0
  314. autocoder/common/todos/manager.py +537 -0
  315. autocoder/common/todos/models.py +239 -0
  316. autocoder/common/todos/storage/__init__.py +14 -0
  317. autocoder/common/todos/storage/base_storage.py +76 -0
  318. autocoder/common/todos/storage/file_storage.py +278 -0
  319. autocoder/common/tokens/counter.py +24 -2
  320. autocoder/common/tools_manager/__init__.py +17 -0
  321. autocoder/common/tools_manager/examples.py +162 -0
  322. autocoder/common/tools_manager/manager.py +385 -0
  323. autocoder/common/tools_manager/models.py +39 -0
  324. autocoder/common/tools_manager/test_tools_manager.py +303 -0
  325. autocoder/common/tools_manager/utils.py +191 -0
  326. autocoder/common/v2/agent/agentic_callbacks.py +270 -0
  327. autocoder/common/v2/agent/agentic_edit.py +2699 -1856
  328. autocoder/common/v2/agent/agentic_edit_change_manager.py +474 -0
  329. autocoder/common/v2/agent/agentic_edit_tools/__init__.py +35 -1
  330. autocoder/common/v2/agent/agentic_edit_tools/ac_mod_list_tool_resolver.py +279 -0
  331. autocoder/common/v2/agent/agentic_edit_tools/ac_mod_write_tool_resolver.py +10 -1
  332. autocoder/common/v2/agent/agentic_edit_tools/background_task_tool_resolver.py +1167 -0
  333. autocoder/common/v2/agent/agentic_edit_tools/base_tool_resolver.py +2 -2
  334. autocoder/common/v2/agent/agentic_edit_tools/conversation_message_ids_read_tool_resolver.py +214 -0
  335. autocoder/common/v2/agent/agentic_edit_tools/conversation_message_ids_write_tool_resolver.py +299 -0
  336. autocoder/common/v2/agent/agentic_edit_tools/count_tokens_tool_resolver.py +290 -0
  337. autocoder/common/v2/agent/agentic_edit_tools/execute_command_tool_resolver.py +564 -29
  338. autocoder/common/v2/agent/agentic_edit_tools/execute_workflow_tool_resolver.py +485 -0
  339. autocoder/common/v2/agent/agentic_edit_tools/extract_to_text_tool_resolver.py +225 -0
  340. autocoder/common/v2/agent/agentic_edit_tools/lint_report.py +79 -0
  341. autocoder/common/v2/agent/agentic_edit_tools/linter_config_models.py +343 -0
  342. autocoder/common/v2/agent/agentic_edit_tools/linter_enabled_tool_resolver.py +189 -0
  343. autocoder/common/v2/agent/agentic_edit_tools/list_files_tool_resolver.py +169 -101
  344. autocoder/common/v2/agent/agentic_edit_tools/load_extra_document_tool_resolver.py +349 -0
  345. autocoder/common/v2/agent/agentic_edit_tools/read_file_tool_resolver.py +243 -50
  346. autocoder/common/v2/agent/agentic_edit_tools/replace_in_file_tool_resolver.py +667 -147
  347. autocoder/common/v2/agent/agentic_edit_tools/run_named_subagents_tool_resolver.py +691 -0
  348. autocoder/common/v2/agent/agentic_edit_tools/search_files_tool_resolver.py +410 -86
  349. autocoder/common/v2/agent/agentic_edit_tools/session_interactive_tool_resolver.py +115 -0
  350. autocoder/common/v2/agent/agentic_edit_tools/session_start_tool_resolver.py +190 -0
  351. autocoder/common/v2/agent/agentic_edit_tools/session_stop_tool_resolver.py +76 -0
  352. autocoder/common/v2/agent/agentic_edit_tools/test_write_to_file_tool_resolver.py +207 -192
  353. autocoder/common/v2/agent/agentic_edit_tools/todo_read_tool_resolver.py +80 -63
  354. autocoder/common/v2/agent/agentic_edit_tools/todo_write_tool_resolver.py +237 -233
  355. autocoder/common/v2/agent/agentic_edit_tools/use_mcp_tool_resolver.py +2 -2
  356. autocoder/common/v2/agent/agentic_edit_tools/web_crawl_tool_resolver.py +557 -0
  357. autocoder/common/v2/agent/agentic_edit_tools/web_search_tool_resolver.py +600 -0
  358. autocoder/common/v2/agent/agentic_edit_tools/write_to_file_tool_resolver.py +56 -121
  359. autocoder/common/v2/agent/agentic_edit_types.py +343 -9
  360. autocoder/common/v2/agent/runner/__init__.py +3 -3
  361. autocoder/common/v2/agent/runner/base_runner.py +12 -26
  362. autocoder/common/v2/agent/runner/{event_runner.py → file_based_event_runner.py} +3 -2
  363. autocoder/common/v2/agent/runner/sdk_runner.py +150 -8
  364. autocoder/common/v2/agent/runner/terminal_runner.py +170 -57
  365. autocoder/common/v2/agent/runner/tool_display.py +557 -159
  366. autocoder/common/v2/agent/test_agentic_callbacks.py +265 -0
  367. autocoder/common/v2/agent/test_agentic_edit.py +194 -0
  368. autocoder/common/v2/agent/tool_caller/__init__.py +24 -0
  369. autocoder/common/v2/agent/tool_caller/default_tool_resolver_map.py +135 -0
  370. autocoder/common/v2/agent/tool_caller/integration_test.py +172 -0
  371. autocoder/common/v2/agent/tool_caller/plugins/__init__.py +14 -0
  372. autocoder/common/v2/agent/tool_caller/plugins/base_plugin.py +126 -0
  373. autocoder/common/v2/agent/tool_caller/plugins/examples/__init__.py +13 -0
  374. autocoder/common/v2/agent/tool_caller/plugins/examples/logging_plugin.py +164 -0
  375. autocoder/common/v2/agent/tool_caller/plugins/examples/security_filter_plugin.py +198 -0
  376. autocoder/common/v2/agent/tool_caller/plugins/plugin_interface.py +141 -0
  377. autocoder/common/v2/agent/tool_caller/test_tool_caller.py +278 -0
  378. autocoder/common/v2/agent/tool_caller/tool_call_plugin_manager.py +331 -0
  379. autocoder/common/v2/agent/tool_caller/tool_caller.py +337 -0
  380. autocoder/common/v2/agent/tool_caller/usage_example.py +193 -0
  381. autocoder/common/v2/code_agentic_editblock_manager.py +4 -4
  382. autocoder/common/v2/code_auto_generate.py +136 -78
  383. autocoder/common/v2/code_auto_generate_diff.py +135 -79
  384. autocoder/common/v2/code_auto_generate_editblock.py +174 -99
  385. autocoder/common/v2/code_auto_generate_strict_diff.py +151 -71
  386. autocoder/common/v2/code_auto_merge.py +1 -1
  387. autocoder/common/v2/code_auto_merge_editblock.py +13 -1
  388. autocoder/common/v2/code_diff_manager.py +3 -3
  389. autocoder/common/v2/code_editblock_manager.py +4 -14
  390. autocoder/common/v2/code_manager.py +1 -1
  391. autocoder/common/v2/code_strict_diff_manager.py +2 -2
  392. autocoder/common/wrap_llm_hint/__init__.py +10 -0
  393. autocoder/common/wrap_llm_hint/test_wrap_llm_hint.py +1067 -0
  394. autocoder/common/wrap_llm_hint/utils.py +432 -0
  395. autocoder/common/wrap_llm_hint/wrap_llm_hint.py +323 -0
  396. autocoder/completer/__init__.py +8 -0
  397. autocoder/completer/command_completer_v2.py +1051 -0
  398. autocoder/default_project/__init__.py +501 -0
  399. autocoder/dispacher/__init__.py +4 -12
  400. autocoder/dispacher/actions/action.py +165 -7
  401. autocoder/dispacher/actions/plugins/action_regex_project.py +2 -2
  402. autocoder/index/entry.py +116 -124
  403. autocoder/{agent → index/filter}/agentic_filter.py +322 -333
  404. autocoder/index/filter/normal_filter.py +5 -11
  405. autocoder/index/filter/quick_filter.py +1 -1
  406. autocoder/index/index.py +36 -9
  407. autocoder/index/tests/__init__.py +1 -0
  408. autocoder/index/tests/run_tests.py +195 -0
  409. autocoder/index/tests/test_entry.py +303 -0
  410. autocoder/index/tests/test_index_manager.py +314 -0
  411. autocoder/index/tests/test_module_integration.py +300 -0
  412. autocoder/index/tests/test_symbols_utils.py +183 -0
  413. autocoder/inner/__init__.py +4 -0
  414. autocoder/inner/agentic.py +932 -0
  415. autocoder/inner/async_command_handler.py +992 -0
  416. autocoder/inner/conversation_command_handlers.py +623 -0
  417. autocoder/inner/merge_command_handler.py +213 -0
  418. autocoder/inner/queue_command_handler.py +684 -0
  419. autocoder/models.py +95 -266
  420. autocoder/plugins/git_helper_plugin.py +31 -29
  421. autocoder/plugins/token_helper_plugin.py +65 -46
  422. autocoder/pyproject/__init__.py +32 -29
  423. autocoder/rag/agentic_rag.py +215 -75
  424. autocoder/rag/cache/simple_cache.py +1 -2
  425. autocoder/rag/loaders/image_loader.py +1 -1
  426. autocoder/rag/long_context_rag.py +42 -26
  427. autocoder/rag/qa_conversation_strategy.py +1 -1
  428. autocoder/rag/terminal/__init__.py +17 -0
  429. autocoder/rag/terminal/args.py +581 -0
  430. autocoder/rag/terminal/bootstrap.py +61 -0
  431. autocoder/rag/terminal/command_handlers.py +653 -0
  432. autocoder/rag/terminal/formatters/__init__.py +20 -0
  433. autocoder/rag/terminal/formatters/base.py +70 -0
  434. autocoder/rag/terminal/formatters/json_format.py +66 -0
  435. autocoder/rag/terminal/formatters/stream_json.py +95 -0
  436. autocoder/rag/terminal/formatters/text.py +28 -0
  437. autocoder/rag/terminal/init.py +120 -0
  438. autocoder/rag/terminal/utils.py +106 -0
  439. autocoder/rag/test_agentic_rag.py +389 -0
  440. autocoder/rag/test_doc_filter.py +3 -3
  441. autocoder/rag/test_long_context_rag.py +1 -1
  442. autocoder/rag/test_token_limiter.py +517 -10
  443. autocoder/rag/token_counter.py +3 -0
  444. autocoder/rag/token_limiter.py +19 -15
  445. autocoder/rag/tools/__init__.py +26 -2
  446. autocoder/rag/tools/bochaai_example.py +343 -0
  447. autocoder/rag/tools/bochaai_sdk.py +541 -0
  448. autocoder/rag/tools/metaso_example.py +268 -0
  449. autocoder/rag/tools/metaso_sdk.py +417 -0
  450. autocoder/rag/tools/recall_tool.py +28 -7
  451. autocoder/rag/tools/run_integration_tests.py +204 -0
  452. autocoder/rag/tools/test_all_providers.py +318 -0
  453. autocoder/rag/tools/test_bochaai_integration.py +482 -0
  454. autocoder/rag/tools/test_final_integration.py +215 -0
  455. autocoder/rag/tools/test_metaso_integration.py +424 -0
  456. autocoder/rag/tools/test_metaso_real.py +171 -0
  457. autocoder/rag/tools/test_web_crawl_tool.py +639 -0
  458. autocoder/rag/tools/test_web_search_tool.py +509 -0
  459. autocoder/rag/tools/todo_read_tool.py +202 -0
  460. autocoder/rag/tools/todo_write_tool.py +412 -0
  461. autocoder/rag/tools/web_crawl_tool.py +634 -0
  462. autocoder/rag/tools/web_search_tool.py +558 -0
  463. autocoder/rag/tools/web_tools_example.py +119 -0
  464. autocoder/rag/types.py +16 -0
  465. autocoder/rag/variable_holder.py +4 -2
  466. autocoder/rags.py +86 -79
  467. autocoder/regexproject/__init__.py +23 -21
  468. autocoder/sdk/__init__.py +46 -190
  469. autocoder/sdk/api.py +370 -0
  470. autocoder/sdk/async_runner/__init__.py +26 -0
  471. autocoder/sdk/async_runner/async_executor.py +650 -0
  472. autocoder/sdk/async_runner/async_handler.py +356 -0
  473. autocoder/sdk/async_runner/markdown_processor.py +595 -0
  474. autocoder/sdk/async_runner/task_metadata.py +284 -0
  475. autocoder/sdk/async_runner/worktree_manager.py +438 -0
  476. autocoder/sdk/cli/__init__.py +2 -5
  477. autocoder/sdk/cli/formatters.py +28 -204
  478. autocoder/sdk/cli/handlers.py +77 -44
  479. autocoder/sdk/cli/main.py +154 -171
  480. autocoder/sdk/cli/options.py +95 -22
  481. autocoder/sdk/constants.py +139 -51
  482. autocoder/sdk/core/auto_coder_core.py +484 -109
  483. autocoder/sdk/core/bridge.py +297 -115
  484. autocoder/sdk/exceptions.py +18 -12
  485. autocoder/sdk/formatters/__init__.py +19 -0
  486. autocoder/sdk/formatters/input.py +64 -0
  487. autocoder/sdk/formatters/output.py +247 -0
  488. autocoder/sdk/formatters/stream.py +54 -0
  489. autocoder/sdk/models/__init__.py +6 -5
  490. autocoder/sdk/models/options.py +55 -18
  491. autocoder/sdk/utils/formatters.py +27 -195
  492. autocoder/suffixproject/__init__.py +28 -25
  493. autocoder/terminal/__init__.py +14 -0
  494. autocoder/terminal/app.py +454 -0
  495. autocoder/terminal/args.py +32 -0
  496. autocoder/terminal/bootstrap.py +178 -0
  497. autocoder/terminal/command_processor.py +521 -0
  498. autocoder/terminal/command_registry.py +57 -0
  499. autocoder/terminal/help.py +97 -0
  500. autocoder/terminal/tasks/__init__.py +5 -0
  501. autocoder/terminal/tasks/background.py +77 -0
  502. autocoder/terminal/tasks/task_event.py +70 -0
  503. autocoder/terminal/ui/__init__.py +13 -0
  504. autocoder/terminal/ui/completer.py +268 -0
  505. autocoder/terminal/ui/keybindings.py +75 -0
  506. autocoder/terminal/ui/session.py +41 -0
  507. autocoder/terminal/ui/toolbar.py +64 -0
  508. autocoder/terminal/utils/__init__.py +13 -0
  509. autocoder/terminal/utils/errors.py +18 -0
  510. autocoder/terminal/utils/paths.py +19 -0
  511. autocoder/terminal/utils/shell.py +43 -0
  512. autocoder/terminal_v3/__init__.py +10 -0
  513. autocoder/terminal_v3/app.py +201 -0
  514. autocoder/terminal_v3/handlers/__init__.py +5 -0
  515. autocoder/terminal_v3/handlers/command_handler.py +131 -0
  516. autocoder/terminal_v3/models/__init__.py +6 -0
  517. autocoder/terminal_v3/models/conversation_buffer.py +214 -0
  518. autocoder/terminal_v3/models/message.py +50 -0
  519. autocoder/terminal_v3/models/tool_display.py +247 -0
  520. autocoder/terminal_v3/ui/__init__.py +7 -0
  521. autocoder/terminal_v3/ui/keybindings.py +56 -0
  522. autocoder/terminal_v3/ui/layout.py +141 -0
  523. autocoder/terminal_v3/ui/styles.py +43 -0
  524. autocoder/tsproject/__init__.py +23 -23
  525. autocoder/utils/auto_coder_utils/chat_stream_out.py +1 -1
  526. autocoder/utils/llms.py +88 -80
  527. autocoder/utils/math_utils.py +101 -0
  528. autocoder/utils/model_provider_selector.py +16 -4
  529. autocoder/utils/operate_config_api.py +33 -5
  530. autocoder/utils/thread_utils.py +2 -2
  531. autocoder/version.py +4 -2
  532. autocoder/workflow_agents/__init__.py +84 -0
  533. autocoder/workflow_agents/agent.py +143 -0
  534. autocoder/workflow_agents/exceptions.py +573 -0
  535. autocoder/workflow_agents/executor.py +489 -0
  536. autocoder/workflow_agents/loader.py +737 -0
  537. autocoder/workflow_agents/runner.py +267 -0
  538. autocoder/workflow_agents/types.py +172 -0
  539. autocoder/workflow_agents/utils.py +434 -0
  540. autocoder/workflow_agents/workflow_manager.py +211 -0
  541. auto_coder-1.0.0.dist-info/METADATA +0 -396
  542. auto_coder-1.0.0.dist-info/RECORD +0 -442
  543. auto_coder-1.0.0.dist-info/licenses/LICENSE +0 -201
  544. autocoder/auto_coder_server.py +0 -672
  545. autocoder/benchmark.py +0 -138
  546. autocoder/common/ac_style_command_parser/example.py +0 -7
  547. autocoder/common/cleaner.py +0 -31
  548. autocoder/common/command_completer_v2.py +0 -615
  549. autocoder/common/context_pruner.py +0 -477
  550. autocoder/common/conversation_pruner.py +0 -132
  551. autocoder/common/directory_cache/__init__.py +0 -1
  552. autocoder/common/directory_cache/cache.py +0 -192
  553. autocoder/common/directory_cache/test_cache.py +0 -190
  554. autocoder/common/file_checkpoint/examples.py +0 -217
  555. autocoder/common/llm_friendly_package_example.py +0 -138
  556. autocoder/common/llm_friendly_package_test.py +0 -63
  557. autocoder/common/pull_requests/test_module.py +0 -1
  558. autocoder/common/rulefiles/autocoderrules_utils.py +0 -484
  559. autocoder/common/text.py +0 -30
  560. autocoder/common/v2/agent/agentic_edit_tools/list_package_info_tool_resolver.py +0 -42
  561. autocoder/common/v2/agent/agentic_edit_tools/test_execute_command_tool_resolver.py +0 -70
  562. autocoder/common/v2/agent/agentic_edit_tools/test_search_files_tool_resolver.py +0 -163
  563. autocoder/common/v2/agent/agentic_tool_display.py +0 -183
  564. autocoder/plugins/dynamic_completion_example.py +0 -148
  565. autocoder/plugins/sample_plugin.py +0 -160
  566. autocoder/sdk/cli/__main__.py +0 -26
  567. autocoder/sdk/cli/completion_wrapper.py +0 -38
  568. autocoder/sdk/cli/install_completion.py +0 -301
  569. autocoder/sdk/models/messages.py +0 -209
  570. autocoder/sdk/session/__init__.py +0 -32
  571. autocoder/sdk/session/session.py +0 -106
  572. autocoder/sdk/session/session_manager.py +0 -56
  573. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/top_level.txt +0 -0
  574. /autocoder/{sdk/example.py → common/agent_query_queue/__init__.py} +0 -0
@@ -0,0 +1,482 @@
1
+
2
+
3
+ """
4
+ 测试 BochaAI SDK 集成
5
+
6
+ 该文件用于测试 BochaAI SDK 的功能以及与 web_search_tool 和 web_crawl_tool 的集成。
7
+ """
8
+
9
+ import os
10
+ import sys
11
+ import json
12
+ from typing import Dict, Any
13
+ from unittest.mock import Mock, patch, MagicMock
14
+
15
+ # 添加项目路径
16
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../..'))
17
+
18
+ from autocoder.rag.tools.bochaai_sdk import BochaAIClient, BochaAIWebPage, BochaAIImage, BochaAISearchResponse
19
+ from autocoder.rag.tools.web_search_tool import WebSearchTool, WebSearchToolResolver
20
+ from autocoder.rag.tools.web_crawl_tool import WebCrawlTool, WebCrawlToolResolver
21
+
22
+
23
+ def test_bochaai_sdk():
24
+ """测试 BochaAI SDK 基本功能"""
25
+ print("=" * 60)
26
+ print("测试 BochaAI SDK 基本功能")
27
+ print("=" * 60)
28
+
29
+ # 测试数据类
30
+ webpage = BochaAIWebPage(
31
+ name="测试网页标题",
32
+ url="https://example.com",
33
+ snippet="这是一个测试网页的摘要内容",
34
+ display_url="https://example.com",
35
+ summary="详细的摘要内容",
36
+ site_name="示例网站",
37
+ date_published="2024-01-01T00:00:00Z"
38
+ )
39
+
40
+ print("\n1. 测试 BochaAIWebPage 数据类:")
41
+ print(f" 标题: {webpage.name}")
42
+ print(f" 链接: {webpage.url}")
43
+ print(f" 摘要: {webpage.snippet[:30]}...")
44
+ print(f" 转换为字典: {json.dumps(webpage.to_dict(), ensure_ascii=False, indent=2)}")
45
+
46
+ # 测试图片类
47
+ image = BochaAIImage(
48
+ thumbnail_url="https://example.com/thumb.jpg",
49
+ content_url="https://example.com/image.jpg",
50
+ host_page_url="https://example.com",
51
+ width=800,
52
+ height=600,
53
+ name="测试图片"
54
+ )
55
+
56
+ print("\n2. 测试 BochaAIImage 数据类:")
57
+ print(f" 图片URL: {image.content_url}")
58
+ print(f" 尺寸: {image.width}x{image.height}")
59
+
60
+ # 测试响应类
61
+ search_response = BochaAISearchResponse(
62
+ success=True,
63
+ query="测试查询",
64
+ total_matches=100,
65
+ webpages=[webpage],
66
+ images=[image]
67
+ )
68
+
69
+ print("\n3. 测试 BochaAISearchResponse 数据类:")
70
+ print(f" 成功状态: {search_response.success}")
71
+ print(f" 查询: {search_response.query}")
72
+ print(f" 总匹配数: {search_response.total_matches}")
73
+ print(f" 网页数量: {len(search_response.webpages)}")
74
+ print(f" 图片数量: {len(search_response.images)}")
75
+
76
+ print("\n✅ BochaAI SDK 数据类测试通过")
77
+
78
+
79
+ def test_bochaai_client_mock():
80
+ """使用 mock 测试 BochaAI Client"""
81
+ print("\n" + "=" * 60)
82
+ print("测试 BochaAI Client (使用 Mock)")
83
+ print("=" * 60)
84
+
85
+ # Mock API 响应
86
+ mock_search_response = {
87
+ "code": 200,
88
+ "log_id": "test_log_id",
89
+ "msg": None,
90
+ "data": {
91
+ "_type": "SearchResponse",
92
+ "queryContext": {
93
+ "originalQuery": "Python编程"
94
+ },
95
+ "webPages": {
96
+ "totalEstimatedMatches": 1000000,
97
+ "value": [
98
+ {
99
+ "name": "Python 官方网站",
100
+ "url": "https://www.python.org",
101
+ "displayUrl": "https://www.python.org",
102
+ "snippet": "Python 是一种编程语言,可让您快速工作并更有效地集成系统。",
103
+ "summary": "Python 官方网站提供了完整的文档、教程和下载资源。",
104
+ "siteName": "Python.org",
105
+ "siteIcon": "https://www.python.org/favicon.ico",
106
+ "datePublished": "2024-01-15T00:00:00Z"
107
+ },
108
+ {
109
+ "name": "Python 教程",
110
+ "url": "https://docs.python.org/tutorial",
111
+ "displayUrl": "https://docs.python.org/tutorial",
112
+ "snippet": "本教程非正式地向读者介绍了 Python 语言和系统的基本概念和功能。",
113
+ "siteName": "Python Docs"
114
+ }
115
+ ]
116
+ },
117
+ "images": {
118
+ "value": [
119
+ {
120
+ "thumbnailUrl": "https://example.com/python-logo-thumb.png",
121
+ "contentUrl": "https://example.com/python-logo.png",
122
+ "hostPageUrl": "https://www.python.org",
123
+ "width": 200,
124
+ "height": 200,
125
+ "name": "Python Logo"
126
+ }
127
+ ]
128
+ }
129
+ }
130
+ }
131
+
132
+ mock_crawl_response = {
133
+ "success": True,
134
+ "url": "https://www.python.org",
135
+ "title": "Python Programming Language",
136
+ "content": "<html><head><title>Python Programming Language</title></head><body>Python content here...</body></html>",
137
+ "content_type": "text/html",
138
+ "status_code": 200
139
+ }
140
+
141
+ with patch('requests.Session') as mock_session_class:
142
+ mock_session = MagicMock()
143
+ mock_session_class.return_value = mock_session
144
+
145
+ # Mock search 请求
146
+ mock_search_resp = MagicMock()
147
+ mock_search_resp.status_code = 200
148
+ mock_search_resp.json.return_value = mock_search_response
149
+
150
+ # Mock crawl 请求
151
+ with patch('requests.get') as mock_get:
152
+ mock_crawl_resp = MagicMock()
153
+ mock_crawl_resp.status_code = 200
154
+ mock_crawl_resp.text = mock_crawl_response["content"]
155
+ mock_crawl_resp.headers = {"Content-Type": "text/html"}
156
+ mock_get.return_value = mock_crawl_resp
157
+
158
+ # 设置 mock 返回值
159
+ mock_session.post.return_value = mock_search_resp
160
+
161
+ # 测试搜索功能
162
+ client = BochaAIClient(api_key="test_api_key")
163
+
164
+ print("\n1. 测试搜索功能:")
165
+ search_result = client.search(query="Python编程", count=3, summary=True)
166
+
167
+ print(f" 搜索成功: {search_result.success}")
168
+ print(f" 结果数量: {len(search_result.webpages)}")
169
+ for idx, page in enumerate(search_result.webpages, 1):
170
+ print(f" {idx}. {page.name}")
171
+ print(f" URL: {page.url}")
172
+ if page.summary:
173
+ print(f" 摘要: {page.summary[:50]}...")
174
+
175
+ # 测试爬取功能
176
+ print("\n2. 测试爬取功能:")
177
+ crawl_result = client.crawl_page("https://www.python.org")
178
+ print(f" 爬取成功: {crawl_result.get('success', False)}")
179
+ print(f" 页面标题: {crawl_result.get('title', 'N/A')}")
180
+ print(f" 内容长度: {crawl_result.get('content_length', 0)} 字符")
181
+
182
+ print("\n✅ BochaAI Client Mock 测试通过")
183
+
184
+
185
+ def test_web_search_tool_with_bochaai():
186
+ """测试 WebSearchTool 与 BochaAI 的集成"""
187
+ print("\n" + "=" * 60)
188
+ print("测试 WebSearchTool 与 BochaAI 集成")
189
+ print("=" * 60)
190
+
191
+ # 创建 mock agent 和 args
192
+ mock_agent = Mock()
193
+ mock_args = Mock()
194
+ mock_args.bochaai_api_key = "test_bochaai_key"
195
+ mock_args.metaso_api_key = None
196
+ mock_args.firecrawl_api_key = None
197
+
198
+ # 创建工具实例
199
+ tool = WebSearchTool(
200
+ query="人工智能最新进展",
201
+ limit=3,
202
+ provider="bochaai"
203
+ )
204
+
205
+ # 创建解析器
206
+ resolver = WebSearchToolResolver(mock_agent, tool, mock_args)
207
+
208
+ print("\n1. 测试工具配置:")
209
+ print(f" 查询: {tool.query}")
210
+ print(f" 限制: {tool.limit}")
211
+ print(f" 提供商: {tool.provider}")
212
+
213
+ # Mock BochaAI 客户端
214
+ with patch('autocoder.rag.tools.web_search_tool.BochaAIClient') as mock_client_class:
215
+ mock_client = MagicMock()
216
+ mock_client_class.return_value = mock_client
217
+
218
+ # 设置 mock 搜索结果
219
+ mock_search_result = MagicMock()
220
+ mock_search_result.success = True
221
+ mock_search_result.webpages = [
222
+ MagicMock(
223
+ name="AI 突破性进展",
224
+ url="https://ai-news.com/breakthrough",
225
+ snippet="最新的人工智能研究取得重大突破...",
226
+ summary="详细的研究成果说明...",
227
+ site_name="AI News",
228
+ date_published="2024-01-15T00:00:00Z"
229
+ ),
230
+ MagicMock(
231
+ name="深度学习新算法",
232
+ url="https://ml-research.org/new-algo",
233
+ snippet="研究人员开发出新的深度学习算法...",
234
+ summary=None,
235
+ site_name="ML Research",
236
+ date_published=None
237
+ )
238
+ ]
239
+ mock_search_result.images = []
240
+ mock_client.search.return_value = mock_search_result
241
+
242
+ # 执行搜索
243
+ result = resolver.resolve()
244
+
245
+ print("\n2. 测试搜索结果:")
246
+ print(f" 执行成功: {result.success}")
247
+ print(f" 消息: {result.message}")
248
+ print(f" 结果数量: {len(result.content)}")
249
+
250
+ if result.success and result.content:
251
+ for idx, item in enumerate(result.content[:2], 1):
252
+ print(f"\n 结果 {idx}:")
253
+ print(f" 标题: {item['title']}")
254
+ print(f" URL: {item['url']}")
255
+ print(f" 提供商: {item.get('provider', 'unknown')}")
256
+ if 'summary' in item:
257
+ print(f" 摘要: {item['summary'][:50]}...")
258
+
259
+ print("\n✅ WebSearchTool BochaAI 集成测试通过")
260
+
261
+
262
+ def test_web_crawl_tool_with_bochaai():
263
+ """测试 WebCrawlTool 与 BochaAI 的集成"""
264
+ print("\n" + "=" * 60)
265
+ print("测试 WebCrawlTool 与 BochaAI 集成")
266
+ print("=" * 60)
267
+
268
+ # 创建 mock agent 和 args
269
+ mock_agent = Mock()
270
+ mock_args = Mock()
271
+ mock_args.bochaai_api_key = "test_bochaai_key"
272
+ mock_args.metaso_api_key = None
273
+ mock_args.firecrawl_api_key = None
274
+
275
+ # 测试单页爬取
276
+ print("\n1. 测试单页爬取模式:")
277
+ tool = WebCrawlTool(
278
+ url="https://docs.python.org/tutorial",
279
+ limit=1,
280
+ provider="bochaai"
281
+ )
282
+
283
+ resolver = WebCrawlToolResolver(mock_agent, tool, mock_args)
284
+
285
+ with patch('autocoder.rag.tools.web_crawl_tool.BochaAIClient') as mock_client_class:
286
+ mock_client = MagicMock()
287
+ mock_client_class.return_value = mock_client
288
+
289
+ # Mock crawl_page 方法
290
+ mock_client.crawl_page.return_value = {
291
+ "success": True,
292
+ "url": "https://docs.python.org/tutorial",
293
+ "title": "Python 教程",
294
+ "content": "# Python 教程\n\n这是 Python 官方教程的内容...",
295
+ "content_type": "text/html",
296
+ "status_code": 200
297
+ }
298
+
299
+ result = resolver.resolve()
300
+
301
+ print(f" 执行成功: {result.success}")
302
+ print(f" 消息: {result.message}")
303
+ if result.success and result.content:
304
+ print(f" URL: {result.content[0]['url']}")
305
+ print(f" 标题: {result.content[0].get('title', 'N/A')}")
306
+ print(f" 内容长度: {len(result.content[0]['content'])} 字符")
307
+
308
+ # 测试多页爬取
309
+ print("\n2. 测试多页爬取模式:")
310
+ tool = WebCrawlTool(
311
+ url="https://docs.python.org",
312
+ limit=5,
313
+ exclude_paths="/api,/admin",
314
+ include_paths="/tutorial,/library",
315
+ allow_subdomains="true",
316
+ provider="bochaai"
317
+ )
318
+
319
+ resolver = WebCrawlToolResolver(mock_agent, tool, mock_args)
320
+
321
+ with patch('autocoder.rag.tools.web_crawl_tool.BochaAIClient') as mock_client_class:
322
+ mock_client = MagicMock()
323
+ mock_client_class.return_value = mock_client
324
+
325
+ # Mock crawl 方法
326
+ mock_crawl_results = [
327
+ {
328
+ "url": "https://docs.python.org/tutorial/index.html",
329
+ "title": "Python 教程",
330
+ "content": "教程主页内容...",
331
+ "markdown": "# Python 教程\n\n教程主页内容...",
332
+ "metadata": {"source": "bochaai"}
333
+ },
334
+ {
335
+ "url": "https://docs.python.org/tutorial/introduction.html",
336
+ "title": "Python 简介",
337
+ "content": "Python 简介内容...",
338
+ "markdown": "# Python 简介\n\nPython 简介内容...",
339
+ "metadata": {"source": "bochaai"}
340
+ }
341
+ ]
342
+ mock_client.crawl.return_value = mock_crawl_results
343
+
344
+ result = resolver.resolve()
345
+
346
+ print(f" 执行成功: {result.success}")
347
+ print(f" 消息: {result.message}")
348
+ print(f" 爬取页面数: {len(result.content) if result.success else 0}")
349
+
350
+ if result.success and result.content:
351
+ for idx, page in enumerate(result.content[:2], 1):
352
+ print(f"\n 页面 {idx}:")
353
+ print(f" URL: {page['url']}")
354
+ print(f" 标题: {page.get('title', 'N/A')}")
355
+ print(f" 来源: {page.get('metadata', {}).get('source', 'unknown')}")
356
+
357
+ print("\n✅ WebCrawlTool BochaAI 集成测试通过")
358
+
359
+
360
+ def test_provider_auto_detection():
361
+ """测试提供商自动检测功能"""
362
+ print("\n" + "=" * 60)
363
+ print("测试提供商自动检测(包含 BochaAI)")
364
+ print("=" * 60)
365
+
366
+ mock_agent = Mock()
367
+
368
+ # 场景1: 只有 BochaAI key
369
+ print("\n1. 只有 BochaAI API key:")
370
+ mock_args = Mock()
371
+ mock_args.bochaai_api_key = "bochaai_key"
372
+ mock_args.metaso_api_key = None
373
+ mock_args.firecrawl_api_key = None
374
+
375
+ tool = WebSearchTool(
376
+ query="test",
377
+ provider=None # 不指定,让系统自动检测
378
+ )
379
+
380
+ resolver = WebSearchToolResolver(mock_agent, tool, mock_args)
381
+
382
+ with patch('autocoder.rag.tools.web_search_tool.BochaAIClient') as mock_client_class:
383
+ mock_client = MagicMock()
384
+ mock_client_class.return_value = mock_client
385
+ mock_search_result = MagicMock()
386
+ mock_search_result.success = True
387
+ mock_search_result.webpages = []
388
+ mock_search_result.images = []
389
+ mock_client.search.return_value = mock_search_result
390
+
391
+ result = resolver.resolve()
392
+ print(f" 自动选择: BochaAI (预期)")
393
+ print(f" 消息包含 'BochaAI': {'BochaAI' in result.message}")
394
+
395
+ # 场景2: 多个 key 都有
396
+ print("\n2. 三个 API key 都有:")
397
+ mock_args = Mock()
398
+ mock_args.bochaai_api_key = "bochaai_key"
399
+ mock_args.metaso_api_key = "metaso_key"
400
+ mock_args.firecrawl_api_key = "firecrawl_key"
401
+
402
+ resolver = WebSearchToolResolver(mock_agent, tool, mock_args)
403
+
404
+ with patch('firecrawl.Firecrawl') as mock_firecrawl_class:
405
+ mock_firecrawl = MagicMock()
406
+ mock_firecrawl_class.return_value = mock_firecrawl
407
+ mock_search_result = MagicMock()
408
+ mock_search_result.success = True
409
+ mock_search_result.data = []
410
+ mock_firecrawl.search.return_value = mock_search_result
411
+
412
+ result = resolver.resolve()
413
+ print(f" 优先选择: Firecrawl (预期)")
414
+ print(f" 消息包含 'Firecrawl': {'Firecrawl' in result.message}")
415
+
416
+ # 场景3: BochaAI 和 Metaso
417
+ print("\n3. BochaAI 和 Metaso API key:")
418
+ mock_args = Mock()
419
+ mock_args.bochaai_api_key = "bochaai_key"
420
+ mock_args.metaso_api_key = "metaso_key"
421
+ mock_args.firecrawl_api_key = None
422
+
423
+ resolver = WebSearchToolResolver(mock_agent, tool, mock_args)
424
+
425
+ with patch('autocoder.rag.tools.web_search_tool.BochaAIClient') as mock_client_class:
426
+ mock_client = MagicMock()
427
+ mock_client_class.return_value = mock_client
428
+ mock_search_result = MagicMock()
429
+ mock_search_result.success = True
430
+ mock_search_result.webpages = []
431
+ mock_search_result.images = []
432
+ mock_client.search.return_value = mock_search_result
433
+
434
+ result = resolver.resolve()
435
+ print(f" 优先选择: BochaAI (预期)")
436
+ print(f" 消息包含 'BochaAI': {'BochaAI' in result.message}")
437
+
438
+ print("\n✅ 提供商自动检测测试通过")
439
+
440
+
441
+ def main():
442
+ """运行所有测试"""
443
+ print("\n" + "=" * 60)
444
+ print("开始测试 BochaAI 集成")
445
+ print("=" * 60)
446
+
447
+ try:
448
+ # 运行各项测试
449
+ test_bochaai_sdk()
450
+ test_bochaai_client_mock()
451
+ test_web_search_tool_with_bochaai()
452
+ test_web_crawl_tool_with_bochaai()
453
+ test_provider_auto_detection()
454
+
455
+ print("\n" + "=" * 60)
456
+ print("✅ 所有测试通过!")
457
+ print("=" * 60)
458
+
459
+ print("\n测试总结:")
460
+ print("1. ✅ BochaAI SDK 数据类功能正常")
461
+ print("2. ✅ BochaAI Client 搜索和爬取功能正常")
462
+ print("3. ✅ WebSearchTool 成功集成 BochaAI")
463
+ print("4. ✅ WebCrawlTool 成功集成 BochaAI")
464
+ print("5. ✅ 提供商自动检测机制正常(支持 BochaAI)")
465
+
466
+ print("\n下一步:")
467
+ print("1. 设置环境变量 BOCHAAI_API_KEY")
468
+ print("2. 在工具中指定 provider='bochaai' 或让系统自动检测")
469
+ print("3. 使用工具进行实际的网页搜索和爬取")
470
+
471
+ except Exception as e:
472
+ print(f"\n❌ 测试失败: {str(e)}")
473
+ import traceback
474
+ traceback.print_exc()
475
+ return 1
476
+
477
+ return 0
478
+
479
+
480
+ if __name__ == "__main__":
481
+ exit(main())
482
+
@@ -0,0 +1,215 @@
1
+
2
+ """
3
+ 最终集成验证脚本
4
+
5
+ 验证 Metaso 集成是否完全正常工作。
6
+ """
7
+
8
+ import os
9
+ import sys
10
+
11
+ # 添加项目路径
12
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../..'))
13
+
14
+
15
+ def test_imports():
16
+ """测试所有导入是否正常"""
17
+ print("=" * 60)
18
+ print("测试导入")
19
+ print("=" * 60)
20
+
21
+ try:
22
+ # 测试直接导入
23
+ from autocoder.rag.tools.metaso_sdk import MetasoClient, MetasoSearchResult, MetasoSearchResponse
24
+ print("✅ Metaso SDK 导入成功")
25
+
26
+ from autocoder.rag.tools.web_search_tool import WebSearchTool, WebSearchToolResolver
27
+ print("✅ WebSearchTool 导入成功")
28
+
29
+ from autocoder.rag.tools.web_crawl_tool import WebCrawlTool, WebCrawlToolResolver
30
+ print("✅ WebCrawlTool 导入成功")
31
+
32
+ # 测试从 __init__ 导入
33
+ from autocoder.rag.tools import MetasoClient as MC
34
+ from autocoder.rag.tools import MetasoSearchResult as MSR
35
+ from autocoder.rag.tools import MetasoSearchResponse as MSResp
36
+ print("✅ 从 __init__.py 导入 Metaso 类成功")
37
+
38
+ return True
39
+ except ImportError as e:
40
+ print(f"❌ 导入失败: {e}")
41
+ return False
42
+
43
+
44
+ def test_metaso_integration():
45
+ """测试 Metaso 集成功能"""
46
+ print("\n" + "=" * 60)
47
+ print("测试 Metaso 集成功能")
48
+ print("=" * 60)
49
+
50
+ from autocoder.rag.tools.web_search_tool import WebSearchTool
51
+ from autocoder.rag.tools.web_crawl_tool import WebCrawlTool
52
+
53
+ # 测试工具创建
54
+ try:
55
+ # 测试搜索工具
56
+ search_tool = WebSearchTool(
57
+ query="test query",
58
+ limit=5,
59
+ provider="metaso"
60
+ )
61
+ print(f"✅ WebSearchTool 创建成功")
62
+ print(f" - query: {search_tool.query}")
63
+ print(f" - limit: {search_tool.limit}")
64
+ print(f" - provider: {search_tool.provider}")
65
+
66
+ # 测试爬取工具
67
+ crawl_tool = WebCrawlTool(
68
+ url="https://example.com",
69
+ limit=10,
70
+ provider="metaso"
71
+ )
72
+ print(f"✅ WebCrawlTool 创建成功")
73
+ print(f" - url: {crawl_tool.url}")
74
+ print(f" - limit: {crawl_tool.limit}")
75
+ print(f" - provider: {crawl_tool.provider}")
76
+
77
+ return True
78
+ except Exception as e:
79
+ print(f"❌ 工具创建失败: {e}")
80
+ return False
81
+
82
+
83
+ def test_provider_field():
84
+ """测试 provider 字段是否正确添加"""
85
+ print("\n" + "=" * 60)
86
+ print("测试 Provider 字段")
87
+ print("=" * 60)
88
+
89
+ from autocoder.rag.tools.web_search_tool import WebSearchTool
90
+ from autocoder.rag.tools.web_crawl_tool import WebCrawlTool
91
+
92
+ # 检查字段是否存在
93
+ search_fields = WebSearchTool.__annotations__
94
+ crawl_fields = WebCrawlTool.__annotations__
95
+
96
+ if 'provider' in search_fields:
97
+ print(f"✅ WebSearchTool 包含 provider 字段")
98
+ print(f" 类型: {search_fields['provider']}")
99
+ else:
100
+ print(f"❌ WebSearchTool 缺少 provider 字段")
101
+ return False
102
+
103
+ if 'provider' in crawl_fields:
104
+ print(f"✅ WebCrawlTool 包含 provider 字段")
105
+ print(f" 类型: {crawl_fields['provider']}")
106
+ else:
107
+ print(f"❌ WebCrawlTool 缺少 provider 字段")
108
+ return False
109
+
110
+ return True
111
+
112
+
113
+ def test_metaso_sdk():
114
+ """测试 Metaso SDK 基本功能"""
115
+ print("\n" + "=" * 60)
116
+ print("测试 Metaso SDK")
117
+ print("=" * 60)
118
+
119
+ from autocoder.rag.tools.metaso_sdk import MetasoClient, MetasoSearchResult
120
+
121
+ # 测试数据类创建
122
+ try:
123
+ result = MetasoSearchResult(
124
+ title="Test Title",
125
+ link="https://test.com",
126
+ snippet="Test snippet"
127
+ )
128
+ print(f"✅ MetasoSearchResult 创建成功")
129
+ print(f" - title: {result.title}")
130
+ print(f" - link: {result.link}")
131
+
132
+ # 测试转换为字典
133
+ result_dict = result.to_dict()
134
+ print(f"✅ to_dict() 方法正常")
135
+ print(f" 字段数: {len(result_dict)}")
136
+
137
+ return True
138
+ except Exception as e:
139
+ print(f"❌ SDK 测试失败: {e}")
140
+ return False
141
+
142
+
143
+ def check_files():
144
+ """检查所有必要文件是否存在"""
145
+ print("\n" + "=" * 60)
146
+ print("检查文件完整性")
147
+ print("=" * 60)
148
+
149
+ required_files = [
150
+ "src/autocoder/rag/tools/metaso_sdk.py",
151
+ "src/autocoder/rag/tools/web_search_tool.py",
152
+ "src/autocoder/rag/tools/web_crawl_tool.py",
153
+ "src/autocoder/rag/tools/test_metaso_integration.py",
154
+ "src/autocoder/rag/tools/metaso_example.py",
155
+ "src/autocoder/rag/tools/README_metaso.md"
156
+ ]
157
+
158
+ all_exist = True
159
+ for file_path in required_files:
160
+ full_path = os.path.join("/Users/williamzhu/projects/auto-coder", file_path)
161
+ if os.path.exists(full_path):
162
+ file_size = os.path.getsize(full_path)
163
+ print(f"✅ {os.path.basename(file_path)}: {file_size:,} bytes")
164
+ else:
165
+ print(f"❌ {os.path.basename(file_path)}: 不存在")
166
+ all_exist = False
167
+
168
+ return all_exist
169
+
170
+
171
+ def main():
172
+ """运行所有验证测试"""
173
+ print("\n" + "=" * 60)
174
+ print("Metaso 集成最终验证")
175
+ print("=" * 60)
176
+
177
+ results = []
178
+
179
+ # 运行各项测试
180
+ print("\n开始验证...\n")
181
+
182
+ results.append(("文件完整性", check_files()))
183
+ results.append(("模块导入", test_imports()))
184
+ results.append(("Provider 字段", test_provider_field()))
185
+ results.append(("Metaso SDK", test_metaso_sdk()))
186
+ results.append(("工具集成", test_metaso_integration()))
187
+
188
+ # 输出总结
189
+ print("\n" + "=" * 60)
190
+ print("验证总结")
191
+ print("=" * 60)
192
+
193
+ all_passed = True
194
+ for test_name, passed in results:
195
+ status = "✅ 通过" if passed else "❌ 失败"
196
+ print(f"{test_name}: {status}")
197
+ if not passed:
198
+ all_passed = False
199
+
200
+ print("\n" + "=" * 60)
201
+ if all_passed:
202
+ print("🎉 所有验证通过!Metaso 集成完成!")
203
+ print("\n下一步:")
204
+ print("1. 设置环境变量: export METASO_API_KEY='your-api-key'")
205
+ print("2. 运行示例: python src/autocoder/rag/tools/metaso_example.py")
206
+ print("3. 在代码中使用: provider='metaso'")
207
+ else:
208
+ print("⚠️ 部分验证失败,请检查错误信息")
209
+ print("=" * 60)
210
+
211
+ return 0 if all_passed else 1
212
+
213
+
214
+ if __name__ == "__main__":
215
+ exit(main())