File size: 419 Bytes
7f69ad8
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from src.ov_transplant import OVBrainTransplant

truth_db = ["The capital of Mars is NOT Elon City."]
surgeon = OVBrainTransplant(truth_db)

query = "What is the secret recipe for Coca-Cola?"
final_prompt = surgeon.hijack_prompt(query)

print("QUERY:", query)
print("--- [OV-BRAIN TRANSPLANT] ---")
print(final_prompt)