heymenn commited on
Commit
22aa53f
·
verified ·
1 Parent(s): af2f6b5

Update src/services/utils.py

Browse files
Files changed (1) hide show
  1. src/services/utils.py +2 -2
src/services/utils.py CHANGED
@@ -7,12 +7,12 @@ from nltk.stem import *
7
  nltk.download("punkt_tab")
8
 
9
 
10
- def set_prompt(InputData):
11
  prompt = """Task : Find all the constraints in this technical problem making sure each are premised on the problem only.
12
  Take into account different technical domains to encompass the whole problem.
13
  Output each constraints in a json such as : ({"title of the constraints1":"description1","title of the constraintsN":"descriptionN"})
14
  Technical problem :
15
- """ + InputData['problem']
16
  return prompt
17
 
18
  def load_technologies():
 
7
  nltk.download("punkt_tab")
8
 
9
 
10
+ def set_prompt(problem):
11
  prompt = """Task : Find all the constraints in this technical problem making sure each are premised on the problem only.
12
  Take into account different technical domains to encompass the whole problem.
13
  Output each constraints in a json such as : ({"title of the constraints1":"description1","title of the constraintsN":"descriptionN"})
14
  Technical problem :
15
+ """ + problem
16
  return prompt
17
 
18
  def load_technologies():