更新 byzerllm 依赖版本至 0.1.171 并增加 LongContextRAG 的 token 统计输出。修改了 require…

…ments.txt 中的 byzerllm 版本号，并在 LongContextRAG 类中增加了 token 统计的 yield 输出。同时更新了版本号至 0.1.282。 auto_coder_000000001857_chat_action.yml_f052739084f2d9a868e866c056cf8d21
allwefantasy · Mar 6, 2025 · 8402555 · 8402555
1 parent dd45704
commit 8402555
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -46,7 +46,7 @@ tokenizers
 
 # camelot-py
 # llama_index
-byzerllm[saas]>=0.1.170
+byzerllm[saas]>=0.1.171
 patch
 diff_match_patch
 GitPython

diff --git a/src/autocoder/rag/long_context_rag.py b/src/autocoder/rag/long_context_rag.py
@@ -788,6 +788,12 @@ def generate_sream():
                                             tokens=request_tokens
                                         )
                                         ))
+
+            yield ("", SingleOutputMeta(input_tokens_count=rag_stat.recall_stat.total_input_tokens + rag_stat.chunk_stat.total_input_tokens,
+                                        generated_tokens_count=rag_stat.recall_stat.total_generated_tokens +
+                                        rag_stat.chunk_stat.total_generated_tokens,
+                                        reasoning_content="qa_model_thinking"
+                                        ))
 
             if LLMComputeEngine is not None and not self.args.disable_inference_enhance:
                 llm_compute_engine = LLMComputeEngine(

diff --git a/src/autocoder/version.py b/src/autocoder/version.py
@@ -1 +1 @@
-__version__ = "0.1.281"
+__version__ = "0.1.282"