diff --git a/src/proxy/middleware/request/preprocessor-factory.ts b/src/proxy/middleware/request/preprocessor-factory.ts index 7358a57..0f9e98b 100644 --- a/src/proxy/middleware/request/preprocessor-factory.ts +++ b/src/proxy/middleware/request/preprocessor-factory.ts @@ -143,8 +143,15 @@ const handleTestMessage: RequestHandler = (req, res) => { completion: "Hello!", // anthropic chat content: [{ type: "text", text: "Hello!" }], + // gemini + candidates: [ + { + content: { parts: [{ text: "Hello!" }] }, + finishReason: "stop", + }, + ], proxy_note: - "This response was generated by the proxy's test message handler and did not go to the API.", + "SillyTavern connection test detected. Your prompt was not sent to the actual model and this response was generated by the proxy.", }); } }; diff --git a/src/shared/key-management/aws/checker.ts b/src/shared/key-management/aws/checker.ts index 395709a..06a3671 100644 --- a/src/shared/key-management/aws/checker.ts +++ b/src/shared/key-management/aws/checker.ts @@ -16,6 +16,7 @@ type AliasModelId = string; type ModuleAliasTuple = [ParentModelId, ...AliasModelId[]]; const KNOWN_MODEL_IDS: ModuleAliasTuple[] = [ + ["anthropic.claude-instant-v1"], ["anthropic.claude-v2", "anthropic.claude-v2:1"], ["anthropic.claude-3-sonnet-20240229-v1:0"], ["anthropic.claude-3-haiku-20240307-v1:0"],