auto-coder 0.1.305__py3-none-any.whl → 0.1.307__py3-none-any.whl

autocoder/dispacher/__init__.py

@@ -1,30 +1,29 @@
-
 from autocoder.common import AutoCoderArgs
 from autocoder.dispacher.actions.copilot import ActionCopilot
-from autocoder.dispacher.actions.action import ActionTSProject, ActionPyScriptProject, ActionPyProject, ActionSuffixProject
-from autocoder.dispacher.actions.plugins.action_translate import ActionTranslate
+from autocoder.dispacher.actions.action import (
+    ActionTSProject,
+    ActionPyProject,
+    ActionSuffixProject,
+)
 from autocoder.dispacher.actions.plugins.action_regex_project import ActionRegexProject
 from typing import Optional
 import byzerllm
-import re
 
-class Dispacher():
-    def __init__(self, args:AutoCoderArgs,llm:Optional[byzerllm.ByzerLLM]=None):
+
+class Dispacher:
+    def __init__(self, args: AutoCoderArgs, llm: Optional[byzerllm.ByzerLLM] = None):
         self.args = args
-        self.llm = llm 
+        self.llm = llm
 
     def dispach(self):
-        args = self.args        
-        actions = [ActionTranslate(args=args,llm=self.llm),
-                    ActionTSProject(args=args,llm=self.llm),
-                    ActionPyScriptProject(args=args,llm=self.llm),
-                    ActionPyProject(args=args,llm=self.llm),
-                    ActionCopilot(args=args,llm=self.llm),
-                    ActionRegexProject(args=args,llm=self.llm),
-                    ActionSuffixProject(args=args,llm=self.llm)
-                    ]
+        args = self.args
+        actions = [            
+            ActionTSProject(args=args, llm=self.llm),
+            ActionPyProject(args=args, llm=self.llm),
+            ActionCopilot(args=args, llm=self.llm),
+            ActionRegexProject(args=args, llm=self.llm),
+            ActionSuffixProject(args=args, llm=self.llm),
+        ]
         for action in actions:
             if action.run():
-                return                
-        
-    
+                return

autocoder/dispacher/actions/action.py

@@ -221,138 +221,6 @@ class ActionTSProject(BaseAction):
                     )
 
 
