|
220 | 220 | "print(f\"User: {user_question}\")\n", |
221 | 221 | "print(\"Assistant: \\n\")\n", |
222 | 222 | "\n", |
223 | | - "for chunk in stream_chat_completion(ELASTICSEARCH_HOST, INFERENCE_ENDPOINT_NAME, messages):\n", |
| 223 | + "for chunk in stream_chat_completion(\n", |
| 224 | + " ELASTICSEARCH_HOST, INFERENCE_ENDPOINT_NAME, messages\n", |
| 225 | + "):\n", |
224 | 226 | " print(chunk, end=\"\", flush=True)" |
225 | 227 | ] |
226 | 228 | }, |
|
397 | 399 | " }\n", |
398 | 400 | "\n", |
399 | 401 | " try:\n", |
400 | | - " response = requests.post(search_url, json=search_body, headers=ELASTICSEARCH_HEADERS)\n", |
| 402 | + " response = requests.post(\n", |
| 403 | + " search_url, json=search_body, headers=ELASTICSEARCH_HEADERS\n", |
| 404 | + " )\n", |
401 | 405 | " response.raise_for_status()\n", |
402 | 406 | "\n", |
403 | 407 | " results = response.json()\n", |
|
468 | 472 | " ]\n", |
469 | 473 | "\n", |
470 | 474 | " full_response = \"\"\n", |
471 | | - " for chunk in stream_chat_completion(ELASTICSEARCH_HOST, INFERENCE_ENDPOINT_NAME, messages):\n", |
| 475 | + " for chunk in stream_chat_completion(\n", |
| 476 | + " ELASTICSEARCH_HOST, INFERENCE_ENDPOINT_NAME, messages\n", |
| 477 | + " ):\n", |
472 | 478 | " print(chunk, end=\"\", flush=True)\n", |
473 | 479 | " full_response += chunk\n", |
474 | 480 | "\n", |
|
0 commit comments