fix accidentally using httpxTimeout

This commit is contained in:
Nexus 2024-04-14 18:42:53 +01:00
parent 38bc54d37f
commit 3a3804fc18
Signed by: nex
GPG key ID: 0FA334385D0B689F

View file

@ -205,7 +205,7 @@ class OllamaClient:
timeout = 10800 timeout = 10800
timeout = aiohttp.ClientTimeout(timeout) timeout = aiohttp.ClientTimeout(timeout)
else: else:
timeout = timeout or httpx.Timeout(60) timeout = timeout or aiohttp.ClientTimeout(120)
return aiohttp.ClientSession(self.base_url, timeout=timeout, auth=self.authorisation) return aiohttp.ClientSession(self.base_url, timeout=timeout, auth=self.authorisation)
async def get_tags(self) -> dict[typing.Literal["models"], dict[str, str, int, dict[str, str, None]]]: async def get_tags(self) -> dict[typing.Literal["models"], dict[str, str, int, dict[str, str, None]]]: