From 91ec9da534d55c5687ee4e7019146cce22eed5c8 Mon Sep 17 00:00:00 2001
From: Erick Friis <erick@langchain.dev>
Date: Tue, 9 Jan 2024 11:54:11 -0800
Subject: [PATCH] openai[patch]: unit test load (#15624)

---
 .../openai/tests/unit_tests/test_load.py      | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 libs/partners/openai/tests/unit_tests/test_load.py

diff --git a/libs/partners/openai/tests/unit_tests/test_load.py b/libs/partners/openai/tests/unit_tests/test_load.py
new file mode 100644
index 0000000000..bb2c1a79d9
--- /dev/null
+++ b/libs/partners/openai/tests/unit_tests/test_load.py
@@ -0,0 +1,46 @@
+from langchain_core.load.dump import dumpd, dumps
+from langchain_core.load.load import load, loads
+
+from langchain_openai import ChatOpenAI, OpenAI
+
+
+def test_loads_openai_llm() -> None:
+    llm = OpenAI(model="davinci", temperature=0.5, openai_api_key="hello")
+    llm_string = dumps(llm)
+    llm2 = loads(llm_string, secrets_map={"OPENAI_API_KEY": "hello"})
+
+    assert llm2 == llm
+    llm_string_2 = dumps(llm2)
+    assert llm_string_2 == llm_string
+    assert isinstance(llm2, OpenAI)
+
+
+def test_load_openai_llm() -> None:
+    llm = OpenAI(model="davinci", temperature=0.5, openai_api_key="hello")
+    llm_obj = dumpd(llm)
+    llm2 = load(llm_obj, secrets_map={"OPENAI_API_KEY": "hello"})
+
+    assert llm2 == llm
+    assert dumpd(llm2) == llm_obj
+    assert isinstance(llm2, OpenAI)
+
+
+def test_loads_openai_chat() -> None:
+    llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.5, openai_api_key="hello")
+    llm_string = dumps(llm)
+    llm2 = loads(llm_string, secrets_map={"OPENAI_API_KEY": "hello"})
+
+    assert llm2 == llm
+    llm_string_2 = dumps(llm2)
+    assert llm_string_2 == llm_string
+    assert isinstance(llm2, ChatOpenAI)
+
+
+def test_load_openai_chat() -> None:
+    llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.5, openai_api_key="hello")
+    llm_obj = dumpd(llm)
+    llm2 = load(llm_obj, secrets_map={"OPENAI_API_KEY": "hello"})
+
+    assert llm2 == llm
+    assert dumpd(llm2) == llm_obj
+    assert isinstance(llm2, ChatOpenAI)