Spaces:
Sleeping
Sleeping
Commit
·
fd17307
1
Parent(s):
ec74f30
Improved LLM check
Browse files
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": "
|
|
|
|
|
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()
|