From 9aa7e5b41783a4ee75dd25bafcf219c625751f79 Mon Sep 17 00:00:00 2001
From: Joshua Li <joshuali925@gmail.com>
Date: Tue, 3 Sep 2024 21:21:28 +0000
Subject: [PATCH] update `type` field in ml-commons response

Signed-off-by: Joshua Li <joshuali925@gmail.com>
---
 server/routes/query_assist/utils/agents.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/server/routes/query_assist/utils/agents.ts b/server/routes/query_assist/utils/agents.ts
index c08550134a..97a668bb88 100644
--- a/server/routes/query_assist/utils/agents.ts
+++ b/server/routes/query_assist/utils/agents.ts
@@ -35,7 +35,8 @@ export const getAgentIdByConfig = async (
       method: 'GET',
       path: `${ML_COMMONS_API_PREFIX}/config/${configName}`,
     })) as ApiResponse<{
-      type: string;
+      type?: string;
+      config_type?: string;
       ml_configuration?: { agent_id?: string };
       configuration?: { agent_id?: string };
     }>;