ydeng/autoBIGS.engine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added optional parameter to change timeout
All checks were successful
autoBIGS.engine/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Harrison Deng 2025-03-14 16:09:01 +00:00
parent f8d92a4aad
commit 14d760a266
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/autobigs/engine/analysis/bigsdb.py
View File

@ -43,12 +43,13 @@ class BIGSdbMLSTProfiler(AbstractAsyncContextManager):
class RemoteBIGSdbMLSTProfiler(BIGSdbMLSTProfiler): class RemoteBIGSdbMLSTProfiler(BIGSdbMLSTProfiler):
def __init__(self, database_api: str, database_name: str, scheme_id: int, retry_requests: int = 5): def __init__(self, database_api: str, database_name: str, scheme_id: int, retry_requests: int = 5, timeout: int = 600):
self._timeout = timeout
self._retry_limit = retry_requests self._retry_limit = retry_requests
self._database_name = database_name self._database_name = database_name
self._scheme_id = scheme_id self._scheme_id = scheme_id
self._base_url = f"{database_api}/db/{self._database_name}/schemes/{self._scheme_id}/" self._base_url = f"{database_api}/db/{self._database_name}/schemes/{self._scheme_id}/"
self._http_client = ClientSession(self._base_url, timeout=ClientTimeout(300)) self._http_client = ClientSession(self._base_url, timeout=ClientTimeout(timeout))
async def __aenter__(self): async def __aenter__(self):
return self return self