mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-10-26 01:54:36 +08:00
XiaoChen
db05f2ec37
Context.__init__ took `kwargs=dict()`. That literal is evaluated once, when
the function is defined, so every Context built without an explicit dict
pointed at the same object -- and the scheduler's four delivery paths, the
evolution push and the cloud client's message push all build one that way
and then write keys through `context["..."] = ...`:
>>> a = Context(ContextType.TEXT, "hello"); a["session_id"] = "conversation-A"
>>> b = Context(ContextType.TEXT, "a different conversation")
>>> b["session_id"]
'conversation-A'
A bare `Context()` after a feishu task even came with that task's
`session_id` and `receiver`, which is the pair a channel uses to decide
where a proactive push goes. bridge/agent_bridge.py and
bridge/agent_event_handler.py read context.kwargs back, so a leaked value
is treated as belonging to the current turn.
Default to None and create a fresh dict per instance. An explicit dict is
still used as given; None is accepted too, where the old code would have
left `.kwargs` as None and turned a missing key into a TypeError.
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
# encoding:utf-8
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class ContextType(Enum):
|
|
TEXT = 1 # 文本消息
|
|
VOICE = 2 # 音频消息
|
|
IMAGE = 3 # 图片消息
|
|
FILE = 4 # 文件信息
|
|
VIDEO = 5 # 视频信息
|
|
SHARING = 6 # 分享信息
|
|
|
|
IMAGE_CREATE = 10 # 创建图片命令
|
|
ACCEPT_FRIEND = 19 # 同意好友请求
|
|
JOIN_GROUP = 20 # 加入群聊
|
|
PATPAT = 21 # 拍了拍
|
|
FUNCTION = 22 # 函数调用
|
|
EXIT_GROUP = 23 #退出
|
|
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Context:
|
|
def __init__(self, type: ContextType = None, content=None, kwargs=None):
|
|
self.type = type
|
|
self.content = content
|
|
# Default to a fresh dict per instance. A literal ``kwargs=dict()``
|
|
# default is evaluated once, when the function is defined, so every
|
|
# Context built without an explicit dict shared that one object: a key
|
|
# a turn wrote through ``context["session_id"]`` (or receiver, agent_id,
|
|
# msg, on_event ...) was still there on the next Context built the same
|
|
# way, in an unrelated conversation.
|
|
self.kwargs = {} if kwargs is None else kwargs
|
|
|
|
def __contains__(self, key):
|
|
if key == "type":
|
|
return self.type is not None
|
|
elif key == "content":
|
|
return self.content is not None
|
|
else:
|
|
return key in self.kwargs
|
|
|
|
def __getitem__(self, key):
|
|
if key == "type":
|
|
return self.type
|
|
elif key == "content":
|
|
return self.content
|
|
else:
|
|
return self.kwargs[key]
|
|
|
|
def get(self, key, default=None):
|
|
try:
|
|
return self[key]
|
|
except KeyError:
|
|
return default
|
|
|
|
def __setitem__(self, key, value):
|
|
if key == "type":
|
|
self.type = value
|
|
elif key == "content":
|
|
self.content = value
|
|
else:
|
|
self.kwargs[key] = value
|
|
|
|
def __delitem__(self, key):
|
|
if key == "type":
|
|
self.type = None
|
|
elif key == "content":
|
|
self.content = None
|
|
else:
|
|
del self.kwargs[key]
|
|
|
|
def __str__(self):
|
|
return "Context(type={}, content={}, kwargs={})".format(self.type, self.content, self.kwargs)
|