smart_eco_footprint / api /llms /llm_config.py
thanhtung09t2's picture
Upload llms settings
0859f1c verified
raw
history blame
914 Bytes
class LLMConfig:
def __init__(self,
system_prompt,
query_context_template,
chat_condense_template,
chat_context_template,
**kwargs
):
"""
system_prompt: act as llm's system instructions
query_context_template: used for QueryEngine - single question-answer pairs without chat history
chat_condense_template: used for ChatEngine - 'condense' & 'condense_plus_context' mode
chat_context_template: used for ChatEngine - 'context' mode
**kwargs: additional keyword arguments for other configurations
"""
self.system_prompt = system_prompt
self.query_context_template = query_context_template
self.chat_condense_template = chat_condense_template
self.chat_context_template = chat_context_template
self.kwargs = kwargs