| from langchain_core.prompts import PromptTemplate | |
| from langchain_folder.llm_helper import llm | |
| def ReturnKeywordsfromPrompt(query): | |
| prompt = """You are a model and your job is the extract the keywords from the query which will be passed to you. | |
| For Eg: If the query is -> Provide me with a combined dataset for house price prediction. | |
| Your answer should be 'house price'. Do not return anything else, just the keywords, note that the queries can be for various datasets, | |
| not jsut for house price prediction, go it? | |
| Query: {query}""" | |
| pt = PromptTemplate.from_template(prompt) | |
| chain = pt | llm | |
| response = chain.invoke({'query': query}) | |
| return response.content |