luanpoppe commited on
Commit
2fa8c74
·
1 Parent(s): d8142ab
Files changed (1) hide show
  1. endpoint_teste/views.py +2 -2
endpoint_teste/views.py CHANGED
@@ -84,9 +84,9 @@ def getPDF(request):
84
  print('temp_file_path: ', temp_file_path)
85
 
86
  resposta_llm = None
87
- if serializer.validated_data['model']:
88
  resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], temp_file_path, model=serializer.validated_data['model'])
89
- else:
90
  resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], temp_file_path)
91
 
92
  os.remove(temp_file_path)
 
84
  print('temp_file_path: ', temp_file_path)
85
 
86
  resposta_llm = None
87
+ try:
88
  resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], temp_file_path, model=serializer.validated_data['model'])
89
+ except:
90
  resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], temp_file_path)
91
 
92
  os.remove(temp_file_path)