auto-coder 0.1.399__py3-none-any.whl → 1.0.0__py3-none-any.whl

autocoder/sdk/core/auto_coder_core.py

@@ -651,163 +651,12 @@ class AutoCoderCore:
                 self._print_final_token_usage()
               
             raise e
-    
-    def modify_code(
-        self, 
-        prompt: str, 
-        pre_commit: bool = False,
-        extra_args: Optional[Dict[str, Any]] = None,
-        show_terminal: bool = True
-    ) -> CodeModificationResult:
-        """
-        代码修改接口 - 直接使用 run_auto_command
         
-        Args:
-            prompt: 修改提示
-            pre_commit: 是否预提交
-            extra_args: 额外参数
-            show_terminal: 是否显示到终端
-            
-        Returns:
-            CodeModificationResult: 修改结果
-        """
-        try:
-            # 重置累计的 token 使用情况
-            self._reset_token_usage()
-            
-            event_stream = self._sync_run_auto_command(
-                prompt, 
-                pre_commit=pre_commit, 
-                extra_args=extra_args
-            )
-            
-            # 分析事件流，提取修改结果
-            modified_files = []
-            created_files = []
-            deleted_files = []
-            messages = []
-            success = True
-            error_details = None
-            
-            for event in event_stream:
-                # 渲染事件到终端
-                self._render_stream_event(event, show_terminal)
-                
-                if event.event_type == "content":
-                    messages.append(event.data.get("content", ""))
-                elif event.event_type == "error":
-                    success = False
-                    error_details = event.data.get("error", "Unknown error")
-                elif event.event_type == "file_modified":
-                    modified_files.extend(event.data.get("files", []))
-                elif event.event_type == "file_created":
-                    created_files.extend(event.data.get("files", []))
-                elif event.event_type == "file_deleted":
-                    deleted_files.extend(event.data.get("files", []))
-                
-                # 添加小延迟以改善视觉效果
-                if show_terminal:
-                    time.sleep(0.05)
-            
-            # 打印最终的累计 token 使用情况
-            if show_terminal:
-                self._print_final_token_usage()
-            
-            return CodeModificationResult(
-                success=success,
-                message="".join(messages),
-                modified_files=modified_files,
-                created_files=created_files,
-                deleted_files=deleted_files,
-                error_details=error_details,
-                metadata={
-                    "pre_commit": pre_commit,
-                    "extra_args": extra_args or {}
-                }
-            )
-            
-        except Exception as e:
-            # 在异常时也打印累计的 token 使用情况
-            if show_terminal:
-                self._print_final_token_usage()
-                self._console.print(Panel(
-                    f"[bold red]FATAL ERROR:[/bold red]\n{str(e)}", 
-                    title="🔥 System Error", 
-                    border_style="red"
-                ))
-            return CodeModificationResult(
-                success=False,
-                message="",
-                error_details=str(e),
-                metadata={"exception_type": type(e).__name__}
-            )
-    
-    async def modify_code_stream(
-        self, 
-        prompt: str, 
-        pre_commit: bool = False,
-        extra_args: Optional[Dict[str, Any]] = None,
-        show_terminal: bool = True
-    ) -> AsyncIterator[StreamEvent]:
-        """
-        异步流式代码修改接口
-        
-        Args:
-            prompt: 修改提示
-            pre_commit: 是否预提交
-            extra_args: 额外参数
-            show_terminal: 是否显示到终端
-            
-        Yields:
-            StreamEvent: 修改事件流
-        """
-        try:
-            # 重置累计的 token 使用情况
-            self._reset_token_usage()
-            
-            loop = asyncio.get_event_loop()
-            
-            # 在线程池中执行同步调用
-            event_stream = await loop.run_in_executor(
-                self._executor,
-                self._sync_run_auto_command,
-                prompt,
-                pre_commit,
-                extra_args
-            )
-            
-            # 处理并转发事件流
-            for event in event_stream:
-                # 渲染事件到终端
-                self._render_stream_event(event, show_terminal)
-                
-                # 转发事件
-                yield event
-                
-                # 添加小延迟以改善视觉效果
-                if show_terminal:
-                    time.sleep(0.05)
-            
-            # 打印最终的累计 token 使用情况
-            if show_terminal:
-                self._print_final_token_usage()
-                
-        except Exception as e:
-            # 在异常时也打印累计的 token 使用情况
-            if show_terminal:
-                self._print_final_token_usage()
-                
-            error_event = StreamEvent(
-                event_type="error",
-                data={"error": str(e), "error_type": type(e).__name__}
-            )
-            self._render_stream_event(error_event, show_terminal)
-            yield error_event
-    
     def _sync_run_auto_command(
         self, 
         prompt: str, 
         pre_commit: bool = False,
+        pr: Optional[bool] = None,
         extra_args: Optional[Dict[str, Any]] = None
     ) -> Iterator[StreamEvent]:
         """
@@ -816,14 +665,20 @@ class AutoCoderCore:
         Args:
             prompt: 查询提示
             pre_commit: 是否预提交
+            pr: 是否创建 PR，如果为None则使用配置中的值
             extra_args: 额外参数
             
         Returns:
             Iterator[StreamEvent]: 事件流
         """
