auto-coder 0.1.229__tar.gz → 0.1.231__tar.gz

{auto_coder-0.1.229 → auto_coder-0.1.231}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.229
+Version: 0.1.231
 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.150
+Requires-Dist: byzerllm[saas]>=0.1.151
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.229
+Version: 0.1.231
 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.150
+Requires-Dist: byzerllm[saas]>=0.1.151
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/auto_coder.egg-info/SOURCES.txt

@@ -127,6 +127,7 @@ src/autocoder/utils/__init__.py
 src/autocoder/utils/_markitdown.py
 src/autocoder/utils/conversation_store.py
 src/autocoder/utils/llm_client_interceptors.py
+src/autocoder/utils/llms.py
 src/autocoder/utils/log_capture.py
 src/autocoder/utils/multi_turn.py
 src/autocoder/utils/operate_config_api.py
@@ -138,6 +139,7 @@ src/autocoder/utils/rest.py
 src/autocoder/utils/tests.py
 src/autocoder/utils/auto_coder_utils/__init__.py
 src/autocoder/utils/auto_coder_utils/chat_stream_out.py
+src/autocoder/utils/chat_auto_coder_utils/__init__.py
 tests/test_action_regex_project.py
 tests/test_chat_auto_coder.py
 tests/test_code_auto_merge_editblock.py

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/auto_coder.egg-info/requires.txt

@@ -16,7 +16,7 @@ tabulate
 jupyter_client
 prompt-toolkit
 tokenizers
-byzerllm[saas]>=0.1.150
+byzerllm[saas]>=0.1.151
 patch
 diff_match_patch
 GitPython

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/autocoder/auto_coder.py

@@ -280,6 +280,7 @@ def main(input_args: Optional[List[str]] = None):
                 byzerllm.connect_cluster(address=args.ray_address)
 
             llm = byzerllm.ByzerLLM(verbose=args.print_request)
+        
         if args.product_mode == "lite":
             llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
             api_key_dir = os.path.expanduser("~/.auto-coder/keys")
@@ -298,7 +299,8 @@ def main(input_args: Optional[List[str]] = None):
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat"
+                    "saas.model": "deepseek-chat",
+                    "saas.is_reasoning": False
                 }
             )
             
@@ -310,7 +312,8 @@ def main(input_args: Optional[List[str]] = None):
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat"
+                    "saas.model": "deepseek-chat",
+                    "saas.is_reasoning": False
                 }
             )
 
@@ -322,7 +325,8 @@ def main(input_args: Optional[List[str]] = None):
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner"
+                    "saas.model": "deepseek-reasoner",
+                    "saas.is_reasoning": True
                 }
             )
 
@@ -334,7 +338,8 @@ def main(input_args: Optional[List[str]] = None):
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner"
+                    "saas.model": "deepseek-reasoner",
+                    "saas.is_reasoning": True
                 }
             )
 
@@ -360,7 +365,8 @@ def main(input_args: Optional[List[str]] = None):
                             infer_params={
                                 "saas.base_url": model_info["base_url"],
                                 "saas.api_key": model_info["api_key"],
-                                "saas.model": model_info["model_name"]
+                                "saas.model": model_info["model_name"],
+                                "saas.is_reasoning": model_info["is_reasoning"]
                             }
                         )
                         models.append(code_model)
@@ -377,7 +383,8 @@ def main(input_args: Optional[List[str]] = None):
                         infer_params={
                             "saas.base_url": model_info["base_url"],
                             "saas.api_key": model_info["api_key"],
-                            "saas.model": model_info["model_name"]
+                            "saas.model": model_info["model_name"],
+                            "saas.is_reasoning": model_info["is_reasoning"]
                         }
                     )
                     llm.setup_sub_client("code_model", code_model)
@@ -397,7 +404,8 @@ def main(input_args: Optional[List[str]] = None):
                             infer_params={
                                 "saas.base_url": model_info["base_url"],
                                 "saas.api_key": model_info["api_key"],
-                                "saas.model": model_info["model_name"]
+                                "saas.model": model_info["model_name"],
+                                "saas.is_reasoning": model_info["is_reasoning"]
                             }
                         )
                         models.append(rerank_model)
@@ -414,7 +422,8 @@ def main(input_args: Optional[List[str]] = None):
                         infer_params={
                             "saas.base_url": model_info["base_url"],
                             "saas.api_key": model_info["api_key"],
-                            "saas.model": model_info["model_name"]
+                            "saas.model": model_info["model_name"],
+                            "saas.is_reasoning": model_info["is_reasoning"]
                         }
                     )
                     llm.setup_sub_client("generate_rerank_model", rerank_model)
