feat: openai socks proxy support

This commit is contained in:
2026-07-17 12:33:14 +07:00
parent 3fe870024d
commit f81b0e3584
2 changed files with 5 additions and 1 deletions

View File

@@ -16,12 +16,14 @@ class OfferReplyGenerator:
model: str,
system_prompt_file: Path,
timeout_seconds: int,
proxy_url: str | None = None,
) -> None:
self._api_base_url = api_base_url.rstrip("/")
self._api_key = api_key
self._model = model
self._system_prompt = system_prompt_file.read_text(encoding="utf-8").strip()
self._timeout_seconds = timeout_seconds
self._proxies = {"http": proxy_url, "https": proxy_url} if proxy_url else None
async def generate_reply(self, offer: Offer) -> str:
return await asyncio.to_thread(self._generate_reply_sync, offer)
@@ -41,6 +43,7 @@ class OfferReplyGenerator:
],
},
timeout=self._timeout_seconds,
proxies=self._proxies,
)
response.raise_for_status()
payload = response.json()

View File

@@ -136,6 +136,7 @@ async def main() -> None:
model=settings.openai_model,
system_prompt_file=settings.openai_system_prompt_file,
timeout_seconds=settings.requests_timeout_seconds,
proxy_url=settings.telegram_proxy_url,
)
dispatcher = build_dispatcher(reply_generator, published_offers_store)