github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-19 05:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain: support api key argument with OpenAI moderation chain (#29140)

Browse Source 
**Description:** Makes it possible to instantiate
`OpenAIModerationChain` with an `openai_api_key` argument only and no
`OPENAI_API_KEY` environment variable defined.

**Issue:** https://github.com/langchain-ai/langchain/issues/25176

**Dependencies:** `openai`

---------

Co-authored-by: ccurme <chester.curme@gmail.com>
This commit is contained in:
Christopher Varjas 2025-01-13 11:00:02 -05:00 committed by GitHub
parent 335ca3a606
commit e156b372fb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/chains/moderation.py
View File

@ -67,8 +67,8 @@ class OpenAIModerationChain(Chain):
if values["openai_pre_1_0"]: if values["openai_pre_1_0"]:
values["client"] = openai.Moderation values["client"] = openai.Moderation
else: else:
values["client"] = openai.OpenAI() values["client"] = openai.OpenAI(api_key=openai_api_key)
values["async_client"] = openai.AsyncOpenAI() values["async_client"] = openai.AsyncOpenAI(api_key=openai_api_key)
except ImportError: except ImportError:
raise ImportError( raise ImportError(

11
libs/langchain/tests/integration_tests/chains/openai_functions/test_openapi.py
View File

@ -2,6 +2,7 @@ import json
import pytest import pytest
from langchain.chains import OpenAIModerationChain
from langchain.chains.openai_functions.openapi import get_openapi_chain from langchain.chains.openai_functions.openapi import get_openapi_chain
api_spec = { api_spec = {
@ -36,3 +37,13 @@ def test_openai_openapi_chain() -> None:
chain = get_openapi_chain(json.dumps(api_spec), llm) chain = get_openapi_chain(json.dumps(api_spec), llm)
output = chain.invoke({"query": "Fetch the top two posts."}) output = chain.invoke({"query": "Fetch the top two posts."})
assert len(output["response"]) == 2 assert len(output["response"]) == 2
@pytest.mark.requires("openai")
def test_openai_moderation_chain_instantiation() -> None:
"""Test OpenAIModerationChain."""
api_key = "foo"
moderation = OpenAIModerationChain(openai_api_key=api_key)
assert isinstance(moderation, OpenAIModerationChain)