Improved variable assignment to use your own azure open ai embedding model

This commit is contained in:
MdNazishArmanShorthillsAI
2025-02-17 12:43:51 +05:30
parent fce24f7611
commit 44ef234002

View File

@@ -165,7 +165,7 @@ async def azure_openai_complete(
) )
async def azure_openai_embed( async def azure_openai_embed(
texts: list[str], texts: list[str],
model: str = "text-embedding-3-small", model: str = os.getenv("EMBEDDING_MODEL", "text-embedding-3-small"),
base_url: str = None, base_url: str = None,
api_key: str = None, api_key: str = None,
api_version: str = None, api_version: str = None,