isayahc commited on
Commit
51a28d2
·
unverified ·
1 Parent(s): d6f22d3

making a chain to generate keywords

Browse files
rag_app/chains/__init__.py CHANGED
@@ -1,3 +1,4 @@
1
  from rag_app.chains.user_response_sentiment_chain import user_response_sentiment_prompt
2
  from rag_app.chains.generate_document_summary import generate_document_summary_prompt
3
- from rag_app.chains.query_rewritten_chain import query_rewritting_prompt
 
 
1
  from rag_app.chains.user_response_sentiment_chain import user_response_sentiment_prompt
2
  from rag_app.chains.generate_document_summary import generate_document_summary_prompt
3
+ from rag_app.chains.query_rewritten_chain import query_rewritting_prompt
4
+ from rag_app.chains.generate_keywords_chain import *
rag_app/chains/generate_keywords_chain.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_core.prompts import PromptTemplate
2
+
3
+
4
+ generate_keywords_template = """
5
+ You will be given meta data for a chunk text
6
+ =================
7
+ {chunk_metadata}
8
+ ====================
9
+
10
+ You will be tasked with creating keywords to help a llm better indentify the correct chunk
11
+ to use. Please only return the comma seperate values such that it can easily be parsed.
12
+
13
+
14
+ """
15
+
16
+ generate_keywords_prompt = PromptTemplate.from_template(generate_keywords_template)
17
+