Chan-Y commited on
Commit
55c526f
1 Parent(s): 0c8e371

Update ChatEngine.py

Browse files
Files changed (1) hide show
  1. ChatEngine.py +3 -2
ChatEngine.py CHANGED
@@ -1,4 +1,3 @@
1
- from llama_index.llms.huggingface import HuggingFaceLLM
2
  from llama_index.core.base.llms.types import ChatMessage, MessageRole
3
 
4
  class ChatEngine:
@@ -8,10 +7,12 @@ class ChatEngine:
8
 
9
  Args:
10
  retriever (HybridRetriever): An instance of a retriever to fetch relevant documents.
 
 
 
11
  """
12
 
13
  self.retriever = retriever
14
-
15
  self.chat_history = []
16
 
17
  def ask_question(self, question, llm):
 
 
1
  from llama_index.core.base.llms.types import ChatMessage, MessageRole
2
 
3
  class ChatEngine:
 
7
 
8
  Args:
9
  retriever (HybridRetriever): An instance of a retriever to fetch relevant documents.
10
+ model_name (str): The name of the language model to be used.
11
+ context_window (int, optional): The maximum context window size for the language model. Defaults to 32000.
12
+ temperature (float, optional): The temperature setting for the language model. Defaults to 0.
13
  """
14
 
15
  self.retriever = retriever
 
16
  self.chat_history = []
17
 
18
  def ask_question(self, question, llm):