auto-coder 0.1.258__tar.gz → 0.1.260__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.
Potentially problematic release.
This version of auto-coder might be problematic. Click here for more details.
- {auto_coder-0.1.258 → auto_coder-0.1.260}/PKG-INFO +1 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/PKG-INFO +1 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/SOURCES.txt +5 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/auto_coder.py +25 -23
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/auto_coder_rag.py +7 -7
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/auto_coder_rag_client_mcp.py +1 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/chat_auto_coder.py +471 -523
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/chat_auto_coder_lang.py +2 -0
- auto_coder-0.1.260/src/autocoder/commands/auto_command.py +1145 -0
- auto_coder-0.1.260/src/autocoder/commands/tools.py +533 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/auto_coder_lang.py +34 -6
- auto_coder-0.1.260/src/autocoder/common/auto_configure.py +304 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_modification_ranker.py +8 -7
- auto_coder-0.1.260/src/autocoder/common/command_completer.py +890 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/command_templates.py +7 -3
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/git_utils.py +82 -1
- auto_coder-0.1.260/src/autocoder/common/result_manager.py +115 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/utils_code_auto_generate.py +2 -2
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/dispacher/actions/action.py +8 -4
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/dispacher/actions/plugins/action_regex_project.py +2 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/entry.py +1 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/filter/quick_filter.py +14 -2
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/raw_rag.py +1 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/auto_coder_utils/chat_stream_out.py +13 -6
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/thread_utils.py +4 -0
- auto_coder-0.1.260/src/autocoder/utils/types.py +0 -0
- auto_coder-0.1.260/src/autocoder/version.py +1 -0
- auto_coder-0.1.258/src/autocoder/common/command_completer.py +0 -324
- auto_coder-0.1.258/src/autocoder/version.py +0 -1
- {auto_coder-0.1.258 → auto_coder-0.1.260}/LICENSE +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/README.md +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/setup.cfg +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/setup.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/dependency_links.txt +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/entry_points.txt +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/requires.txt +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/auto_coder.egg-info/top_level.txt +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/auto_demand_organizer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/auto_filegroup.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/auto_guess_query.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/auto_review_commit.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/auto_tool.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/coder.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/designer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/planner.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/agent/project_reader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/auto_coder_rag_mcp.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/auto_coder_server.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/benchmark.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/chat/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/command_args.py +0 -0
- {auto_coder-0.1.258/src/autocoder/common/mcp_servers → auto_coder-0.1.260/src/autocoder/commands}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/JupyterClient.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/ShellClient.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/anything2images.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/anything2img.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/audio.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/buildin_tokenizer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/chunk_validation.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/cleaner.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_execute.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_generate.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_generate_diff.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_generate_editblock.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_generate_strict_diff.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_merge.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_merge_diff.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_merge_editblock.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/code_auto_merge_strict_diff.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/command_generator.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/const.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/files.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/global_cancel.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/image_to_page.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/interpreter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/llm_rerank.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/mcp_hub.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/mcp_server.py +0 -0
- {auto_coder-0.1.258/src/autocoder/db → auto_coder-0.1.260/src/autocoder/common/mcp_servers}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/mcp_servers/mcp_server_perplexity.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/mcp_tools.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/memory_manager.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/model_speed_test.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/printer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/recall_validation.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/screenshots.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/search.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/search_replace.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/shells.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/sys_prompt.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/text.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/common/types.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/data/byzerllm.md +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/data/tokenizer.json +0 -0
- {auto_coder-0.1.258/src/autocoder/dispacher/actions → auto_coder-0.1.260/src/autocoder/db}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/db/store.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/dispacher/__init__.py +0 -0
- {auto_coder-0.1.258/src/autocoder/dispacher/actions/plugins → auto_coder-0.1.260/src/autocoder/dispacher/actions}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/dispacher/actions/copilot.py +0 -0
- {auto_coder-0.1.258/src/autocoder/index → auto_coder-0.1.260/src/autocoder/dispacher/actions/plugins}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/dispacher/actions/plugins/action_translate.py +0 -0
- {auto_coder-0.1.258/src/autocoder/index/filter → auto_coder-0.1.260/src/autocoder/index}/__init__.py +0 -0
- {auto_coder-0.1.258/src/autocoder/rag → auto_coder-0.1.260/src/autocoder/index/filter}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/filter/normal_filter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/for_command.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/index.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/symbols_utils.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/index/types.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/lang.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/models.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/privacy/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/privacy/model_filter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/pyproject/__init__.py +0 -0
- {auto_coder-0.1.258/src/autocoder/rag/cache → auto_coder-0.1.260/src/autocoder/rag}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/api_server.py +0 -0
- {auto_coder-0.1.258/src/autocoder/rag/stream_event → auto_coder-0.1.260/src/autocoder/rag/cache}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/cache/base_cache.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/cache/byzer_storage_cache.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/cache/file_monitor_cache.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/cache/simple_cache.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/doc_filter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/document_retriever.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/llm_wrapper.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/loaders/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/loaders/docx_loader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/loaders/excel_loader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/loaders/pdf_loader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/loaders/ppt_loader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/long_context_rag.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/rag_config.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/rag_entry.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/relevant_utils.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/simple_directory_reader.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/simple_rag.py +0 -0
- {auto_coder-0.1.258/src/autocoder/utils/auto_coder_utils → auto_coder-0.1.260/src/autocoder/rag/stream_event}/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/stream_event/event_writer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/stream_event/types.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/token_checker.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/token_counter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/token_limiter.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/types.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/utils.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/rag/variable_holder.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/regexproject/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/suffixproject/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/tsproject/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/_markitdown.py +0 -0
- {auto_coder-0.1.258/src/autocoder/utils/chat_auto_coder_utils → auto_coder-0.1.260/src/autocoder/utils/auto_coder_utils}/__init__.py +0 -0
- /auto_coder-0.1.258/src/autocoder/utils/types.py → /auto_coder-0.1.260/src/autocoder/utils/chat_auto_coder_utils/__init__.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/conversation_store.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/llm_client_interceptors.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/llms.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/log_capture.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/model_provider_selector.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/multi_turn.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/operate_config_api.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/print_table.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/queue_communicate.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/request_event_queue.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/request_queue.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/rest.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/src/autocoder/utils/tests.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_action_regex_project.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_chat_auto_coder.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_code_auto_merge_editblock.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_command_completer.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_planner.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_privacy.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_queue_communicate.py +0 -0
- {auto_coder-0.1.258 → auto_coder-0.1.260}/tests/test_symbols_utils.py +0 -0
|
@@ -31,6 +31,9 @@ src/autocoder/agent/designer.py
|
|
|
31
31
|
src/autocoder/agent/planner.py
|
|
32
32
|
src/autocoder/agent/project_reader.py
|
|
33
33
|
src/autocoder/chat/__init__.py
|
|
34
|
+
src/autocoder/commands/__init__.py
|
|
35
|
+
src/autocoder/commands/auto_command.py
|
|
36
|
+
src/autocoder/commands/tools.py
|
|
34
37
|
src/autocoder/common/JupyterClient.py
|
|
35
38
|
src/autocoder/common/ShellClient.py
|
|
36
39
|
src/autocoder/common/__init__.py
|
|
@@ -38,6 +41,7 @@ src/autocoder/common/anything2images.py
|
|
|
38
41
|
src/autocoder/common/anything2img.py
|
|
39
42
|
src/autocoder/common/audio.py
|
|
40
43
|
src/autocoder/common/auto_coder_lang.py
|
|
44
|
+
src/autocoder/common/auto_configure.py
|
|
41
45
|
src/autocoder/common/buildin_tokenizer.py
|
|
42
46
|
src/autocoder/common/chunk_validation.py
|
|
43
47
|
src/autocoder/common/cleaner.py
|
|
@@ -68,6 +72,7 @@ src/autocoder/common/memory_manager.py
|
|
|
68
72
|
src/autocoder/common/model_speed_test.py
|
|
69
73
|
src/autocoder/common/printer.py
|
|
70
74
|
src/autocoder/common/recall_validation.py
|
|
75
|
+
src/autocoder/common/result_manager.py
|
|
71
76
|
src/autocoder/common/screenshots.py
|
|
72
77
|
src/autocoder/common/search.py
|
|
73
78
|
src/autocoder/common/search_replace.py
|
|
@@ -48,6 +48,7 @@ from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
|
|
|
48
48
|
from autocoder.common.printer import Printer
|
|
49
49
|
from autocoder.rag.token_counter import count_tokens
|
|
50
50
|
from autocoder.privacy.model_filter import ModelPathFilter
|
|
51
|
+
from autocoder.common.result_manager import ResultManager
|
|
51
52
|
|
|
52
53
|
console = Console()
|
|
53
54
|
|
|
@@ -282,28 +283,7 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
282
283
|
)
|
|
283
284
|
byzerllm.connect_cluster(address=args.ray_address)
|
|
284
285
|
|
|
285
|
-
llm = byzerllm.ByzerLLM(verbose=args.print_request)
|
|
286
|
-
|
|
287
|
-
# code_model,index_filter_model,generate_rerank_model,chat_model
|
|
288
|
-
# 这四个模型如果用户没有设置,就会使用默认的
|
|
289
|
-
# 如果用户随便填写 deepseek 官方key,就会导致 Authentic(No User) 的错误
|
|
290
|
-
# 或者 Insuffient Balance 之类的错误
|
|
291
|
-
|
|
292
|
-
code_model = byzerllm.ByzerLLM()
|
|
293
|
-
code_model.setup_default_model_name(args.model)
|
|
294
|
-
llm.setup_sub_client("code_model", code_model)
|
|
295
|
-
|
|
296
|
-
index_filter_model = byzerllm.ByzerLLM()
|
|
297
|
-
index_filter_model.setup_default_model_name("deepseek_r1_chat")
|
|
298
|
-
llm.setup_sub_client("index_filter_model", index_filter_model)
|
|
299
|
-
|
|
300
|
-
generate_rerank_model = byzerllm.ByzerLLM()
|
|
301
|
-
generate_rerank_model.setup_default_model_name("deepseek_r1_chat")
|
|
302
|
-
llm.setup_sub_client("generate_rerank_model", generate_rerank_model)
|
|
303
|
-
|
|
304
|
-
chat_model = byzerllm.ByzerLLM()
|
|
305
|
-
chat_model.setup_default_model_name("deepseek_r1_chat")
|
|
306
|
-
llm.setup_sub_client("chat_model", chat_model)
|
|
286
|
+
llm = byzerllm.ByzerLLM(verbose=args.print_request)
|
|
307
287
|
|
|
308
288
|
if args.product_mode == "lite":
|
|
309
289
|
default_model = args.model
|
|
@@ -1083,6 +1063,7 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
1083
1063
|
os.makedirs(memory_dir, exist_ok=True)
|
|
1084
1064
|
memory_file = os.path.join(memory_dir, "chat_history.json")
|
|
1085
1065
|
console = Console()
|
|
1066
|
+
result_manager = ResultManager()
|
|
1086
1067
|
if args.new_session:
|
|
1087
1068
|
if os.path.exists(memory_file):
|
|
1088
1069
|
with open(memory_file, "r") as f:
|
|
@@ -1098,6 +1079,13 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
1098
1079
|
"conversation_history": []}
|
|
1099
1080
|
with open(memory_file, "w") as f:
|
|
1100
1081
|
json.dump(chat_history, f, ensure_ascii=False)
|
|
1082
|
+
|
|
1083
|
+
result_manager.add_result(content=get_message("new_session_started"), meta={
|
|
1084
|
+
"action": "chat",
|
|
1085
|
+
"input": {
|
|
1086
|
+
"query": args.query
|
|
1087
|
+
}
|
|
1088
|
+
})
|
|
1101
1089
|
console.print(
|
|
1102
1090
|
Panel(
|
|
1103
1091
|
get_message("new_session_started"),
|
|
@@ -1289,6 +1277,12 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
1289
1277
|
with open(args.target_file, "w") as f:
|
|
1290
1278
|
f.write(chat_content)
|
|
1291
1279
|
|
|
1280
|
+
result_manager = ResultManager()
|
|
1281
|
+
result_manager.append(content=result,
|
|
1282
|
+
meta={"action": "chat","input":{
|
|
1283
|
+
"query":args.query
|
|
1284
|
+
}})
|
|
1285
|
+
|
|
1292
1286
|
# Update chat history with user's response
|
|
1293
1287
|
chat_history["ask_conversation"].append(
|
|
1294
1288
|
{"role": "assistant", "content": result}
|
|
@@ -1363,7 +1357,15 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
1363
1357
|
model_name=model_name,
|
|
1364
1358
|
args=args
|
|
1365
1359
|
)
|
|
1366
|
-
|
|
1360
|
+
|
|
1361
|
+
result_manager = ResultManager()
|
|
1362
|
+
result_manager.append(content=assistant_response, meta={
|
|
1363
|
+
"action": "chat",
|
|
1364
|
+
"input": {
|
|
1365
|
+
"query": args.query
|
|
1366
|
+
}
|
|
1367
|
+
})
|
|
1368
|
+
|
|
1367
1369
|
# 打印耗时和token统计
|
|
1368
1370
|
if last_meta:
|
|
1369
1371
|
elapsed_time = time.time() - start_time
|
|
@@ -66,7 +66,7 @@ def initialize_system(args):
|
|
|
66
66
|
print_status(get_message("checking_model"), "")
|
|
67
67
|
try:
|
|
68
68
|
result = subprocess.run(
|
|
69
|
-
["easy-byzerllm", "chat", "
|
|
69
|
+
["easy-byzerllm", "chat", "v3_chat", "你好"],
|
|
70
70
|
capture_output=True,
|
|
71
71
|
text=True,
|
|
72
72
|
timeout=30,
|
|
@@ -115,7 +115,7 @@ def initialize_system(args):
|
|
|
115
115
|
"--infer_params",
|
|
116
116
|
f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
|
|
117
117
|
"--model",
|
|
118
|
-
"
|
|
118
|
+
"v3_chat",
|
|
119
119
|
]
|
|
120
120
|
|
|
121
121
|
try:
|
|
@@ -129,7 +129,7 @@ def initialize_system(args):
|
|
|
129
129
|
print_status(get_message("validating_deploy"), "")
|
|
130
130
|
try:
|
|
131
131
|
validation_result = subprocess.run(
|
|
132
|
-
["easy-byzerllm", "chat", "
|
|
132
|
+
["easy-byzerllm", "chat", "v3_chat", "你好"],
|
|
133
133
|
capture_output=True,
|
|
134
134
|
text=True,
|
|
135
135
|
timeout=30,
|
|
@@ -139,7 +139,7 @@ def initialize_system(args):
|
|
|
139
139
|
except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
|
|
140
140
|
print_status(get_message("validation_fail"), "error")
|
|
141
141
|
print_status(get_message("manual_start"), "warning")
|
|
142
|
-
print_status("easy-byzerllm chat
|
|
142
|
+
print_status("easy-byzerllm chat v3_chat 你好", "")
|
|
143
143
|
|
|
144
144
|
print_status(get_message("init_complete_final"), "success")
|
|
145
145
|
|
|
@@ -168,7 +168,7 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
168
168
|
)
|
|
169
169
|
build_index_parser.add_argument("--file", default="", help=desc["file"])
|
|
170
170
|
build_index_parser.add_argument(
|
|
171
|
-
"--model", default="
|
|
171
|
+
"--model", default="v3_chat", help=desc["model"]
|
|
172
172
|
)
|
|
173
173
|
build_index_parser.add_argument(
|
|
174
174
|
"--index_model", default="", help=desc["index_model"]
|
|
@@ -199,7 +199,7 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
199
199
|
"--quick", action="store_true", help="Skip system initialization"
|
|
200
200
|
)
|
|
201
201
|
serve_parser.add_argument("--file", default="", help=desc["file"])
|
|
202
|
-
serve_parser.add_argument("--model", default="
|
|
202
|
+
serve_parser.add_argument("--model", default="v3_chat", help=desc["model"])
|
|
203
203
|
serve_parser.add_argument("--index_model", default="", help=desc["index_model"])
|
|
204
204
|
serve_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
|
|
205
205
|
serve_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
|
|
@@ -366,7 +366,7 @@ def main(input_args: Optional[List[str]] = None):
|
|
|
366
366
|
"benchmark", help="Benchmark LLM client performance"
|
|
367
367
|
)
|
|
368
368
|
benchmark_parser.add_argument(
|
|
369
|
-
"--model", default="
|
|
369
|
+
"--model", default="v3_chat", help="Model to benchmark"
|
|
370
370
|
)
|
|
371
371
|
benchmark_parser.add_argument(
|
|
372
372
|
"--parallel", type=int, default=10, help="Number of parallel requests"
|
|
@@ -147,7 +147,7 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
|
|
|
147
147
|
parser = argparse.ArgumentParser(description="Auto Coder RAG Client MCP Server")
|
|
148
148
|
parser.add_argument("--rag_url", required=True, help="RAG server URL")
|
|
149
149
|
parser.add_argument("--rag_token", required=True, help="RAG server token")
|
|
150
|
-
parser.add_argument("--model", default="
|
|
150
|
+
parser.add_argument("--model", default="v3_chat", help=desc["model"])
|
|
151
151
|
parser.add_argument("--rag_params_max_tokens", type=int, default=4096, help="Max tokens for RAG response")
|
|
152
152
|
|
|
153
153
|
args = parser.parse_args(input_args)
|