wellborgmann commited on
Commit
61f28aa
·
verified ·
1 Parent(s): d8bc9c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -23
app.py CHANGED
@@ -15,6 +15,7 @@ def respond(
15
  temperature,
16
  top_p,
17
  ):
 
18
  messages = [{"role": "system", "content": system_message}]
19
 
20
  for val in history:
@@ -25,34 +26,37 @@ def respond(
25
 
26
  messages.append({"role": "user", "content": message})
27
 
28
- response = ""
29
 
30
- try:
31
- for message in client.chat_completion(
32
- messages,
33
- max_tokens=max_tokens,
34
- stream=True,
35
- temperature=temperature,
36
- top_p=top_p,
37
- ):
38
- # Verifica se a resposta contém o conteúdo esperado
39
- if 'choices' not in message or len(message.choices) == 0 or 'delta' not in message.choices[0]:
40
- raise ValueError("Resposta inesperada do modelo.")
 
41
 
42
- token = message.choices[0].delta.content
 
43
 
44
- response += token
45
- yield response
46
 
47
- except ValueError as e:
48
- print(f"Erro de valor: {e}")
49
- except ConnectionError as e:
50
- print(f"Erro de conexão: {e}")
51
- except TimeoutError as e:
52
- print(f"Erro de tempo: {e}")
53
- except Exception as e:
54
- print(f"Erro inesperado: {e}")
55
 
 
56
 
57
 
58
  """
 
15
  temperature,
16
  top_p,
17
  ):
18
+ # Prepara as mensagens para a API
19
  messages = [{"role": "system", "content": system_message}]
20
 
21
  for val in history:
 
26
 
27
  messages.append({"role": "user", "content": message})
28
 
29
+ response = "" # Resposta acumulada
30
 
31
+ try:
32
+ # Chama a API de completamento com streaming
33
+ for message in client.chat_completion(
34
+ messages,
35
+ max_tokens=max_tokens,
36
+ stream=True,
37
+ temperature=temperature,
38
+ top_p=top_p,
39
+ ):
40
+ # Verifica se a resposta contém o conteúdo esperado
41
+ if 'choices' not in message or len(message.choices) == 0 or 'delta' not in message.choices[0]:
42
+ raise ValueError("Resposta inesperada do modelo.")
43
 
44
+ token = message.choices[0].delta.content
45
+ response += token # Acumula o conteúdo
46
 
47
+ # Retorna a resposta incrementalmente
48
+ yield response
49
 
50
+ except ValueError as e:
51
+ print(f"Erro de valor: {e}")
52
+ except ConnectionError as e:
53
+ print(f"Erro de conexão: {e}")
54
+ except TimeoutError as e:
55
+ print(f"Erro de tempo: {e}")
56
+ except Exception as e:
57
+ print(f"Erro inesperado: {e}")
58
 
59
+ return response # Retorna a resposta final ao final do processamento
60
 
61
 
62
  """