Skip to content

Commit 141461b

Browse files
committed
debug
1 parent a206624 commit 141461b

File tree

3 files changed

+24
-3
lines changed

3 files changed

+24
-3
lines changed

apps/agentfabric/server.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,7 @@ def save_builder_config(uuid_str):
353353
error=str(e),
354354
details={'error_traceback': traceback.format_exc()})
355355
save_builder_configuration(builder_cfg=builder_config, uuid_str=uuid_str)
356-
# app.session_manager.clear_user_bot(uuid_str)
356+
app.session_manager.renew_user_bot(uuid_str)
357357

358358
return jsonify({'success': True, 'request_id': request_id_var.get('')})
359359

apps/agentfabric/server_utils.py

+18-2
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,8 @@ def get_builder_bot(
120120
builder_id,
121121
renew=False) -> Tuple[AgentBuilder, MemoryWithRetrievalKnowledge]:
122122
builder_agent = self.builder_bots[builder_id]
123+
print(f'builder_id: {builder_id}')
124+
print(f'builder_bots: {self.builder_bots.keys()}')
123125
if renew or builder_agent is None:
124126
logger.info(f'init_builder_chatbot_agent: {builder_id} ')
125127
builder_agent = init_builder_chatbot_agent(builder_id)
@@ -137,17 +139,31 @@ def get_user_bot(
137139
self,
138140
builder_id,
139141
session,
140-
renew=False,
141142
user_token=None) -> Tuple[RolePlay, MemoryWithRetrievalKnowledge]:
142143
unique_id = builder_id + '_' + session
144+
print(f'unique_id: {unique_id}')
143145
user_agent = self.user_bots[unique_id]
144-
if renew or user_agent is None:
146+
print(f'self.user_bots: {self.user_bots}')
147+
print(f'user_agent: {user_agent}')
148+
if user_agent is None:
145149
logger.info(f'init_user_chatbot_agent: {builder_id} {session}')
146150
user_agent = init_user_chatbot_agent(
147151
builder_id, session, use_tool_api=True, user_token=user_token)
148152
self.user_bots[unique_id] = user_agent
149153
return user_agent
150154

155+
def renew_user_bot(
156+
self,
157+
builder_id,
158+
session='default',
159+
user_token=None):
160+
self.clear_user_bot(builder_id, session)
161+
unique_id = builder_id + '_' + session
162+
logger.info(f'reinit_user_chatbot_agent: {builder_id} {session}')
163+
user_agent = init_user_chatbot_agent(
164+
builder_id, session, use_tool_api=True, user_token=user_token)
165+
self.user_bots[unique_id] = user_agent
166+
151167
def clear_user_bot(self, builder_id, session):
152168
unique_id = builder_id + '_' + session
153169
user_agent = self.user_bots[unique_id]

apps/agentfabric/user_core.py

+5
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,11 @@ def init_user_chatbot_agent(uuid_str='',
5454
user_token=user_token,
5555
openapi_list=openapi_plugin_list)
5656

57+
logger.query_info(
58+
uuid=uuid_str,
59+
message=
60+
f'init {builder_cfg.name} chatbot end.')
61+
5762
# build memory
5863
preview_history_dir = get_user_preview_history_dir(uuid_str, session)
5964
storage_path = preview_history_dir

0 commit comments

Comments
 (0)