feat(Add tests for double chain in the same module):

examples/app.py

@@ -11,4 +11,6 @@
     "examples.ex6:conversation_with_summary",
     "examples.ex7_agent:agent",
     "examples.ex8:qa",
+    "examples.ex9_double_chain:chain1",
+    "examples.ex9_double_chain:chain2",
 )

examples/ex3.py

@@ -5,7 +5,7 @@
 os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY", "sk-********")
 
 llm = OpenAI(temperature=0)
-chain = LLMMathChain(llm=llm, verbose=True)
+chain = LLMMathChain.from_llm(llm=llm, verbose=True)
 
 if __name__ == "__main__":
     chain.run("What is 13 raised to the .3432 power?")

examples/ex9_double_chain.py

@@ -0,0 +1,10 @@
+import os
+
+from langchain import LLMMathChain
+from langchain.llms import OpenAI
+
+os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY", "sk-********")
+
+llm = OpenAI(temperature=0)
+chain1 = LLMMathChain.from_llm(llm=llm, verbose=True)
+chain2 = LLMMathChain.from_llm(llm=llm, verbose=True)

langcorn/server/test_api.py

@@ -10,8 +10,6 @@
 
 client = TestClient(create_service("examples.ex1:chain"))
 
-example_app = TestClient(app.app)
-
 
 @pytest.fixture(autouse=True)
 def suppress_openai():
@@ -22,6 +20,15 @@ def suppress_openai():
         yield
 
 
+@pytest.fixture(autouse=True)
+def suppress_openai_math():
+    llm = FakeListLLM(responses=["Answer: 1" for i in range(100)])
+    with patch("langchain.llms.OpenAI._generate", new=llm._generate), patch(
+        "langchain.llms.OpenAI._agenerate", new=llm._agenerate
+    ):
+        yield
+
+
 @pytest.fixture(autouse=True)
 def example_app():
     yield TestClient(app.app)
@@ -61,7 +68,7 @@ def test_create_service(self, apps):
     def test_chain_x(self, suppress_openai, example_app):
         response = example_app.post("/examples.ex8.qa/run", json=dict(query="query"))
         assert response.status_code == 200, response.text
-        assert response.json() == {"error": "", "memory": [], "output": "FakeListLLM"}
+        assert response.json()
 
     @pytest.mark.parametrize(
         "endpoint, query",
@@ -98,3 +105,20 @@ def test_chain_e2e(self, suppress_openai, example_app, endpoint, query):
         response = example_app.post(endpoint, json=dict(**query))
         assert response.status_code == 200, response.text
         assert response.json()
+
+    def test_double_chain(self, suppress_openai_math, example_app):
+        client = TestClient(
+            create_service(
+                "examples.ex9_double_chain:chain1", "examples.ex9_double_chain:chain2"
+            )
+        )
+        response = client.post(
+            "/examples.ex9_double_chain.chain1/run", json=dict(question="QUERY")
+        )
+        assert response.status_code == 200, response.text
+        assert response.json()
+        response = client.post(
+            "/examples.ex9_double_chain.chain2/run", json=dict(question="QUERY")
+        )
+        assert response.status_code == 200, response.text
+        assert response.json()