mirror of
https://github.com/MIDORIBIN/langchain-gpt4free.git
synced 2024-12-25 03:54:41 +03:00
28 lines
685 B
Python
28 lines
685 B
Python
|
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()
|