File size: 1,148 Bytes
cf5c1e2 60532a1 cf5c1e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from mixpanel import Mixpanel
from datetime import datetime
from enum import Enum
from knowlang.configs.chat_config import ChatbotAnalyticsConfig
class ChatFeedback(Enum):
POSITIVE = "positive"
NEGATIVE = "negative"
class ChatAnalytics:
def __init__(self, config: ChatbotAnalyticsConfig):
self.mp = Mixpanel(config.api_key)
def track_query(
self,
query: str,
client_ip: str
):
"""Track query event in Mixpanel"""
self.mp.track(
distinct_id=hash(client_ip), # Hash for privacy
event_name="chat_query",
properties={
"query": query,
}
)
def track_feedback(
self,
like: bool,
query: str,
client_ip: str
):
"""Track feedback event in Mixpanel"""
self.mp.track(
distinct_id=hash(client_ip), # Hash for privacy
event_name="chat_feedback",
properties={
"feedback": ChatFeedback.POSITIVE.value if like else ChatFeedback.NEGATIVE.value,
"query": query,
}
) |