highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Favor tz aware objects (#562)

Browse Source
This commit is contained in:
Yuhong Sun 2023-10-11 21:19:09 -07:00 committed by GitHub
parent e6866c92cf
commit 51490b5cd9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/danswer/connectors/github/connector.py
View File

@ -141,8 +141,8 @@ class GithubConnector(LoadConnector, PollConnector):
def poll_source(
self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch
) -> GenerateDocumentsOutput:
start_datetime = datetime.fromtimestamp(start)
end_datetime = datetime.fromtimestamp(end)
start_datetime = datetime.utcfromtimestamp(start)
end_datetime = datetime.utcfromtimestamp(end)
return self._fetch_from_github(start_datetime, end_datetime)

4
backend/danswer/connectors/hubspot/connector.py
View File

@ -123,8 +123,8 @@ class HubSpotConnector(LoadConnector, PollConnector):
def poll_source(
self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch
) -> GenerateDocumentsOutput:
start_datetime = datetime.fromtimestamp(start)
end_datetime = datetime.fromtimestamp(end)
start_datetime = datetime.utcfromtimestamp(start)
end_datetime = datetime.utcfromtimestamp(end)
return self._process_tickets(start_datetime, end_datetime)

7
backend/danswer/server/manage.py
View File

@ -1,5 +1,6 @@
from datetime import datetime
from datetime import timedelta
from datetime import timezone
from typing import cast
from fastapi import APIRouter
@ -459,9 +460,11 @@ def validate_existing_genai_api_key(
# Only validate every so often
check_key_time = "genai_api_key_last_check_time"
kv_store = get_dynamic_config_store()
curr_time = datetime.now()
curr_time = datetime.now(tz=timezone.utc)
try:
last_check = datetime.fromtimestamp(cast(float, kv_store.load(check_key_time)))
last_check = datetime.fromtimestamp(
cast(float, kv_store.load(check_key_time)), tz=timezone.utc
)
check_freq_sec = timedelta(seconds=GENERATIVE_MODEL_ACCESS_CHECK_FREQ)
if curr_time - last_check < check_freq_sec:
return

2
web/src/app/admin/connectors/github/page.tsx
View File

@ -182,7 +182,7 @@ const Main = () => {
`${values.repo_owner}/${values.repo_name}`
}
source="github"
inputType="load_state"
inputType="poll"
formBody={
<>
<TextFormField name="repo_owner" label="Repository Owner:" />