From eef60e84631c7ccc1030a0fdccfcca03e5af6c55 Mon Sep 17 00:00:00 2001 From: "Blessing Jones, Joshua Divine" Date: Mon, 19 May 2025 11:27:18 +0530 Subject: [PATCH] Update model.py Handling dict key --- python/ai-server/src/ai_server/py_client/gaas/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ai-server/src/ai_server/py_client/gaas/model.py b/python/ai-server/src/ai_server/py_client/gaas/model.py index ecb7fea..64d3363 100644 --- a/python/ai-server/src/ai_server/py_client/gaas/model.py +++ b/python/ai-server/src/ai_server/py_client/gaas/model.py @@ -283,6 +283,8 @@ def get_chat_history( history = self.model_engine.get_conversation_history() messages = [] for msg in sorted(history, key=lambda x: x["DATE_CREATED"]): + if "MESSAGE_DATA" not in msg: + continue if msg["MESSAGE_TYPE"] == "INPUT": messages.append(HumanMessage(content=msg["MESSAGE_DATA"])) elif msg["MESSAGE_TYPE"] == "RESPONSE": @@ -333,7 +335,8 @@ def _create_chat_result(self, response: Dict[str, Any]) -> ChatResult: # if this is a tool # need to do a different return - if response["messageType"] == "TOOL": + # if response["messageType"] == "TOOL": + if response.get("messageType") == "TOOL": # Returns None in case "messageType" is not present tool_response = [] for m in message: tool_response.append(