Files
langchain/tests/unit_tests/embeddings/test_openai.py
T
2023-07-18 12:14:51 -07:00

21 lines
514 B
Python

import os
import pytest
from langchain.embeddings.openai import OpenAIEmbeddings
os.environ["OPENAI_API_KEY"] = "foo"
@pytest.mark.requires("openai")
def test_openai_invalid_model_kwargs() -> None:
with pytest.raises(ValueError):
OpenAIEmbeddings(model_kwargs={"model": "foo"})
@pytest.mark.requires("openai")
def test_openai_incorrect_field() -> None:
with pytest.warns(match="not default parameter"):
llm = OpenAIEmbeddings(foo="bar")
assert llm.model_kwargs == {"foo": "bar"}