github-mirrors/kibana
1
0
Fork 0
mirror of https://github.com/elastic/kibana.git synced 2025-04-23 17:28:26 -04:00
Code Issues Projects Releases Packages Wiki Activity

[Obs AI Assistant] Use ELSER v2 (#170092)

Browse source 
Co-authored-by: Kibana Machine <42973632+kibanamachine@users.noreply.github.com>
This commit is contained in:
Dario Gieselaar 2023-10-30 20:03:35 +01:00 • committed by GitHub
parent 8fe512b8c2
commit beae7b1448
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
x-pack/plugins/observability_ai_assistant/server/service/index.ts
View file

@ -25,6 +25,8 @@ function getResourceName(resource: string) {
return `.kibana-observability-ai-assistant-${resource}`;
}
export const ELSER_MODEL_ID = '.elser_model_2';
export const INDEX_QUEUED_DOCUMENTS_TASK_ID = 'observabilityAIAssistant:indexQueuedDocumentsTask';
export const INDEX_QUEUED_DOCUMENTS_TASK_TYPE = INDEX_QUEUED_DOCUMENTS_TASK_ID + 'Type';
@ -120,6 +122,11 @@ export class ObservabilityAIAssistantService {
auto_expand_replicas: '0-1',
hidden: true,
},
mappings: {
_meta: {
model: ELSER_MODEL_ID,
},
},
},
});
@ -150,7 +157,7 @@ export class ObservabilityAIAssistantService {
processors: [
{
inference: {
model_id: '.elser_model_1',
model_id: ELSER_MODEL_ID,
target_field: 'ml',
field_map: {
text: 'text_field',

10
x-pack/plugins/observability_ai_assistant/server/service/kb_service/index.ts
View file

@ -13,7 +13,11 @@ import type { TaskManagerStartContract } from '@kbn/task-manager-plugin/server';
import pLimit from 'p-limit';
import pRetry from 'p-retry';
import { map } from 'lodash';
import { INDEX_QUEUED_DOCUMENTS_TASK_ID, INDEX_QUEUED_DOCUMENTS_TASK_TYPE } from '..';
import {
ELSER_MODEL_ID,
INDEX_QUEUED_DOCUMENTS_TASK_ID,
INDEX_QUEUED_DOCUMENTS_TASK_TYPE,
} from '..';
import type { KnowledgeBaseEntry } from '../../../common/types';
import type { ObservabilityAIAssistantResourceNames } from '../types';
import { getAccessQuery } from '../util/get_access_query';
@ -42,8 +46,6 @@ function isAlreadyExistsError(error: Error) {
);
}
const ELSER_MODEL_ID = '.elser_model_1';
function throwKnowledgeBaseNotReady(body: any) {
throw serverUnavailable(`Knowledge base is not ready yet`, body);
}
@ -199,7 +201,7 @@ export class KnowledgeBaseService {
text_expansion: {
'ml.tokens': {
model_text: text,
model_id: '.elser_model_1',
model_id: ELSER_MODEL_ID,
},
} as unknown as QueryDslTextExpansionQuery,
})),