Merge pull request #802 from MdNazishArmanShorthillsAI/main

Improved variable assignment to use your own azure open ai embedding …
This commit is contained in:
zrguo
2025-02-17 15:53:37 +08:00
committed by GitHub

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,