auto-coder 0.1.220__py3-none-any.whl → 0.1.222__py3-none-any.whl

{auto_coder-0.1.220.dist-info → auto_coder-0.1.222.dist-info}/METADATA

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

{auto_coder-0.1.220.dist-info → auto_coder-0.1.222.dist-info}/RECORD

@@ -1,14 +1,14 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=xWEEptvSMPu0tEtqOo2V-GoCzL1402NveoxjyYrsw_g,41929
+autocoder/auto_coder.py,sha256=BQvCwOc-4XgOtpwQWHCTvcx97XEejdwRordyrCYSf0s,43554
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=illKgzP2bv-Tq50ujsofJnOHdI4pzr0ALtfR8NHHWdQ,22351
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=NTCWQKBQJluEhay5NGoTl5tdv00Zmu8oFioCjCpJCd8,94294
+autocoder/chat_auto_coder.py,sha256=x98afu7PCzYtf2545tIdJP13tI3lixFJg4sSSFtRjeM,95346
 autocoder/chat_auto_coder_lang.py,sha256=ReWukXKVvuzVvpbYk5O9kc1ev7XNmAv3DnuQhmpLmnc,8717
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/version.py,sha256=AkV-rQzPFI1YGmw3_1zhV77j25aK6xZpk7snIDE9-KU,24
+autocoder/version.py,sha256=tEdfS6Yw3kxd6NglROcL14L5i85YA9bN5_7H9HyePxA,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
@@ -21,7 +21,7 @@ autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMb
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=Itx7HWrrMEMszf4hpUZ7YhIGbpWQVtJjQOWOEqRsa-Q,11458
+autocoder/common/__init__.py,sha256=jwjsx9qT_5fJmdlbNt3I5UiDtDKcdeLqQ8s_Tw_sT0U,11492
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/anything2img.py,sha256=4TREa-sOA-iargieUy7MpyCYVUE-9Mmq0wJtwomPqnE,7662
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
@@ -38,7 +38,7 @@ autocoder/common/code_auto_merge_diff.py,sha256=5SI6ggklJ0QDHvsS0cpNXFuIkFRQxp1i
 autocoder/common/code_auto_merge_editblock.py,sha256=l6yEiZqXyIlUNIIPXvkHOnLCIInXR78TzSjF-jtJkkg,17035
 autocoder/common/code_auto_merge_strict_diff.py,sha256=ABYOTDUQYA4Bn4BwT1Rw812y49cHW3UH_JSpM9uJ6ig,9399
 autocoder/common/code_modification_ranker.py,sha256=DFlbwgdg8GK47zVcvfZSzkyniEKmTVLTOWejjcVIgaw,5121
-autocoder/common/command_completer.py,sha256=10nNnazkzZwya-jScPLlhpCBR2oYsOiVfwofqGvIUEM,9145
+autocoder/common/command_completer.py,sha256=xXDatBOkjV78jsgG1eF8pcMFC0glTm1poesjd_Q6kSw,9169
 autocoder/common/command_generator.py,sha256=v4LmU7sO-P7jEZIXCWHUC6P-vT7AvBi_x_PTwCqBAE8,1323
 autocoder/common/command_templates.py,sha256=3G-pCNbL6iHbnkG6v1JZpbIK3Mc9d373_RYGmCcDPMY,8548
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
@@ -57,7 +57,7 @@ autocoder/common/sys_prompt.py,sha256=JlexfjZt554faqbgkCmzOJqYUzDHfbnxly5ugFfHfE
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/types.py,sha256=oQKPE1TG0O7DQQLaBSMp6CP3Bbvg0K8elFdidlV52Lg,631
 autocoder/common/mcp_servers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/common/mcp_servers/mcp_server_perplexity.py,sha256=OV_orQ6NsVnLxGt0SkVt9wsxu-nw7gOuwa4hgtdastI,5546
+autocoder/common/mcp_servers/mcp_server_perplexity.py,sha256=jz0LkCgZcqKkNdLZ9swNOu9Besoba4JOyHDedoZnWHo,5546
 autocoder/data/tokenizer.json,sha256=QfO_ZCE9qMAS2L0IcaWKH99wRj6PCPEQ3bsQgvUp9mk,4607451
 autocoder/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/db/store.py,sha256=tFT66bP2ZKIqZip-uhLkHRSLaaOAUUDZfozJwcqix3c,1908
