31 lines
898 B
SYSTEMD
31 lines
898 B
SYSTEMD
|
[Unit]
|
||
|
|
Description=ZhiXi RAG Worker
|
||
|
|
After=network.target
|
||
|
|
|
||
|
|
[Service]
|
||
|
|
Type=simple
|
||
|
|
User=root
|
||
|
|
WorkingDirectory=/opt/zhixi/backend/rag-worker
|
||
|
|
Environment="PYTHONUNBUFFERED=1"
|
||
|
|
EnvironmentFile=/opt/zhixi/backend/rag-worker/.env
|
||
|
|
|
||
|
|
# Install deps and self-test before starting main loop
|
||
|
|
ExecStartPre=/usr/bin/python3.11 -m pip install -q -r /opt/zhixi/backend/rag-worker/requirements.txt -i https://mirrors.tencentyun.com/pypi/simple
|
||
|
|
ExecStartPre=/usr/bin/python3.11 -c "
|
||
|
|
import asyncio, sys
|
||
|
|
from reranker import rerank
|
||
|
|
async def t():
|
||
|
|
r = await rerank('测试', ['这是测试文本', '无关内容'], top_n=1)
|
||
|
|
if not r: raise RuntimeError('reranker returned empty')
|
||
|
|
print(f'[self-test] reranker OK, score={r[0][\"score\"]:.4f}')
|
||
|
|
asyncio.run(t())
|
||
|
|
print('[self-test] all checks passed')
|
||
|
|
"
|
||
|
|
|
||
|
|
ExecStart=/usr/bin/python3.11 -u main.py
|
||
|
|
Restart=always
|
||
|
|
RestartSec=5
|
||
|
|
|
||
|
|
[Install]
|
||
|
|
WantedBy=multi-user.target
|