mirror of
https://github.com/MIDORIBIN/langchain-gpt4free.git
synced 2024-12-23 19:22:58 +03:00
138abc4a30
fix #1
25 lines
923 B
Python
25 lines
923 B
Python
from g4f import Provider, models
|
|
from langchain import PromptTemplate
|
|
from langchain.chains import LLMChain
|
|
from langchain.llms.base import LLM
|
|
|
|
from langchain_g4f import G4FLLM
|
|
|
|
|
|
def main():
|
|
llm: LLM = G4FLLM(
|
|
model=models.gpt_35_turbo,
|
|
provider=Provider.Aichat,
|
|
)
|
|
prompt_template = PromptTemplate(
|
|
input_variables=["location"],
|
|
template="Where is the best tourist attraction in {location}?",
|
|
)
|
|
chain = LLMChain(llm=llm, prompt=prompt_template)
|
|
print(chain("tokyo"))
|
|
# {'location': 'tokyo', 'text': 'Tokyo has many amazing tourist attractions, but one of the most popular is the Tokyo Tower. Standing at 333 meters tall, it offers stunning views of the city skyline. Additionally, the historic Meiji Shrine and the bustling and vibrant neighborhood of Shibuya with its famous scramble crossing are also top attractions.'}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|