Spaces:
Running
Running
luanpoppe
commited on
Commit
·
4cd3056
1
Parent(s):
4dcf767
fix: system prompt
Browse files- 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:
|