cappuch commited on
Commit
1832559
1 Parent(s): 2bcefc7
Files changed (1) hide show
  1. app.py +7 -15
app.py CHANGED
@@ -7,23 +7,15 @@ client = Groq(
7
  api_key = os.environ.get("Groq_Api_Key")
8
  )
9
 
 
 
 
 
10
 
11
  def generate_response(prompt, history, model, temperature, max_tokens, top_p, seed):
12
- messages = []
13
- for i, data in enumerate(history):
14
- if i % 2 == 0:
15
- role = 'user'
16
- else:
17
- role = 'assistant'
18
- message = {}
19
- message["role"] = role
20
- message["content"] = data
21
- messages.append(message)
22
-
23
- message = {}
24
- message["role"] = "user"
25
- message["content"] = prompt
26
- messages.append(message)
27
 
28
  if seed == 0:
29
  seed = random.randint(1, 100000)
 
7
  api_key = os.environ.get("Groq_Api_Key")
8
  )
9
 
10
+ def create_history_messages(history):
11
+ history_messages = [{"role": "user", "content": m[0]} for m in history]
12
+ history_messages.extend([{"role": "assistant", "content": m[1]} for m in history])
13
+ return history_messages
14
 
15
  def generate_response(prompt, history, model, temperature, max_tokens, top_p, seed):
16
+ messages = create_history_messages(history)
17
+ messages.append({"role": "user", "content": prompt})
18
+ print(messages)
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  if seed == 0:
21
  seed = random.randint(1, 100000)