@@ -78,7 +78,7 @@ autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7
 autocoder/rag/doc_filter.py,sha256=B99Qcy3tcNLuSz2kWbpgfBj2_Igme91zWKOJ2Niq2UY,6652
 autocoder/rag/document_retriever.py,sha256=5oThtxukGuRFF96o3pHKsk306a8diXbhgSrbqyU2BvM,8894
 autocoder/rag/llm_wrapper.py,sha256=sbDxCANiZyWb_ocqNgqu2oy3c2t8orPNRGleEs-Uwl8,2649
-autocoder/rag/long_context_rag.py,sha256=jEhil33y2ryAF393zKdUhrk_QEHUlZ30gZcvPH0Z3MY,25074
+autocoder/rag/long_context_rag.py,sha256=F-ulTwSZ9ogIGL9jTXDUnI44zStH02EjrztZgRkW9-g,25246
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=6TKtErZ0Us9XSV6HgRKXA6yR3SiZGPHpynOKSaR1wgE,2463
 autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
@@ -122,9 +122,9 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=HawagAap3wMIDROGhY1730zSZrJR_EycODAA5qOj83c,8807
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.220.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.220.dist-info/METADATA,sha256=uQi6g28X2Lzv4rVNmWu0j3yEQ0W2uo_YQOqdXmpSHwc,2615
-auto_coder-0.1.220.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.220.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.220.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.220.dist-info/RECORD,,
+auto_coder-0.1.222.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.222.dist-info/METADATA,sha256=T3YPAkMEoTSY8J8xgPi-f-Rn_szn6-A4wJhbQfn5Gvs,2615
+auto_coder-0.1.222.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.222.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.222.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.222.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -577,6 +577,9 @@ def main(input_args: Optional[List[str]] = None):
             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:
                 live.update(
                     Panel(
@@ -689,6 +692,9 @@ def main(input_args: Optional[List[str]] = None):
             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:
                 live.update(
                     Panel(
@@ -698,8 +704,8 @@ def main(input_args: Optional[List[str]] = None):
                         expand=False,
                     )
                 )
-            return
 
+            return
         elif raw_args.agent_command == "designer":
             from autocoder.agent.designer import SVGDesigner, SDDesigner, LogoDesigner
 
@@ -938,11 +944,24 @@ def main(input_args: Optional[List[str]] = None):
 
                 return {}
 
-            if args.enable_rag_search or args.enable_rag_context:
+            if args.action == "rag":
+                args.enable_rag_search = True
+                args.enable_rag_context = False
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
                 response = rag.stream_chat_oai(
-                    conversations=loaded_conversations)[0]
+                    conversations=[{"role": "user", "content": args.query}])[0]                                                
                 v = ([item, None] for item in response)
+                
+            elif args.action == "mcp": 
+                from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRequest, McpRemoveRequest, McpListRequest, McpListRunningRequest, McpRefreshRequest                               
+                mcp_server = get_mcp_server()
+                response = mcp_server.send_request(
+                    McpRequest(
+                        query=args.query,
+                        model=args.inference_model or args.model
+                    )
+                )
+                v = [[response.result,None]]
             else:
                 v = chat_llm.stream_chat_oai(
                     conversations=loaded_conversations, delta_mode=True
@@ -953,8 +972,11 @@ def main(input_args: Optional[List[str]] = None):
 
             try:
                 with Live(
-                    Panel("", title="Response"),
+                    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]
@@ -975,6 +997,14 @@ def main(input_args: Optional[List[str]] = None):
                                 expand=False,
                             )
                         )
+                    live.update(
+                            Panel(
+                                Markdown(markdown_content),
+                                title="Response",
+                                border_style="green",
+                                expand=False,
+                            )
+                        )        
             except Exception as e:
                 request_queue.add_request(
                     args.request_id,
@@ -989,7 +1019,7 @@ def main(input_args: Optional[List[str]] = None):
                         value=StreamValue(value=[""]), status=RequestOption.COMPLETED
                     ),
                 )
-
+            
             chat_history["ask_conversation"].append(
                 {"role": "assistant", "content": assistant_response}
             )

autocoder/chat_auto_coder.py

@@ -38,6 +38,7 @@ from rich.table import Table
 from rich.live import Live
 from rich.text import Text
 from rich.live import Live
+from rich.markdown import Markdown
 from byzerllm.utils.nontext import Image
 import git
 from autocoder.common import git_utils
