Skip to content

Commit 726eb64

Browse files
committed
fix: attach source_ids as keys in json logs
Signed-off-by: Anupam Kumar <kyteinsky@gmail.com>
1 parent 286db22 commit 726eb64

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

context_chat_backend/task_fetcher.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,9 @@ def files_indexing_thread(app_config: TConfig, app_enabled: Event) -> None:
8181

8282
def _load_sources(source_items: Mapping[int, SourceItem | ReceivedFileItem]) -> Mapping[int, IndexingError | None]:
8383
source_refs = [s.reference for s in source_items.values()]
84-
LOGGER.info('Starting embed_sources subprocess for %d source(s): %s', len(source_items), source_refs)
84+
LOGGER.info('Starting embed_sources subprocess for %d source(s)', len(source_items), extra={
85+
'source_ids': source_refs,
86+
})
8587
try:
8688
result = exec_in_proc(
8789
target=embed_sources,
@@ -96,8 +98,10 @@ def _load_sources(source_items: Mapping[int, SourceItem | ReceivedFileItem]) ->
9698
return result
9799
except SubprocessKilledError as e:
98100
LOGGER.error(
99-
'embed_sources subprocess was killed for %d source(s) with exitcode %s: %s',
100-
len(source_items), e.exitcode, source_refs, exc_info=e,
101+
'embed_sources subprocess was killed for %d source(s) with exitcode %s',
102+
len(source_items), e.exitcode, exc_info=e, extra={
103+
'source_ids': source_refs,
104+
},
101105
)
102106
if len(source_items) == 1:
103107
return dict.fromkeys(
@@ -120,8 +124,10 @@ def _load_sources(source_items: Mapping[int, SourceItem | ReceivedFileItem]) ->
120124
retryable=True,
121125
)
122126
LOGGER.error(
123-
'embed_sources subprocess raised a %s error for sources %s, marking all as retryable',
124-
e.__class__.__name__, source_refs, exc_info=e,
127+
'embed_sources subprocess raised a %s error for %d sources, marking all as retryable',
128+
e.__class__.__name__, len(source_refs), exc_info=e, extra={
129+
'source_ids': source_refs,
130+
}
125131
)
126132
return dict.fromkeys(source_items, err)
127133

0 commit comments

Comments
 (0)