compass.services.threaded.JurisdictionUpdater#
- class JurisdictionUpdater(jurisdiction_fp, tpe_kwargs=None)[source]#
Bases:
ThreadedService
Service that updates jurisdiction info into a file
- Parameters:
jurisdiction_fp (path-like) – Path to JSON file where jurisdictions should be tracked.
tpe_kwargs (
dict
, optional) – Keyword-value argument pairs to pass toconcurrent.futures.ThreadPoolExecutor
. By default,None
.
Methods
Open thread pool and temp directory
call
(*args, **kwargs)Call the service
process
(county, doc, seconds_elapsed[, ...])Add usage from tracker to file
process_using_futures
(fut, *args, **kwargs)Process a call to the service
Shutdown thread pool and cleanup temp directory
Attributes
Max number of concurrent job submissions.
True
if file not currently being written toService name used to pull the correct queue object
- async process(county, doc, seconds_elapsed, usage_tracker=None)[source]#
Add usage from tracker to file
Any existing usage info in the file will remain unchanged EXCEPT for anything under the label of the input tracker, all of which will be replaced with info from the tracker itself.
- Parameters:
county (
compass.utilities.location.Location
) – County to record.doc (
elm.web.document.Document
) – Document containing meta information about the jurisdiction. Must have relevant processing keys in theattrs
dict, otherwise the jurisdiction may not be recorded properly. IfNone
, the jurisdiction is assumed not to have been found.seconds_elapsed (
int
orfloat
) – Total number of seconds it took to look for (and possibly parse) this document.usage_tracker (
compass.services.usage.UsageTracker
, optional) – Optional tracker instance to monitor token usage during LLM calls. By default,None
.
- MAX_CONCURRENT_JOBS = 10000#
Max number of concurrent job submissions.
- acquire_resources()#
Open thread pool and temp directory
- async classmethod call(*args, **kwargs)#
Call the service
- Parameters:
*args, **kwargs – Positional and keyword arguments to be passed to the underlying service processing function.
- Returns:
obj
– A response object from the underlying service.
- async process_using_futures(fut, *args, **kwargs)#
Process a call to the service
- Parameters:
fut (
asyncio.Future
) – A future object that should get the result of the processing operation. If the processing function returnsanswer
, this method should callfut.set_result(answer)
.**kwargs – Keyword arguments to be passed to the underlying processing function.
- release_resources()#
Shutdown thread pool and cleanup temp directory