switch to ml_configuration in agent response as backend changed

Signed-off-by: Heng Qian <[email protected]>
qianheng-aws · Jul 24, 2024 · b2610ca · b2610ca
1 parent beac8cc
commit b2610ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/routes/get_agent.ts b/server/routes/get_agent.ts
@@ -14,10 +14,10 @@ export const getAgent = async (id: string, client: OpenSearchClient['transport']
       path,
     });
 
-    if (!response || !response.body.configuration?.agent_id) {
+    if (!response || !response.body.ml_configuration?.agent_id) {
       throw new Error(`cannot get agent ${id} by calling the api: ${path}`);
     }
-    return response.body.configuration.agent_id;
+    return response.body.ml_configuration.agent_id;
   } catch (error) {
     const errorMessage = JSON.stringify(error.meta?.body) || error;
     throw new Error(`get agent ${id} failed, reason: ${errorMessage}`);