matteopilotto commited on
Commit
56cba7c
·
1 Parent(s): 5ca8659

update utils

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -26,10 +26,10 @@ def generate_workout(system_prompt, query, knowledge):
26
  return response
27
 
28
 
29
- def run(gender, muscle_group, equipment, level, duration, vectorstore, system_prompt, k=5, randomize=True):
30
  query = f"{duration}-minute {muscle_group} workout for {gender} {level} level {equipment}"
31
  knowledge = retrieve_knowledge(query, vectorstore, k, randomize)
32
- response = generate_workout(system_prompt, query, knowledge)
33
 
34
  for i in range(len(response)):
35
  time.sleep(0.01)
 
26
  return response
27
 
28
 
29
+ def run(gender, muscle_group, equipment, level, duration, vectorstore, system_prompt, llm, k=5, randomize=True):
30
  query = f"{duration}-minute {muscle_group} workout for {gender} {level} level {equipment}"
31
  knowledge = retrieve_knowledge(query, vectorstore, k, randomize)
32
+ response = generate_workout(system_prompt, query, knowledge, llm)
33
 
34
  for i in range(len(response)):
35
  time.sleep(0.01)