luanpoppe commited on
Commit
4cd3056
·
1 Parent(s): 4dcf767

fix: system prompt

Browse files
Files changed (1) hide show
  1. resumos/views.py +3 -1
resumos/views.py CHANGED
@@ -1,6 +1,8 @@
1
  from rest_framework.views import APIView
2
  import tempfile, os
3
  from rest_framework.response import Response
 
 
4
  from .serializer import ResumoPDFSerializer
5
  from langchain_backend.main import get_llm_answer_summary
6
  from setup.environment import default_model
@@ -32,7 +34,7 @@ class ResumoView(APIView):
32
  listaPDFs.append(temp_file_path)
33
  # print('listaPDFs: ', listaPDFs)
34
 
35
- system_prompt = data.get("system_prompt", )
36
  resposta_llm = get_llm_answer_summary(system_prompt, data["user_message"], listaPDFs, model=model, isIterativeRefinement=data["iterative_refinement"])
37
 
38
  for file in listaPDFs:
 
1
  from rest_framework.views import APIView
2
  import tempfile, os
3
  from rest_framework.response import Response
4
+
5
+ from langchain_backend.utils import DEFAULT_SYSTEM_PROMPT
6
  from .serializer import ResumoPDFSerializer
7
  from langchain_backend.main import get_llm_answer_summary
8
  from setup.environment import default_model
 
34
  listaPDFs.append(temp_file_path)
35
  # print('listaPDFs: ', listaPDFs)
36
 
37
+ system_prompt = data.get("system_prompt", DEFAULT_SYSTEM_PROMPT)
38
  resposta_llm = get_llm_answer_summary(system_prompt, data["user_message"], listaPDFs, model=model, isIterativeRefinement=data["iterative_refinement"])
39
 
40
  for file in listaPDFs: