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

update utils

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -15,7 +15,7 @@ def retrieve_knowledge(query, vectorstore, k=10, randomize=True):
15
  return knowledge
16
 
17
 
18
- def generate_workout(system_prompt, query, knowledge):
19
  messages = [
20
  SystemMessage(content=system_prompt.format(workout_context=knowledge)),
21
  HumanMessage(content=query)
 
15
  return knowledge
16
 
17
 
18
+ def generate_workout(system_prompt, query, knowledge, llm):
19
  messages = [
20
  SystemMessage(content=system_prompt.format(workout_context=knowledge)),
21
  HumanMessage(content=query)