Skip to content

Commit fe056c1

Browse files
committed
encoder for message sender types for conversations
1 parent fa36164 commit fe056c1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

sdk/chatfaq_sdk/agents/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import inspect
22
import os
3+
import json
34
from enum import Enum
45
from logging import getLogger
56
from typing import List, Callable, Any
@@ -16,6 +17,11 @@ class MessageSender(Enum):
1617
assistant = "assistant"
1718
user = "user"
1819

20+
class MessageSenderEncoder(json.JSONEncoder):
21+
def default(self, obj):
22+
if isinstance(obj, MessageSender):
23+
return obj.value
24+
return super().default(obj)
1925

2026
class StreamingMessageWithReferences(Layer):
2127
"""

0 commit comments

Comments
 (0)