SetAIMode set the ai mode for the session.
(sessionId string, aiMode openai.AIMode)
| 119 | |
| 120 | // SetAIMode set the ai mode for the session. |
| 121 | func (s *SessionService) SetAIMode(sessionId string, aiMode openai.AIMode) { |
| 122 | maxCacheTime := time.Hour * 12 |
| 123 | sessionContext, ok := s.cache.Get(sessionId) |
| 124 | if !ok { |
| 125 | sessionMeta := &SessionMeta{AIMode: aiMode} |
| 126 | s.cache.Set(sessionId, sessionMeta, maxCacheTime) |
| 127 | return |
| 128 | } |
| 129 | sessionMeta := sessionContext.(*SessionMeta) |
| 130 | sessionMeta.AIMode = aiMode |
| 131 | s.cache.Set(sessionId, sessionMeta, maxCacheTime) |
| 132 | } |
| 133 | |
| 134 | func (s *SessionService) GetMsg(sessionId string) (msg []openai.Messages) { |
| 135 | sessionContext, ok := s.cache.Get(sessionId) |