-class ActionPyScriptProject(BaseAction):
-    def __init__(
-        self, args: AutoCoderArgs, llm: Optional[byzerllm.ByzerLLM] = None
-    ) -> None:
-        self.args = args
-        self.llm = llm
-        self.printer = Printer()
-
-    def run(self) -> bool:
-        args = self.args
-        if args.project_type != "py-script":
-            return False
-        pp = Level1PyProject(
-            script_path=args.script_path, package_name=args.package_name
-        )
-        pp.run()
-        source_code_list = SourceCodeList(pp.sources)
-        self.process_content(source_code_list)
-        return True
-
-    def process_content(self, source_code_list: SourceCodeList):
-        args = self.args
-        global_cancel.check_and_raise()
-
-        if args.execute:
-            self.printer.print_in_terminal("code_generation_start")
-            start_time = time.time()
-            if args.auto_merge == "diff":
-                generate = CodeAutoGenerateDiff(
-                    llm=self.llm, args=self.args, action=self
-                )
-            elif args.auto_merge == "strict_diff":
-                generate = CodeAutoGenerateStrictDiff(
-                    llm=self.llm, args=self.args, action=self
-                )
-            elif args.auto_merge == "editblock":
-                generate = CodeAutoGenerateEditBlock(
-                    llm=self.llm, args=self.args, action=self
-                )
-            else:
-                generate = CodeAutoGenerate(
-                    llm=self.llm, args=self.args, action=self)
-            if self.args.enable_multi_round_generate:
-                generate_result = generate.multi_round_run(
-                    query=args.query, source_code_list=source_code_list
-                )
-            else:
-                generate_result = generate.single_round_run(
-                    query=args.query, source_code_list=source_code_list
-                )
-
-            elapsed_time = time.time() - start_time
-            speed = generate_result.metadata.get(
-                'generated_tokens_count', 0) / elapsed_time if elapsed_time > 0 else 0
-            model_names = ",".join(get_llm_names(generate.llms))
-            input_tokens_cost = generate_result.metadata.get(
-                'input_tokens_cost', 0)
-            generated_tokens_cost = generate_result.metadata.get(
-                'generated_tokens_cost', 0)
-            self.printer.print_in_terminal(
-                "code_generation_complete",
-                duration=elapsed_time,
-                input_tokens=generate_result.metadata.get(
-                    'input_tokens_count', 0),
-                output_tokens=generate_result.metadata.get(
-                    'generated_tokens_count', 0),
-                input_cost=input_tokens_cost,
-                output_cost=generated_tokens_cost,
-                speed=round(speed, 2),
-                model_names=model_names,
-                sampling_count=len(generate_result.contents)
-            )
-
-            get_event_manager(self.args.event_file).write_result(
-                EventContentCreator.create_result(content=EventContentCreator.ResultTokenStatContent(
-                    model_name=model_names,
-                    elapsed_time=elapsed_time,
-                    input_tokens=generate_result.metadata.get(
-                        'input_tokens_count', 0),
-                    output_tokens=generate_result.metadata.get(
-                        'generated_tokens_count', 0),
-                    input_cost=input_tokens_cost,
-                    output_cost=generated_tokens_cost,
-                    speed=round(speed, 2)
-                )).to_dict(),metadata=EventMetadata(
-                    action_file=self.args.file
-                ).to_dict())
-
-            global_cancel.check_and_raise()
-
-            merge_result = None
-            if args.execute and args.auto_merge:
-                self.printer.print_in_terminal("code_merge_start")
-                if args.auto_merge == "diff":
-                    code_merge = CodeAutoMergeDiff(
-                        llm=self.llm, args=self.args)
-                    merge_result = code_merge.merge_code(
-                        generate_result=generate_result)
-                elif args.auto_merge == "strict_diff":
-                    code_merge = CodeAutoMergeStrictDiff(
-                        llm=self.llm, args=self.args)
-                    merge_result = code_merge.merge_code(
-                        generate_result=generate_result)
-                elif args.auto_merge == "editblock":
-                    code_merge = CodeAutoMergeEditBlock(
-                        llm=self.llm, args=self.args)
-                    merge_result = code_merge.merge_code(
-                        generate_result=generate_result)
-                else:
-                    code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                    merge_result = code_merge.merge_code(
-                        generate_result=generate_result)
-
-                content = merge_result.contents[0]
-
-                store_code_model_conversation(
-                    args=self.args,
-                    instruction=self.args.query,
-                    conversations=merge_result.conversations[0],
-                    model=self.llm.default_model_name,
-                )
-            else:
-                content = generate_result.contents[0]
-
-                store_code_model_conversation(
-                    args=self.args,
-                    instruction=self.args.query,
-                    conversations=generate_result.conversations[0],
-                    model=self.llm.default_model_name,
-                )
-
-
 class ActionPyProject(BaseAction):
     def __init__(
         self, args: AutoCoderArgs, llm: Optional[byzerllm.ByzerLLM] = None

autocoder/index/filter/quick_filter.py

@@ -120,7 +120,8 @@ class QuickFilter():
                         self.index_manager.index_filter_llm,
                         [{"role": "user", "content": self.quick_filter_files.prompt(
                             chunk, self.args.query)}],
-                        {}
+                        {},
+                        self.args
                     )
                     full_response, last_meta = stream_out(
                         stream_generator,
@@ -485,7 +486,8 @@ class QuickFilter():
             stream_generator = stream_chat_with_continue(
                 self.index_manager.index_filter_llm,
                 [{"role": "user", "content": query}],
-                {}
+                {},
+                self.args
             )
 
             def extract_file_number_list(content: str) -> str:
@@ -760,7 +762,8 @@ class QuickFilter():
                 stream_generator = stream_chat_with_continue(
                     self.index_manager.index_filter_llm,
                     [{"role": "user", "content": prompt}],
-                    {}
+                    {},
+                    self.args
                 )
                 
                 def extract_file_number_list(content: str) -> str:

autocoder/memory/__init__.py

@@ -0,0 +1,7 @@
+"""
+记忆子系统 - 用于跟踪和记录代码变更上下文
+"""
+
+from autocoder.memory.active_context_manager import ActiveContextManager
+
+__all__ = ["ActiveContextManager"]