@@ -120,6 +120,8 @@ def get_builder_bot(
120
120
builder_id ,
121
121
renew = False ) -> Tuple [AgentBuilder , MemoryWithRetrievalKnowledge ]:
122
122
builder_agent = self .builder_bots [builder_id ]
123
+ print (f'builder_id: { builder_id } ' )
124
+ print (f'builder_bots: { self .builder_bots .keys ()} ' )
123
125
if renew or builder_agent is None :
124
126
logger .info (f'init_builder_chatbot_agent: { builder_id } ' )
125
127
builder_agent = init_builder_chatbot_agent (builder_id )
@@ -137,17 +139,31 @@ def get_user_bot(
137
139
self ,
138
140
builder_id ,
139
141
session ,
140
- renew = False ,
141
142
user_token = None ) -> Tuple [RolePlay , MemoryWithRetrievalKnowledge ]:
142
143
unique_id = builder_id + '_' + session
144
+ print (f'unique_id: { unique_id } ' )
143
145
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 :
145
149
logger .info (f'init_user_chatbot_agent: { builder_id } { session } ' )
146
150
user_agent = init_user_chatbot_agent (
147
151
builder_id , session , use_tool_api = True , user_token = user_token )
148
152
self .user_bots [unique_id ] = user_agent
149
153
return user_agent
150
154
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
+
151
167
def clear_user_bot (self , builder_id , session ):
152
168
unique_id = builder_id + '_' + session
153
169
user_agent = self .user_bots [unique_id ]
0 commit comments