-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.agent
41 lines (36 loc) · 977 Bytes
/
config.agent
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
llm:
type: "OpenAI"
model: "gpt-3.5-turbo"
knowledgebase:
data_source: "./test_data/meteoric"
data_loader: "SimpleDirectoryReader"
data_transformer:
type: "CharacterTextSplitter"
chunk_overlap: 40
chunk_size: 1024
vector_store:
type: "Chroma"
embedding_function: "OpenAIEmbeddings"
agent:
type: "ChatAgent"
prompt_template: |
{{#user~}}
You will use this FORMAT only to answer user's QUERY
FORMAT: {{format}}
QUERY: {{input}}
Use the below knowledge to answer QUERY in given FORMAT:-
{{RETRIEVED_KNOWLEDGE}}
{{~/user}}
{{#assistant~}}
Yes, I will tell you about with that
{{~/assistant}}
{{#user~}}
Yes, tell me
{{~/user}}
{{#assistant~}}
{{gen 'response' temperature=0 max_tokens=300}}
{{~/assistant}}
input_variables:
knowledge_variable: "input"
extras: "format"
output_key: "response"