from g4f import Provider, Model
from langchain.llms.base import LLM
from langchain import PromptTemplate
from langchain.chains import LLMChain

from langchain_g4f import G4FLLM


def main():
    llm: LLM = G4FLLM(
        model=Model.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()