dlflannery commited on
Commit
4c09467
·
verified ·
1 Parent(s): 1e70f16

Update app.py

Browse files

added o1-mini model

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -662,6 +662,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
662
  if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
663
  chatType = 'normal'
664
  deepseek = False
 
665
  prompt = prompt.strip()
666
  if prompt.lower().startswith('dsr1 '):
667
  deepseek = True
@@ -679,6 +680,10 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
679
  deepseek = True
680
  ds_model = 'deepseek-ai/DeepSeek-R1-Distill-Llama-70B'
681
  prompt = prompt[5:]
 
 
 
 
682
  elif prompt.lower().startswith('solve'):
683
  prompt = 'How do I solve ' + prompt[5:] + ' Do not use Latex for math expressions.'
684
  chatType = 'math'
@@ -726,7 +731,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
726
  if tokens > 40000:
727
  response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE RESTART CONVERSATION SOON."
728
  past.append({"role":"assistant", "content": final_text})
729
- if not deepseek:
730
  accessOk = False
731
  for i in range(3):
732
  try:
 
662
  if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
663
  chatType = 'normal'
664
  deepseek = False
665
+ reasoning = False
666
  prompt = prompt.strip()
667
  if prompt.lower().startswith('dsr1 '):
668
  deepseek = True
 
680
  deepseek = True
681
  ds_model = 'deepseek-ai/DeepSeek-R1-Distill-Llama-70B'
682
  prompt = prompt[5:]
683
+ elif prompt.lower().startswith('o1m '):
684
+ reasoning = True
685
+ gptModel = 'o1-mini'
686
+ prompt = prompt[4:] + '. Provide a detailed step-by-step description of your reasoning.'
687
  elif prompt.lower().startswith('solve'):
688
  prompt = 'How do I solve ' + prompt[5:] + ' Do not use Latex for math expressions.'
689
  chatType = 'math'
 
731
  if tokens > 40000:
732
  response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE RESTART CONVERSATION SOON."
733
  past.append({"role":"assistant", "content": final_text})
734
+ if not deepseek and not reasoning:
735
  accessOk = False
736
  for i in range(3):
737
  try: