Spaces:
Build error
Build error
made changes to the get context function
Browse files
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(
|
| 69 |
-
queried_results = retriever.invoke(
|
| 70 |
query_embedding = genai.embed_content(model=embedding_model,
|
| 71 |
-
content=
|
| 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,
|
|
|
|
| 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,
|