IliaLarchenko commited on
Commit
fd17307
1 Parent(s): ec74f30

Improved LLM check

Browse files
Files changed (1) hide show
  1. llm.py +3 -1
llm.py CHANGED
@@ -19,7 +19,9 @@ def test_connection():
19
  response = client_LLM.chat.completions.create(
20
  model=LLM_NAME,
21
  messages=[
22
- {"role": "system", "content": "Test connection"},
 
 
23
  ],
24
  )
25
  return response.choices[0].message.content.strip()
 
19
  response = client_LLM.chat.completions.create(
20
  model=LLM_NAME,
21
  messages=[
22
+ {"role": "system", "content": "You just help me test the connection."},
23
+ {"role": "user", "content": "Hi!"},
24
+ {"role": "user", "content": "Ping!"},
25
  ],
26
  )
27
  return response.choices[0].message.content.strip()