Shamik commited on
Commit
11d325c
·
verified ·
1 Parent(s): 70579ee

fix: reverting back to the original prompt for the VLM.

Browse files
src/insurance_assistants/agents.py CHANGED
@@ -16,7 +16,7 @@ from smolagents import (
16
  )
17
 
18
  from src.insurance_assistants.complex_rag import RAG
19
- from src.insurance_assistants.consts import PROMPT_PREFIX, PROMPT
20
 
21
  # _ = load_dotenv(dotenv_path=find_dotenv())
22
  rag_app = RAG()
@@ -45,11 +45,7 @@ class InsuranceInfoRetriever(Tool):
45
  chat_template = [
46
  {
47
  "role": "system",
48
- "content": PROMPT
49
- # """You find answers from the relevant documents. Answer only
50
- # from these documents. If answer isn't available return 'Question cannot be answered based
51
- # on the documents provided.' """
52
- ,
53
  },
54
  {
55
  "role": "user",
 
16
  )
17
 
18
  from src.insurance_assistants.complex_rag import RAG
19
+ from src.insurance_assistants.consts import PROMPT_PREFIX, VLM_PROMPT
20
 
21
  # _ = load_dotenv(dotenv_path=find_dotenv())
22
  rag_app = RAG()
 
45
  chat_template = [
46
  {
47
  "role": "system",
48
+ "content": VLM_PROMPT,
 
 
 
 
49
  },
50
  {
51
  "role": "user",
src/insurance_assistants/consts.py CHANGED
@@ -11,6 +11,8 @@ You can answer using information contained in plots and figures if necessary.
11
  Answer in the same language as the query.
12
 
13
  """
 
 
14
 
15
  PRIMARY_HEADING = """# Multi-agent Health Insurance Policy Expert
16
  #### This agent answers health insurance related questions from pre-ingested set \
 
11
  Answer in the same language as the query.
12
 
13
  """
14
+ VLM_PROMPT = """
15
+ You find answers from the relevant documents. Answer only from these documents. If answer isn't available return 'Question cannot be answered based on the documents provided.' """
16
 
17
  PRIMARY_HEADING = """# Multi-agent Health Insurance Policy Expert
18
  #### This agent answers health insurance related questions from pre-ingested set \