auto-coder 0.1.301__py3-none-any.whl → 0.1.303__py3-none-any.whl

autocoder/common/auto_configure.py

@@ -156,13 +156,18 @@ def config_readme() -> str:
     2. py
     3. 代码文件后缀名列表（比如.java,.py,.go,.js,.ts），多个按逗号分割    
 
-    推荐使用 3 选项，因为项目类型通常为多种后缀名混合。
-
-    ## include_project_structure: 是否包含项目结构
-    是否包含项目结构。推荐设置为 true。默认为true,但是项目结构也可能很大，如果项目结构很大，那么可以设置为 false。
+    推荐使用 3 选项，因为项目类型通常为多种后缀名混合。    
 
     ## conversation_prune_safe_zone_tokens: 对话剪枝安全区token数量
     在对话剪枝时，会根据对话的token数量，如果token数量超过该值，那么会剪枝掉一部分对话。
+
+    ## enable_task_history
+    该参数对 /auto 指令有效， 当你使用 /auto 指令的时候，系统会自动将近期对项目的变更(commit) 信息给到大模型，方便大模型对用户的需求有更好的理解。
+    默认为 false
+
+    ## include_project_structure
+    使用 /chat , /coding 等指令时，是否将项目的目录结构也放到上下文中。
+    默认为true, 如果你项目很大，请设置为 false
     """
 
 class ConfigAutoTuner:

autocoder/common/code_auto_merge.py

@@ -197,7 +197,7 @@ class CodeAutoMerge:
 
         self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git and not self.args.skip_commit:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}")
             
             action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
             action_file_name = os.path.basename(self.args.file)

autocoder/common/code_auto_merge_diff.py

@@ -580,7 +580,7 @@ class CodeAutoMergeDiff:
 
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git and not self.args.skip_commit:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}")
 
             action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
             action_file_name = os.path.basename(self.args.file)

autocoder/common/code_auto_merge_editblock.py

@@ -427,7 +427,7 @@ class CodeAutoMergeEditBlock:
                 try:
                     commit_result = git_utils.commit_changes(
                         self.args.source_dir,
-                        f"{self.args.query}\nauto_coder_{file_name}_{md5}",
+                        f"{self.args.query}\nauto_coder_{file_name}",
                     )
                     
                     action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)

autocoder/common/code_auto_merge_strict_diff.py

@@ -287,7 +287,7 @@ class CodeAutoMergeStrictDiff:
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git and not self.args.skip_commit:
             commit_result = git_utils.commit_changes(
-                self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}"
+                self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}"
             )
             action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
             action_file_name = os.path.basename(self.args.file)

autocoder/common/stream_out_type.py

@@ -0,0 +1,7 @@
+from enum import Enum
+
+class AutoCommandStreamOutType(Enum):
+    COMMAND_SUGGESTION = "command_suggestion"
+
+class IndexFilterStreamOutType(Enum):
+    FILE_NUMBER_LIST = "file_number_list"

autocoder/dispacher/actions/action.py

@@ -30,17 +30,22 @@ from autocoder.utils.llms import get_llm_names
 from autocoder.privacy.model_filter import ModelPathFilter
 from autocoder.common import SourceCodeList
 from autocoder.common.global_cancel import global_cancel
+from autocoder.events.event_manager_singleton import get_event_manager
+from autocoder.events import event_content as EventContentCreator
+from autocoder.events.event_types import EventMetadata
 
 
-class BaseAction:    
+class BaseAction:
     def _get_content_length(self, content: str) -> int:
         try:
             tokenizer = BuildinTokenizer()
             return tokenizer.count_tokens(content)
         except Exception as e:
-            logger.warning(f"Failed to use tokenizer to count tokens, fallback to len(): {e}")
+            logger.warning(
+                f"Failed to use tokenizer to count tokens, fallback to len(): {e}")
             return len(content)
 
+
 class ActionTSProject(BaseAction):
     def __init__(
         self, args: AutoCoderArgs, llm: Optional[byzerllm.ByzerLLM] = None
@@ -66,7 +71,7 @@ class ActionTSProject(BaseAction):
                 args.query = (args.context or "") + "\n\n" + args.query
             source_code_list = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
-            )            
+            )
             if args.in_code_apply:
                 args.query = old_query
 
@@ -86,13 +91,13 @@ class ActionTSProject(BaseAction):
                 max_iter=self.args.image_max_iter,
             )
             html_code = ""
-            with open(html_path, "r",encoding="utf-8") as f:
+            with open(html_path, "r", encoding="utf-8") as f:
                 html_code = f.read()
-            
-            source_code_list.sources.append(SourceCode( 
+
+            source_code_list.sources.append(SourceCode(
                 module_name=html_path,
                 source_code=html_code,
-                tag="IMAGE"))            
+                tag="IMAGE"))
 
         self.process_content(source_code_list)
         return True
@@ -105,13 +110,14 @@ class ActionTSProject(BaseAction):
             if content_length > self.args.model_max_input_length:
                 logger.warning(
                     f"Content(send to model) is {content_length} tokens, which is larger than the maximum input length {self.args.model_max_input_length}"
-                )                
+                )
 
         if global_cancel.requested:
-            printer = Printer()            
-            raise Exception(printer.get_message_from_key("generation_cancelled")) 
-                             
-        if args.execute:             
+            printer = Printer()
+            raise Exception(printer.get_message_from_key(
+                "generation_cancelled"))
+
+        if args.execute:
             self.printer.print_in_terminal("code_generation_start")
             start_time = time.time()
             if args.auto_merge == "diff":
@@ -127,8 +133,9 @@ class ActionTSProject(BaseAction):
                     llm=self.llm, args=self.args, action=self
                 )
             else:
-                generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
-                        
+                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
@@ -138,41 +145,69 @@ class ActionTSProject(BaseAction):
                     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
-            input_tokens_cost = generate_result.metadata.get('input_tokens_cost', 0)
-            generated_tokens_cost = generate_result.metadata.get('generated_tokens_cost', 0)
+            speed = generate_result.metadata.get(
+                'generated_tokens_count', 0) / elapsed_time if elapsed_time > 0 else 0
+            input_tokens_cost = generate_result.metadata.get(
+                'input_tokens_cost', 0)
+            generated_tokens_cost = generate_result.metadata.get(
+                'generated_tokens_cost', 0)
             model_names = ",".join(get_llm_names(generate.llms))
             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_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())
+
             if global_cancel.requested:
-                printer = Printer()            
-                raise Exception(printer.get_message_from_key("generation_cancelled")) 
-            
+                printer = Printer()
+                raise Exception(printer.get_message_from_key(
+                    "generation_cancelled"))
+
             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)
+                    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)
+                    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)
+                    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)
+                    merge_result = code_merge.merge_code(
+                        generate_result=generate_result)
 
                 if merge_result is not None:
                     content = merge_result.contents[0]
@@ -190,7 +225,6 @@ class ActionTSProject(BaseAction):
                         conversations=generate_result.conversations[0],
                         model=self.llm.default_model_name,
                     )
-                
 
 
 class ActionPyScriptProject(BaseAction):
@@ -216,9 +250,10 @@ class ActionPyScriptProject(BaseAction):
     def process_content(self, source_code_list: SourceCodeList):
         args = self.args
         if global_cancel.requested:
-            printer = Printer()            
-            raise Exception(printer.get_message_from_key("generation_cancelled")) 
-        
+            printer = Printer()
+            raise Exception(printer.get_message_from_key(
+                "generation_cancelled"))
+
         if args.execute:
             self.printer.print_in_terminal("code_generation_start")
             start_time = time.time()
@@ -235,7 +270,8 @@ class ActionPyScriptProject(BaseAction):
                     llm=self.llm, args=self.args, action=self
                 )
             else:
-                generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
+                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
@@ -246,15 +282,20 @@ class ActionPyScriptProject(BaseAction):
                 )
 
             elapsed_time = time.time() - start_time
-            speed = generate_result.metadata.get('generated_tokens_count', 0) / elapsed_time if elapsed_time > 0 else 0
+            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)
+            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_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),
@@ -262,25 +303,48 @@ class ActionPyScriptProject(BaseAction):
                 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())
+
             if global_cancel.requested:
-                printer = Printer()            
-                raise Exception(printer.get_message_from_key("generation_cancelled")) 
-            
+                printer = Printer()
+                raise Exception(printer.get_message_from_key(
+                    "generation_cancelled"))
+
             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)
+                    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)
+                    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)
+                    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)
+                    merge_result = code_merge.merge_code(
+                        generate_result=generate_result)
 
                 content = merge_result.contents[0]
 
@@ -298,7 +362,7 @@ class ActionPyScriptProject(BaseAction):
                     instruction=self.args.query,
                     conversations=generate_result.conversations[0],
                     model=self.llm.default_model_name,
-                )                        
+                )
 
 
 class ActionPyProject(BaseAction):
@@ -316,8 +380,9 @@ class ActionPyProject(BaseAction):
             return False
         pp = PyProject(args=self.args, llm=self.llm)
         self.pp = pp
-        pp.run(packages=args.py_packages.split(",") if args.py_packages else [])            
-        source_code_list = SourceCodeList(pp.sources)                
+        pp.run(packages=args.py_packages.split(
+            ",") if args.py_packages else [])
+        source_code_list = SourceCodeList(pp.sources)
 
         if self.llm:
             old_query = args.query
@@ -339,15 +404,16 @@ class ActionPyProject(BaseAction):
             content_length = self._get_content_length(content)
             if content_length > self.args.model_max_input_length:
                 self.printer.print_in_terminal(
-                "code_execution_warning",
-                style="yellow",
-                content_length=content_length,
-                max_length=self.args.model_max_input_length
-            )
-        
+                    "code_execution_warning",
+                    style="yellow",
+                    content_length=content_length,
+                    max_length=self.args.model_max_input_length
+                )
+
         if global_cancel.requested:
-            printer = Printer()            
-            raise Exception(printer.get_message_from_key("generation_cancelled")) 
+            printer = Printer()
+            raise Exception(printer.get_message_from_key(
+                "generation_cancelled"))
 
         if args.execute:
             self.printer.print_in_terminal("code_generation_start")
@@ -365,8 +431,8 @@ class ActionPyProject(BaseAction):
                     llm=self.llm, args=self.args, action=self
                 )
             else:
-                generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
-
+                generate = CodeAutoGenerate(
+                    llm=self.llm, args=self.args, action=self)
 
             if self.args.enable_multi_round_generate:
                 generate_result = generate.multi_round_run(
@@ -377,15 +443,20 @@ class ActionPyProject(BaseAction):
                     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
+            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)
+            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_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),
@@ -393,25 +464,48 @@ class ActionPyProject(BaseAction):
                 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())
+
             if global_cancel.requested:
-                printer = Printer()            
-                raise Exception(printer.get_message_from_key("generation_cancelled")) 
-            
+                printer = Printer()
+                raise Exception(printer.get_message_from_key(
+                    "generation_cancelled"))
+
             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)
+                    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)
+                    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)
+                    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)
+                    merge_result = code_merge.merge_code(
+                        generate_result=generate_result)
 
                 content = merge_result.contents[0]
 
@@ -429,7 +523,7 @@ class ActionPyProject(BaseAction):
                     instruction=self.args.query,
                     conversations=generate_result.conversations[0],
                     model=self.llm.default_model_name,
-                )                        
+                )
 
 
 class ActionSuffixProject(BaseAction):
@@ -467,11 +561,12 @@ class ActionSuffixProject(BaseAction):
             if content_length > self.args.model_max_input_length:
                 logger.warning(
                     f"Content(send to model) is {content_length} tokens, which is larger than the maximum input length {self.args.model_max_input_length}"
-                )                
+                )
 
         if global_cancel.requested:
-            printer = Printer()            
-            raise Exception(printer.get_message_from_key("generation_cancelled")) 
+            printer = Printer()
+            raise Exception(printer.get_message_from_key(
+                "generation_cancelled"))
 
         if args.execute:
             self.printer.print_in_terminal("code_generation_start")
@@ -489,7 +584,8 @@ class ActionSuffixProject(BaseAction):
                     llm=self.llm, args=self.args, action=self
                 )
             else:
-                generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
+                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
@@ -498,43 +594,69 @@ class ActionSuffixProject(BaseAction):
                 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
+        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)
+        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),
+            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())
+
         if global_cancel.requested:
-            printer = Printer()            
-            raise Exception(printer.get_message_from_key("generation_cancelled")                    )
-        
+            printer = Printer()
+            raise Exception(printer.get_message_from_key(
+                "generation_cancelled"))
+
         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)
+                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)
+                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)
+                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)
+                merge_result = code_merge.merge_code(
+                    generate_result=generate_result)
 
         if merge_result is not None:
             content = merge_result.contents[0]
@@ -543,7 +665,7 @@ class ActionSuffixProject(BaseAction):
                 instruction=self.args.query,
                 conversations=merge_result.conversations[0],
                 model=self.llm.default_model_name,
-            )            
+            )
         else:
             content = generate_result.contents[0]
 
@@ -553,5 +675,3 @@ class ActionSuffixProject(BaseAction):
                 conversations=generate_result.conversations[0],
                 model=self.llm.default_model_name,
             )
-            
-