Similarity

similarity.py

from dotenv import load_dotenv
from langchain import PromptTemplate, LLMChain
from langchain.llms import OpenAI


load_dotenv()


template = """Given the following true sentence:
 
    A sign that says W3C is on a white background

Tell me how similar is that sentence to the following one:

	{sentence}


Write your answer in a JSON object with one key: similarity (a float from 0 to 1) 
"""
prompt = PromptTemplate(
template=template,
input_variables=['sentence']
)
llm = OpenAI(temperature=0)


llm_chain = LLMChain(
prompt=prompt,
llm=llm,
)

sentence = input("Enter a sentence: ")
print(llm_chain.run(sentence))