nagarmayank commited on
Commit
cb8a8b6
·
1 Parent(s): dcb385e

Switch primary model in write_message method to ChatGroq with fallback to ChatGoogleGenerativeAI

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -143,11 +143,12 @@ def write_message(data: dict, header: str = Header()):
143
  message = data['message']
144
 
145
  try:
146
- model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, callbacks = [OpikTracer()])
147
  transaction_bot = Agent(model, system=prompt)
148
  transaction_bot.graph.invoke({"messages": [message]})
149
  except Exception as e: #fallback model
150
- model = ChatGroq(model=GROQ_MODEL, temperature=1, callbacks = [OpikTracer()])
 
151
  transaction_bot = Agent(model, system=prompt)
152
  transaction_bot.graph.invoke({"messages": [message]})
153
 
 
143
  message = data['message']
144
 
145
  try:
146
+ model = ChatGroq(model=GROQ_MODEL, temperature=1, callbacks = [OpikTracer()])
147
  transaction_bot = Agent(model, system=prompt)
148
  transaction_bot.graph.invoke({"messages": [message]})
149
  except Exception as e: #fallback model
150
+
151
+ model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, callbacks = [OpikTracer()])
152
  transaction_bot = Agent(model, system=prompt)
153
  transaction_bot.graph.invoke({"messages": [message]})
154