Spaces:
Paused
Paused
frdel commited on
Commit ·
0540020
1
Parent(s): bc38658
fix default prompt profile
Browse files- python/tools/call_subordinate.py +11 -12
python/tools/call_subordinate.py
CHANGED
|
@@ -10,26 +10,25 @@ class Delegation(Tool):
|
|
| 10 |
self.agent.get_data(Agent.DATA_NAME_SUBORDINATE) is None
|
| 11 |
or str(reset).lower().strip() == "true"
|
| 12 |
):
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
sub.set_data(Agent.DATA_NAME_SUPERIOR, self.agent)
|
| 17 |
self.agent.set_data(Agent.DATA_NAME_SUBORDINATE, sub)
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
agent_prompt_profile = self.agent.config.prompts_subdir
|
| 21 |
-
if agent_prompt_profile != prompt_profile:
|
| 22 |
-
self.agent.config.prompts_subdir = prompt_profile
|
| 23 |
|
| 24 |
# add user message to subordinate agent
|
| 25 |
subordinate: Agent = self.agent.get_data(Agent.DATA_NAME_SUBORDINATE)
|
| 26 |
subordinate.hist_add_user_message(UserMessage(message=message, attachments=[]))
|
| 27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
# run subordinate monologue
|
| 29 |
result = await subordinate.monologue()
|
| 30 |
|
| 31 |
-
# reset prompt profile
|
| 32 |
-
self.agent.config.prompts_subdir = agent_prompt_profile
|
| 33 |
-
|
| 34 |
# result
|
| 35 |
-
return Response(message=result, break_loop=False)
|
|
|
|
| 10 |
self.agent.get_data(Agent.DATA_NAME_SUBORDINATE) is None
|
| 11 |
or str(reset).lower().strip() == "true"
|
| 12 |
):
|
| 13 |
+
# crate agent
|
| 14 |
+
sub = Agent(self.agent.number + 1, self.agent.config, self.agent.context)
|
| 15 |
+
# register superior/subordinate
|
| 16 |
sub.set_data(Agent.DATA_NAME_SUPERIOR, self.agent)
|
| 17 |
self.agent.set_data(Agent.DATA_NAME_SUBORDINATE, sub)
|
| 18 |
+
# set default prompt profile to new agents
|
| 19 |
+
sub.config.prompts_subdir = "default"
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
# add user message to subordinate agent
|
| 22 |
subordinate: Agent = self.agent.get_data(Agent.DATA_NAME_SUBORDINATE)
|
| 23 |
subordinate.hist_add_user_message(UserMessage(message=message, attachments=[]))
|
| 24 |
|
| 25 |
+
# set subordinate prompt profile if provided, if not, keep original
|
| 26 |
+
prompt_profile = kwargs.get("prompt_profile")
|
| 27 |
+
if prompt_profile:
|
| 28 |
+
subordinate.config.prompts_subdir = prompt_profile
|
| 29 |
+
|
| 30 |
# run subordinate monologue
|
| 31 |
result = await subordinate.monologue()
|
| 32 |
|
|
|
|
|
|
|
|
|
|
| 33 |
# result
|
| 34 |
+
return Response(message=result, break_loop=False)
|