update chat_completion
Browse files- appStore/rag.py +1 -1
appStore/rag.py
CHANGED
|
@@ -53,7 +53,7 @@ def run_query(context, label, model_sel_name):
|
|
| 53 |
client = InferenceClient(model_sel_name, token=hf_token)
|
| 54 |
|
| 55 |
# Instantiate ChatCompletion as a generator object (stream is set to True)
|
| 56 |
-
chat_completion = client.
|
| 57 |
messages=messages,
|
| 58 |
stream=True
|
| 59 |
)
|
|
|
|
| 53 |
client = InferenceClient(model_sel_name, token=hf_token)
|
| 54 |
|
| 55 |
# Instantiate ChatCompletion as a generator object (stream is set to True)
|
| 56 |
+
chat_completion = client.chat_completion(
|
| 57 |
messages=messages,
|
| 58 |
stream=True
|
| 59 |
)
|