auto-coder 0.1.196__py3-none-any.whl → 0.1.198__py3-none-any.whl

autocoder/command_args.py

@@ -14,14 +14,18 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     subparsers = parser.add_subparsers(dest="command")
 
     parser.add_argument("--request_id", default="", help=desc["request_id"])
-    parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
+    parser.add_argument("--source_dir", required=False,
+                        help=desc["source_dir"])
     parser.add_argument("--git_url", help=desc["git_url"])
-    parser.add_argument("--target_file", required=False, help=desc["target_file"])
+    parser.add_argument("--target_file", required=False,
+                        help=desc["target_file"])
     parser.add_argument("--query", help=desc["query"])
     parser.add_argument("--template", default="common", help=desc["template"])
-    parser.add_argument("--project_type", default="py", help=desc["project_type"])
+    parser.add_argument("--project_type", default="py",
+                        help=desc["project_type"])
     parser.add_argument("--execute", action="store_true", help=desc["execute"])
-    parser.add_argument("--package_name", default="", help=desc["package_name"])
+    parser.add_argument("--package_name", default="",
+                        help=desc["package_name"])
     parser.add_argument("--script_path", default="", help=desc["script_path"])
 
     parser.add_argument("--model", default="", help=desc["model"])
@@ -39,8 +43,10 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     parser.add_argument("--vl_model", default="", help=desc["vl_model"])
     parser.add_argument("--sd_model", default="", help=desc["sd_model"])
     parser.add_argument("--emb_model", default="", help=desc["emb_model"])
-    parser.add_argument("--text2voice_model", default="", help=desc["text2voice_model"])
-    parser.add_argument("--voice2text_model", default="", help=desc["voice2text_model"])
+    parser.add_argument("--text2voice_model", default="",
+                        help=desc["text2voice_model"])
+    parser.add_argument("--voice2text_model", default="",
+                        help=desc["voice2text_model"])
 
     parser.add_argument("--index_model", default="", help=desc["index_model"])
     parser.add_argument(
@@ -76,10 +82,13 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     parser.add_argument(
         "--index_build_workers", type=int, default=1, help=desc["index_build_workers"]
     )
-    parser.add_argument("--rag_doc_filter_relevance", type=int, default=5, help="")
+    parser.add_argument("--rag_doc_filter_relevance",
+                        type=int, default=5, help="")
 
-    parser.add_argument("--file", default=None, required=False, help=desc["file"])
-    parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
+    parser.add_argument("--file", default=None,
+                        required=False, help=desc["file"])
+    parser.add_argument("--ray_address", default="auto",
+                        help=desc["ray_address"])
     parser.add_argument(
         "--anti_quota_limit", type=int, default=1, help=desc["anti_quota_limit"]
     )
@@ -93,7 +102,8 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
         "--print_request", action="store_true", help=desc["print_request"]
     )
     parser.add_argument("--code_model", default="", help=desc["code_model"])
-    parser.add_argument("--planner_model", default="", help=desc["planner_model"])
+    parser.add_argument("--planner_model", default="",
+                        help=desc["planner_model"])
     parser.add_argument(
         "--py_packages", required=False, default="", help=desc["py_packages"]
     )
@@ -103,13 +113,17 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     parser.add_argument("--urls", default="", help=desc["urls"])
     parser.add_argument(
         "--urls_use_model", action="store_true", help=desc["urls_use_model"]
-    )    
-    parser.add_argument("--designer_model", default="", help=desc["designer_model"])
-    parser.add_argument("--query_prefix", default=None, help=desc["query_prefix"])
-    parser.add_argument("--query_suffix", default=None, help=desc["query_suffix"])
+    )
+    parser.add_argument("--designer_model", default="",
+                        help=desc["designer_model"])
+    parser.add_argument("--query_prefix", default=None,
+                        help=desc["query_prefix"])
+    parser.add_argument("--query_suffix", default=None,
+                        help=desc["query_suffix"])
 
     parser.add_argument("--search", default="", help="")
-    parser.add_argument("--search_engine", default="", help=desc["search_engine"])
+    parser.add_argument("--search_engine", default="",
+                        help=desc["search_engine"])
     parser.add_argument(
         "--search_engine_token", default="", help=desc["search_engine_token"]
     )
