|
1 | | -from config.paths import get_flows_path |
2 | 1 | import asyncio |
| 2 | +import concurrent.futures |
3 | 3 | import os |
4 | 4 | import threading |
5 | | -import concurrent.futures |
6 | | -from utils.env_utils import get_env_int, get_env_float |
7 | 5 |
|
8 | 6 | import httpx |
9 | 7 | from agentd.patch import patch_openai_with_mcp |
10 | 8 | from dotenv import load_dotenv |
11 | 9 | from openai import AsyncOpenAI |
12 | 10 | from opensearchpy import AsyncOpenSearch |
13 | 11 | from opensearchpy._async.http_aiohttp import AIOHttpConnection |
14 | | -from config.embedding_constants import OPENAI_DEFAULT_EMBEDDING_MODEL |
15 | 12 |
|
16 | | -from utils.container_utils import get_container_host, determine_docling_host |
| 13 | +from config.embedding_constants import OPENAI_DEFAULT_EMBEDDING_MODEL |
| 14 | +from config.paths import get_flows_path |
| 15 | +from utils.container_utils import determine_docling_host, get_container_host |
17 | 16 | from utils.embedding_fields import build_knn_vector_field |
| 17 | +from utils.env_utils import get_env_float, get_env_int |
18 | 18 | from utils.logging_config import get_logger |
| 19 | + |
19 | 20 | # Import configuration manager |
20 | 21 | from .config_manager import config_manager |
21 | 22 |
|
@@ -549,7 +550,7 @@ async def probe_http2(): |
549 | 550 | ) |
550 | 551 | logger.info(f"HTTP/2 probe successful with {model_name}") |
551 | 552 | return True |
552 | | - except (asyncio.TimeoutError, Exception) as probe_error: |
| 553 | + except (TimeoutError, Exception) as probe_error: |
553 | 554 | logger.warning(f"HTTP/2 probe failed with {model_name}, falling back to HTTP/1.1", error=str(probe_error)) |
554 | 555 | return False |
555 | 556 | finally: |
|
0 commit comments