fix: update callback function to accurately use LlmResponse and adjust Part order insertion

Author: Rasic2

agents/matmaster_agent/callback.py

@@ -6,7 +6,7 @@
 
 import litellm
 from google.adk.agents.callback_context import CallbackContext
-from google.adk.models import LlmResponse, LlmRequest
+from google.adk.models import LlmResponse
 from google.genai import types
 from google.genai.types import FunctionCall, Part, FunctionResponse
 
@@ -53,7 +53,8 @@ async def matmaster_set_lang(callback_context: CallbackContext) -> Optional[type
     callback_context.state['target_language'] = language
 
 
-async def matmaster_check_job_status(callback_context: CallbackContext, llm_response: LlmRequest) -> Optional[
+# after_model_callback
+async def matmaster_check_job_status(callback_context: CallbackContext, llm_response: LlmResponse) -> Optional[
     LlmResponse]:
     if (
             (jobs_dict := callback_context.state['long_running_jobs']) and
@@ -76,15 +77,15 @@ async def matmaster_check_job_status(callback_context: CallbackContext, llm_resp
                 llm_response.content.parts.insert(0, Part(text=job_complete_intro.format(job_id=job_id),
                                                           function_call=FunctionCall(id=function_call_id,
                                                                                      name='transfer_to_agent',
-                                                                                     args={'agent_name': agent_name}),
-                                                          function_response=FunctionResponse(id=function_call_id,
-                                                                                             name='transfer_to_agent',
-                                                                                             response=None)
+                                                                                     args={'agent_name': agent_name})
                                                           )
                                                   )
+                llm_response.content.parts.insert(1, Part(function_response=FunctionResponse(id=function_call_id,
+                                                                                             name='transfer_to_agent',
+                                                                                             response=None)))
+        return llm_response
 
 
-# after_model_callback
 async def matmaster_check_transfer(callback_context: CallbackContext, llm_response: LlmResponse) -> Optional[
     LlmResponse]:
     # 检查响应是否有效

agents/matmaster_agent/main.py

@@ -12,6 +12,7 @@
 from agents.matmaster_agent.logger import logger
 import litellm
 
+
 # litellm._turn_on_debug()
 
 
@@ -67,7 +68,8 @@ async def agent_main() -> None:
     # user_input = "帮我用DPA优化这个结构：https://dp-storage-test2.oss-cn-zhangjiakou.aliyuncs.com/bohrium-test/110663/12791/store/7ba41529-5af4-4e38-a6fb-c569cd769dd9/outputs/structure_paths/structure_bulk.cif"
     # user_input = "帮我检索TiO2"
     # user_input = "请你为我搭建一个氯化钠的结构"
-    user_input = '用openlam查找一个TiO2'
+    user_input = "我想要一个bandgap 小于0.5ev的结构，空间群225，生成数量1"
+    # user_input = '用openlam查找一个TiO2'
     print(f"🧑 用户：{user_input}")
 
     # Create the initial content with user input