github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-01 04:29:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed DeprecationWarning for PromptTemplate.from_file module-level calls (#14468)

Browse Source 
Resolves https://github.com/langchain-ai/langchain/issues/14467
This commit is contained in:
James Braza 2023-12-12 20:43:27 -05:00 committed by GitHub
parent df95abb7e7
commit b9ef92f2f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/langchain/langchain/chains/llm_summarization_checker/base.py
View File

@ -17,18 +17,10 @@ from langchain.chains.sequential import SequentialChain
PROMPTS_DIR = Path(__file__).parent / "prompts"
CREATE_ASSERTIONS_PROMPT = PromptTemplate.from_file(
PROMPTS_DIR / "create_facts.txt", ["summary"]
)
CHECK_ASSERTIONS_PROMPT = PromptTemplate.from_file(
PROMPTS_DIR / "check_facts.txt", ["assertions"]
)
REVISED_SUMMARY_PROMPT = PromptTemplate.from_file(
PROMPTS_DIR / "revise_summary.txt", ["checked_assertions", "summary"]
)
ARE_ALL_TRUE_PROMPT = PromptTemplate.from_file(
PROMPTS_DIR / "are_all_true_prompt.txt", ["checked_assertions"]
)
CREATE_ASSERTIONS_PROMPT = PromptTemplate.from_file(PROMPTS_DIR / "create_facts.txt")
CHECK_ASSERTIONS_PROMPT = PromptTemplate.from_file(PROMPTS_DIR / "check_facts.txt")
REVISED_SUMMARY_PROMPT = PromptTemplate.from_file(PROMPTS_DIR / "revise_summary.txt")
ARE_ALL_TRUE_PROMPT = PromptTemplate.from_file(PROMPTS_DIR / "are_all_true_prompt.txt")
def _load_sequential_chain(

7
libs/langchain/tests/unit_tests/chains/test_llm_summarization_checker.py
View File

@ -14,6 +14,13 @@ from langchain.chains.llm_summarization_checker.base import (
from tests.unit_tests.llms.fake_llm import FakeLLM
def test_input_variables() -> None:
assert CREATE_ASSERTIONS_PROMPT.input_variables == ["summary"]
assert CHECK_ASSERTIONS_PROMPT.input_variables == ["assertions"]
assert REVISED_SUMMARY_PROMPT.input_variables == ["checked_assertions", "summary"]
assert ARE_ALL_TRUE_PROMPT.input_variables == ["checked_assertions"]
@pytest.fixture
def fake_llm_summarization_checker_chain() -> LLMSummarizationCheckerChain:
"""Fake LLMCheckerChain for testing."""