@@ -236,8 +236,30 @@ def _convert_pr_to_document(pull_request: PullRequest) -> Document:
236
236
else None
237
237
),
238
238
metadata = {
239
- "merged" : str (pull_request .merged ),
240
- "state" : pull_request .state ,
239
+ k : str (v )
240
+ for k , v in {
241
+ "merged" : pull_request .merged ,
242
+ "state" : pull_request .state ,
243
+ "assignee" : (
244
+ pull_request .assignee .login if pull_request .assignee else None
245
+ ),
246
+ "created_at" : pull_request .created_at .replace (tzinfo = timezone .utc ),
247
+ "updated_at" : (
248
+ pull_request .updated_at .replace (tzinfo = timezone .utc )
249
+ if pull_request .updated_at
250
+ else None
251
+ ),
252
+ "merged_at" : (
253
+ pull_request .merged_at .replace (tzinfo = timezone .utc )
254
+ if pull_request .merged_at
255
+ else None
256
+ ),
257
+ "merged_by" : (
258
+ pull_request .merged_by .login if pull_request .merged_by else None
259
+ ),
260
+ "url" : pull_request .url ,
261
+ }.items ()
262
+ if v is not None
241
263
},
242
264
)
243
265
@@ -256,7 +278,28 @@ def _convert_issue_to_document(issue: Issue) -> Document:
256
278
# updated_at is UTC time but is timezone unaware
257
279
doc_updated_at = issue .updated_at .replace (tzinfo = timezone .utc ),
258
280
metadata = {
259
- "state" : issue .state ,
281
+ k : str (v )
282
+ for k , v in {
283
+ "state" : issue .state ,
284
+ "assignee" : issue .assignee .login if issue .assignee else None ,
285
+ "created_at" : issue .created_at .replace (tzinfo = timezone .utc ),
286
+ "updated_at" : (
287
+ issue .updated_at .replace (tzinfo = timezone .utc )
288
+ if issue .updated_at
289
+ else None
290
+ ),
291
+ "closed_at" : (
292
+ issue .closed_at .replace (tzinfo = timezone .utc )
293
+ if issue .closed_at
294
+ else None
295
+ ),
296
+ "closed_by" : issue .closed_by .login if issue .closed_by else None ,
297
+ "labels" : (
298
+ [label .name for label in issue .labels ] if issue .labels else []
299
+ ),
300
+ "url" : issue .url ,
301
+ }.items ()
302
+ if v is not None
260
303
},
261
304
)
262
305
0 commit comments