highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-10 21:26:01 +02: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
4 changed files with 10 additions and 7 deletions
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( def poll_source(
self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch
) -> GenerateDocumentsOutput: ) -> GenerateDocumentsOutput:
start_datetime = datetime.fromtimestamp(start) start_datetime = datetime.utcfromtimestamp(start)
end_datetime = datetime.fromtimestamp(end) end_datetime = datetime.utcfromtimestamp(end)
return self._fetch_from_github(start_datetime, end_datetime) 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( def poll_source(
self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch self, start: SecondsSinceUnixEpoch, end: SecondsSinceUnixEpoch
) -> GenerateDocumentsOutput: ) -> GenerateDocumentsOutput:
start_datetime = datetime.fromtimestamp(start) start_datetime = datetime.utcfromtimestamp(start)
end_datetime = datetime.fromtimestamp(end) end_datetime = datetime.utcfromtimestamp(end)
return self._process_tickets(start_datetime, end_datetime) 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 datetime
from datetime import timedelta from datetime import timedelta
from datetime import timezone
from typing import cast from typing import cast
from fastapi import APIRouter from fastapi import APIRouter
@@ -459,9 +460,11 @@ def validate_existing_genai_api_key(
# Only validate every so often # Only validate every so often
check_key_time = "genai_api_key_last_check_time" check_key_time = "genai_api_key_last_check_time"
kv_store = get_dynamic_config_store() kv_store = get_dynamic_config_store()
curr_time = datetime.now() curr_time = datetime.now(tz=timezone.utc)
try: 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) check_freq_sec = timedelta(seconds=GENERATIVE_MODEL_ACCESS_CHECK_FREQ)
if curr_time - last_check < check_freq_sec: if curr_time - last_check < check_freq_sec:
return return

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

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