Merge pull request #802 from MdNazishArmanShorthillsAI/main
Improved variable assignment to use your own azure open ai embedding …
This commit is contained in:
commit
5581df0b06
1 changed files with 1 additions and 1 deletions
|
|
@ -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,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue