auto-coder 0.1.206__tar.gz → 0.1.207__tar.gz

{auto_coder-0.1.206 → auto_coder-0.1.207}/PKG-INFO

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

{auto_coder-0.1.206 → auto_coder-0.1.207}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.206 → auto_coder-0.1.207}/src/autocoder/auto_coder_rag.py

@@ -358,6 +358,9 @@ def main(input_args: Optional[List[str]] = None):
     benchmark_parser.add_argument(
         "--base_url", default="", help="Base URL for OpenAI client"
     )
+    benchmark_parser.add_argument(
+        "--query", default="Hello, how are you?", help="Query to use for benchmarking"
+    )
 
     # Tools command
     tools_parser = subparsers.add_parser("tools", help="Various tools")
@@ -413,11 +416,11 @@ def main(input_args: Optional[List[str]] = None):
                 return
             asyncio.run(
                 benchmark_openai(
-                    args.model, args.parallel, args.api_key, args.base_url, args.rounds
+                    args.model, args.parallel, args.api_key, args.base_url, args.rounds, args.query
                 )
             )
         else:  # byzerllm
-            benchmark_byzerllm(args.model, args.parallel, args.rounds)
+            benchmark_byzerllm(args.model, args.parallel, args.rounds, args.query)
 
     elif args.command == "serve":
         if not args.quick:

{auto_coder-0.1.206 → auto_coder-0.1.207}/src/autocoder/benchmark.py

@@ -9,16 +9,19 @@ from loguru import logger
 import byzerllm
 from concurrent.futures import ThreadPoolExecutor
 
-async def benchmark_openai(model: str, parallel: int, api_key: str, base_url: str = None, rounds: int = 1):
+
+async def benchmark_openai(
+    model: str, parallel: int, api_key: str, base_url: str = None, rounds: int = 1, query: str = "Hello, how are you?"
+):
     client = AsyncOpenAI(api_key=api_key, base_url=base_url if base_url else None)
     start_time = time.time()
-    
+
     async def single_request():
         try:
             t1 = time.time()
             response = await client.chat.completions.create(
                 model=model,
-                messages=[{"role": "user", "content": "Hello, how are you?"}]
+                messages=[{"role": "user", "content": query}],
             )
             t2 = time.time()
             return t2 - t1
