|
|
@@ -20,6 +20,7 @@ import { prettyObject } from "../utils/format";
|
|
|
import { estimateTokenLength } from "../utils/token";
|
|
|
import { nanoid } from "nanoid";
|
|
|
import { createPersistStore } from "../utils/store";
|
|
|
+import { identifyDefaultClaudeModel } from "../utils/checkers";
|
|
|
|
|
|
export type ChatMessage = RequestMessage & {
|
|
|
date: string;
|
|
|
@@ -353,7 +354,7 @@ export const useChatStore = createPersistStore(
|
|
|
var api: ClientApi;
|
|
|
if (modelConfig.model.startsWith("gemini")) {
|
|
|
api = new ClientApi(ModelProvider.GeminiPro);
|
|
|
- } else if (modelConfig.model.startsWith("claude")) {
|
|
|
+ } else if (identifyDefaultClaudeModel(modelConfig.model)) {
|
|
|
api = new ClientApi(ModelProvider.Claude);
|
|
|
} else {
|
|
|
api = new ClientApi(ModelProvider.GPT);
|
|
|
@@ -539,7 +540,7 @@ export const useChatStore = createPersistStore(
|
|
|
var api: ClientApi;
|
|
|
if (modelConfig.model.startsWith("gemini")) {
|
|
|
api = new ClientApi(ModelProvider.GeminiPro);
|
|
|
- } else if (modelConfig.model.startsWith("claude")) {
|
|
|
+ } else if (identifyDefaultClaudeModel(modelConfig.model)) {
|
|
|
api = new ClientApi(ModelProvider.Claude);
|
|
|
} else {
|
|
|
api = new ClientApi(ModelProvider.GPT);
|