Skip to content

Commit 8e06232

Browse files
committed
feat: updated github metadata
1 parent e822afd commit 8e06232

File tree

1 file changed

+46
-3
lines changed

1 file changed

+46
-3
lines changed

backend/onyx/connectors/github/connector.py

Lines changed: 46 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -236,8 +236,30 @@ def _convert_pr_to_document(pull_request: PullRequest) -> Document:
236236
else None
237237
),
238238
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
241263
},
242264
)
243265

@@ -256,7 +278,28 @@ def _convert_issue_to_document(issue: Issue) -> Document:
256278
# updated_at is UTC time but is timezone unaware
257279
doc_updated_at=issue.updated_at.replace(tzinfo=timezone.utc),
258280
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
260303
},
261304
)
262305

0 commit comments

Comments
 (0)