Skip to content

Commit

Permalink
Summarization model types support
Browse files Browse the repository at this point in the history
  • Loading branch information
Samiul-TheSoccerFan committed Feb 23, 2024
1 parent 521deb2 commit 798d3c2
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 20 deletions.
9 changes: 7 additions & 2 deletions packages/kbn-ai-playground/components/chat.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,12 @@ import { i18n } from '@kbn/i18n';

import { ChatSidebar } from './chat_sidebar';
import { useChat } from '../hooks/useChat';
import { ChatForm, ChatFormFields, MessageRole } from '../types';
import {
ChatForm,
ChatFormFields,
MessageRole,
SummarizationModelName
} from '../types';

import { MessageList } from './message_list/message_list';
import { QuestionInput } from './question_input';
Expand Down Expand Up @@ -55,7 +60,7 @@ export const Chat = () => {
api_key: data[ChatFormFields.openAIKey],
citations: data[ChatFormFields.citations].toString(),
elasticsearchQuery: JSON.stringify(data[ChatFormFields.elasticsearchQuery]),
summarization_model: data[ChatFormFields.summarizationModel],
summarization_model: data[ChatFormFields.summarizationModel] ?? SummarizationModelName.gpt3_5_turbo_1106,
},
}
);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
* 2.0.
*/

import React, { useState } from "react";
import React from "react";

import {
EuiButtonEmpty,
Expand All @@ -19,9 +19,11 @@ import {
} from "@elastic/eui";

import { i18n } from "@kbn/i18n";
import { SummarizationModelName } from "@kbn/ai-playground/types";

import { OpenAIIcon } from "./open_ai_icon";


const renderSelectOptions = (label: string) => (
<EuiFlexGroup alignItems="center">
<EuiFlexItem grow={false}>
Expand All @@ -33,16 +35,7 @@ const renderSelectOptions = (label: string) => (
</EuiFlexGroup>
);


enum SummarizationModelName {
gpt3_5 = 'gpt-3.5-turbo',
gpt3_5_turbo_1106 = 'gpt-3.5-turbo-1106',
gpt3_5_turbo_16k = 'gpt-3.5-turbo-16k',
gpt3_5_turbo_16k_0613 = 'gpt-3.5-turbo-16k-0613',
gpt3_5_turbo = 'gpt-3.5-turbo-instruct',
}

const Summarization_Model: EuiSuperSelectOption<string>[] = [
const SummarizationModel: EuiSuperSelectOption<string>[] = [
{
value: SummarizationModelName.gpt3_5,
inputDisplay: renderSelectOptions(SummarizationModelName.gpt3_5),
Expand Down Expand Up @@ -72,12 +65,9 @@ interface OpenAISummarizationModelProps {
onSelect: (key: string) => void;
}

export const OpenAISummarizationModel: React.FC<OpenAISummarizationModelProps> = ({ model, onSelect, openAIFlyOutOpen }) => {

const [selectedModel, setSelectedModel] = useState(model ?? SummarizationModelName.gpt3_5_turbo_1106);
export const OpenAISummarizationModel: React.FC<OpenAISummarizationModelProps> = ({ model = SummarizationModelName.gpt3_5_turbo_1106, onSelect, openAIFlyOutOpen }) => {

const onChange = (value: string) => {
setSelectedModel(value);
onSelect(value);
};

Expand Down Expand Up @@ -109,9 +99,9 @@ export const OpenAISummarizationModel: React.FC<OpenAISummarizationModelProps> =
}
>
<EuiSuperSelect
options={Summarization_Model}
valueOfSelected={selectedModel}
onChange={(value) => onChange(value)}
options={SummarizationModel}
valueOfSelected={model}
onChange={onChange}
/>
</EuiFormRow>
);
Expand Down
8 changes: 8 additions & 0 deletions packages/kbn-ai-playground/types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -99,3 +99,11 @@ export interface QuerySourceFields {
bm25_query_fields: string[];
source_fields: string[];
}

export enum SummarizationModelName {
gpt3_5 = 'gpt-3.5-turbo',
gpt3_5_turbo_1106 = 'gpt-3.5-turbo-1106',
gpt3_5_turbo_16k = 'gpt-3.5-turbo-16k',
gpt3_5_turbo_16k_0613 = 'gpt-3.5-turbo-16k-0613',
gpt3_5_turbo = 'gpt-3.5-turbo-instruct',
}

0 comments on commit 798d3c2

Please sign in to comment.