github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-10 15:33:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix chat openai unit test (#17124)

Browse Source 
- **Description:** 
Actually the test named `test_openai_apredict` isn't testing the
apredict method from ChatOpenAI.
  - **Twitter handle:**
  https://twitter.com/OAlmofadas
This commit is contained in:
Luiz Ferreira
2024-02-08 00:08:26 -03:00
committed by GitHub
parent f92738a6f6
commit 34d2daffb3
3 changed files with 12 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/tests/unit_tests/llms/test_openai.py
View File

@@ -1,12 +1,8 @@
import os
import pytest
from langchain_community.llms.openai import OpenAI
from langchain_community.utils.openai import is_openai_v1
os.environ["OPENAI_API_KEY"] = "foo"
def _openai_v1_installed() -> bool:
try:
@@ -17,15 +13,15 @@ def _openai_v1_installed() -> bool:
@pytest.mark.requires("openai")
def test_openai_model_param() -> None:
llm = OpenAI(model="foo")
llm = OpenAI(model="foo", openai_api_key="foo")
assert llm.model_name == "foo"
llm = OpenAI(model_name="foo")
llm = OpenAI(model_name="foo", openai_api_key="foo")
assert llm.model_name == "foo"
@pytest.mark.requires("openai")
def test_openai_model_kwargs() -> None:
llm = OpenAI(model_kwargs={"foo": "bar"})
llm = OpenAI(model_kwargs={"foo": "bar"}, openai_api_key="foo")
assert llm.model_kwargs == {"foo": "bar"}
@@ -42,7 +38,7 @@ def test_openai_invalid_model_kwargs() -> None:
@pytest.mark.requires("openai")
def test_openai_incorrect_field() -> None:
with pytest.warns(match="not default parameter"):
llm = OpenAI(foo="bar")
llm = OpenAI(foo="bar", openai_api_key="foo")
assert llm.model_kwargs == {"foo": "bar"}