Add getting a TEMPLATE from an environment variable

This commit is contained in:
HardAndHeavy 2024-12-02 01:24:43 +03:00
parent 33da359c6d
commit 3af8b1310f

View File

@ -62,7 +62,7 @@ class Pipeline:
),
"OPENAI_API_KEY": os.getenv("OPENAI_API_KEY", "YOUR_OPENAI_API_KEY"),
"TASK_MODEL": os.getenv("TASK_MODEL", "gpt-3.5-turbo"),
"TEMPLATE": """Use the following context as your learned knowledge, inside <context></context> XML tags.
"TEMPLATE": os.getenv("TEMPLATE", """Use the following context as your learned knowledge, inside <context></context> XML tags.
<context>
{{CONTEXT}}
</context>
@ -71,7 +71,7 @@ When answer to user:
- If you don't know, just say that you don't know.
- If you don't know when you are not sure, ask for clarification.
Avoid mentioning that you obtained the information from the context.
And answer according to the language of the user's question.""",
And answer according to the language of the user's question."""),
}
)