auto-coder 0.1.228__py3-none-any.whl → 0.1.229__py3-none-any.whl

{auto_coder-0.1.228.dist-info → auto_coder-0.1.229.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.228
+Version: 0.1.229
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter-client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas] >=0.1.149
+Requires-Dist: byzerllm[saas] >=0.1.150
 Requires-Dist: patch
 Requires-Dist: diff-match-patch
 Requires-Dist: GitPython
@@ -35,6 +35,7 @@ Requires-Dist: anthropic
 Requires-Dist: google-generativeai
 Requires-Dist: protobuf
 Requires-Dist: azure-cognitiveservices-speech
+Requires-Dist: real-agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image

{auto_coder-0.1.228.dist-info → auto_coder-0.1.229.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=dggZdWFCXACD-5nObadXgA-JADDdwMwNkuyBhtqU_50,59164
+autocoder/auto_coder.py,sha256=TGauh4UJQoPTL8rplB_DzziXnVw37A6-vs0TpT6VtVA,57054
 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
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=ZKj_VdoEAFPsdInf_dQeBK6lQtz8Mzz7L4nLO6l8pEY,101456
+autocoder/chat_auto_coder.py,sha256=eufxzEhopHzLPZUhE8epP5d3JVvcUWXrl_DdSh2-Sfs,101889
 autocoder/chat_auto_coder_lang.py,sha256=YJsFi8an0Kjbo9X7xKZfpdbHS3rbhrvChZNjWqEQ5Sw,11032
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/models.py,sha256=DX30r_VjQK9cvYHPLYyEXuifkoJat8sRCvbOytrXwtY,5054
-autocoder/version.py,sha256=Wi8xJHc_sggKQwapQNkMrXbzUQskh-gmUlZ7t1VxJmc,24
+autocoder/models.py,sha256=YIvJZJ_Vrmff9mLyDuHvo8zd4YOnNP3v0uYG8oSu7ZM,5162
+autocoder/version.py,sha256=sfyghLRAQ6aSLULneoCBX_LJssf68smbuYMGOKVEtDI,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
@@ -129,9 +129,11 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=opE_kBEdNQdxh350M5lUTMk5TViRfpuKP_qWc0B1lks,8861
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.228.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.228.dist-info/METADATA,sha256=EUX3bowx5PPKjP_eJj8nV1PnpRQhhjpPOWjujzpnWM8,2615
-auto_coder-0.1.228.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.228.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.228.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.228.dist-info/RECORD,,
+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
+auto_coder-0.1.229.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.229.dist-info/METADATA,sha256=TAvwuRtI5_rO2QiAbGV_ug6L6bgkj8Jkj1p_p_zjz88,2641
+auto_coder-0.1.229.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.229.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.229.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.229.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -42,6 +42,7 @@ from rich.live import Live
 from autocoder.auto_coder_lang import get_message
 from autocoder.common.memory_manager import save_to_memory_file
 from autocoder import models as models_module
+from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
 
 console = Console()
 
@@ -1257,64 +1258,11 @@ def main(input_args: Optional[List[str]] = None):
                 v = chat_llm.stream_chat_oai(
                     conversations=loaded_conversations, delta_mode=True
                 )                
-
-            assistant_response = ""
-            markdown_content = ""
-
-            try:
-                with Live(
-                    Panel("", title="Response", border_style="green", expand=False),
-                    refresh_per_second=4,
-                    auto_refresh=True,
-                    vertical_overflow="visible",
-                    console=Console(force_terminal=True, color_system="auto", height=None)
-                ) as live:
-                    for res in v:
-                        markdown_content += res[0]
-                        assistant_response += res[0]
-                        if args.request_id:
-                            request_queue.add_request(
-                                args.request_id,
-                                RequestValue(
-                                    value=StreamValue(value=[res[0]]),
-                                    status=RequestOption.RUNNING,
-                                ),
-                            )
-                        live.update(
-                            Panel(
-                                Markdown(markdown_content),
-                                title="Response",
-                                border_style="green",
-                                expand=False,
-                            )
-                        )
-                    live.update(
-                            Panel(
-                                Markdown(markdown_content),
-                                title="Response",
-                                border_style="green",
-                                expand=False,
-                            )
-                        )        
-            except Exception as e:
-                ##MARK
-                console.print(Panel(
-                    f"Error: {str(e)}",  
-                    title="Error",
-                    border_style="red"
-                ))
-                request_queue.add_request(
-                    args.request_id,
-                    RequestValue(
-                        value=StreamValue(value=[str(e)]), status=RequestOption.FAILED
-                    ),
-                )
-            finally:
-                request_queue.add_request(
-                    args.request_id,
-                    RequestValue(
-                        value=StreamValue(value=[""]), status=RequestOption.COMPLETED
-                    ),
+            
+            assistant_response, last_meta = stream_out(
+                    v, 
+                    request_id=args.request_id,                    
+                    console=console
                 )
             
             chat_history["ask_conversation"].append(

autocoder/chat_auto_coder.py

@@ -2173,7 +2173,8 @@ def manage_models(params, query: str):
             "model_name": data_dict.get("model_name", data_dict["name"]),
             "base_url": data_dict.get("base_url", "https://api.openai.com/v1"),
             "api_key_path": data_dict.get("api_key_path", "api.openai.com"),
-            "description": data_dict.get("description", "")
+            "description": data_dict.get("description", ""),
+            "is_reasoning": data_dict.get("is_reasoning", "false") in ["true", "True", "TRUE", "1"]
         }
 
         models_data.append(final_model)
@@ -2476,6 +2477,9 @@ def lib_command(args: List[str]):
     else:
         console.print(f"Unknown subcommand: {subcommand}")
 
+def agent(query: str):
+    console.print(f"Agent query: {query}")
+
 
 def main():
     ARGS = parse_arguments()
@@ -2650,14 +2654,20 @@ def main():
                 if not query:
                     print("Please enter your query.")
                 else:
-                    manage_models(ARGS,query)    
+                    manage_models(ARGS,query) 
+            elif user_input.startswith("/agent"):
+                query = user_input[len("/agent"):].strip()
+                if not query:
+                    print("Please enter your query.")
+                else:
+                    agent(query)
 
             elif user_input.startswith("/mode"):
                 conf = user_input[len("/mode"):].strip()
                 if not conf:
                     print(memory["mode"])
                 else:
-                    memory["mode"] = conf
+                    memory["mode"] = conf                    
 
             elif user_input.startswith("/conf"):
                 conf = user_input[len("/conf"):].strip()

autocoder/models.py

@@ -13,7 +13,8 @@ default_models_list = [
         "model_name": "deepseek-reasoner",
         "model_type": "saas/openai",
         "base_url": "https://api.deepseek.com/v1",
-        "api_key_path": "api.deepseek.com"
+        "api_key_path": "api.deepseek.com",
+        "is_reasoning": True
     },    
     {
         "name": "deepseek_chat",
@@ -21,7 +22,8 @@ default_models_list = [
         "model_name": "deepseek-chat",
         "model_type": "saas/openai",
         "base_url": "https://api.deepseek.com/v1",
-        "api_key_path": "api.deepseek.com"
+        "api_key_path": "api.deepseek.com",
+        "is_reasoning": False
     },
     {
         "name":"o1",
@@ -29,7 +31,8 @@ default_models_list = [
         "model_name": "o1-2024-12-17",
         "model_type": "saas/openai",
         "base_url": "https://api.openai.com/v1",
-        "api_key_path": ""
+        "api_key_path": "",
+        "is_reasoning": False
     }
 ]
 
@@ -123,7 +126,7 @@ def update_model_with_api_key(name: str, api_key: str) -> Dict:
     # 在现有模型中查找
     found_model = None
     for model in models:
-        if model["name"] == name:
+        if model["name"] == name.strip():
             found_model = model
             break
                     
@@ -140,7 +143,7 @@ def update_model_with_api_key(name: str, api_key: str) -> Dict:
         os.makedirs(api_key_dir, exist_ok=True)
         api_key_file = os.path.join(api_key_dir, api_key_path)
         with open(api_key_file, "w") as f:
-            f.write(api_key)
+            f.write(api_key.strip())
         
         # 如果是新模型，添加到模型列表中
         if all(model["name"] != name for model in models):

autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -0,0 +1,120 @@
+from rich.console import Console
+from rich.live import Live
+from rich.panel import Panel
+from rich.markdown import Markdown
+from typing import Generator, List, Dict, Any, Optional, Tuple
+from autocoder.utils.request_queue import RequestValue, RequestOption, StreamValue
+from autocoder.utils.request_queue import request_queue
+
+MAX_HISTORY_LINES = 40  # 最大保留历史行数
+
+def stream_out(
+    stream_generator: Generator[Tuple[str, Dict[str, Any]], None, None],
+    request_id: Optional[str] = None,    
+    console: Optional[Console] = None
+) -> Tuple[str, Optional[Dict[str, Any]]]:
+    """
+    处理流式输出事件并在终端中展示
+    
+    Args:
+        stream_generator: 生成流式输出的生成器
+        request_id: 请求ID,用于更新请求队列        
+        console: Rich Console对象
+        
+    Returns:
+        Tuple[str, Dict[str, Any]]: 返回完整的响应内容和最后的元数据
+    """
+    if console is None:
+        console = Console(force_terminal=True, color_system="auto", height=None)
+        
+    lines_buffer = []  # 存储历史行
+    current_line = ""  # 当前行
+    assistant_response = ""
+    last_meta = None
+    
+    try:
+        with Live(
+            Panel("", title="Response", border_style="green"),
+            refresh_per_second=4,
+            console=console
+        ) as live:
+            for res in stream_generator:
+                last_meta = res[1]
+                content = res[0]
+                assistant_response += content
+                
+                # 处理所有行
+                parts = (current_line + content).split("\n")
+                
+                # 最后一部分是未完成的新行
+                if len(parts) > 1:
+                    # 将完整行加入缓冲区
+                    lines_buffer.extend(parts[:-1])
+                    # 保留最大行数限制
+                    if len(lines_buffer) > MAX_HISTORY_LINES:
+                        del lines_buffer[0:len(lines_buffer) - MAX_HISTORY_LINES]
+                
+                # 更新当前行
+                current_line = parts[-1]
+                
+                # 构建显示内容 = 历史行 + 当前行
+                display_content = "\n".join(lines_buffer[-MAX_HISTORY_LINES:] + [current_line])
+                
+                if request_id and request_queue:
+                    request_queue.add_request(
+                        request_id,
+                        RequestValue(
+                            value=StreamValue(value=[content]),
+                            status=RequestOption.RUNNING,
+                        ),
+                    )
+                    
+                live.update(
+                    Panel(
+                        Markdown(display_content),
+                        title="Response",
+                        border_style="green",
+                        height=min(50, live.console.height - 4)
+                    )
+                )
+            
+            # 处理最后一行的内容
+            if current_line:
+                lines_buffer.append(current_line)
+            
+            # 最终显示结果
+            live.update(
+                Panel(
+                    Markdown(assistant_response),
+                    title="Final Response",
+                    border_style="blue"
+                )
+            )
+            
+    except Exception as e:
+        console.print(Panel(
+            f"Error: {str(e)}",  
+            title="Error",
+            border_style="red"
+        ))
+        
+        if request_id and request_queue:
+            request_queue.add_request(
+                request_id,
+                RequestValue(
+                    value=StreamValue(value=[str(e)]), 
+                    status=RequestOption.FAILED
+                ),
+            )
+            
+    finally:
+        if request_id and request_queue:
+            request_queue.add_request(
+                request_id,
+                RequestValue(
+                    value=StreamValue(value=[""]), 
+                    status=RequestOption.COMPLETED
+                ),
+            )
+            
+    return assistant_response, last_meta

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.228"
+__version__ = "0.1.229"