@@ -32,15 +35,15 @@ async def benchmark_openai(model: str, parallel: int, api_key: str, base_url: st
         tasks = [single_request() for _ in range(parallel)]
         results = await asyncio.gather(*tasks)
         all_results.extend(results)
-    
+
     results = all_results
-    
+
     # Filter out None values from failed requests
     results = [r for r in results if r is not None]
-    
+
     end_time = time.time()
     total_time = end_time - start_time
-    
+
     if not results:
         print("All requests failed")
         return
@@ -51,14 +54,14 @@ async def benchmark_openai(model: str, parallel: int, api_key: str, base_url: st
     p90 = np.percentile(results, 90)
     p95 = np.percentile(results, 95)
     p99 = np.percentile(results, 99)
-    
+
     # Create rich table for output
     console = Console()
     table = Table(title=f"OpenAI Client Benchmark Results (Parallel={parallel})")
-    
+
     table.add_column("Metric", style="cyan")
     table.add_column("Value (seconds)", style="magenta")
-    
+
     table.add_row("Total Time", f"{total_time:.2f}")
     table.add_row("Average Response Time", f"{avg_time:.2f}")
     table.add_row("Median (P50)", f"{p50:.2f}")
@@ -66,21 +69,21 @@ async def benchmark_openai(model: str, parallel: int, api_key: str, base_url: st
     table.add_row("P95", f"{p95:.2f}")
     table.add_row("P99", f"{p99:.2f}")
     table.add_row("Requests/Second", f"{parallel/total_time:.2f}")
-    
+
     console.print(table)
 
-def benchmark_byzerllm(model: str, parallel: int, rounds: int = 1):
+
+def benchmark_byzerllm(model: str, parallel: int, rounds: int = 1, query: str = "Hello, how are you?"):
     byzerllm.connect_cluster(address="auto")
     llm = byzerllm.ByzerLLM()
-    llm.setup_default_model_name(model)    
+    llm.setup_default_model_name(model)
 
     def single_request(llm):
         try:
             t1 = time.time()
-            llm.chat_oai(conversations=[{
-                "role": "user",
-                "content": "Hello, how are you?"
-            }])
+            llm.chat_oai(
+                conversations=[{"role": "user", "content": query}]
+            )
             t2 = time.time()
             return t2 - t1
         except Exception as e:
@@ -97,39 +100,39 @@ def benchmark_byzerllm(model: str, parallel: int, rounds: int = 1):
             # get results from futures
             results = [future.result() for future in futures]
             all_results.extend(results)
-    
+
         results = all_results
-        
+
         # Filter out None values from failed requests
         results = [r for r in results if r is not None]
-        
+
         end_time = time.time()
         total_time = end_time - start_time
 
-        if not results:
-            print("All requests failed")
-            return
-
-        # Calculate statistics
-        avg_time = np.mean(results)
-        p50 = np.percentile(results, 50)
-        p90 = np.percentile(results, 90)
-        p95 = np.percentile(results, 95)
-        p99 = np.percentile(results, 99)
-        
-        # Create rich table for output
-        console = Console()
-        table = Table(title=f"ByzerLLM Client Benchmark Results (Parallel={parallel})")
-        
-        table.add_column("Metric", style="cyan")
-        table.add_column("Value (seconds)", style="magenta")
-        
-        table.add_row("Total Time", f"{total_time:.2f}")
-        table.add_row("Average Response Time", f"{avg_time:.2f}")
-        table.add_row("Median (P50)", f"{p50:.2f}")
-        table.add_row("P90", f"{p90:.2f}")
-        table.add_row("P95", f"{p95:.2f}")
-        table.add_row("P99", f"{p99:.2f}")
-        table.add_row("Requests/Second", f"{parallel/total_time:.2f}")
-        
-        console.print(table)
+    if not results:
+        print("All requests failed")
+        return
+
+    # Calculate statistics
+    avg_time = np.mean(results)
+    p50 = np.percentile(results, 50)
+    p90 = np.percentile(results, 90)
+    p95 = np.percentile(results, 95)
+    p99 = np.percentile(results, 99)
+
+    # Create rich table for output
+    console = Console()
+    table = Table(title=f"ByzerLLM Client Benchmark Results (Parallel={parallel})")
+
+    table.add_column("Metric", style="cyan")
+    table.add_column("Value (seconds)", style="magenta")
+
+    table.add_row("Total Time", f"{total_time:.2f}")
+    table.add_row("Average Response Time", f"{avg_time:.2f}")
+    table.add_row("Median (P50)", f"{p50:.2f}")
+    table.add_row("P90", f"{p90:.2f}")
+    table.add_row("P95", f"{p95:.2f}")
+    table.add_row("P99", f"{p99:.2f}")
+    table.add_row("Requests/Second", f"{parallel/total_time:.2f}")
+
+    console.print(table)

{auto_coder-0.1.206 → auto_coder-0.1.207}/src/autocoder/chat_auto_coder.py

@@ -1427,12 +1427,16 @@ def convert_yaml_to_config(yaml_file: str):
                 setattr(args, key, value)
     return args
 
-def commit():
+def commit(query: str):
     def prepare_commit_yaml():
         auto_coder_main(["next", "chat_action"])
 
     prepare_commit_yaml()
 
+    # no_diff = query.strip().startswith("/no_diff")
+    # if no_diff:
+    #     query = query.replace("/no_diff", "", 1).strip()            
+
     latest_yaml_file = get_last_yaml_file("actions")
     
     conf = memory.get("conf", {})
@@ -1472,7 +1476,7 @@ def commit():
                     os.remove(temp_yaml)
             
             llm = byzerllm.ByzerLLM.from_default_model(args.code_model or args.model)
-            uncommitted_changes = git_utils.get_uncommitted_changes(".")            
+            uncommitted_changes = git_utils.get_uncommitted_changes(".")                       
             commit_message = git_utils.generate_commit_message.with_llm(
                 llm).run(uncommitted_changes)
             memory["conversation"].append({"role": "user", "content": commit_message})
@@ -2305,7 +2309,8 @@ def main():
             elif user_input.startswith("/revert"):
                 revert()
             elif user_input.startswith("/commit"):
-                commit()
+                query = user_input[len("/commit"):].strip()
+                commit(query)
             elif user_input.startswith("/help"):
                 show_help()
             elif user_input.startswith("/exclude_dirs"):

auto_coder-0.1.207/src/autocoder/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.207"

auto_coder-0.1.206/src/autocoder/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.206"