auto-coder 0.1.275__py3-none-any.whl → 0.1.277__py3-none-any.whl

{auto_coder-0.1.275.dist-info → auto_coder-0.1.277.dist-info}/METADATA

@@ -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.dist-info → auto_coder-0.1.277.dist-info}/RECORD

@@ -4,7 +4,7 @@ autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,
 autocoder/auto_coder_rag.py,sha256=mX-szIG9T7Mzwoc4QwKp_GyYBcVf6dfsNJnKzYHHl6U,30329
 autocoder/auto_coder_rag_client_mcp.py,sha256=QRxUbjc6A8UmDMQ8lXgZkjgqtq3lgKYeatJbDY6rSo0,6270
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
-autocoder/auto_coder_runner.py,sha256=1M74XhbmmJg0z5rC_TjjlPaqyEPfg2z6N7mLN7z8qFw,100773
+autocoder/auto_coder_runner.py,sha256=w-4MCKhOFaoABcDfVoZoonF59UyRso3kghimQYLz3NA,100851
 autocoder/auto_coder_server.py,sha256=6YQweNEKUrGAZ3yPvw8_qlNZJYLVSVUXGrn1K6udLts,20413
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
 autocoder/chat_auto_coder.py,sha256=G7_CIgDOTdGGPzRQDo0hEOh5p8A36oJQaYJc514xBkk,16842
@@ -12,7 +12,7 @@ autocoder/chat_auto_coder_lang.py,sha256=ShOQVOnMA-WlT-fB9OrOer-xQkbcWxJGl-WMPuZ
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
 autocoder/models.py,sha256=PlG1tKHSHwB57cKLOl5gTl5yTzFUDzCgeHPJU3N9F6Q,9106
-autocoder/version.py,sha256=zCL4o733jNf4GPlcthnHlKI-WRYh8ZEhNOgsrud1D_E,23
+autocoder/version.py,sha256=aQmPFbK421hxX_q_qH6lzGAzBJ-yTN3E_wgJvqVGg9k,23
 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
@@ -111,7 +111,7 @@ autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7
 autocoder/rag/doc_filter.py,sha256=yEXaBw1XJH57Gtvk4-RFQtd5eawA6SBjzxeRZrIsQew,11623
 autocoder/rag/document_retriever.py,sha256=5oThtxukGuRFF96o3pHKsk306a8diXbhgSrbqyU2BvM,8894
 autocoder/rag/llm_wrapper.py,sha256=wf56ofQNOaBkLhnoxK9VoVnHWD0gsj0pP8mUBfS92RI,2737
-autocoder/rag/long_context_rag.py,sha256=CzPC-ct6PVIKBkHsKon4s92YXmi8jZOlGgcquOwWQlI,31802
+autocoder/rag/long_context_rag.py,sha256=qFlNmbgQnstCSCb0SxfkMEYtZRr8p6YEc6u0jpve4Q0,32002
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=6TKtErZ0Us9XSV6HgRKXA6yR3SiZGPHpynOKSaR1wgE,2463
 autocoder/rag/raw_rag.py,sha256=BOr0YGf3umjqXOIDVO1LXQ0bIHx8hzBdiubND2ezyxc,2946
@@ -165,9 +165,9 @@ autocoder/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=lkJ_A-sYU36JMzjFWkk3pR6uos8oZHYt9GPsPe_CPAo,11766
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.275.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.275.dist-info/METADATA,sha256=jbvivKV7U2ukHABw1YawGuUFGastAiiG06xblNioQ5w,2643
-auto_coder-0.1.275.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.275.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.275.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.275.dist-info/RECORD,,
+auto_coder-0.1.277.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.277.dist-info/METADATA,sha256=m2MjOLFknaEjczW5V_NfTL4jj7bikJe0jbn_tuYRfdQ,2643
+auto_coder-0.1.277.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.277.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.277.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.277.dist-info/RECORD,,

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():

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"

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.275"
+__version__ = "0.1.277"