Update rag_engine.py

rag_engine.py

@@ -2,23 +2,28 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
 
-model = SentenceTransformer('all-MiniLM-L6-v2')
+model = SentenceTransformer("all-MiniLM-L6-v2")
 
 knowledge = [
     "Hello! I am ZXAI.",
-    "I am an AI built by ZX Labs.",
-    "I can help you with coding, reasoning, and live data."
+    "I am an AI developed by ZX.",
+    "I can help you with coding, reasoning, and live data.",
+    "To generate an image, type /image followed by your prompt."
 ]
 
 embeddings = model.encode(knowledge)
 index = faiss.IndexFlatL2(384)
 index.add(np.array(embeddings))
 
-def rag_response(query):
+
+def rag_response(query, threshold=0.75):
+
     query_embedding = model.encode([query])
     D, I = index.search(np.array(query_embedding), 1)
 
-    if D[0][0] < 0.8:
+    similarity_score = D[0][0]
+
+    if similarity_score < threshold:
         return knowledge[I[0][0]]
+
     return None
-