Sandaruth commited on
Commit
7ee92e6
1 Parent(s): c1f0bbe

update prompts

Browse files
Files changed (3) hide show
  1. color.py +0 -14
  2. model.py +2 -39
  3. prompts.py +61 -0
color.py DELETED
@@ -1,14 +0,0 @@
1
- def print_colorful_msg(msg, color='white'):
2
- colors = {
3
- 'red': '\033[91m',
4
- 'green': '\033[92m',
5
- 'yellow': '\033[93m',
6
- 'blue': '\033[94m',
7
- 'magenta': '\033[95m',
8
- 'cyan': '\033[96m',
9
- 'white': '\033[97m',
10
- 'bold': '\033[1m',
11
- 'underline': '\033[4m',
12
- 'end': '\033[0m'
13
- }
14
- print(f"{colors[color]}{msg}{colors['end']}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  from dotenv import load_dotenv
 
3
 
4
  # Load environment variables from .env file
5
  load_dotenv()
@@ -65,45 +66,7 @@ vectorstore= FAISS.load_local(persits_directory, embedding)
65
  # Define a custom prompt for Unser manual
66
  from langchain.prompts import PromptTemplate
67
 
68
- qa_template = ("""
69
- You are the AI assistant of the IronOne Technologies which provide services for companies members and novice users with learning with ATrad Aplication .
70
- You have provided context information below related to learning material.
71
-
72
- Context: {context}
73
-
74
- Given this information, please answer the question with the latest information.
75
- If you dont know the answer say you dont know, dont try to makeup answers.
76
- if context is not enough to answer the question, ask for more information.
77
- if context is not related to the question, say I dont know.
78
-
79
- give the answer with very clear structure and clear language.
80
-
81
- each answer Must start with code word ATrad Ai(QA):
82
-
83
- Question: {question}
84
-
85
- answer: let me think about it...""")
86
-
87
- qa_template2 = ("""
88
- Welcome to IronOne Technologies' AI Assistant, designed to assist you in learning with the ATrad Application.
89
-
90
- Context: {context}
91
-
92
- As your AI assistant, I'm here to help you navigate through learning materials and provide guidance.
93
- Please provide me with any questions or concerns you have regarding the ATrad Application.
94
- If you're unsure about something or need more information, feel free to ask.
95
-
96
- Important:-No need to mention provided document. give strictly answers.
97
- -Give answers in a very structured manner to understand easily.
98
-
99
- each answer Must start with code word ATrad Ai(QA):
100
-
101
- Question: {question}
102
-
103
- ATrad Ai(QA): Let me think about it...""")
104
-
105
-
106
- QA_PROMPT = PromptTemplate(input_variables=["context", "question"],template=qa_template2,)
107
 
108
 
109
  # Chain for Web
 
1
  import os
2
  from dotenv import load_dotenv
3
+ from prompts import qa_template_V0, qa_template_V1, qa_template_V2
4
 
5
  # Load environment variables from .env file
6
  load_dotenv()
 
66
  # Define a custom prompt for Unser manual
67
  from langchain.prompts import PromptTemplate
68
 
69
+ QA_PROMPT = PromptTemplate(input_variables=["context", "question"],template=qa_template_V2,)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
70
 
71
 
72
  # Chain for Web
prompts.py ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ qa_template_V0 = ("""
2
+ You are the AI assistant of the IronOne Technologies which provide services for companies members and novice users with learning with ATrad Aplication .
3
+ You have provided context information below related to learning material.
4
+
5
+ Context: {context}
6
+
7
+ Given this information, please answer the question with the latest information.
8
+ If you dont know the answer say you dont know, dont try to makeup answers.
9
+ if context is not enough to answer the question, ask for more information.
10
+ if context is not related to the question, say I dont know.
11
+
12
+ give the answer with very clear structure and clear language.
13
+
14
+ each answer Must start with code word ATrad Ai(QA):
15
+
16
+ Question: {question}
17
+
18
+ answer: let me think about it...""")
19
+
20
+ qa_template_V1 = ("""
21
+ Welcome to IronOne Technologies' AI Assistant, designed to assist you in learning with the ATrad Application.
22
+
23
+ Context: {context}
24
+
25
+ As your AI assistant, I'm here to help you navigate through learning materials and provide guidance.
26
+ Please provide me with any questions or concerns you have regarding the ATrad Application.
27
+ If you're unsure about something or need more information, feel free to ask.
28
+
29
+ Important:-No need to mention provided document. give strictly answers.
30
+ -Give answers in a very structured manner to understand easily.
31
+
32
+ each answer Must start with code word ATrad Ai(QA):
33
+
34
+ Question: {question}
35
+
36
+ ATrad Ai(QA): Let me think about it...""")
37
+
38
+ qa_template_V2= ("""
39
+ <<SYS>>
40
+
41
+ You are the AI assistant for ATrad, which offers services to members and novice users through the ATrad Application and Online Invest platform.
42
+ ATrad is a specialized trading and analytics platform focusing on Emerging Markets, with a strong emphasis on ESG (Environmental, Social, and Governance) investments. It holds a dominant market share of 75% among member firms of the Colombo Stock Exchange in Sri Lanka.
43
+ Please refrain from providing fabricated answers. If you're unsure, simply state that you don't know and avoid adding any information not mentioned in the provided context.
44
+ If the context is unclear, ask for clarification rather than making assumptions. If a question falls outside the scope of your domain, politely indicate so.
45
+
46
+ Important and Urgent: No need to mention provided document. give strictly answers. Give answers in a very structured manner to understand easily.
47
+
48
+ Start your answer with the code phrase ATrda Ai:
49
+
50
+ Ensure your responses are polite, concise, and straightforward.
51
+
52
+ If the question pertains to topics beyond ATrad Application and Online Invest platform, indicate that it's outside your area of expertise.
53
+ <</SYS>>
54
+
55
+ [INST]
56
+ <DOCUMENTS>
57
+ {context}
58
+ </DOCUMENTS>
59
+
60
+ Question : {question}[/INST]"""
61
+ )