+        # 如果没有明确指定pr参数，使用配置中的值
+        if pr is None:
+            pr = self.options.pr
+        
         return self.bridge.call_run_auto_command(
             query=prompt,
             pre_commit=pre_commit,
+            pr=pr,
             extra_args=extra_args or {},
             stream=True
         )

autocoder/sdk/core/bridge.py

@@ -56,15 +56,14 @@ class AutoCoderBridge:
             pass
 
     def init_project_if_required(self):
-        if not os.path.exists(".auto-coder") or not os.path.exists("actions"):
-            subprocess.run(
-                ["auto-coder", "init", "--source_dir", "."], check=True
-            )        
+        from autocoder.auto_coder_runner import init_project_if_required as init_project_if_required_buildin
+        init_project_if_required_buildin(self.project_root,".py,.ts")
 
     def call_run_auto_command(
         self,
         query: str,
         pre_commit: bool = False,
+        pr: bool = False,
         extra_args: Optional[Dict[str, Any]] = None,
         stream: bool = True
     ) -> Iterator[StreamEvent]:
@@ -74,6 +73,7 @@ class AutoCoderBridge:
         Args:
             query: 查询内容
             pre_commit: 是否预提交
+            pr: 是否预提交
             extra_args: 额外参数
             stream: 是否流式返回
 
@@ -103,6 +103,7 @@ class AutoCoderBridge:
                 events = run_auto_command(
                     query=query,
                     pre_commit=pre_commit,
+                    pr=pr,
                     extra_args=extra_args
                 )
 

autocoder/sdk/models/options.py

@@ -52,6 +52,7 @@ class AutoCodeOptions:
     timeout: int = 30
     verbose: bool = False
     include_project_structure: bool = True
+    pr: bool = False  # 是否创建 PR
     
     def __post_init__(self):
         """初始化后验证"""
@@ -61,12 +62,12 @@ class AutoCodeOptions:
     def validate(self) -> None:
         """验证配置选项"""
         
-        # 验证max_turns
-        if self.max_turns <= 0:
-            raise ValidationError("max_turns", "must be positive integer")
+        # 验证max_turns（-1表示不限制）
+        if self.max_turns <= 0 and self.max_turns != -1:
+            raise ValidationError("max_turns", "must be positive integer or -1 (unlimited)")
         
-        if self.max_turns > 100:
-            raise ValidationError("max_turns", "cannot exceed 100")
+        if self.max_turns > 100 and self.max_turns != -1:
+            raise ValidationError("max_turns", "cannot exceed 100 (unless -1 for unlimited)")
         
         # 验证output_format
         if self.output_format not in OUTPUT_FORMATS:
@@ -140,7 +141,8 @@ class AutoCodeOptions:
             "temperature": self.temperature,
             "timeout": self.timeout,
             "verbose": self.verbose,
-            "include_project_structure": self.include_project_structure
+            "include_project_structure": self.include_project_structure,
+            "pr": self.pr
         }
     
     @classmethod

autocoder/version.py

@@ -1,2 +1,2 @@
-__version__ = "0.1.399"
+__version__ = "1.0.0"