0
0
mirror of https://github.com/MIDORIBIN/langchain-gpt4free.git synced 2024-12-24 11:34:39 +03:00
langchain-gpt4free/sample/prompt_template_sample.py

28 lines
685 B
Python
Raw Normal View History

2023-07-23 04:20:03 +03:00
from g4f import Provider, Model
from langchain.llms.base import LLM
from langchain import PromptTemplate
from langchain_g4f import G4FLLM
def main():
template = "What color is the {fruit}?"
prompt_template = PromptTemplate(template=template, input_variables=["fruit"])
llm: LLM = G4FLLM(
model=Model.gpt_35_turbo,
provider=Provider.Aichat,
)
res = llm(prompt_template.format(fruit="apple"))
print(res)
# The color of an apple can vary, but it is typically red, green, or yellow.
res = llm(prompt_template.format(fruit="lemon"))
print(res)
# The color of a lemon is typically yellow.
if __name__ == "__main__":
main()