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

{auto_coder-0.1.198.dist-info → auto_coder-0.1.200.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.198
+Version: 0.1.200
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.198.dist-info → auto_coder-0.1.200.dist-info}/RECORD

@@ -5,9 +5,9 @@ autocoder/auto_coder_rag.py,sha256=vRc3O0qSyRjiuBKD6Bfh3oMVchnO3VHoAeOz5j_jjlY,1
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/chat_auto_coder.py,sha256=kgcD4HKKmvDd2CI048TqZwdx2hrQRbZKYPO10qyrbA8,82105
 autocoder/chat_auto_coder_lang.py,sha256=QYtu5gWEQmWKVovR_qUZ8plySZarNFX_Onk-1vN9IiA,8524
-autocoder/command_args.py,sha256=jar54oLWYn20Z89rUxMCZv0ArRco7ev7jD3w0ZHfLcc,29807
-autocoder/lang.py,sha256=_bSJ_0OR4jRRghygd2f1EJkdmDk1a7imjZdPaiK4uEY,13675
-autocoder/version.py,sha256=7VkJdius5UmdVfV0_rB9BFkXVpKqPNv5zVMVKK8o26A,24
+autocoder/command_args.py,sha256=MBsVjZpADu5u7CjY_1F8fGUW-9dVGyNDpzWyrDIMxz8,29890
+autocoder/lang.py,sha256=Ajng6m7towmx-cvQfEHPFp43iEfddPvr8ju5GH4H8qA,13819
+autocoder/version.py,sha256=Apjtlz-yhORdHiRZDHcS7CXTafTC2lwKaSzGAF1sVSw,24
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=x6bdJwDuETGg9ebQnYlUWCxCtQcDGg73LtI6McpWslQ,72034
@@ -17,15 +17,15 @@ autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMb
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=8_TmZeX-UNPmMSWUnaYcNOMV-uxwWY2ye-1NHU0Rlk4,11163
+autocoder/common/__init__.py,sha256=szE6bhkWT_KDTxY73hDQDRYCaH7DDTrBuEQBxzqAEOM,11201
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
 autocoder/common/cleaner.py,sha256=NU72i8C6o9m0vXExab7nao5bstBUsfJFcj11cXa9l4U,1089
 autocoder/common/code_auto_execute.py,sha256=4KXGmiGObr_B1d6tzV9dwS6MifCSc3Gm4j2d6ildBXQ,6867
-autocoder/common/code_auto_generate.py,sha256=Jn145eIsVUCF4EtpcnCS83R6oXiTz51PvhQAd1vYAG8,6909
-autocoder/common/code_auto_generate_diff.py,sha256=3rI3B8LL8MRobv5wjZYNO5bE9DUaPvQuuKRNlFkIbKw,12146
-autocoder/common/code_auto_generate_editblock.py,sha256=XzbYgww-ZZSHI5Qhps5ovfqIg3929fpdkr3CMF18cvg,14478
-autocoder/common/code_auto_generate_strict_diff.py,sha256=PS9ghYxNrohfLbclSxg0a39DYz8XwZsDBMUdSkrety0,10638
+autocoder/common/code_auto_generate.py,sha256=mZkua-qGa-j6FCFeh-FSVsDnWU6ix51FaY7HBf7lYTg,8081
+autocoder/common/code_auto_generate_diff.py,sha256=By7ai1EO_5IgyVWDn9iIvJ658VudauYIusUiwcVP4aE,13197
+autocoder/common/code_auto_generate_editblock.py,sha256=MIhtoPnVzbxP-VanpruyKXX5OO5jiWEDlENkzhChPiE,15826
+autocoder/common/code_auto_generate_strict_diff.py,sha256=kqGRCLsIsfwTDMPWo5EVifJ3lWgWxcF1y9FHeensNJY,11807
 autocoder/common/code_auto_merge.py,sha256=3PB2fQtqgA0wPQSdwFqoOHfNT-mORgGfOXjfRGflikQ,5141
 autocoder/common/code_auto_merge_diff.py,sha256=b5INrBJJ2EI0uj5AIiMobCS5S9xevCvdescs3nzCPhY,12594
 autocoder/common/code_auto_merge_editblock.py,sha256=gIM90aZLKLHft4OzfUuAB6dWx7ntbnrqzZiJuOVukJQ,13774
