Skip to content

Commit 5a4505b

Browse files
committed
75: Improved logging
1 parent 97d4c96 commit 5a4505b

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

service/src/ai_document_plugin_service/ai/common/llm_client.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -165,19 +165,21 @@ async def completion(
165165
raise RuntimeError(msg)
166166
req_id = uuid.uuid4().hex[:8]
167167
wait_start = time.perf_counter()
168-
logger.debug('[llm] req=%s model=%s queueing', req_id, self.model)
168+
logger.debug('[llm] tenant=%s req=%s model=%s queueing', self.tenant_uuid, req_id, self.model)
169169
async with self.semaphore:
170170
wait_s = time.perf_counter() - wait_start
171171
logger.debug(
172-
'[llm] req=%s acquired semaphore after %.3fs (limit=%s)',
172+
'[llm] tenant=%s req=%s acquired semaphore after %.3fs (limit=%s)',
173+
self.tenant_uuid,
173174
req_id,
174175
wait_s,
175176
self.semaphore.limit,
176177
)
177178
call_start = time.perf_counter()
178179
result = await self.client.chat.completions.create(*args, model=self.model, **kwargs)
179180
logger.debug(
180-
'[llm] req=%s completed in %.3fs (releasing semaphore)',
181+
'[llm] tenant=%s req=%s completed in %.3fs (releasing semaphore)',
182+
self.tenant_uuid,
181183
req_id,
182184
time.perf_counter() - call_start,
183185
)

0 commit comments

Comments
 (0)