diff --git a/templates/types/streaming/fastapi/app/settings.py b/templates/types/streaming/fastapi/app/settings.py index 41ab539584d0201444f609477d620696e793b998..e4225a40141126d935738b8d9fb305865545170a 100644 --- a/templates/types/streaming/fastapi/app/settings.py +++ b/templates/types/streaming/fastapi/app/settings.py @@ -40,10 +40,10 @@ def init_openai(): } Settings.llm = OpenAI(**config) - dimension = os.getenv("EMBEDDING_DIM") + dimensions = os.getenv("EMBEDDING_DIM") config = { "model": os.getenv("EMBEDDING_MODEL"), - "dimension": int(dimension) if dimension is not None else None, + "dimensions": int(dimensions) if dimensions is not None else None, } Settings.embed_model = OpenAIEmbedding(**config)