anekameni commited on
Commit
e7ac6dd
·
1 Parent(s): 0bcd21d

Refactor ChatInterface to improve message history handling and update requirements for langchain packages

Browse files
Files changed (2) hide show
  1. app.py +3 -5
  2. 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