diff --git a/src/proxy/middleware/request/preprocessors/validate-context-size.ts b/src/proxy/middleware/request/preprocessors/validate-context-size.ts index e639b37..887b039 100644 --- a/src/proxy/middleware/request/preprocessors/validate-context-size.ts +++ b/src/proxy/middleware/request/preprocessors/validate-context-size.ts @@ -68,9 +68,11 @@ export const validateContextSize: RequestPreprocessor = async (req) => { modelMax = 131072; } else if (model.match(/^gpt-4(-\d{4})?-vision(-preview)?$/)) { modelMax = 131072; + } else if (model.match(/^o1(-\d{4}-\d{2}-\d{2})?$/)) { + modelMax = 200000; } else if (model.match(/^o1-mini(-\d{4}-\d{2}-\d{2})?$/)) { modelMax = 128000; - } else if (model.match(/^o1(-preview)?(-\d{4}-\d{2}-\d{2})?$/)) { + } else if (model.match(/^o1-preview(-\d{4}-\d{2}-\d{2})?$/)) { modelMax = 128000; } else if (model.match(/gpt-3.5-turbo/)) { modelMax = 16384;