@@ -430,7 +439,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("inference_model", inference_model)        
@@ -610,7 +620,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("chat_model", chat_model)
@@ -626,7 +637,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("vl_model", vl_model)
@@ -642,7 +654,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("sd_model", sd_model)
@@ -658,7 +671,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("text2voice_model", text2voice_model)
@@ -674,7 +688,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("voice2text_model", voice2text_model)
@@ -690,7 +705,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("planner_model", planner_model)
@@ -706,7 +722,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("designer_model", designer_model)
@@ -722,7 +739,8 @@ def main(input_args: Optional[List[str]] = None):
                     infer_params={
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
-                        "saas.model": model_info["model_name"]
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
                 llm.setup_sub_client("emb_model", emb_model)

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/autocoder/chat_auto_coder.py

@@ -51,7 +51,7 @@ from byzerllm.utils import format_str_jinja2
 from autocoder.common.memory_manager import get_global_memory_file_paths 
 from autocoder import models
 import shlex
-
+from autocoder.utils.llms import get_single_llm
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -1316,6 +1316,9 @@ def ask(query: str):
     if "code_model" in conf:
         yaml_config["code_model"] = conf["code_model"]
 
+    if "product_mode" in conf:
+        yaml_config["product_mode"] = conf["product_mode"]
+
     yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
@@ -1589,18 +1592,9 @@ def code_next(query: str):
     )
 
 
-def get_single_llm(model_names: str):
-    if "," in model_names:
-        # Multiple code models specified
-        model_names = model_names.split(",")
-        for _, model_name in enumerate(model_names):
-            return byzerllm.ByzerLLM.from_default_model(model_name)
-    else:
-        # Single code model
-        return byzerllm.ByzerLLM.from_default_model(model_names)
-
-
 def commit(query: str):
+    conf = memory.get("conf", {})
+    product_mode = conf.get("product_mode", "lite")
     def prepare_commit_yaml():
         auto_coder_main(["next", "chat_action"])
 
@@ -1652,7 +1646,7 @@ def commit(query: str):
                 if os.path.exists(temp_yaml):
                     os.remove(temp_yaml)
 
-            llm = get_single_llm(args.code_model or args.model)
+            llm = get_single_llm(args.code_model or args.model, product_mode)
             uncommitted_changes = git_utils.get_uncommitted_changes(".")
             commit_message = git_utils.generate_commit_message.with_llm(llm).run(
                 uncommitted_changes
@@ -1672,6 +1666,8 @@ def commit(query: str):
             )
             git_utils.print_commit_info(commit_result=commit_result)
         except Exception as e:
+            import traceback
+            traceback.print_exc()            
             print(f"Failed to commit: {e}")
             if execute_file:
                 os.remove(execute_file)
@@ -1949,6 +1945,9 @@ def summon(query: str):
     if "model" in conf:
         yaml_config["model"] = conf["model"]
 
+    if "product_mode" in conf:
+        yaml_config["product_mode"] = conf["product_mode"]
+
     yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
@@ -2477,9 +2476,6 @@ 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()
@@ -2655,12 +2651,6 @@ def main():
                     print("Please enter your query.")
                 else:
                     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()

{auto_coder-0.1.229 → auto_coder-0.1.231}/src/autocoder/models.py

@@ -32,7 +32,7 @@ default_models_list = [
         "model_type": "saas/openai",
         "base_url": "https://api.openai.com/v1",
         "api_key_path": "",
-        "is_reasoning": False
+        "is_reasoning": True
     }
 ]
 
@@ -54,9 +54,12 @@ def load_models() -> List[Dict]:
                 custom_models = json.load(f)
                 # Custom models will override defaults with same name
                 for model in custom_models:
+                    model["is_reasoning"] = model.get("is_reasoning", False)
                     models_dict[model["name"]] = model
+
         except json.JSONDecodeError:
             # If JSON is invalid, just use defaults
+            print("JSON is invalid, using defaults")
             save_models(default_models_list)
     else:
         # If file doesn't exist, create it with defaults

auto_coder-0.1.231/src/autocoder/utils/llms.py

@@ -0,0 +1,52 @@
+import byzerllm
+from autocoder.auto_coder import models_module
+
+def get_single_llm(model_names: str, product_mode: str):
+    if product_mode == "pro":
+        if "," in model_names:
+            # Multiple code models specified
+            model_names = model_names.split(",")
+            for _, model_name in enumerate(model_names):
+                return byzerllm.ByzerLLM.from_default_model(model_name)
+        else:
+            # Single code model
+            return byzerllm.ByzerLLM.from_default_model(model_names)
+
+    if product_mode == "lite":
+        if "," in model_names:
+            # Multiple code models specified
+            model_names = model_names.split(",")            
+            for _, model_name in enumerate(model_names):
+                model_name = model_name.strip()
+                model_info = models_module.get_model_by_name(model_name)
+                target_llm = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                target_llm.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=model_name,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
+                return target_llm
+            
+        else:
+            # Single code model
+            model_info = models_module.get_model_by_name(model_names)
+            model_name = model_names
+            target_llm = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+            target_llm.deploy(
+                model_path="",
+                pretrained_model_type=model_info["model_type"],
+                udf_name=model_name,
+                infer_params={
+                    "saas.base_url": model_info["base_url"],
+                    "saas.api_key": model_info["api_key"],
+                    "saas.model": model_info["model_name"],
+                    "saas.is_reasoning": model_info["is_reasoning"]
+                }
+            )
+            return target_llm

auto_coder-0.1.231/src/autocoder/version.py

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

auto_coder-0.1.229/src/autocoder/version.py

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