@@ -123,12 +137,12 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
 
     parser.add_argument(
-            "--rag_context_window_limit",
-            nargs="?",
-            const=True,
-            default=False,
-            help="",
-        )
+        "--rag_context_window_limit",
+        nargs="?",
+        const=True,
+        default=False,
+        help="",
+    )
 
     parser.add_argument(
         "--enable_rag_context",
@@ -140,6 +154,7 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     parser.add_argument("--rag_token", default="", help="")
     parser.add_argument("--rag_url", default="", help="")
+    parser.add_argument("--rag_params_max_tokens", default=4096, help="")
     parser.add_argument(
         "--rag_type", default="storage", help="RAG type, default is storage"
     )
@@ -160,7 +175,8 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
 
     parser.add_argument("--image_file", default="", help=desc["image_file"])
-    parser.add_argument("--image_mode", default="direct", help=desc["image_mode"])
+    parser.add_argument("--image_mode", default="direct",
+                        help=desc["image_mode"])
     parser.add_argument(
         "--image_max_iter", type=int, default=1, help=desc["image_max_iter"]
     )
@@ -181,27 +197,39 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     revert_parser = subparsers.add_parser("revert", help=desc["revert_desc"])
     revert_parser.add_argument("--file", help=desc["revert_desc"])
-    revert_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    revert_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
 
     store_parser = subparsers.add_parser("store", help=desc["store_desc"])
-    store_parser.add_argument("--source_dir", default=".", help=desc["source_dir"])
-    store_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
-    store_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    store_parser.add_argument(
+        "--source_dir", default=".", help=desc["source_dir"])
+    store_parser.add_argument(
+        "--ray_address", default="auto", help=desc["ray_address"])
+    store_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
 
     index_parser = subparsers.add_parser("index", help=desc["index_desc"])
     index_parser.add_argument("--file", help=desc["file"])
     index_parser.add_argument("--model", default="", help=desc["model"])
-    index_parser.add_argument("--index_model", default="", help=desc["index_model"])
-    index_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
-    index_parser.add_argument("--project_type", default="py", help=desc["project_type"])
-    index_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
-    index_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    index_parser.add_argument(
+        "--index_model", default="", help=desc["index_model"])
+    index_parser.add_argument(
+        "--source_dir", required=False, help=desc["source_dir"])
+    index_parser.add_argument(
+        "--project_type", default="py", help=desc["project_type"])
+    index_parser.add_argument(
+        "--ray_address", default="auto", help=desc["ray_address"])
+    index_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
 
     index_query_parser = subparsers.add_parser(
         "index-query", help=desc["index_query_desc"]
     )  # New subcommand
     index_query_parser.add_argument("--file", help=desc["file"])
-    index_query_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    index_query_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    index_query_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
     index_query_parser.add_argument("--model", default="", help=desc["model"])
     index_query_parser.add_argument(
         "--index_model", default="", help=desc["index_model"]
@@ -218,26 +246,34 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
 
     doc_parser = subparsers.add_parser("doc", help=desc["doc_desc"])
-    doc_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    doc_parser.add_argument("--request_id", default="",
+                            help=desc["request_id"])
+    doc_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
     doc_parser.add_argument("--urls", default="", help=desc["urls"])
     doc_parser.add_argument("--model", default="", help=desc["model"])
-    doc_parser.add_argument("--target_file", default="", help=desc["target_file"])
+    doc_parser.add_argument("--target_file", default="",
+                            help=desc["target_file"])
     doc_parser.add_argument("--file", default="", help=desc["file"])
-    doc_parser.add_argument("--source_dir", required=False, help=desc["source_dir"])
+    doc_parser.add_argument(
+        "--source_dir", required=False, help=desc["source_dir"])
     doc_parser.add_argument(
         "--human_as_model", action="store_true", help=desc["human_as_model"]
     )
     doc_parser.add_argument(
         "--urls_use_model", action="store_true", help=desc["urls_use_model"]
     )
-    doc_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
+    doc_parser.add_argument(
+        "--ray_address", default="auto", help=desc["ray_address"])
 
     doc_subparsers = doc_parser.add_subparsers(dest="doc_command")
     doc_build_parse = doc_subparsers.add_parser("build", help="")
-    doc_build_parse.add_argument("--request_id", default="", help=desc["request_id"])
+    doc_build_parse.add_argument(
+        "--request_id", default="", help=desc["request_id"])
     doc_build_parse.add_argument("--source_dir", default="", help="")
     doc_build_parse.add_argument("--model", default="", help=desc["model"])
-    doc_build_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
+    doc_build_parse.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     doc_build_parse.add_argument("--file", default="", help=desc["file"])
     doc_build_parse.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
@@ -256,16 +292,21 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
 
     doc_query_parse = doc_subparsers.add_parser("query", help="")
-    doc_query_parse.add_argument("--request_id", default="", help=desc["request_id"])
+    doc_query_parse.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    doc_query_parse.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
     doc_query_parse.add_argument("--query", default="", help="")
     doc_query_parse.add_argument("--source_dir", default=".", help="")
     doc_query_parse.add_argument("--model", default="", help=desc["model"])
-    doc_query_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
+    doc_query_parse.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     doc_query_parse.add_argument("--file", default="", help=desc["file"])
     doc_query_parse.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
-    doc_query_parse.add_argument("--execute", action="store_true", help=desc["execute"])
+    doc_query_parse.add_argument(
+        "--execute", action="store_true", help=desc["execute"])
     doc_query_parse.add_argument(
         "--collections",
         default="",
@@ -281,10 +322,14 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
 
     doc_chat_parse = doc_subparsers.add_parser("chat", help="")
-    doc_chat_parse.add_argument("--request_id", default="", help=desc["request_id"])
+    doc_chat_parse.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    doc_chat_parse.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
     doc_chat_parse.add_argument("--file", default="", help=desc["file"])
     doc_chat_parse.add_argument("--model", default="", help=desc["model"])
-    doc_chat_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
+    doc_chat_parse.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     doc_chat_parse.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
@@ -306,8 +351,10 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     doc_serve_parse = doc_subparsers.add_parser("serve", help="")
     doc_serve_parse.add_argument("--file", default="", help=desc["file"])
     doc_serve_parse.add_argument("--model", default="", help=desc["model"])
-    doc_serve_parse.add_argument("--index_model", default="", help=desc["index_model"])
-    doc_serve_parse.add_argument("--emb_model", default="", help=desc["emb_model"])
+    doc_serve_parse.add_argument(
+        "--index_model", default="", help=desc["index_model"])
+    doc_serve_parse.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     doc_serve_parse.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
@@ -323,7 +370,7 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
         default=3,
         help=desc["index_filter_file_num"],
     )
-    
+
     doc_serve_parse.add_argument(
         "--rag_context_window_limit",
         type=int,
@@ -343,16 +390,24 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
         default=5,
         help=desc["filter_batch_size"],
     )
+    parser.add_argument(
+        "--skip_events",
+        action="store_true",
+        help=desc["skip_events"],
+    )
 
     doc_serve_parse.add_argument(
         "--required_exts", default="", help=desc["doc_build_parse_required_exts"]
     )
-    doc_serve_parse.add_argument("--rag_doc_filter_relevance", type=int, default=5, help="")
+    doc_serve_parse.add_argument(
+        "--rag_doc_filter_relevance", type=int, default=5, help="")
     doc_serve_parse.add_argument("--source_dir", default=".", help="")
     doc_serve_parse.add_argument("--host", default="", help="")
     doc_serve_parse.add_argument("--port", type=int, default=8000, help="")
-    doc_serve_parse.add_argument("--uvicorn_log_level", default="info", help="")
-    doc_serve_parse.add_argument("--allow_credentials", action="store_true", help="")
+    doc_serve_parse.add_argument(
+        "--uvicorn_log_level", default="info", help="")
+    doc_serve_parse.add_argument(
+        "--allow_credentials", action="store_true", help="")
     doc_serve_parse.add_argument("--allowed_origins", default=["*"], help="")
     doc_serve_parse.add_argument("--allowed_methods", default=["*"], help="")
     doc_serve_parse.add_argument("--allowed_headers", default=["*"], help="")
@@ -361,12 +416,18 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     doc_serve_parse.add_argument("--prompt_template", default="", help="")
     doc_serve_parse.add_argument("--ssl_keyfile", default="", help="")
     doc_serve_parse.add_argument("--ssl_certfile", default="", help="")
-    doc_serve_parse.add_argument("--response_role", default="assistant", help="")
+    doc_serve_parse.add_argument(
+        "--response_role", default="assistant", help="")
     doc_serve_parse.add_argument("--doc_dir", default="", help="")
     doc_serve_parse.add_argument("--tokenizer_path", default="", help="")
     doc_serve_parse.add_argument(
         "--collections", default="", help="Collection name for indexing"
     )
+    doc_serve_parse.add_argument(
+        "--skip_events",
+        action="store_true",
+        help=desc["skip_events"],
+    )
     doc_serve_parse.add_argument(
         "--base_dir",
         default="",
@@ -381,16 +442,23 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     agent_parser = subparsers.add_parser("agent", help="Run an agent")
     agent_subparsers = agent_parser.add_subparsers(dest="agent_command")
 
-    chat_parser = agent_subparsers.add_parser("chat", help="Run the chat agent")
-    chat_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    chat_parser = agent_subparsers.add_parser(
+        "chat", help="Run the chat agent")
+    chat_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    chat_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
 
-    chat_parser.add_argument("--source_dir", default=".", help="Source directory")
+    chat_parser.add_argument(
+        "--source_dir", default=".", help="Source directory")
     chat_parser.add_argument("--query", help="Query for the planner")
     chat_parser.add_argument("--model", default="", help=desc["model"])
     chat_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
     chat_parser.add_argument("--file", default="", help=desc["file"])
-    chat_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
-    chat_parser.add_argument("--execute", action="store_true", help=desc["execute"])
+    chat_parser.add_argument(
+        "--ray_address", default="auto", help=desc["ray_address"])
+    chat_parser.add_argument(
+        "--execute", action="store_true", help=desc["execute"])
     chat_parser.add_argument(
         "--collections",
         default="",
@@ -416,6 +484,7 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     chat_parser.add_argument("--rag_token", default="", help="")
     chat_parser.add_argument("--rag_url", default="", help="")
+    chat_parser.add_argument("--rag_params_max_tokens", default=4096, help="")
     chat_parser.add_argument(
         "--rag_type",
         default="storage",
@@ -441,13 +510,16 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     read_project_parser.add_argument(
         "--request_id", default="", help=desc["request_id"]
     )
+    read_project_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
 
     read_project_parser.add_argument(
         "--source_dir", default=".", help="Source directory"
     )
     read_project_parser.add_argument("--query", help="Query for the planner")
     read_project_parser.add_argument("--model", default="", help=desc["model"])
-    read_project_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
+    read_project_parser.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     read_project_parser.add_argument("--file", default="", help=desc["file"])
     read_project_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
@@ -480,23 +552,30 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     read_project_parser.add_argument("--rag_token", default="", help="")
     read_project_parser.add_argument("--rag_url", default="", help="")
+    read_project_parser.add_argument("--rag_params_max_tokens", default=4096, help="")
     read_project_parser.add_argument(
         "--rag_type", default="storage", help="RAG type, default is storage"
     )
-    read_project_parser.add_argument("--target_file", default="./output.txt", help="")
+    read_project_parser.add_argument(
+        "--target_file", default="./output.txt", help="")
 
     voice2text_parser = agent_subparsers.add_parser(
         "voice2text", help="Convert voice to text"
     )
-    voice2text_parser.add_argument("--request_id", default="", help=desc["request_id"])
+    voice2text_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    voice2text_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
     voice2text_parser.add_argument("--model", default="", help=desc["model"])
-    voice2text_parser.add_argument("--voice2text_model", default="", help=desc["model"])
+    voice2text_parser.add_argument(
+        "--voice2text_model", default="", help=desc["model"])
     voice2text_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
     voice2text_parser.add_argument("--source_dir", default=".", help="")
     voice2text_parser.add_argument("--file", default="", help=desc["file"])
-    voice2text_parser.add_argument("--target_file", default="./output.txt", help="")
+    voice2text_parser.add_argument(
+        "--target_file", default="./output.txt", help="")
 
     generate_shell_parser = agent_subparsers.add_parser(
         "generate_command", help="Convert text to shell command"
@@ -504,24 +583,33 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     generate_shell_parser.add_argument(
         "--request_id", default="", help=desc["request_id"]
     )
-    generate_shell_parser.add_argument("--model", default="", help=desc["model"])
-    generate_shell_parser.add_argument("--query", default="", help=desc["query"])
+    generate_shell_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
+    generate_shell_parser.add_argument(
+        "--model", default="", help=desc["model"])
+    generate_shell_parser.add_argument(
+        "--query", default="", help=desc["query"])
     generate_shell_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
     generate_shell_parser.add_argument("--source_dir", default=".", help="")
     generate_shell_parser.add_argument("--file", default="", help=desc["file"])
-    generate_shell_parser.add_argument("--target_file", default="./output.txt", help="")
+    generate_shell_parser.add_argument(
+        "--target_file", default="./output.txt", help="")
 
     auto_tool_parser = agent_subparsers.add_parser(
         "auto_tool", help="Run the chat agent"
     )
-    auto_tool_parser.add_argument("--request_id", default="", help=desc["request_id"])
-
-    auto_tool_parser.add_argument("--source_dir", default=".", help="Source directory")
+    auto_tool_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    auto_tool_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
+    auto_tool_parser.add_argument(
+        "--source_dir", default=".", help="Source directory")
     auto_tool_parser.add_argument("--query", help="Query for the planner")
     auto_tool_parser.add_argument("--model", default="", help=desc["model"])
-    auto_tool_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
+    auto_tool_parser.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     auto_tool_parser.add_argument("--file", default="", help=desc["file"])
     auto_tool_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
@@ -554,34 +642,47 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     auto_tool_parser.add_argument("--rag_token", default="", help="")
     auto_tool_parser.add_argument("--rag_url", default="", help="")
+    auto_tool_parser.add_argument("--rag_params_max_tokens", default=4096, help="")
     auto_tool_parser.add_argument(
         "--rag_type", default="storage", help="RAG type, default is storage"
     )
-    auto_tool_parser.add_argument("--target_file", default="./output.txt", help="")
+    auto_tool_parser.add_argument(
+        "--target_file", default="./output.txt", help="")
 
     designer_parser = agent_subparsers.add_parser(
         "designer", help="Run the designer agent"
     )
-    designer_parser.add_argument("--request_id", default="", help=desc["request_id"])
-    designer_parser.add_argument("--source_dir", default=".", help="Source directory")
+    designer_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    designer_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
+    designer_parser.add_argument(
+        "--source_dir", default=".", help="Source directory")
     designer_parser.add_argument("--query", help="Query for the designer")
     designer_parser.add_argument("--model", default="", help=desc["model"])
     designer_parser.add_argument("--file", default="", help=desc["file"])
-    designer_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
+    designer_parser.add_argument(
+        "--ray_address", default="auto", help=desc["ray_address"])
 
     planner_parser = agent_subparsers.add_parser(
         "planner", help="Run the planner agent"
     )
-    planner_parser.add_argument("--request_id", default="", help=desc["request_id"])
-    planner_parser.add_argument("--source_dir", default=".", help="Source directory")
+    planner_parser.add_argument(
+        "--request_id", default="", help=desc["request_id"])
+    planner_parser.add_argument(
+        "--skip_events", action="store_true", help=desc["skip_events"])
+    planner_parser.add_argument(
+        "--source_dir", default=".", help="Source directory")
     planner_parser.add_argument("--query", help="Query for the planner")
     planner_parser.add_argument("--model", default="", help=desc["model"])
-    planner_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
+    planner_parser.add_argument(
+        "--emb_model", default="", help=desc["emb_model"])
     planner_parser.add_argument("--file", default="", help=desc["file"])
     planner_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )
-    planner_parser.add_argument("--execute", action="store_true", help=desc["execute"])
+    planner_parser.add_argument(
+        "--execute", action="store_true", help=desc["execute"])
     planner_parser.add_argument(
         "--collections",
         default="",
@@ -607,18 +708,22 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
 
     planner_parser.add_argument("--rag_token", default="", help="")
     planner_parser.add_argument("--rag_url", default="", help="")
+    planner_parser.add_argument("--rag_params_max_tokens", default=4096, help="")   
     planner_parser.add_argument(
         "--rag_type", default="storage", help="RAG type, default is storage"
     )
-    planner_parser.add_argument("--target_file", default="./output.txt", help="")
+    planner_parser.add_argument(
+        "--target_file", default="./output.txt", help="")
 
     init_parser = subparsers.add_parser("init", help=desc["init_desc"])
-    init_parser.add_argument("--source_dir", required=True, help=desc["init_dir"])
+    init_parser.add_argument(
+        "--source_dir", required=True, help=desc["init_dir"])
 
     screenshot_parser = subparsers.add_parser(
         "screenshot", help=desc["screenshot_desc"]
     )
-    screenshot_parser.add_argument("--urls", required=True, help=desc["screenshot_url"])
+    screenshot_parser.add_argument(
+        "--urls", required=True, help=desc["screenshot_url"])
     screenshot_parser.add_argument(
         "--output", required=True, help=desc["screenshot_output"]
     )
@@ -636,7 +741,8 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     )
     doc2html_parser.add_argument("--file", default="", help=desc["file"])
     doc2html_parser.add_argument("--model", default="", help=desc["model"])
-    doc2html_parser.add_argument("--vl_model", default="", help=desc["vl_model"])
+    doc2html_parser.add_argument(
+        "--vl_model", default="", help=desc["vl_model"])
     doc2html_parser.add_argument(
         "--ray_address", default="auto", help=desc["ray_address"]
     )

autocoder/common/__init__.py

@@ -98,6 +98,9 @@ class EnvInfo(pydantic.BaseModel):
     conda_env: Optional[str]
     virtualenv: Optional[str]
     has_bash: bool
+    default_shell: Optional[str]
+    home_dir: Optional[str]
+    cwd: Optional[str]
 
 
 def has_sufficient_content(file_content, min_line_count=10):
@@ -155,8 +158,19 @@ def detect_env() -> EnvInfo:
     )
 
     conda_env = os.environ.get("CONDA_DEFAULT_ENV")
-
     virtualenv = os.environ.get("VIRTUAL_ENV")
+    
+    # Get default shell
+    if os_name == "win32":
+        default_shell = os.environ.get("COMSPEC", "cmd.exe")
+    else:
+        default_shell = os.environ.get("SHELL", "/bin/sh")
+
+    # Get home directory
+    home_dir = os.path.expanduser("~")
+    
+    # Get current working directory
+    cwd = os.getcwd()
 
     has_bash = True
     try:
@@ -171,6 +185,9 @@ def detect_env() -> EnvInfo:
         conda_env=conda_env,
         virtualenv=virtualenv,
         has_bash=has_bash,
+        default_shell=default_shell,
+        home_dir=home_dir,
+        cwd=cwd,
     )
 
 
@@ -261,7 +278,8 @@ class AutoCoderArgs(pydantic.BaseModel):
         
     rag_url: Optional[str] = ""
     rag_token: Optional[str] = ""
-    rag_type: Optional[str] = "storage"    
+    rag_type: Optional[str] = "storage"   
+    rag_params_max_tokens: Optional[int] = 4096 
     rag_doc_filter_relevance: Optional[int] = 5
     rag_context_window_limit: Optional[int] = 120000    
     verify_file_relevance_score: int = 6
@@ -321,6 +339,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     inference_slow_without_deep_thought: Optional[bool] = False
     inference_compute_precision: int = 64
     without_contexts: Optional[bool] = False
+    skip_events: Optional[bool] = False
 
     class Config:
         protected_namespaces = ()

autocoder/common/code_auto_generate_editblock.py

@@ -2,6 +2,7 @@ from typing import List, Dict, Tuple
 from autocoder.common.types import Mode
 from autocoder.common import AutoCoderArgs
 import byzerllm
+from autocoder.common import sys_prompt
 
 
 class CodeAutoGenerateEditBlock:
@@ -379,8 +380,11 @@ class CodeAutoGenerateEditBlock:
 
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
+        
 
-        conversations = [{"role": "user", "content": init_prompt}]
+        conversations = []
+        # conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        conversations.append({"role": "user", "content": init_prompt})
 
         t = self.llm.chat_oai(conversations=conversations, llm_config=llm_config)
         conversations.append({"role": "assistant", "content": t[0].output})
@@ -401,7 +405,9 @@ class CodeAutoGenerateEditBlock:
                 instruction=query, content=source_content
             )
 
-        conversations = [{"role": "user", "content": init_prompt}]
+        conversations = []
+        # conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        conversations.append({"role": "user", "content": init_prompt})
 
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)