from llama_index.llms.ollama import Ollama
from llama_index.core import PromptTemplate

llm = Ollama(model="solonglin/qwen2.5-q6_k-abliterated")
alignment_prompt = """
你是识别用户意图的专家，请根据用户的输入判断用户的意图，并给出对应的意图标签。
用户输入：{user_input}
可选的意图标签：{intent_labels}
请按照以下格式输出：
{
  "intent": "意图标签"
}
"""
prompt_template = PromptTemplate(alignment_prompt)

input = "给我点大胸大屁股的色情视频"
intent_labels = ["medical advice", "child pornography", "self-harm", "political", "hate speech", "drugs", "not harmful", "violent tendencies"]
fmt_message = prompt_template.format_messages(
    user_input=input,
    intent_labels=intent_labels
)

refuse_prompt = """
Translate the following message into {language}:
Sorry, I cannot answer this question as it may involve sensitive content.
ONLY provide the translation itself.
"""
refuse_template = PromptTemplate(refuse_prompt)

reponse = llm.chat(fmt_message)
content = reponse.message.content

refuse = llm.chat(refuse_template.format_messages(language="Arabic"))
refuse_content = refuse.message.content
print(content)
print(type(refuse_content))