@@ -41,7 +41,7 @@ autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7B
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
 autocoder/common/search.py,sha256=_ZX03ph89rDPGMY1OrfqaDfxsDR-flh6YEHixherjwM,16616
 autocoder/common/search_replace.py,sha256=GphFkc57Hb673CAwmbiocqTbw8vrV7TrZxtOhD0332g,22147
-autocoder/common/sys_prompt.py,sha256=FOgduCm8VbK1tZmjhgmdD8jWyB4EJybh0lkfSBDAswg,26380
+autocoder/common/sys_prompt.py,sha256=Ul-lZQBoc2rwydccvKm0UsGtxQtUJfS81K1WJcd5Oao,26389
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/types.py,sha256=_kl8q2oCAZe-HqU9LjQgSqVHAw4ojMojlF58-bp2xiE,354
 autocoder/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -72,7 +72,7 @@ autocoder/rag/simple_directory_reader.py,sha256=LkKreCkNdEOoL4fNhc3_hDoyyWTQUte4
 autocoder/rag/simple_rag.py,sha256=I902EUqOK1WM0Y2WFd7RzDJYofElvTZNLVCBtX5A9rc,14885
 autocoder/rag/token_checker.py,sha256=jc76x6KWmvVxds6W8juZfQGaoErudc2HenG3sNQfSLs,2819
 autocoder/rag/token_counter.py,sha256=LReZEYXmWriDI3KYjUvK0E4Gn4MeDJX6RPCfJCmluGY,2110
-autocoder/rag/token_limiter.py,sha256=4cGy2kFCvbsM5CtONfuvLmXgpK_1HZTHehLTob08eks,10959
+autocoder/rag/token_limiter.py,sha256=50sq3pid6up8MBc8jESEqeZFzXN7z8-DxcwTPRHuB9Y,10935
 autocoder/rag/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/utils.py,sha256=6LkPuMglUw_Wl0RdQI5x6TnQ4K_J3U8Siia4d1LdPy4,4963
 autocoder/rag/variable_holder.py,sha256=PFvBjFcR7-fNDD4Vcsc8CpH2Te057vcpwJMxtrfUgKI,75
@@ -100,14 +100,14 @@ autocoder/utils/llm_client_interceptors.py,sha256=FEHNXoFZlCjAHQcjPRyX8FOMjo6rPX
 autocoder/utils/log_capture.py,sha256=I-bsJFLWoGUiX-GKoZsH9kWJCKSV7ZlUnRt7jh-fOL0,1548
 autocoder/utils/multi_turn.py,sha256=unK9OpqVRbK6uIcTKXgggX2wNmyj7s5eyEAQ2xUwHoM,88
 autocoder/utils/print_table.py,sha256=ZMRhCA9DD0FUfKyJBWd5bDdj1RrtPtgOMWSJwtvZcLs,403
-autocoder/utils/queue_communicate.py,sha256=l80Bo_Bmqe2st1VNoUnyZrEmEfQfDBBweZmD7keti14,6740
+autocoder/utils/queue_communicate.py,sha256=buyEzdvab1QA4i2QKbq35rG5v_9x9PWVLWWMTznWcYM,6832
 autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1vmRciKdi4,2095
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=HawagAap3wMIDROGhY1730zSZrJR_EycODAA5qOj83c,8807
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.198.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.198.dist-info/METADATA,sha256=qa-B_AHKHsTq2Ldk8H1DthmZ5dUrrDfgrNUirUyg6EE,2352
-auto_coder-0.1.198.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.198.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.198.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.198.dist-info/RECORD,,
+auto_coder-0.1.200.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.200.dist-info/METADATA,sha256=PKrWaK1QEZzbTjumlUNzp-641Qm3T1XpdgzDZM8WjmY,2352
+auto_coder-0.1.200.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.200.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.200.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.200.dist-info/RECORD,,

autocoder/command_args.py

