From e3b7effc8f39333076c2bedd7306de81ce988de6 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Fri, 30 Jun 2023 09:26:05 -0700
Subject: [PATCH] Beef up import test (#6979)

---
 tests/unit_tests/test_dependencies.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/unit_tests/test_dependencies.py b/tests/unit_tests/test_dependencies.py
index b17af63910a..72ca1b793f4 100644
--- a/tests/unit_tests/test_dependencies.py
+++ b/tests/unit_tests/test_dependencies.py
@@ -79,6 +79,8 @@ def test_test_group_dependencies(poetry_conf: Mapping[str, Any]) -> None:
 
 def test_imports() -> None:
     """Test that you can import all top level things okay."""
+    from langchain.agents import OpenAIFunctionsAgent  # noqa: F401
+    from langchain.callbacks import OpenAICallbackHandler  # noqa: F401
     from langchain.chains import LLMChain  # noqa: F401
     from langchain.chat_models import ChatOpenAI  # noqa: F401
     from langchain.document_loaders import BSHTMLLoader  # noqa: F401