Spaces:
Running
Running
anekameni
commited on
Commit
·
e7ac6dd
1
Parent(s):
0bcd21d
Refactor ChatInterface to improve message history handling and update requirements for langchain packages
Browse files- app.py +3 -5
- requirements.txt +2 -2
app.py
CHANGED
@@ -8,10 +8,12 @@ os.environ["TOKENIZERS_PARALLELISM"]="true"
|
|
8 |
class ChatInterface:
|
9 |
def __init__(self, rag_system: RAGSystem):
|
10 |
self.rag_system = rag_system
|
11 |
-
self.chat_history: List[Dict] = []
|
12 |
|
13 |
def respond(self, message: str, history: List[List[str]]):
|
14 |
result = ""
|
|
|
|
|
|
|
15 |
for text in self.rag_system.query(message, history):
|
16 |
result += text
|
17 |
yield result
|
@@ -23,10 +25,6 @@ class ChatInterface:
|
|
23 |
type="messages",
|
24 |
title="Medivocate",
|
25 |
description="Medivocate est une application qui offre des informations claires et structurées sur l'histoire de l'Afrique et sa médecine traditionnelle, en s'appuyant exclusivement sur un contexte issu de documentaires sur l'histoire du continent africain.",
|
26 |
-
# retry_btn=None,
|
27 |
-
# undo_btn=None,
|
28 |
-
# clear_btn="Clear",
|
29 |
-
# chatbot=gr.Chatbot(show_copy_button=True),
|
30 |
)
|
31 |
return chat_interface
|
32 |
|
|
|
8 |
class ChatInterface:
|
9 |
def __init__(self, rag_system: RAGSystem):
|
10 |
self.rag_system = rag_system
|
|
|
11 |
|
12 |
def respond(self, message: str, history: List[List[str]]):
|
13 |
result = ""
|
14 |
+
history = [
|
15 |
+
(turn["role"], turn["content"]) for turn in history
|
16 |
+
]
|
17 |
for text in self.rag_system.query(message, history):
|
18 |
result += text
|
19 |
yield result
|
|
|
25 |
type="messages",
|
26 |
title="Medivocate",
|
27 |
description="Medivocate est une application qui offre des informations claires et structurées sur l'histoire de l'Afrique et sa médecine traditionnelle, en s'appuyant exclusivement sur un contexte issu de documentaires sur l'histoire du continent africain.",
|
|
|
|
|
|
|
|
|
28 |
)
|
29 |
return chat_interface
|
30 |
|
requirements.txt
CHANGED
@@ -2,8 +2,8 @@ langchain-ollama==0.2.2
|
|
2 |
langchain-groq==0.2.2
|
3 |
langchain-community==0.3.14
|
4 |
langchain-chroma==0.1.4
|
5 |
-
langchain
|
6 |
-
langchain-huggingface
|
7 |
ollama==0.4.5
|
8 |
chromadb==0.5.23
|
9 |
tqdm==4.67.1
|
|
|
2 |
langchain-groq==0.2.2
|
3 |
langchain-community==0.3.14
|
4 |
langchain-chroma==0.1.4
|
5 |
+
langchain==0.3.14
|
6 |
+
langchain-huggingface==0.1.2
|
7 |
ollama==0.4.5
|
8 |
chromadb==0.5.23
|
9 |
tqdm==4.67.1
|