auto-coder 0.1.198__tar.gz → 0.1.200__tar.gz

{auto_coder-0.1.198 → auto_coder-0.1.200}/PKG-INFO

@@ -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 → auto_coder-0.1.200}/src/auto_coder.egg-info/PKG-INFO

@@ -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 → auto_coder-0.1.200}/src/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(

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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] = ""
     

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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(

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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(

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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})

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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(

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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.
 

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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": "要使用的设计模型的名称。默认为空",

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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:

{auto_coder-0.1.198 → auto_coder-0.1.200}/src/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"

auto_coder-0.1.200/src/autocoder/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.200"

auto_coder-0.1.198/src/autocoder/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.198"