@@ -46,6 +47,7 @@ from autocoder.utils import operate_config_api
 from autocoder.agent.auto_guess_query import AutoGuessQuery
 from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRequest, McpRemoveRequest, McpListRequest, McpListRunningRequest, McpRefreshRequest
 import byzerllm
+from byzerllm.utils import format_str_jinja2
 
 
 class SymbolItem(BaseModel):
@@ -1499,11 +1501,32 @@ def mcp(query: str):
             model=args.inference_model or args.model
         )
     )
-
+    console = Console()
     if response.error:
-        print(f"Error from MCP server: {response.error}")
+        console.print(Panel(
+            f"Error from MCP server: {response.error}",
+            title="Error",
+            border_style="red"
+        ))
     else:
-        print(response.result)
+        # Save conversation
+        mcp_dir = os.path.join(".auto-coder", "mcp", "conversations")
+        os.makedirs(mcp_dir, exist_ok=True)
+        timestamp = str(int(time.time()))
+        file_path = os.path.join(mcp_dir, f"{timestamp}.md")
+        
+        # Format response as markdown
+        markdown_content = f"# MCP Response\n\n{response.result}"
+        
+        # Save to file
+        with open(file_path, "w", encoding="utf-8") as f:
+            f.write(markdown_content)
+            
+        # Print with markdown formatting
+        console.print(Panel(
+            Markdown(markdown_content),            
+            border_style="green"
+        ))
 
 
 def code_next(query: str):
@@ -1837,10 +1860,18 @@ def chat(query: str):
     if "emb_model" in conf:
         yaml_config["emb_model"] = conf["emb_model"]
 
-    is_new = query.strip().startswith("/new")
+    is_new = "/new" in query
     if is_new:
         query = query.replace("/new", "", 1).strip()
 
+    if "/mcp " in query:
+        yaml_config["action"] = "mcp"
+        query = query.replace("/mcp ", "", 1).strip()
+
+    if "/rag " in query:
+        yaml_config["action"] = "rag"
+        query = query.replace("/rag ", "", 1).strip()
+
     is_review = query.strip().startswith("/review")
     if is_review:
         query = query.replace("/review", "", 1).strip()

autocoder/common/__init__.py

@@ -348,5 +348,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     data_cells_max_num: Optional[int] = 2000
     generate_times_same_model: Optional[int] = 1
 
+    action: Optional[str] = None
+
     class Config:
         protected_namespaces = ()

autocoder/common/command_completer.py

@@ -13,7 +13,7 @@ COMMANDS = {
         "/sd": {},
     },
     "/coding": {"/apply": {}, "/next": {}},
-    "/chat": {"/new": {}, "/review": {}, "/no_context": {}},
+    "/chat": {"/new": {}, "/mcp": {}, "/rag": {}, "/review": {}, "/no_context": {}},
     "/mcp": {
         "/add": "",         
         "/remove": "",

autocoder/common/mcp_servers/mcp_server_perplexity.py

@@ -51,11 +51,11 @@ async def handle_list_tools() -> list[types.Tool]:
                         "type": "string",
                         "description": "The name of the model that will complete your prompt.",
                         "enum": [
-                            "llama-3.1-sonar-small-128k-online",
+                            # "llama-3.1-sonar-small-128k-online",
                             # Commenting out larger models,which have higher risks of timing out,
                             # until Claude Desktop can handle long-running tasks effectively.
                             # "llama-3.1-sonar-large-128k-online",
-                            # "llama-3.1-sonar-huge-128k-online",
+                            "llama-3.1-sonar-huge-128k-online",
                         ],
                     },
                     "messages": {

autocoder/rag/long_context_rag.py

@@ -104,6 +104,8 @@ class LongContextRAG:
                 raise ValueError(
                     "You are in client mode, please provide the RAG token. e.g. rag_token: your_token_here"
                 )
+            if not args.rag_url.endswith("/v1"):
+                args.rag_url = args.rag_url.rstrip("/") + "/v1"
             self.client = OpenAI(api_key=args.rag_token, base_url=args.rag_url)
         else:
             self.client = None
@@ -340,6 +342,7 @@ class LongContextRAG:
                 model=model,
                 messages=conversations,
                 stream=True,
+                max_tokens=self.args.rag_params_max_tokens
             )
 
             def response_generator():

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.220"
+__version__ = "0.1.222"