github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 16:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

prompt nit (#141)

Browse Source 
doing some cleanup, and i think this just simplifies things...
This commit is contained in:
Harrison Chase 2022-11-14 21:30:33 -08:00 committed by GitHub
parent bbb405a492
commit 1835e8a681
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/prompts/prompt.py
View File

@ -94,8 +94,7 @@ class Prompt(BaseModel, BasePrompt):
Returns: Returns:
The final prompt generated. The final prompt generated.
""" """
example_str = example_separator.join(examples) template = example_separator.join([prefix, *examples, suffix])
template = prefix + example_str + suffix
return cls(input_variables=input_variables, template=template) return cls(input_variables=input_variables, template=template)
@classmethod @classmethod

4
tests/unit_tests/test_prompt.py
View File

@ -51,8 +51,8 @@ Question: {question}
Answer:""" Answer:"""
input_variables = ["question"] input_variables = ["question"]
example_separator = "\n\n" example_separator = "\n\n"
prefix = """Test Prompt:\n\n""" prefix = """Test Prompt:"""
suffix = """\n\nQuestion: {question}\nAnswer:""" suffix = """Question: {question}\nAnswer:"""
examples = [ examples = [
"""Question: who are you?\nAnswer: foo""", """Question: who are you?\nAnswer: foo""",
"""Question: what are you?\nAnswer: bar""", """Question: what are you?\nAnswer: bar""",