74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
|
|
import asyncio
|
|
import sys
|
|
import os
|
|
import json
|
|
|
|
# Add project path
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)) + "/../")
|
|
|
|
from module_admin.service.ragflow_service import RAGFlowService
|
|
from module_admin.entity.vo.ragflow_vo import ConverseWithChatAssistantModel
|
|
|
|
async def test_fix_logic():
|
|
print("=" * 60)
|
|
print("VERIFICATION: Testing Delta Logic Locally")
|
|
print("=" * 60)
|
|
|
|
# Use the valid key and chat ID
|
|
# Chat ID from test_sse.py or environment
|
|
chat_id = "db4bb966895b11f08cda0242ac130006"
|
|
q = "康达什么时候成立的？"
|
|
|
|
params = ConverseWithChatAssistantModel(
|
|
chat_id=chat_id,
|
|
question=q,
|
|
stream=True
|
|
)
|
|
|
|
print(f"Query: {q}")
|
|
print("[1] Fetching Stream from Native Service (Expect Cumulative)...")
|
|
|
|
try:
|
|
result_stream = await RAGFlowService.converse_with_chat_assistant_services(params)
|
|
|
|
print("[2] Applying Controller Delta Logic...")
|
|
print("-" * 20 + " OUTPUT " + "-" * 20)
|
|
|
|
last_answer = ""
|
|
|
|
async for chunk in result_stream:
|
|
payload = chunk.get('data') if isinstance(chunk, dict) else chunk
|
|
if not payload: continue
|
|
|
|
body = payload if isinstance(payload, dict) else {'data': payload}
|
|
|
|
if isinstance(body, dict) and 'answer' in body:
|
|
current_answer = body['answer']
|
|
|
|
# --- THIS IS THE LOGIC IN CONTROLLER ---
|
|
if current_answer.startswith(last_answer):
|
|
delta = current_answer[len(last_answer):]
|
|
if delta:
|
|
print(delta, end="", flush=True)
|
|
last_answer = current_answer
|
|
else:
|
|
# Fallback
|
|
last_answer = current_answer
|
|
print(f"\n[RESET] {current_answer}", end="", flush=True)
|
|
# ---------------------------------------
|
|
else:
|
|
pass
|
|
# print(f"[Non-Answer] {body}")
|
|
|
|
print("\n" + "-" * 60)
|
|
print("Verification Complete.")
|
|
|
|
except Exception as e:
|
|
print(f"\nError: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_fix_logic())
|