777e027553
Permission denied on git mv, used robocopy instead. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import paramiko, sys
|
|
sys.stdout.reconfigure(encoding='utf-8', errors='replace')
|
|
|
|
client = paramiko.SSHClient()
|
|
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
client.connect('101.79.17.164', username='root', password='1q2w3e!Q', timeout=30)
|
|
|
|
def run(label, cmd, timeout=60):
|
|
print(f'\n[{label}]')
|
|
_, o, e = client.exec_command(cmd, timeout=timeout)
|
|
out = o.read().decode('utf-8', errors='replace').strip()
|
|
err = e.read().decode('utf-8', errors='replace').strip()
|
|
if out: print(out[:400])
|
|
if err and 'warn' not in err.lower(): print('ERR:', err[:200])
|
|
|
|
run('pgvector 타입 정확성 테스트', """
|
|
/opt/guardia/venv/bin/python3 << 'PYEOF'
|
|
import psycopg2
|
|
from pgvector.psycopg2 import register_vector
|
|
import numpy as np
|
|
|
|
conn = psycopg2.connect(
|
|
host='localhost', port=5432,
|
|
dbname='guardia_db', user='guardia', password='G@urd1a_2026!'
|
|
)
|
|
register_vector(conn) # vector 타입 등록 필수
|
|
cur = conn.cursor()
|
|
|
|
# numpy array로 삽입 (register_vector 후 자동 변환)
|
|
emb = np.random.rand(768).astype(np.float32)
|
|
cur.execute(
|
|
'INSERT INTO tb_vector_sr (sr_id, title, content, embedding) VALUES (%s,%s,%s,%s) ON CONFLICT (sr_id) DO UPDATE SET embedding=EXCLUDED.embedding',
|
|
('SR-TEST-001', 'pgvector 연동 테스트', 'pgvector 설치 확인', emb)
|
|
)
|
|
conn.commit()
|
|
|
|
# 유사 검색
|
|
query = np.random.rand(768).astype(np.float32)
|
|
cur.execute(
|
|
'SELECT sr_id, title, 1 - (embedding <=> %s::vector) AS similarity FROM tb_vector_sr ORDER BY embedding <=> %s::vector LIMIT 3',
|
|
(emb.tolist(), query.tolist())
|
|
)
|
|
rows = cur.fetchall()
|
|
for row in rows:
|
|
print('SR:', row[0], '| 유사도:', round(float(row[2]), 4))
|
|
|
|
cur.close(); conn.close()
|
|
print('SUCCESS: pgvector 연동 완료!')
|
|
PYEOF
|
|
""")
|
|
|
|
run('pgvector 버전 확인', """
|
|
sudo -u postgres psql guardia_db -c "SELECT extversion FROM pg_extension WHERE extname='vector';" 2>&1
|
|
""")
|
|
|
|
run('인덱스 확인', """
|
|
sudo -u postgres psql guardia_db -c "SELECT indexname, indexdef FROM pg_indexes WHERE tablename LIKE 'tb_vector%';" 2>&1
|
|
""")
|
|
|
|
client.close()
|