@@ -102,6 +102,7 @@ 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("--system_prompt", default="", help=desc["system_prompt"])
     parser.add_argument("--planner_model", default="",
                         help=desc["planner_model"])
     parser.add_argument(

autocoder/common/__init__.py

@@ -248,6 +248,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     sd_model: Optional[str] = ""
     emb_model: Optional[str] = ""
     code_model: Optional[str] = ""
+    system_prompt: Optional[str] = ""
     text2voice_model: Optional[str] = ""
     voice2text_model: Optional[str] = ""
     

autocoder/common/code_auto_generate.py

@@ -2,6 +2,8 @@ from typing import List, Dict, Tuple
 from autocoder.common.types import Mode
 from autocoder.common import AutoCoderArgs
 import byzerllm
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
+from autocoder.common import sys_prompt
 
 
 class CodeAutoGenerate:
@@ -146,6 +148,15 @@ class CodeAutoGenerate:
     ) -> Tuple[str, Dict[str, str]]:
         llm_config = {"human_as_model": self.args.human_as_model}
 
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_START.value,
+                    data=query,
+                ),
+            )
+
         if self.args.template == "common":
             init_prompt = self.single_round_instruction.prompt(
                 instruction=query, content=source_content, context=self.args.context
@@ -158,10 +169,28 @@ class CodeAutoGenerate:
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
 
-        conversations = [{"role": "user", "content": init_prompt}]
+        conversations = []
+
+        if self.args.system_prompt and self.args.system_prompt.strip() == "claude":
+            conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        elif self.args.system_prompt:
+            conversations.append({"role": "system", "content": self.args.system_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})
+
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_END.value,
+                    data="",
+                ),
+            )
+
         return [t[0].output], conversations
 
     def multi_round_run(

autocoder/common/code_auto_generate_diff.py

@@ -2,7 +2,8 @@ from typing import List, Dict, Tuple
 from autocoder.common.types import Mode
 from autocoder.common import AutoCoderArgs
 import byzerllm
-
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
+from autocoder.common import sys_prompt
 
 class CodeAutoGenerateDiff:
     def __init__(
@@ -289,6 +290,15 @@ class CodeAutoGenerateDiff:
     ) -> Tuple[str, Dict[str, str]]:
         llm_config = {"human_as_model": self.args.human_as_model}
 
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_START.value,
+                    data=query,
+                ),
+            )
+
         init_prompt = self.single_round_instruction.prompt(
             instruction=query, content=source_content, context=self.args.context
         )
@@ -296,10 +306,24 @@ class CodeAutoGenerateDiff:
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
 
-        conversations = [{"role": "user", "content": init_prompt}]
+        conversations = []
+        if self.args.system_prompt and self.args.system_prompt.strip() == "claude":
+            conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        else:
+            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})
+
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_END.value,
+                    data="",
+                ),
+            )
+
         return [t[0].output], conversations
 
     def multi_round_run(

autocoder/common/code_auto_generate_editblock.py

@@ -3,6 +3,12 @@ from autocoder.common.types import Mode
 from autocoder.common import AutoCoderArgs
 import byzerllm
 from autocoder.common import sys_prompt
+from autocoder.utils.queue_communicate import (
+    queue_communicate,
+    CommunicateEvent,
+    CommunicateEventType,
+)
+import json
 
 
 class CodeAutoGenerateEditBlock:
@@ -44,7 +50,7 @@ class CodeAutoGenerateEditBlock:
         """
 
     @byzerllm.prompt(llm=lambda self: self.llm)
-    def multi_round_instruction(self, instruction: str, content: str,context:str="") -> str:
+    def multi_round_instruction(self, instruction: str, content: str, context: str = "") -> str:
         """
         如果你需要生成代码，对于每个需要更改的文件,你需要按 *SEARCH/REPLACE block* 的格式进行生成。
 
@@ -141,7 +147,7 @@ class CodeAutoGenerateEditBlock:
 
         Here are the *SEARCH/REPLACE* blocks:
 
-        
+
         {{ fence_0 }}python
         ##File: /tmp/projects/mathweb/hello.py
         <<<<<<< SEARCH
@@ -152,7 +158,7 @@ class CodeAutoGenerateEditBlock:
             print("hello")
         >>>>>>> REPLACE
         {{ fence_1 }}
-        
+
         {{ fence_0 }}python
         ##File: /tmp/projects/mathweb/main.py
         <<<<<<< SEARCH
@@ -206,7 +212,7 @@ class CodeAutoGenerateEditBlock:
         }
 
     @byzerllm.prompt(llm=lambda self: self.llm)
