auto-coder 0.1.231__py3-none-any.whl → 0.1.232__py3-none-any.whl

{auto_coder-0.1.231.dist-info → auto_coder-0.1.232.dist-info}/METADATA

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

{auto_coder-0.1.231.dist-info → auto_coder-0.1.232.dist-info}/RECORD

@@ -1,5 +1,5 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=DEAtEcOHw82tbmX64VeaUpsR6eG_mM2FTm_fwyKZ7IE,58226
+autocoder/auto_coder.py,sha256=REnZJUy4Dx9b7Z8gM9D4SWMkbHXl0JjF-DFiod4aB-A,58357
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=illKgzP2bv-Tq50ujsofJnOHdI4pzr0ALtfR8NHHWdQ,22351
 autocoder/auto_coder_rag_client_mcp.py,sha256=WV7j5JUiQge0x4-B7Hp5-pSAFXLbvLpzQMcCovbauIM,6276
@@ -11,7 +11,7 @@ autocoder/chat_auto_coder_lang.py,sha256=YJsFi8an0Kjbo9X7xKZfpdbHS3rbhrvChZNjWqE
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
 autocoder/models.py,sha256=FlBrF6HhGao_RiCSgYhCmP7vs0KlG4hI_BI6dyZiL9s,5292
-autocoder/version.py,sha256=kae9Q0JtG-0yFEVujzjGt1HsRxp6oJAjpsB8Fkk3hbY,24
+autocoder/version.py,sha256=pvV6K1v2HfuQywwyaRYD8uMohj-o0EfkRauHjMIrYtk,24
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=NWSAEsEk94vT3lGjfo25kKLMwYdPcpy9e-i21txPasQ,6942
 autocoder/agent/auto_filegroup.py,sha256=CW7bqp0FW1GIEMnl-blyAc2UGT7O9Mom0q66ITz1ckM,6635
@@ -61,7 +61,7 @@ autocoder/common/search_replace.py,sha256=GphFkc57Hb673CAwmbiocqTbw8vrV7TrZxtOhD
 autocoder/common/sys_prompt.py,sha256=JlexfjZt554faqbgkCmzOJqYUzDHfbnxly5ugFfHfEE,26403
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/types.py,sha256=PXTETrsTvhLE49jqAeUKGySvxBN9pjeyCgRHLDYdd9U,664
-autocoder/common/utils_code_auto_generate.py,sha256=c_N4ZdMLKZlkyPC7RL5hSVjUz8R_2xv7unqTmQ04haw,1586
+autocoder/common/utils_code_auto_generate.py,sha256=QwJYr6QhjlQbSo231IWuI6c8G0kXkYDDoDHMo5NSXqI,3348
 autocoder/common/mcp_servers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/mcp_servers/mcp_server_perplexity.py,sha256=jz0LkCgZcqKkNdLZ9swNOu9Besoba4JOyHDedoZnWHo,5546
 autocoder/data/tokenizer.json,sha256=QfO_ZCE9qMAS2L0IcaWKH99wRj6PCPEQ3bsQgvUp9mk,4607451
@@ -131,11 +131,11 @@ autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-
 autocoder/utils/rest.py,sha256=opE_kBEdNQdxh350M5lUTMk5TViRfpuKP_qWc0B1lks,8861
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=pBOyWa1qwCcsAag1XsLIeTMv_D4QN4ppGo5jFiKzIkE,4165
+autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=6D_SIa5hHSwIHC1poO_ztK7IVugAqNHu-jQySd7EnfQ,4181
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.231.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.231.dist-info/METADATA,sha256=Qt4XlOfi5lr8v_1fhYJJqVCoQlDHDzLvI9GawDN3NeI,2641
-auto_coder-0.1.231.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.231.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.231.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.231.dist-info/RECORD,,
+auto_coder-0.1.232.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.232.dist-info/METADATA,sha256=5XhWylx2N9N4DNDoJpF3GUWHJzpNVg2wwgBZ5m_rBDU,2641
+auto_coder-0.1.232.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.232.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.232.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.232.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -1272,10 +1272,13 @@ def main(input_args: Optional[List[str]] = None):
                     )
                 )
                 v = [[response.result,None]]
-            else:                
-                v = chat_llm.stream_chat_oai(
-                    conversations=loaded_conversations, delta_mode=True
-                )                
+            else:
+                from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+                v = stream_chat_with_continue(
+                    llm=chat_llm,
+                    conversations=loaded_conversations,
+                    llm_config={}
+                )              
             
             assistant_response, last_meta = stream_out(
                     v, 

autocoder/common/utils_code_auto_generate.py

@@ -1,14 +1,15 @@
-from byzerllm import ByzerLLM
-from typing import List,Any,Union
+from byzerllm import ByzerLLM,SimpleByzerLLM
+from typing import Generator, List, Any, Union
 from pydantic import BaseModel
 from loguru import logger
+
 class ChatWithContinueResult(BaseModel):
     content: str
     input_tokens_count: int
     generated_tokens_count: int
     
 
-def chat_with_continue(llm: ByzerLLM, conversations: List[dict], llm_config: dict) -> ChatWithContinueResult:
+def chat_with_continue(llm: Union[ByzerLLM,SimpleByzerLLM], conversations: List[dict], llm_config: dict) -> ChatWithContinueResult:
     final_result = ChatWithContinueResult(content="", input_tokens_count=0, generated_tokens_count=0)
     v = llm.chat_oai(
         conversations=conversations, llm_config=llm_config)
@@ -36,3 +37,54 @@ def chat_with_continue(llm: ByzerLLM, conversations: List[dict], llm_config: dic
     #   logger.info(f"The code generation is exceed the max length, continue to generate the code {count -1 } times")
     final_result.content = single_result    
     return final_result
+
+def stream_chat_with_continue(
+    llm: Union[ByzerLLM, SimpleByzerLLM], 
+    conversations: List[dict], 
+    llm_config: dict
+) -> Generator[Any, None, None]:
+    """
+    流式处理并继续生成内容，直到完成。
+    
+    Args:
+        llm (Union[ByzerLLM, SimpleByzerLLM]): LLM实例
+        conversations (List[dict]): 对话历史
+        llm_config (dict): LLM配置参数
+        
+
+    """
+    
+    count = 0
+    temp_conversations = conversations
+    current_metadata = None
+    
+    while True:
+        # 使用流式接口获取生成内容
+        stream_generator = llm.stream_chat_oai(
+            conversations=temp_conversations,
+            delta_mode=True,
+            llm_config={**llm_config, "gen.response_prefix": True if count > 0 else False}
+        )
+        
+        current_content = ""        
+        for res in stream_generator:
+            content = res[0]
+            current_content += content 
+            if current_metadata is None:
+                current_metadata = res[1]            
+            else:
+                current_metadata.generated_tokens_count += res[1].generated_tokens_count
+                current_metadata.input_tokens_count += res[1].input_tokens_count
+                current_metadata.finish_reason = res[1].finish_reason                
+            
+            # Yield 当前的 StreamChatWithContinueResult
+            yield (content,current_metadata)
+        
+        # 更新对话历史
+        temp_conversations.append({"role": "assistant", "content": current_content})
+        
+        # 检查是否需要继续生成
+        if current_metadata.finish_reason != "length" or count >= 5:
+            break
+        
+        count += 1

autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -39,7 +39,7 @@ def stream_out(
             console=console
         ) as live:
             for res in stream_generator:
-                last_meta = res[1]
+                last_meta = res[1]                
                 content = res[0]
                 assistant_response += content
                 

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.231"
+__version__ = "0.1.232"