alihan/KG_RAG
1
0
Fork 0
mirror of https://github.com/BaranziniLab/KG_RAG.git synced 2024-06-08 14:12:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

changed the system prmpts

Browse Source
This commit is contained in:
Karthik Soman
2024-03-13 15:52:28 -07:00
parent c1d8be6cbf
commit 35bc915602
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kg_rag/rag_based_generation/GPT/drug_repurposing_v2.py
View File

@@ -51,7 +51,7 @@ Question:
def main():
if not INTERACTIVE:
print("Retrieving context from SPOKE graph...")
context = retrieve_context(question_template, vectorstore, embedding_function_for_context_retrieval, node_context_df, CONTEXT_VOLUME, QUESTION_VS_CONTEXT_SIMILARITY_PERCENTILE_THRESHOLD, QUESTION_VS_CONTEXT_MINIMUM_SIMILARITY, EDGE_EVIDENCE)
context = retrieve_context(question, vectorstore, embedding_function_for_context_retrieval, node_context_df, CONTEXT_VOLUME, QUESTION_VS_CONTEXT_SIMILARITY_PERCENTILE_THRESHOLD, QUESTION_VS_CONTEXT_MINIMUM_SIMILARITY, EDGE_EVIDENCE)
print("Here is the KG-RAG based answer:")
print("")
enriched_prompt = "Context: "+ context + "\n" + "Question: " + question