diff --git a/semantic_router/encoders/openai.py b/semantic_router/encoders/openai.py index e46e6c496f2e8ae77a7a6121b528b2107e2b654f..bd68727f1a0d616932486d5221cd96763a4bb7ae 100644 --- a/semantic_router/encoders/openai.py +++ b/semantic_router/encoders/openai.py @@ -19,8 +19,12 @@ model_configs = { "text-embedding-ada-002": EncoderInfo( name="text-embedding-ada-002", token_limit=8192 ), - "text-embedding-3-small": EncoderInfo(name="text-embedding-3-small", token_limit=8192), - "text-embedding-3-large": EncoderInfo(name="text-embedding-3-large", token_limit=8192), + "text-embedding-3-small": EncoderInfo( + name="text-embedding-3-small", token_limit=8192 + ), + "text-embedding-3-large": EncoderInfo( + name="text-embedding-3-large", token_limit=8192 + ), }