29 lines
729 B
Python
29 lines
729 B
Python
#!/usr/bin/env python3
|
||
"""
|
||
PreMemoryPrompt Hook - 用户上下文加载器示例
|
||
|
||
在记忆提取提示词（FACT_RETRIEVAL_PROMPT）加载时执行，
|
||
根据环境变量决定是否启用禁止使用模型自身知识的 retrieval policy。
|
||
"""
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
|
||
def main():
|
||
enable_self_knowledge = (
|
||
os.getenv("ENABLE_SELF_KNOWLEDGE", "false").lower() == "true"
|
||
)
|
||
policy_name = (
|
||
"retrieval-policy.md"
|
||
if enable_self_knowledge
|
||
else "retrieval-policy-forbidden-self-knowledge.md"
|
||
)
|
||
prompt_file = Path(__file__).parent / policy_name
|
||
print(prompt_file.read_text(encoding="utf-8"))
|
||
return 0
|
||
|
||
|
||
if __name__ == '__main__':
|
||
sys.exit(main())
|