fix: correct indentation for publish_snapshot calls in usage functions

This fix ensures that the snapshot publishing happens within the usage lock context, maintaining proper synchronization of usage counts.
This commit is contained in:
Alpha Nerd 2026-01-29 10:32:59 +01:00
parent d4b2558116
commit a1276e3de8

View file

@ -532,7 +532,7 @@ def dedupe_on_keys(dicts, key_fields):
async def increment_usage(endpoint: str, model: str) -> None:
async with usage_lock:
usage_counts[endpoint][model] += 1
await publish_snapshot()
await publish_snapshot()
async def decrement_usage(endpoint: str, model: str) -> None:
async with usage_lock:
@ -545,7 +545,7 @@ async def decrement_usage(endpoint: str, model: str) -> None:
usage_counts[endpoint].pop(model, None)
#if not usage_counts[endpoint]:
# usage_counts.pop(endpoint, None)
await publish_snapshot()
await publish_snapshot()
async def _make_chat_request(endpoint: str, model: str, messages: list, tools=None, stream: bool = False, think: bool = False, format=None, options=None, keep_alive: str = None) -> ollama.ChatResponse:
"""