1package tools
 2
 3import (
 4	"context"
 5)
 6
 7type (
 8	sessionIDContextKey string
 9	messageIDContextKey string
10)
11
12const (
13	SessionIDContextKey sessionIDContextKey = "session_id"
14	MessageIDContextKey messageIDContextKey = "message_id"
15)
16
17func GetSessionFromContext(ctx context.Context) string {
18	sessionID := ctx.Value(SessionIDContextKey)
19	if sessionID == nil {
20		return ""
21	}
22	s, ok := sessionID.(string)
23	if !ok {
24		return ""
25	}
26	return s
27}
28
29func GetMessageFromContext(ctx context.Context) string {
30	messageID := ctx.Value(MessageIDContextKey)
31	if messageID == nil {
32		return ""
33	}
34	s, ok := messageID.(string)
35	if !ok {
36		return ""
37	}
38	return s
39}