Skip to content

Commit 2b841f3

Browse files
committed
Fixed load. Comment cleanup.
1 parent 95e6a44 commit 2b841f3

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

repository_manager/utils.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -252,9 +252,11 @@ def load_skills_from_directory(directory: str) -> List[Skill]:
252252
return skills
253253

254254

255-
def get_http_client(ssl_verify: bool = True) -> httpx.AsyncClient | None:
255+
def get_http_client(
256+
ssl_verify: bool = True, timeout: float = 300.0
257+
) -> httpx.AsyncClient | None:
256258
if not ssl_verify:
257-
return httpx.AsyncClient(verify=False)
259+
return httpx.AsyncClient(verify=False, timeout=timeout)
258260
return None
259261

260262

@@ -264,6 +266,7 @@ def create_model(
264266
base_url: Optional[str],
265267
api_key: Optional[str],
266268
ssl_verify: bool = True,
269+
timeout: float = 300.0,
267270
):
268271
"""
269272
Create a Pydantic AI model with the specified provider and configuration.
@@ -280,7 +283,7 @@ def create_model(
280283
"""
281284
http_client = None
282285
if not ssl_verify:
283-
http_client = httpx.AsyncClient(verify=False)
286+
http_client = httpx.AsyncClient(verify=False, timeout=timeout)
284287

285288
if provider == "openai":
286289
target_base_url = base_url

0 commit comments

Comments
 (0)