cli[patch]: integration template (#14571)

libs/cli/langchain_cli/integration_template/tests/integration_tests/test_chat_models.py

@@ -0,0 +1,63 @@
+"""Test Chat__ModuleName__ chat model."""
+from __module_name__.chat_models import Chat__ModuleName__
+
+
+def test_stream() -> None:
+    """Test streaming tokens from OpenAI."""
+    llm = Chat__ModuleName__()
+
+    for token in llm.stream("I'm Pickle Rick"):
+        assert isinstance(token.content, str)
+
+
+async def test_astream() -> None:
+    """Test streaming tokens from OpenAI."""
+    llm = Chat__ModuleName__()
+
+    async for token in llm.astream("I'm Pickle Rick"):
+        assert isinstance(token.content, str)
+
+
+async def test_abatch() -> None:
+    """Test streaming tokens from Chat__ModuleName__."""
+    llm = Chat__ModuleName__()
+
+    result = await llm.abatch(["I'm Pickle Rick", "I'm not Pickle Rick"])
+    for token in result:
+        assert isinstance(token.content, str)
+
+
+async def test_abatch_tags() -> None:
+    """Test batch tokens from Chat__ModuleName__."""
+    llm = Chat__ModuleName__()
+
+    result = await llm.abatch(
+        ["I'm Pickle Rick", "I'm not Pickle Rick"], config={"tags": ["foo"]}
+    )
+    for token in result:
+        assert isinstance(token.content, str)
+
+
+def test_batch() -> None:
+    """Test batch tokens from Chat__ModuleName__."""
+    llm = Chat__ModuleName__()
+
+    result = llm.batch(["I'm Pickle Rick", "I'm not Pickle Rick"])
+    for token in result:
+        assert isinstance(token.content, str)
+
+
+async def test_ainvoke() -> None:
+    """Test invoke tokens from Chat__ModuleName__."""
+    llm = Chat__ModuleName__()
+
+    result = await llm.ainvoke("I'm Pickle Rick", config={"tags": ["foo"]})
+    assert isinstance(result.content, str)
+
+
+def test_invoke() -> None:
+    """Test invoke tokens from Chat__ModuleName__."""
+    llm = Chat__ModuleName__()
+
+    result = llm.invoke("I'm Pickle Rick", config=dict(tags=["foo"]))
+    assert isinstance(result.content, str)

libs/cli/langchain_cli/integration_template/tests/integration_tests/test_compile.py

@@ -0,0 +1,7 @@
+import pytest
+
+
+@pytest.mark.compile
+def test_placeholder() -> None:
+    """Used for compiling integration tests without running any real tests."""
+    pass

libs/cli/langchain_cli/integration_template/tests/integration_tests/test_llms.py

@@ -0,0 +1,63 @@
+"""Test __ModuleName__LLM llm."""
+from __module_name__.llms import __ModuleName__LLM
+
+
+def test_stream() -> None:
+    """Test streaming tokens from OpenAI."""
+    llm = __ModuleName__LLM()
+
+    for token in llm.stream("I'm Pickle Rick"):
+        assert isinstance(token, str)
+
+
+async def test_astream() -> None:
+    """Test streaming tokens from OpenAI."""
+    llm = __ModuleName__LLM()
+
+    async for token in llm.astream("I'm Pickle Rick"):
+        assert isinstance(token, str)
+
+
+async def test_abatch() -> None:
+    """Test streaming tokens from __ModuleName__LLM."""
+    llm = __ModuleName__LLM()
+
+    result = await llm.abatch(["I'm Pickle Rick", "I'm not Pickle Rick"])
+    for token in result:
+        assert isinstance(token, str)
+
+
+async def test_abatch_tags() -> None:
+    """Test batch tokens from __ModuleName__LLM."""
+    llm = __ModuleName__LLM()
+
+    result = await llm.abatch(
+        ["I'm Pickle Rick", "I'm not Pickle Rick"], config={"tags": ["foo"]}
+    )
+    for token in result:
+        assert isinstance(token, str)
+
+
+def test_batch() -> None:
+    """Test batch tokens from __ModuleName__LLM."""
+    llm = __ModuleName__LLM()
+
+    result = llm.batch(["I'm Pickle Rick", "I'm not Pickle Rick"])
+    for token in result:
+        assert isinstance(token, str)
+
+
+async def test_ainvoke() -> None:
+    """Test invoke tokens from __ModuleName__LLM."""
+    llm = __ModuleName__LLM()
+
+    result = await llm.ainvoke("I'm Pickle Rick", config={"tags": ["foo"]})
+    assert isinstance(result, str)
+
+
+def test_invoke() -> None:
+    """Test invoke tokens from __ModuleName__LLM."""
+    llm = __ModuleName__LLM()
+
+    result = llm.invoke("I'm Pickle Rick", config=dict(tags=["foo"]))
+    assert isinstance(result, str)

libs/cli/langchain_cli/integration_template/tests/unit_tests/test_chat_models.py

@@ -0,0 +1,9 @@
+"""Test chat model integration."""
+
+
+from __module_name__.chat_models import Chat__ModuleName__
+
+
+def test_initialization() -> None:
+    """Test chat model initialization."""
+    Chat__ModuleName__()

libs/cli/langchain_cli/integration_template/tests/unit_tests/test_imports.py

@@ -0,0 +1,7 @@
+from __module_name__ import __all__
+
+EXPECTED_ALL = ["__ModuleName__LLM", "Chat__ModuleName__", "__ModuleName__VectorStore"]
+
+
+def test_all_imports() -> None:
+    assert sorted(EXPECTED_ALL) == sorted(__all__)

libs/cli/langchain_cli/integration_template/tests/unit_tests/test_llms.py

@@ -0,0 +1,7 @@
+"""Test __ModuleName__ Chat API wrapper."""
+from __module_name__ import __ModuleName__LLM
+
+
+def test_initialization() -> None:
+    """Test integration initialization."""
+    __ModuleName__LLM()

libs/cli/langchain_cli/integration_template/tests/unit_tests/test_vectorstores.py

@@ -0,0 +1,6 @@
+from __module_name__.vectorstores import __ModuleName__VectorStore
+
+
+def test_initialization() -> None:
+    """Test integration vectorstore initialization."""
+    __ModuleName__VectorStore()