wessamelden commited on
Commit
f2c87c6
·
1 Parent(s): 780ba71

add API key protection

Browse files
Files changed (1) hide show
  1. app/retrieval/reranker.py +1 -1
app/retrieval/reranker.py CHANGED
@@ -166,7 +166,7 @@ async def warmup_reranker() -> bool:
166
  try:
167
  await _client.chat.completions.create(
168
  model=settings.groq_model,
169
- messages=[{"role": "user", "content": 'Return {"test": 1}'}],
170
  temperature=0.0,
171
  max_tokens=10,
172
  response_format={"type": "json_object"}
 
166
  try:
167
  await _client.chat.completions.create(
168
  model=settings.groq_model,
169
+ messages=[{"role": "user", "content": 'Return {"test": 1} in JSON format'}],
170
  temperature=0.0,
171
  max_tokens=10,
172
  response_format={"type": "json_object"}