From 100807900b35d18202a3a58929841e7101368ff9 Mon Sep 17 00:00:00 2001
From: Steph Milovic <stephanie.milovic@elastic.co>
Date: Mon, 7 Oct 2024 15:49:57 -0600
Subject: [PATCH] update isOpenSourceModel

---
 .../server/routes/utils.test.ts               | 12 +++++++++
 .../elastic_assistant/server/routes/utils.ts  | 26 ++++++++++++-------
 2 files changed, 28 insertions(+), 10 deletions(-)

diff --git a/x-pack/plugins/elastic_assistant/server/routes/utils.test.ts b/x-pack/plugins/elastic_assistant/server/routes/utils.test.ts
index 3ca1b8edb5036..9a77e645686dd 100644
--- a/x-pack/plugins/elastic_assistant/server/routes/utils.test.ts
+++ b/x-pack/plugins/elastic_assistant/server/routes/utils.test.ts
@@ -65,5 +65,17 @@ describe('Utils', () => {
       const isOpenModel = isOpenSourceModel(connector);
       expect(isOpenModel).toEqual(true);
     });
+
+    it('should return `true` when apiProvider of OpenAiProviderType.Other is specified', async () => {
+      const connector = {
+        actionTypeId: '.gen-ai',
+        config: {
+          apiUrl: OPENAI_CHAT_URL,
+          apiProvider: OpenAiProviderType.Other,
+        },
+      } as unknown as Connector;
+      const isOpenModel = isOpenSourceModel(connector);
+      expect(isOpenModel).toEqual(true);
+    });
   });
 });
diff --git a/x-pack/plugins/elastic_assistant/server/routes/utils.ts b/x-pack/plugins/elastic_assistant/server/routes/utils.ts
index 5811109b94ede..19fa2c84cc785 100644
--- a/x-pack/plugins/elastic_assistant/server/routes/utils.ts
+++ b/x-pack/plugins/elastic_assistant/server/routes/utils.ts
@@ -201,19 +201,25 @@ export const isOpenSourceModel = (connector?: Connector): boolean => {
   }
 
   const llmType = getLlmType(connector.actionTypeId);
-  const connectorApiUrl = connector.config?.apiUrl
-    ? (connector.config.apiUrl as string)
-    : undefined;
+  const isOpenAiType = llmType === 'openai';
+
+  if (!isOpenAiType) {
+    return false;
+  }
   const connectorApiProvider = connector.config?.apiProvider
     ? (connector.config?.apiProvider as OpenAiProviderType)
     : undefined;
+  if (connectorApiProvider === OpenAiProviderType.Other) {
+    return true;
+  }
 
-  const isOpenAiType = llmType === 'openai';
-  const isOpenAI =
-    isOpenAiType &&
-    (!connectorApiUrl ||
-      connectorApiUrl === OPENAI_CHAT_URL ||
-      connectorApiProvider === OpenAiProviderType.AzureAi);
+  const connectorApiUrl = connector.config?.apiUrl
+    ? (connector.config.apiUrl as string)
+    : undefined;
 
-  return isOpenAiType && !isOpenAI;
+  return (
+    !!connectorApiUrl &&
+    connectorApiUrl !== OPENAI_CHAT_URL &&
+    connectorApiProvider !== OpenAiProviderType.AzureAi
+  );
 };