auto-coder 0.1.284__py3-none-any.whl → 0.1.286__py3-none-any.whl

{auto_coder-0.1.284.dist-info → auto_coder-0.1.286.dist-info}/METADATA

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

@@ -11,8 +11,8 @@ autocoder/chat_auto_coder.py,sha256=G7_CIgDOTdGGPzRQDo0hEOh5p8A36oJQaYJc514xBkk,
 autocoder/chat_auto_coder_lang.py,sha256=ShOQVOnMA-WlT-fB9OrOer-xQkbcWxJGl-WMPuZcUkM,19572
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/models.py,sha256=PlG1tKHSHwB57cKLOl5gTl5yTzFUDzCgeHPJU3N9F6Q,9106
-autocoder/version.py,sha256=CdPfaa9UyiMW7CWw6BaV5azX5klLdPvg_B_GcYjlyFk,23
+autocoder/models.py,sha256=AyoZ-Pzy0oyYUmWCxOIRiOImsqboSfRET7LO9-UOuxI,11172
+autocoder/version.py,sha256=IEWJd9XFdzl6BiBXmhjT4zjl4SdS7u6oVsHUCg0Zq0A,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
@@ -131,7 +131,7 @@ autocoder/rag/cache/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 autocoder/rag/cache/base_cache.py,sha256=EaYYYbclMBvnlOUoM7qonnluwZX5oSvUjdvGvFun8_8,742
 autocoder/rag/cache/byzer_storage_cache.py,sha256=gK90pf741CgccCzQ73urBorCqVyAfwU1FAqMtSorWVk,17232
 autocoder/rag/cache/file_monitor_cache.py,sha256=2TnOW8Y81Zc0WA1upRrkmQH18IMdv40CeNccmnTvd3c,4981
-autocoder/rag/cache/local_byzer_storage_cache.py,sha256=Uhmu5JK0tfZ8NvlcjJzcwtQRhZDpbGp_U6qLXZxVwss,17495
+autocoder/rag/cache/local_byzer_storage_cache.py,sha256=kgGMYP4MIc4upXcwTmeGbWYW4dDnt-hhJKmnjsFPKU4,18977
 autocoder/rag/cache/simple_cache.py,sha256=8FMmBAfhAPcdSNUWC6Ga43LBFGXD-klwabVbzm_bciI,9347
 autocoder/rag/loaders/__init__.py,sha256=EQHEZ5Cmz-mGP2SllUTvcIbYCnF7W149dNpNItfs0yE,304
 autocoder/rag/loaders/docx_loader.py,sha256=ZswPqiiLngUEpzLhNNm1nmwEYV7ZHFEfIoXoG7c5GDU,614
@@ -169,9 +169,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.284.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.284.dist-info/METADATA,sha256=NYN7m8jbf2aikPm1nXpD_hTBHMJOmVcpb0-Y8DbhveE,2643
-auto_coder-0.1.284.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.284.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.284.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.284.dist-info/RECORD,,
+auto_coder-0.1.286.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.286.dist-info/METADATA,sha256=U258OIH53mmFGbXDNu1KEIiI2lWExnsf-S1X8u1r0NY,2643
+auto_coder-0.1.286.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.286.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.286.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.286.dist-info/RECORD,,

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
+

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]:

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.284"
+__version__ = "0.1.286"