Skip to content

Commit

Permalink
mind id
Browse files Browse the repository at this point in the history
  • Loading branch information
femto committed Feb 17, 2025
1 parent 6dc9f42 commit 86b8f3b
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 1 deletion.
2 changes: 2 additions & 0 deletions examples/smart_minion/aime/evalute_aime.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,6 +218,8 @@ async def solve_question(item):

answer, score, *_ = await brain.step(
query=item["problem"],
mind_id="left_mind", #deepseek r1 skips choose mind

execution_config=load_execution_config(ensemble_logic_path),
)
return answer
Expand Down
2 changes: 1 addition & 1 deletion minion/main/brain.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ async def step(self, input=None, query="", query_type="", **kwargs):
input.query_id = input.query_id or uuid.uuid4()
input.images = self.process_image_input(input) # normalize image format to base64

mind_id = await self.choose_mind(input)
mind_id = input.mind_id or await self.choose_mind(input)
if mind_id == "left_mind":
self.llm.config.temperature = 0.1
elif mind_id == "right_mind":
Expand Down
4 changes: 4 additions & 0 deletions minion/main/input.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,10 @@ class Input(BaseModel):
default="",
description="System prompt for LLM models"
)
mind_id : str = Field(
default="",
description="mind if to choose, left_mind/right_mind/hippocampus_mind"
)

def update_execution_state(self, **kwargs):
"""Update execution state with the provided key-value pairs.
Expand Down

0 comments on commit 86b8f3b

Please sign in to comment.