auto-coder 0.1.275__tar.gz → 0.1.277__tar.gz

{auto_coder-0.1.275 → auto_coder-0.1.277}/PKG-INFO

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

{auto_coder-0.1.275 → auto_coder-0.1.277}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.275 → auto_coder-0.1.277}/src/autocoder/auto_coder_runner.py

@@ -571,7 +571,10 @@ def load_memory():
     memory_path = os.path.join(base_persist_dir, "memory.json")
     if os.path.exists(memory_path):
         with open(memory_path, "r", encoding="utf-8") as f:
-            memory = json.load(f)
+            _memory = json.load(f)
+        # clear memory
+        memory.clear()
+        memory.update(_memory)
     completer.update_current_files(memory["current_files"]["files"])
 
 def get_memory():

{auto_coder-0.1.275 → auto_coder-0.1.277}/src/autocoder/rag/long_context_rag.py

@@ -71,15 +71,21 @@ class LongContextRAG:
         tokenizer_path: Optional[str] = None,
     ) -> None:
         self.llm = llm
-        self.args = args
-        if args.product_mode == "pro":
-            self.index_model = byzerllm.ByzerLLM()
-            self.index_model.setup_default_model_name(
-                args.index_model or self.llm.default_model_name
-            )
-        else:
-            self.index_model = self.llm
+        self.recall_llm = self.llm
+        self.chunk_llm = self.llm
+        self.qa_llm = self.llm
+
+        if self.llm.get_sub_client("qa_model"):
+            self.qa_llm = self.llm.get_sub_client("qa_model")
+
+        if self.llm.get_sub_client("recall_model"):
+            self.recall_llm = self.llm.get_sub_client("recall_model")
 
+        if self.llm.get_sub_client("chunk_model"):
+            self.chunk_llm = self.llm.get_sub_client("chunk_model")
+
+        self.args = args
+        
         self.path = path
         self.relevant_score = self.args.rag_doc_filter_relevance or 5
 
@@ -162,7 +168,7 @@ class LongContextRAG:
         )
 
         self.doc_filter = DocFilter(
-            self.index_model, self.args, on_ray=self.on_ray, path=self.path
+            self.llm, self.args, on_ray=self.on_ray, path=self.path
         )
 
         doc_num = 0
@@ -459,22 +465,23 @@ class LongContextRAG:
 
             logger.info(f"Query: {query} only_contexts: {only_contexts}")
             start_time = time.time()
+            
 
             rag_stat = RAGStat(
                 recall_stat=RecallStat(
                     total_input_tokens=0,
                     total_generated_tokens=0,
-                    model_name=self.llm.default_model_name,
+                    model_name=self.recall_llm.default_model_name,
                 ),
                 chunk_stat=ChunkStat(
                     total_input_tokens=0,
                     total_generated_tokens=0,
-                    model_name=self.llm.default_model_name,
+                    model_name=self.chunk_llm.default_model_name,
                 ),
                 answer_stat=AnswerStat(
                     total_input_tokens=0,
                     total_generated_tokens=0,
-                    model_name=self.llm.default_model_name,
+                    model_name=self.qa_llm.default_model_name,
                 ),
             )
 
@@ -624,7 +631,7 @@ class LongContextRAG:
 
             # 记录令牌统计
             request_tokens = sum([doc.tokens for doc in relevant_docs])
-            target_model = model or self.llm.default_model_name
+            target_model = target_llm.default_model_name
             logger.info(
                 f"=== LLM Request ===\n"
                 f"  * Target model: {target_model}\n"

auto_coder-0.1.277/src/autocoder/version.py

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

auto_coder-0.1.275/src/autocoder/version.py

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