github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-22 11:00:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/improve usability of api chain (#247)

Browse Source 
improve usability of api chain
This commit is contained in:
Harrison Chase
2022-12-02 15:44:10 -08:00
committed by GitHub
parent c897bd6cbd
commit a9ce04201f
3 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit_tests/chains/test_api.py
View File

@@ -68,11 +68,11 @@ def fake_llm_api_chain(test_api_data: dict) -> APIChain:
fake_llm = FakeLLM(queries=queries)
api_request_chain = LLMChain(llm=fake_llm, prompt=API_URL_PROMPT)
api_answer_chain = LLMChain(llm=fake_llm, prompt=API_RESPONSE_PROMPT)
requests_chain = FakeRequestsChain(output=TEST_API_RESPONSE)
requests_wrapper = FakeRequestsChain(output=TEST_API_RESPONSE)
return APIChain(
api_request_chain=api_request_chain,
api_answer_chain=api_answer_chain,
requests_chain=requests_chain,
requests_wrapper=requests_wrapper,
api_docs=TEST_API_DOCS,
)