Sebunya commited on
Commit
e576071
·
verified ·
1 Parent(s): 5e1198e

made changes to the get context function

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -65,10 +65,10 @@ def generate_xeno_response(context, question):
65
  response = model.generate_content(user_prompt)
66
  return response.text.strip()
67
 
68
- def get_context_and_answer(query):
69
- queried_results = retriever.invoke(query)
70
  query_embedding = genai.embed_content(model=embedding_model,
71
- content=query,
72
  task_type="retrieval_query")['embedding']
73
  cosine_scores = []
74
  for doc in queried_results:
@@ -81,7 +81,8 @@ def get_context_and_answer(query):
81
  top_docs = sorted(zip(queried_results, cosine_scores), key=lambda x: x[1], reverse=True)[:2]
82
  for i, (result, score) in enumerate(top_docs, 1):
83
  context += f"Knowledge Entry {i}:\nQ: {result.metadata['question']}\nA: {result.metadata['content']}\n{'-'*40}\n"
84
- return generate_xeno_response(context, query)
 
85
 
86
  iface = gr.ChatInterface(
87
  fn=get_context_and_answer,
 
65
  response = model.generate_content(user_prompt)
66
  return response.text.strip()
67
 
68
+ def get_context_and_answer(message, history):
69
+ queried_results = retriever.invoke(message)
70
  query_embedding = genai.embed_content(model=embedding_model,
71
+ content=message,
72
  task_type="retrieval_query")['embedding']
73
  cosine_scores = []
74
  for doc in queried_results:
 
81
  top_docs = sorted(zip(queried_results, cosine_scores), key=lambda x: x[1], reverse=True)[:2]
82
  for i, (result, score) in enumerate(top_docs, 1):
83
  context += f"Knowledge Entry {i}:\nQ: {result.metadata['question']}\nA: {result.metadata['content']}\n{'-'*40}\n"
84
+ return generate_xeno_response(context, message)
85
+
86
 
87
  iface = gr.ChatInterface(
88
  fn=get_context_and_answer,