Update main.py
Browse filesthe retrieved chunks are changed to telugu from english
main.py
CHANGED
|
@@ -204,7 +204,8 @@ async def retrieve_top_k_hybrid(query, k, sem_weight,syn_weight,bm25):
|
|
| 204 |
|
| 205 |
top_results = df_expanded.sort_values(by="similarity", ascending=False).head(k)
|
| 206 |
|
| 207 |
-
return top_results["Chunks"].to_list()
|
|
|
|
| 208 |
|
| 209 |
|
| 210 |
os.makedirs("/tmp/transformers_cache", exist_ok=True)
|
|
@@ -332,7 +333,7 @@ def tts_chunk_stream(text_chunk: str, lang: str = "en"):
|
|
| 332 |
async def get_rag_response(user_message: str):
|
| 333 |
global chat_messages
|
| 334 |
Chunks = await retrieve_top_k_hybrid(user_message,15, 0.9, 0.1,bm25)
|
| 335 |
-
context = "======================================================================================================\n".join(Chunks)
|
| 336 |
chat_messages.append({"role": "user", "content": f'''
|
| 337 |
Context : {context}
|
| 338 |
User Query: {user_message}'''})
|
|
|
|
| 204 |
|
| 205 |
top_results = df_expanded.sort_values(by="similarity", ascending=False).head(k)
|
| 206 |
|
| 207 |
+
# return top_results["Chunks"].to_list()
|
| 208 |
+
return top_results["telugu_chunk"].to_list()
|
| 209 |
|
| 210 |
|
| 211 |
os.makedirs("/tmp/transformers_cache", exist_ok=True)
|
|
|
|
| 333 |
async def get_rag_response(user_message: str):
|
| 334 |
global chat_messages
|
| 335 |
Chunks = await retrieve_top_k_hybrid(user_message,15, 0.9, 0.1,bm25)
|
| 336 |
+
context = "======================================================================================================\n".join(map(str,Chunks))
|
| 337 |
chat_messages.append({"role": "user", "content": f'''
|
| 338 |
Context : {context}
|
| 339 |
User Query: {user_message}'''})
|