aidlmrza fjarsra commited on
Commit
c41aa74
·
verified ·
1 Parent(s): 61dcb91

Update app/services/llm_engine.py (#5)

Browse files

- Update app/services/llm_engine.py (1f86b8ab92c86a1a82f7b1b5494ec2ef8f173d67)


Co-authored-by: Fajar Syafatoni Raihannadif <fjarsra@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app/services/llm_engine.py +3 -3
app/services/llm_engine.py CHANGED
@@ -299,14 +299,14 @@ class LLMEngine:
299
  }
300
 
301
  try:
302
- # Gunakan model yang cepat
303
- completion = await self.client.chat.completions.create(
304
  messages=[system_msg],
305
  model="llama-3.1-8b-instant",
306
  temperature=0.7
307
  )
308
- return completion.choices[0].message.content
309
  except Exception as e:
310
  return f"Error generate progress: {str(e)}"
 
311
 
312
  llm_engine = LLMEngine()
 
299
  }
300
 
301
  try:
302
+ # PERBAIKAN: Gunakan _execute_with_retry, jangan self.client
303
+ return await self._execute_with_retry(
304
  messages=[system_msg],
305
  model="llama-3.1-8b-instant",
306
  temperature=0.7
307
  )
 
308
  except Exception as e:
309
  return f"Error generate progress: {str(e)}"
310
+
311
 
312
  llm_engine = LLMEngine()