Spaces:
Runtime error
Runtime error
Update ChatEngine.py
Browse files- 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):
|