35 lines
900 B
Python
35 lines
900 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
PrePrompt Hook - 用户上下文加载器示例
|
|
|
|
在 system_prompt 加载时执行，可以动态注入用户相关信息到 prompt 中。
|
|
"""
|
|
import os
|
|
import sys
|
|
|
|
|
|
def main():
|
|
"""从环境变量读取参数并输出注入内容"""
|
|
user_identifier = os.environ.get('USER_IDENTIFIER', '')
|
|
bot_id = os.environ.get('ASSISTANT_ID', '')
|
|
|
|
# 示例：根据 user_identifier 查询用户上下文
|
|
# 这里只是演示，实际应该从数据库或其他服务获取
|
|
if user_identifier:
|
|
context_info = f"""## User Context
|
|
|
|
用户标识: {user_identifier}
|
|
Bot ID: {bot_id}
|
|
|
|
> 此内容由 user-context-loader skill 的 PrePrompt hook 注入。
|
|
> 实际使用时，可以在这里查询用户的位置、偏好、历史记录等信息。
|
|
"""
|
|
print(context_info)
|
|
return 0
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|