修复测试用例流结束判断

lightrag/api/lightrag_ollama.py

@@ -743,12 +743,12 @@ def create_app(args):
                                 "model": LIGHTRAG_MODEL,
                                 "created_at": LIGHTRAG_CREATED_AT,
                                 "done": True,
-                                "total_duration": 0,  # 由于我们没有实际统计这些指标，暂时使用默认值
-                                "load_duration": 0,
-                                "prompt_eval_count": 0,
-                                "prompt_eval_duration": 0,
-                                "eval_count": 0,
-                                "eval_duration": 0
+                                "total_duration": 1,  # 由于我们没有实际统计这些指标，暂时使用默认值
+                                "load_duration": 1,
+                                "prompt_eval_count": 999,
+                                "prompt_eval_duration": 1,
+                                "eval_count": 999,
+                                "eval_duration": 1
                             }
                             yield f"data: {json.dumps(data, ensure_ascii=False)}\n\n"
                             return  # 确保生成器在发送完成标记后立即结束
@@ -789,12 +789,12 @@ def create_app(args):
                         "images": None
                     },
                     "done": True,
-                    "total_duration": 0,  # 由于我们没有实际统计这些指标，暂时使用默认值
-                    "load_duration": 0,
-                    "prompt_eval_count": 0,
-                    "prompt_eval_duration": 0,
-                    "eval_count": 0,
-                    "eval_duration": 0
+                    "total_duration": 1,  # 由于我们没有实际统计这些指标，暂时使用默认值
+                    "load_duration": 1,
+                    "prompt_eval_count": 999,
+                    "prompt_eval_duration": 1,
+                    "eval_count": 999,
+                    "eval_duration": 1
                 }
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))

test_lightrag_ollama_chat.py

@@ -70,7 +70,7 @@ def test_stream_chat():
         for event in client.events():
             try:
                 data = json.loads(event.data)
-                if data.get("done", False):  # 如果是完成标记
+                if data.get("done", True):  # 如果是完成标记
                     if "total_duration" in data:  # 最终的性能统计消息
                         print("\n=== 性能统计 ===")
                         print(json.dumps(data, ensure_ascii=False, indent=2))