diff --git a/src/LLMProviders/chatModelManager.ts b/src/LLMProviders/chatModelManager.ts index 038f953d..40b14b19 100644 --- a/src/LLMProviders/chatModelManager.ts +++ b/src/LLMProviders/chatModelManager.ts @@ -189,6 +189,6 @@ export default class ChatModelManager { } async countTokens(inputStr: string): Promise { - return ChatModelManager.chatOpenAI.getNumTokens(inputStr); + return ChatModelManager.chatModel.getNumTokens(inputStr); } } \ No newline at end of file diff --git a/src/components/ChatComponents/ChatIcons.tsx b/src/components/ChatComponents/ChatIcons.tsx index 918243b1..77faad53 100644 --- a/src/components/ChatComponents/ChatIcons.tsx +++ b/src/components/ChatComponents/ChatIcons.tsx @@ -113,14 +113,7 @@ const ChatIcons: React.FC = ({ - {/* - - - */} - - - - + diff --git a/src/constants.ts b/src/constants.ts index fabb0b5e..f5427fe7 100644 --- a/src/constants.ts +++ b/src/constants.ts @@ -11,8 +11,6 @@ export enum ChatModels { GPT_4 = 'gpt-4', GPT_4_TURBO = 'gpt-4-turbo-preview', GPT_4_32K = 'gpt-4-32k', - AZURE_GPT_35_TURBO = 'gpt-35-turbo', - AZURE_GPT_35_TURBO_16K = 'gpt-35-turbo-16k', GEMINI_PRO = 'gemini-pro', OLLAMA = 'ollama', } @@ -23,10 +21,7 @@ export enum ChatModelDisplayNames { GPT_4 = 'GPT-4', GPT_4_TURBO = 'GPT-4 TURBO', GPT_4_32K = 'GPT-4 32K', - AZURE_GPT_35_TURBO = 'AZURE GPT-3.5', - AZURE_GPT_35_TURBO_16K = 'AZURE GPT-3.5-16K', - AZURE_GPT_4 = 'AZURE GPT-4', - AZURE_GPT_4_32K = 'AZURE GPT-4 32K', + AZURE_OPENAI = 'AZURE OPENAI', GEMINI_PRO = 'GEMINI PRO', OPENROUTERAI = 'OPENROUTER.AI', OLLAMA = 'OLLAMA (LOCAL)', @@ -43,10 +38,7 @@ export const OPENAI_MODELS = new Set([ ]); export const AZURE_MODELS = new Set([ - ChatModelDisplayNames.AZURE_GPT_35_TURBO, - ChatModelDisplayNames.AZURE_GPT_35_TURBO_16K, - ChatModelDisplayNames.AZURE_GPT_4, - ChatModelDisplayNames.AZURE_GPT_4_32K, + ChatModelDisplayNames.AZURE_OPENAI, ]); export const GOOGLE_MODELS = new Set([ @@ -71,10 +63,7 @@ export const DISPLAY_NAME_TO_MODEL: Record = { [ChatModelDisplayNames.GPT_4]: ChatModels.GPT_4, [ChatModelDisplayNames.GPT_4_TURBO]: ChatModels.GPT_4_TURBO, [ChatModelDisplayNames.GPT_4_32K]: ChatModels.GPT_4_32K, - [ChatModelDisplayNames.AZURE_GPT_35_TURBO]: ChatModels.AZURE_GPT_35_TURBO, - [ChatModelDisplayNames.AZURE_GPT_35_TURBO_16K]: ChatModels.AZURE_GPT_35_TURBO_16K, - [ChatModelDisplayNames.AZURE_GPT_4]: ChatModels.GPT_4, - [ChatModelDisplayNames.AZURE_GPT_4_32K]: ChatModels.GPT_4_32K, + [ChatModelDisplayNames.AZURE_OPENAI]: 'azure_openai', [ChatModelDisplayNames.GEMINI_PRO]: ChatModels.GEMINI_PRO, }; diff --git a/src/settings/components/ApiSettings.tsx b/src/settings/components/ApiSettings.tsx index 50b9c427..7bbf9c46 100644 --- a/src/settings/components/ApiSettings.tsx +++ b/src/settings/components/ApiSettings.tsx @@ -146,6 +146,7 @@ const ApiSettings: React.FC = ({ /> 0 + ? `\ntags: ${noteWithTag.tags.join(",")}` + : "") + ); } function getChatContextStr(chatNoteContextPath: string, chatNoteContextTags: string[]): string { const pathStr = (chatNoteContextPath ? `\nChat context by path: ${chatNoteContextPath}` : ''); - const tagsStr = (chatNoteContextTags ? `\nChat context by tags: ${chatNoteContextTags}` : ''); + const tagsStr = (chatNoteContextTags?.length > 0 ? `\nChat context by tags: ${chatNoteContextTags}` : ''); return pathStr + tagsStr; }