basic.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. package plugin_daemon
  2. type PluginAccessType string
  3. const (
  4. PLUGIN_ACCESS_TYPE_TOOL PluginAccessType = "tool"
  5. PLUGIN_ACCESS_TYPE_MODEL PluginAccessType = "model"
  6. )
  7. type PluginAccessAction string
  8. const (
  9. PLUGIN_ACCESS_ACTION_INVOKE_TOOL PluginAccessAction = "invoke_tool"
  10. PLUGIN_ACCESS_ACTION_INVOKE_LLM PluginAccessAction = "invoke_llm"
  11. PLUGIN_ACCESS_ACTION_INVOKE_TEXT_EMBEDDING PluginAccessAction = "invoke_text_embedding"
  12. PLUGIN_ACCESS_ACTION_INVOKE_RERANK PluginAccessAction = "invoke_rerank"
  13. PLUGIN_ACCESS_ACTION_INVOKE_TTS PluginAccessAction = "invoke_tts"
  14. PLUGIN_ACCESS_ACTION_INVOKE_SPEECH2TEXT PluginAccessAction = "invoke_speech2text"
  15. PLUGIN_ACCESS_ACTION_INVOKE_MODERATION PluginAccessAction = "invoke_moderation"
  16. )
  17. const (
  18. PLUGIN_IN_STREAM_EVENT = "request"
  19. )
  20. func getBasicPluginAccessMap(session_id string, user_id string, access_type PluginAccessType, action PluginAccessAction) map[string]any {
  21. return map[string]any{
  22. "session_id": session_id,
  23. "event": PLUGIN_IN_STREAM_EVENT,
  24. "data": map[string]any{
  25. "user_id": user_id,
  26. "type": access_type,
  27. "action": action,
  28. },
  29. }
  30. }