artificialguybr commited on
Commit
7ecd090
·
verified ·
1 Parent(s): 24155be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -2
app.py CHANGED
@@ -27,10 +27,25 @@ def user(message, history):
27
  history.append({"role": "user", "content": message})
28
  return "", history
29
 
30
- def call_api(history, max_tokens, temperature, top_p, seed=42):
31
  print("Calling API...")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  payload = {
33
- "messages": history,
34
  "temperature": temperature,
35
  "top_p": top_p,
36
  "max_tokens": max_tokens,
 
27
  history.append({"role": "user", "content": message})
28
  return "", history
29
 
30
+ def call_api(history, max_tokens, temperature, top_p, system_message="", seed=42):
31
  print("Calling API...")
32
+ messages = []
33
+
34
+ # Adicionando a mensagem de sistema, se presente
35
+ if system_message:
36
+ messages.append({"role": "system", "content": system_message})
37
+
38
+ # Adicionando mensagens de usuário e assistente da conversa
39
+ for msg in history:
40
+ messages.append({"role": msg["role"], "content": msg["content"]})
41
+
42
+ # Verificando se a última mensagem é do usuário, se não, ajuste conforme necessário
43
+ if not messages or messages[-1]["role"] != "user":
44
+ print("A última mensagem deve ser do usuário com conteúdo preenchido.")
45
+ return ""
46
+
47
  payload = {
48
+ "messages": messages,
49
  "temperature": temperature,
50
  "top_p": top_p,
51
  "max_tokens": max_tokens,