Update services/nlp.py

services/nlp.py

@@ -1,6 +1,6 @@
 import os
 from openai import OpenAI
-from models.query import Query
+from models.input import Input
 import re
 from transformers import pipeline
 from services.utils import clean_text, encode_and_normalize
@@ -63,8 +63,8 @@ def extract_keywords(text):
     return cleaned_text.split()
 
 
-def generate_response(query: Query, context_from_elasticsearch):
-    prompt = query.history
+def generate_response(input: Input, context_from_elasticsearch):
+    prompt = input.history
     prompt.append({"role": "assistant", "content": str(context_from_elasticsearch)})
 
     print(prompt)