mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-14 13:40:54 +00:00
feat(runtime): Execute codes in a sandbox environment (#2119)
This commit is contained in:
@@ -35,7 +35,7 @@ class GptsMemory:
|
||||
message_memory if message_memory is not None else DefaultGptsMessageMemory()
|
||||
)
|
||||
|
||||
self.messages_cache: defaultdict = defaultdict(List[GptsMessage])
|
||||
self.messages_cache: defaultdict = defaultdict(list)
|
||||
self.channels: defaultdict = defaultdict(Queue)
|
||||
self.enable_vis_map: defaultdict = defaultdict(bool)
|
||||
self.start_round_map: defaultdict = defaultdict(int)
|
||||
@@ -374,9 +374,9 @@ class GptsMemory:
|
||||
"receiver": message.receiver,
|
||||
"model": message.model_name,
|
||||
"markdown": view_info,
|
||||
"resource": message.resource_info
|
||||
if message.resource_info
|
||||
else None,
|
||||
"resource": (
|
||||
message.resource_info if message.resource_info else None
|
||||
),
|
||||
}
|
||||
)
|
||||
return await vis_client.get(VisAgentMessages.vis_tag()).display(
|
||||
@@ -427,3 +427,20 @@ class GptsMemory:
|
||||
else:
|
||||
param["status"] = Status.COMPLETE.value
|
||||
return await vis_client.get(VisAppLink.vis_tag()).display(content=param)
|
||||
|
||||
async def chat_messages(
|
||||
self,
|
||||
conv_id: str,
|
||||
):
|
||||
"""Get chat messages."""
|
||||
while True:
|
||||
queue = self.queue(conv_id)
|
||||
if not queue:
|
||||
break
|
||||
item = await queue.get()
|
||||
if item == "[DONE]":
|
||||
queue.task_done()
|
||||
break
|
||||
else:
|
||||
yield item
|
||||
await asyncio.sleep(0.005)
|
||||
|
Reference in New Issue
Block a user