auto-coder 0.1.284__tar.gz → 0.1.286__tar.gz

{auto_coder-0.1.284 → auto_coder-0.1.286}/PKG-INFO

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

{auto_coder-0.1.284 → auto_coder-0.1.286}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.284 → auto_coder-0.1.286}/src/autocoder/models.py

@@ -286,3 +286,61 @@ def update_model_with_api_key(name: str, api_key: str) -> Dict:
     
     return found_model
 
+def update_model(name: str, model_data: Dict) -> Dict:
+    """
+    更新模型信息
+    
+    Args:
+        name: 要更新的模型名称
+        model_data: 包含模型新信息的字典，可以包含以下字段:
+            - name: 模型名称
+            - description: 模型描述
+            - model_name: 模型实际名称
+            - model_type: 模型类型
+            - base_url: 基础URL
+            - api_key: API密钥
+            - is_reasoning: 是否为推理模型
+            - input_price: 输入价格
+            - output_price: 输出价格
+            
+    Returns:
+        Dict: 更新后的模型信息，如果未找到则返回None
+    """
+    models = load_models()
+    
+    # 查找要更新的模型
+    found = False
+    for i, model in enumerate(models):
+        if model["name"] == name:
+            # 更新模型字段
+            if "description" in model_data:
+                model["description"] = model_data["description"]
+            if "model_name" in model_data:
+                model["model_name"] = model_data["model_name"]
+            if "model_type" in model_data:
+                model["model_type"] = model_data["model_type"]
+            if "base_url" in model_data:
+                model["base_url"] = model_data["base_url"]
+            if "is_reasoning" in model_data:
+                model["is_reasoning"] = model_data["is_reasoning"]
+            if "input_price" in model_data:
+                model["input_price"] = float(model_data["input_price"])
+            if "output_price" in model_data:
+                model["output_price"] = float(model_data["output_price"])
+            
+            # 保存更新后的模型
+            models[i] = model
+            found = True
+            
+            # 如果提供了API密钥，则更新
+            if "api_key" in model_data and model_data["api_key"]:
+                update_model_with_api_key(name, model_data["api_key"])
+            
+            break
+    
+    if found:
+        save_models(models)
+        return models[i]
+    
+    return None
+

{auto_coder-0.1.284 → auto_coder-0.1.286}/src/autocoder/rag/cache/local_byzer_storage_cache.py

@@ -66,13 +66,15 @@ class LocalByzerStorageCache(BaseCacheManager):
         required_exts,
         extra_params: Optional[AutoCoderArgs] = None,
         emb_llm: Union[ByzerLLM, SimpleByzerLLM] = None,
+        host: str = "127.0.0.1",
+        port: int = 33333,
     ):
         self.path = path
         self.ignore_spec = ignore_spec
         self.required_exts = required_exts
         self.rag_build_name = extra_params.rag_build_name
         self.storage = LocalByzerStorage("byzerai_store",
-            "rag_test", self.rag_build_name, host="127.0.0.1", port=33333,emb_llm=emb_llm)
+            "rag_test", self.rag_build_name, host=host, port=port,emb_llm=emb_llm)
         self.queue = []
         self.chunk_size = 1000
         self._init_schema()
@@ -397,6 +399,27 @@ class LocalByzerStorageCache(BaseCacheManager):
 
         results = query_builder.execute()
 
+        logger.info(f"从缓存获取: {len(results)} 条数据")
+        # Preview first 5 results with all fields but limited content size
+        preview_results = []
+        for r in results[:5]:
+            # Create a copy of the entire result
+            preview = r.copy()            
+            # Similarly limit raw_content if it exists
+            if "raw_content" in preview and isinstance(preview["raw_content"], str):
+                preview["raw_content"] = preview["raw_content"][:100] + "..." if len(preview["raw_content"]) > 100 else preview["raw_content"]
+            preview_results.append(preview)
+        logger.info(f"预览前5条数据:")
+        
+        for r in preview_results:
+            logger.info(f"文件路径: {r['file_path']}")            
+            logger.info(f"原始内容: {r['raw_content']}")
+            # 打印其他字段
+            for k, v in r.items():
+                if k not in ["file_path", "raw_content"]:
+                    logger.info(f"{k}: {v}")
+            logger.info("-"*100)
+
         # Group results by file_path and reconstruct documents while preserving order
         # 这里还可以有排序优化，综合考虑一篇内容出现的次数以及排序位置
         file_paths = []
@@ -414,10 +437,12 @@ class LocalByzerStorageCache(BaseCacheManager):
                 cached_data = self.cache[file_path]
                 for doc in cached_data.content:
                     if total_tokens + doc["tokens"] > self.max_output_tokens:
+                        logger.info(f"用户tokens设置为:{self.max_output_tokens}，累计tokens: {total_tokens} 当前文件: {file_path} tokens: {doc['tokens']}，数据条数变化: {len(results)} -> {len(result)}")
                         return result
                     total_tokens += doc["tokens"]
                 result[file_path] = cached_data.model_dump()
-
+        
+        logger.info(f"用户tokens设置为:{self.max_output_tokens}，累计tokens: {total_tokens}，数据条数变化: {len(results)} -> {len(result)}")
         return result
 
     def get_all_files(self) -> List[FileInfo]:

auto_coder-0.1.286/src/autocoder/version.py

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

auto_coder-0.1.284/src/autocoder/version.py

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