feat: add matmaster_set_lang callback to set language before agent execution

Author: Rasic2

agents/matmaster_agent/agent.py

@@ -14,7 +14,7 @@
 from agents.matmaster_agent.MrDice_agent.agent import init_MrDice_agent
 from agents.matmaster_agent.apex_agent.agent import init_apex_agent
 from agents.matmaster_agent.base_agents.io_agent import HandleFileUploadLlmAgent
-from agents.matmaster_agent.callback import matmaster_prepare_state, matmaster_check_transfer
+from agents.matmaster_agent.callback import matmaster_prepare_state, matmaster_check_transfer, matmaster_set_lang
 from agents.matmaster_agent.chembrain_agent.agent import init_chembrain_agent
 from agents.matmaster_agent.constant import MATMASTER_AGENT_NAME
 from agents.matmaster_agent.llm_config import MatMasterLlmConfig
@@ -76,7 +76,7 @@ def __init__(self, llm_config):
             global_instruction=GlobalInstruction,
             instruction=AgentInstruction,
             description=AgentDescription,
-            before_agent_callback=matmaster_prepare_state,
+            before_agent_callback=[matmaster_prepare_state, matmaster_set_lang],
             after_model_callback=matmaster_check_transfer,
         )
 

agents/matmaster_agent/callback.py

@@ -38,6 +38,8 @@ async def matmaster_prepare_state(callback_context: CallbackContext) -> Optional
     callback_context.state['invocation_id_with_tool_call'] = callback_context.state.get('invocation_id_with_tool_call',
                                                                                         None)
 
+
+async def matmaster_set_lang(callback_context: CallbackContext) -> Optional[types.Content]:
     user_content = callback_context.user_content.parts[0].text
     prompt = get_user_content_lang().format(user_content=user_content)
     response = litellm.completion(model='azure/gpt-4o', messages=[{'role': 'user', 'content': prompt}],