caarleexx commited on
Commit
d24bb0a
·
verified ·
1 Parent(s): 757d621

Update llm/llm_manager.py

Browse files
Files changed (1) hide show
  1. llm/llm_manager.py +9 -1
llm/llm_manager.py CHANGED
@@ -161,9 +161,17 @@ class LLMManager:
161
  else:
162
  raise ValueError(f"Client {type(client)} não suporta geração")
163
 
 
 
 
 
 
 
 
 
164
  return {
165
  'status': 'success',
166
- 'result': response,
167
  'mode': 'immediate'
168
  }
169
 
 
161
  else:
162
  raise ValueError(f"Client {type(client)} não suporta geração")
163
 
164
+ # Extrair conteúdo do GroqResponse
165
+ if hasattr(response, 'content'):
166
+ result_content = response.content
167
+ elif isinstance(response, dict):
168
+ result_content = response.get('content', response)
169
+ else:
170
+ result_content = str(response)
171
+
172
  return {
173
  'status': 'success',
174
+ 'result': result_content,
175
  'mode': 'immediate'
176
  }
177