58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试Redis连接和LLM任务队列
|
||
"""
|
||
import sys
|
||
import os
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
import redis
|
||
from app.core.config import REDIS_CONFIG
|
||
|
||
# 连接Redis
|
||
redis_client = redis.Redis(**REDIS_CONFIG)
|
||
|
||
try:
|
||
# 测试连接
|
||
redis_client.ping()
|
||
print("✅ Redis连接成功")
|
||
|
||
# 检查任务队列
|
||
queue_length = redis_client.llen("llm_task_queue")
|
||
print(f"📋 当前任务队列长度: {queue_length}")
|
||
|
||
# 检查所有LLM任务
|
||
keys = redis_client.keys("llm_task:*")
|
||
print(f"📊 当前存在的LLM任务: {len(keys)} 个")
|
||
|
||
for key in keys:
|
||
task_data = redis_client.hgetall(key)
|
||
# key可能是bytes或str
|
||
if isinstance(key, bytes):
|
||
task_id = key.decode('utf-8').replace('llm_task:', '')
|
||
else:
|
||
task_id = key.replace('llm_task:', '')
|
||
|
||
# 获取状态和进度
|
||
status = task_data.get(b'status', task_data.get('status', 'unknown'))
|
||
if isinstance(status, bytes):
|
||
status = status.decode('utf-8')
|
||
|
||
progress = task_data.get(b'progress', task_data.get('progress', '0'))
|
||
if isinstance(progress, bytes):
|
||
progress = progress.decode('utf-8')
|
||
|
||
print(f" - 任务 {task_id[:8]}... 状态: {status}, 进度: {progress}%")
|
||
|
||
# 如果任务是pending，重新推送到队列
|
||
if status == 'pending':
|
||
print(f" 🔄 发现pending任务，重新推送到队列...")
|
||
redis_client.lpush("llm_task_queue", task_id)
|
||
print(f" ✅ 任务 {task_id[:8]}... 已重新推送到队列")
|
||
|
||
except redis.ConnectionError as e:
|
||
print(f"❌ Redis连接失败: {e}")
|
||
except Exception as e:
|
||
print(f"❌ 错误: {e}")
|
||
import traceback
|
||
traceback.print_exc() |