Demosthene-OR commited on
Commit
daf791e
1 Parent(s): c217630

Update chatbot_tab.py

Browse files
Files changed (1) hide show
  1. tabs/chatbot_tab.py +3 -1
tabs/chatbot_tab.py CHANGED
@@ -13,6 +13,7 @@ import numpy as np
13
  from translate_app import tr
14
  import getpass
15
  from langchain_mistralai import ChatMistralAI
 
16
  from langgraph.checkpoint.memory import MemorySaver
17
  from langgraph.graph import START, END, MessagesState, StateGraph
18
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
@@ -42,6 +43,7 @@ os.getenv("MISTRAL_API_KEY")
42
 
43
 
44
  model = ChatMistralAI(model="mistral-large-latest")
 
45
 
46
 
47
  language = "French"
@@ -118,7 +120,7 @@ et mon équipe de vente n'est pas performante.
118
 
119
  Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
120
  """
121
- ai_message1 = "J'ai bien compris, je suis un Directeur Commercial prospecté et je réponds seulement à tes questions. Je réponds à une seule question à la fois"
122
 
123
  context = st.text_area(label=tr("Contexte:"), value=context)
124
  human_message1 = st.text_area(label=tr("Consigne"), value=human_message1,height=300)
 
13
  from translate_app import tr
14
  import getpass
15
  from langchain_mistralai import ChatMistralAI
16
+ from langchain_openai import ChatOpenAI
17
  from langgraph.checkpoint.memory import MemorySaver
18
  from langgraph.graph import START, END, MessagesState, StateGraph
19
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 
43
 
44
 
45
  model = ChatMistralAI(model="mistral-large-latest")
46
+ # model = ChatOpenAI(model="gpt-3.5-turbo")
47
 
48
 
49
  language = "French"
 
120
 
121
  Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
122
  """
123
+ ai_message1 = "J'ai bien compris, je suis un Directeur Commercial prospecté et je réponds seulement à tes questions. Je réponds à une seule question à la fois, sans commencer mes réponses par 'En tant que Directeur Commercial'"
124
 
125
  context = st.text_area(label=tr("Contexte:"), value=context)
126
  human_message1 = st.text_area(label=tr("Consigne"), value=human_message1,height=300)