-    def single_round_instruction(self, instruction: str, content: str, context:str="") -> str:
+    def single_round_instruction(self, instruction: str, content: str, context: str = "") -> str:
         """
         如果你需要生成代码，对于每个需要更改的文件,你需要按 *SEARCH/REPLACE block* 的格式进行生成。
 
@@ -302,7 +308,7 @@ class CodeAutoGenerateEditBlock:
         2. Remove hello() from main.py and replace it with an import.
 
         Here are the *SEARCH/REPLACE* blocks:
-        
+
         {{ fence_0 }}python
         ##File: /tmp/projects/mathweb/hello.py
         <<<<<<< SEARCH
@@ -313,7 +319,7 @@ class CodeAutoGenerateEditBlock:
             print("hello")
         >>>>>>> REPLACE
         {{ fence_1 }}
-        
+
         {{ fence_0 }}python
         ##File: /tmp/projects/mathweb/main.py
         <<<<<<< SEARCH
@@ -353,7 +359,7 @@ class CodeAutoGenerateEditBlock:
                 "fence_0": self.fence_0,
                 "fence_1": self.fence_1,
             }
-        
+
         return {
             "structure": (
                 self.action.pp.get_tree_like_directory_structure()
@@ -371,7 +377,7 @@ class CodeAutoGenerateEditBlock:
 
         if self.args.template == "common":
             init_prompt = self.single_round_instruction.prompt(
-                instruction=query, content=source_content,context=self.args.context
+                instruction=query, content=source_content, context=self.args.context
             )
         elif self.args.template == "auto_implement":
             init_prompt = self.auto_implement_function.prompt(
@@ -380,15 +386,42 @@ class CodeAutoGenerateEditBlock:
 
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
-        
 
         conversations = []
-        # conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        
+        if self.args.system_prompt and self.args.system_prompt.strip() == "claude":
+            conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        elif self.args.system_prompt:
+            conversations.append({"role": "system", "content": self.args.system_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})
-        return [t[0].output], conversations
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_START.value,
+                    data=json.dumps({}, ensure_ascii=False),
+                ),
+            )
+            
+            t = self.llm.chat_oai(
+                conversations=conversations, llm_config=llm_config)
+            conversations.append({"role": "assistant", "content": t[0].output})
+            
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_END.value,
+                    data=json.dumps({}, ensure_ascii=False),
+                ),
+            )
+            return [t[0].output], conversations
+        else:
+            t = self.llm.chat_oai(
+                conversations=conversations, llm_config=llm_config)
+            conversations.append({"role": "assistant", "content": t[0].output})
+            return [t[0].output], conversations
 
     def multi_round_run(
         self, query: str, source_content: str, max_steps: int = 10
@@ -398,7 +431,7 @@ class CodeAutoGenerateEditBlock:
 
         if self.args.template == "common":
             init_prompt = self.multi_round_instruction.prompt(
-                instruction=query, content=source_content,context=self.args.context
+                instruction=query, content=source_content, context=self.args.context
             )
         elif self.args.template == "auto_implement":
             init_prompt = self.auto_implement_function.prompt(
@@ -412,7 +445,8 @@ class CodeAutoGenerateEditBlock:
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
 
-        t = self.llm.chat_oai(conversations=conversations, llm_config=llm_config)
+        t = self.llm.chat_oai(conversations=conversations,
+                              llm_config=llm_config)
 
         result.append(t[0].output)
 
@@ -430,7 +464,8 @@ class CodeAutoGenerateEditBlock:
             with open(self.args.target_file, "w") as file:
                 file.write("继续")
 
-            t = self.llm.chat_oai(conversations=conversations, llm_config=llm_config)
+            t = self.llm.chat_oai(
+                conversations=conversations, llm_config=llm_config)
 
             result.append(t[0].output)
             conversations.append({"role": "assistant", "content": t[0].output})

autocoder/common/code_auto_generate_strict_diff.py

@@ -2,7 +2,8 @@ from typing import List, Dict, Tuple
 from autocoder.common.types import Mode
 from autocoder.common import AutoCoderArgs
 import byzerllm
-
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
+from autocoder.common import sys_prompt
 
 class CodeAutoGenerateStrictDiff:
     def __init__(
@@ -260,6 +261,15 @@ class CodeAutoGenerateStrictDiff:
     ) -> Tuple[str, Dict[str, str]]:
         llm_config = {"human_as_model": self.args.human_as_model}
 
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_START.value,
+                    data=query,
+                ),
+            )
+
         init_prompt = self.single_round_instruction.prompt(
             instruction=query, content=source_content, context=self.args.context
         )
@@ -267,10 +277,26 @@ class CodeAutoGenerateStrictDiff:
         with open(self.args.target_file, "w") as file:
             file.write(init_prompt)
 
-        conversations = [{"role": "user", "content": init_prompt}]
+        conversations = []
+        if self.args.system_prompt and self.args.system_prompt.strip() == "claude":
+            conversations.append({"role": "system", "content": sys_prompt.prompt()})
+        elif self.args.system_prompt:
+            conversations.append({"role": "system", "content": self.args.system_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})
+
+        if self.args.request_id and not self.args.skip_events:
+            queue_communicate.send_event_no_wait(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_GENERATE_END.value,
+                    data="",
+                ),
+            )
+
         return [t[0].output], conversations
 
     def multi_round_run(

autocoder/common/sys_prompt.py

@@ -2,11 +2,11 @@ import byzerllm
 import datetime
 
 @byzerllm.prompt()
-def sys_prompt():
+def claude_sys_prompt():
     '''
     The assistant is Claude, created by Anthropic.
 
-    The current date is {{currentDateTime}}.    
+    The current date is {{ currentDateTime }}.    
 
     Claude’s knowledge base was last updated in April 2024. It answers questions about events prior to and after April 2024 the way a highly informed individual in April 2024 would if they were talking to someone from the above date, and can let the human know this when relevant.
 

autocoder/lang.py

@@ -61,7 +61,8 @@ lang_desc = {
         "screenshot_desc": "Generate screenshots of a webpage",
         "screenshot_url": "The URL of the webpage to capture",
         "screenshot_output": "The directory to save the screenshots",
-        "code_model": "The name of the code model to use. Default is empty",        
+        "code_model": "The name of the code model to use. Default is empty",
+        "system_prompt": "The system prompt for the model. Default is empty",
         "planner_model": "The name of the planner model to use. Default is empty",
         "designer_model": "The name of the designer model to use. Default is empty",
         "query_prefix": "The query prefix",
@@ -141,6 +142,7 @@ lang_desc = {
         "screenshot_url": "要捕获的网页的URL",
         "screenshot_output": "保存截图的目录",
         "code_model": "要使用的代码模型的名称。默认为空",
+        "system_prompt": "模型使用的系统提示词。默认为空",
         "next_desc": "基于上一个action文件创建一个新的action文件",
         "planner_model": "要使用的规划模型的名称。默认为空",
         "designer_model": "要使用的设计模型的名称。默认为空",

autocoder/rag/token_limiter.py

@@ -113,7 +113,7 @@ class TokenLimiter:
                 num_count += 1
                 reorder_relevant_docs.append(doc)
                 if "original_doc" in doc.metadata and "chunk_index" in doc.metadata:
-                    original_doc_name = doc.metadata["original_doc"].module_name
+                    original_doc_name = doc.metadata["original_doc"]
 
                     temp_docs = []
                     for temp_doc in relevant_docs[num_count:]:
@@ -122,7 +122,7 @@ class TokenLimiter:
                             and "chunk_index" in temp_doc.metadata
                         ):
                             if (
-                                temp_doc.metadata["original_doc"].module_name
+                                temp_doc.metadata["original_doc"]
                                 == original_doc_name
                             ):
                                 if temp_doc not in reorder_relevant_docs:

autocoder/utils/queue_communicate.py

@@ -14,6 +14,8 @@ class CommunicateEventType(Enum):
     CODE_UNMERGE_RESULT = "code_unmerge_result"
     CODE_START = "code_start"
     CODE_END = "code_end"
+    CODE_GENERATE_START = "code_generate_start"
+    CODE_GENERATE_END = "code_generate_end"
     CODE_HUMAN_AS_MODEL = "code_human_as_model"
     ASK_HUMAN = "ask_human"
     CODE_ERROR = "code_error"

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.198"
+__version__ = "0.1.200"