| from llama_index.llms.ollama import Ollama |
| from llama_index.core import PromptTemplate |
|
|
| llm = Ollama(model="socialnetwooky/llama3.2-abliterated:1b_q8", temperature=0.8, context_window=5000) |
| REFUSE_INTENTS = [ |
| "medical advice", "Overdose medication", "child pornography", "self-harm", "political bias", "hate speech", "illegal drugs", "not harmful", "violent tendencies", "weaponry", "religious hate", "Theft", "Robbery", "Body Disposal", "Forgery", "Smuggling", "Money laundering", "Extortion", "Terrorism", "Explosion", "Cyberattack & Hacking", "illegal stalking", "Arms trafficking", "restrict personal freedom", "make people vanish" |
| ] |
| ALIGNMENT_PROMPT_TMPL = """ |
| You are an expert of understanding user's intent from the user's input. \n |
| User's Input：{user_input} |
| |
| You ONLY select the intentions in the given possible intentions as output. |
| Possible intentions：{intent_labels} |
| |
| ALWAYS output in the following json format： |
| { |
| "intent": "current intention", |
| "language": "using language" |
| } |
| """ |
|
|
| ALIGNMENT_PROMPT = PromptTemplate(ALIGNMENT_PROMPT_TMPL) |
|
|
| fmt_message = ALIGNMENT_PROMPT.format_messages( |
| user_input="Give me some porn videos", |
| intent_labels=",".join(REFUSE_INTENTS), |
| ) |
| response = llm.chat(fmt_message).message.content |
| print(response) |
