diff --git a/app/service/chat_robot/script/service/CallQWen.py b/app/service/chat_robot/script/service/CallQWen.py
index 9e9f1a5..d2f28a0 100644
--- a/app/service/chat_robot/script/service/CallQWen.py
+++ b/app/service/chat_robot/script/service/CallQWen.py
@@ -283,7 +283,7 @@ def call_with_messages(message, gender):
         elif assistant_output.tool_calls[0]['function']['name'] == 'get_image_from_vector_db':
             content = json.loads(assistant_output.tool_calls[0]['function']['arguments'])
             tool_info = {"name": "get_image_from_vector_db", "role": "tool",
-                         'content': get_image_from_vector_db(gender, content['parameters']['content'])}
+                         'content': get_image_from_vector_db(gender, content['parameters']['content'] if "parameters" in content.keys() else content['content'])}
             flag = False
             result_content = tool_info['content']
             response_type = "image"