[8.19] [Obs AI Assistant] Minor cleanup api tests (#215181) | [Obs AI Assistant] Replace Task Manager usage with Lock Manager (#216916) | update knowledge base installation flow (#214133) | [Obs AI Assistant] Specify embedding model during onboarding for the Knowledge Base (#218448)

packages/kbn-lock-manager/src/lock_manager_client.ts

@@ -256,6 +256,19 @@ export class LockManager {
   }
 }
 
+export async function getLock({
+  esClient,
+  logger,
+  lockId,
+}: {
+  esClient: ElasticsearchClient;
+  logger: Logger;
+  lockId: LockId;
+}): Promise<LockDocument | undefined> {
+  const lockManager = new LockManager(lockId, esClient, logger);
+  return lockManager.get();
+}
+
 export async function withLock<T>(
   {
     esClient,
@@ -280,9 +293,7 @@ export async function withLock<T>(
 
   // extend the ttl periodically
   const extendInterval = Math.floor(ttl / 4);
-  logger.debug(
-    `Lock "${lockId}" acquired. Extending TTL every ${prettyMilliseconds(extendInterval)}`
-  );
+  logger.debug(`Extending TTL for lock "${lockId}" every ${prettyMilliseconds(extendInterval)}`);
 
   let extendTTlPromise = Promise.resolve(true);
   const intervalId = setInterval(() => {

packages/kbn-lock-manager/src/lock_manager_service.ts

@@ -8,7 +8,7 @@
  */
 
 import { CoreSetup, Logger } from '@kbn/core/server';
-import { LockId, withLock } from './lock_manager_client';
+import { LockId, withLock, getLock } from './lock_manager_client';
 
 export class LockManagerService {
   constructor(private readonly coreSetup: CoreSetup<any>, private readonly logger: Logger) {}
@@ -35,8 +35,16 @@ export class LockManagerService {
   ) {
     const [coreStart] = await this.coreSetup.getStartServices();
     const esClient = coreStart.elasticsearch.client.asInternalUser;
-    const logger = this.logger.get('LockManager');
+    const logger = this.logger.get('lock-manager');
 
     return withLock<T>({ esClient, logger, lockId, metadata }, callback);
   }
+
+  async getLock(lockId: LockId) {
+    const [coreStart] = await this.coreSetup.getStartServices();
+    const esClient = coreStart.elasticsearch.client.asInternalUser;
+    const logger = this.logger.get('lock-manager');
+
+    return getLock({ esClient, logger, lockId });
+  }
 }

src/dev/run_check_file_casing.ts

@@ -34,6 +34,9 @@ run(async ({ log }) => {
       'src/platform/**',
       'x-pack/platform/**',
       'x-pack/solutions/**',
+
+      // ignore autogenerated snapshots
+      'x-pack/test/api_integration/deployment_agnostic/apis/observability/ai_assistant/knowledge_base/snapshots',
     ],
   });
 

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/welcome_message.tsx

@@ -16,7 +16,7 @@ import type { UseKnowledgeBaseResult } from '../hooks/use_knowledge_base';
 import type { UseGenAIConnectorsResult } from '../hooks/use_genai_connectors';
 import { Disclaimer } from './disclaimer';
 import { WelcomeMessageConnectors } from './welcome_message_connectors';
-import { WelcomeMessageKnowledgeBase } from './welcome_message_knowledge_base';
+import { WelcomeMessageKnowledgeBase } from '../knowledge_base/welcome_message_knowledge_base';
 import { StarterPrompts } from './starter_prompts';
 import { useKibana } from '../hooks/use_kibana';
 
@@ -60,10 +60,6 @@ export function WelcomeMessage({
     if (isSupportedConnectorType(createdConnector.actionTypeId)) {
       connectors.reloadConnectors();
     }
-
-    if (!knowledgeBase.status.value || knowledgeBase.status.value?.ready === false) {
-      knowledgeBase.install();
-    }
   };
 
   const ConnectorFlyout = useMemo(

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/welcome_message_connectors.tsx

@@ -21,7 +21,7 @@ import { euiThemeVars } from '@kbn/ui-theme';
 import { isHttpFetchError } from '@kbn/core-http-browser';
 import type { UseGenAIConnectorsResult } from '../hooks/use_genai_connectors';
 
-const fadeInAnimation = keyframes`
+export const fadeInAnimation = keyframes`
   from {
     opacity: 0;
   }