Update openai_call.py

openai_call.py

@@ -1,10 +1,10 @@
-from src.helper import download_hugging_face_embeddings
+from helper import download_hugging_face_embeddings
 from langchain_pinecone import PineconeVectorStore
 from langchain.prompts import PromptTemplate
 from langchain_openai import OpenAI
 from langchain.chains import RetrievalQA
 from dotenv import load_dotenv
-from src.prompt import prompt_template
+from prompt import prompt_template
 import os
 
 load_dotenv()
@@ -43,8 +43,3 @@ qa = RetrievalQA.from_chain_type(
 def openai_call(input):
     result = qa.invoke({"query": input})
     return str(result["result"])
-
-
-if __name__ == "__main__":
-    msg = "If a previous owner of a land had allowed a neighbour or neighbour to walk or drive over his land in a shortcut and this has been going on for say a decade or so can I as the new owner stop them now from using the shortcut?"
-    print(f"response: {openai_call(msg)}")