klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Address mypy typing issues

Browse Source
This commit is contained in:
sabaimran
2023-12-16 09:24:26 +05:30
parent 5f6dcf9f2e
commit 3065cea562
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/database/adapters/__init__.py
View File

@@ -401,7 +401,7 @@ class ConversationAdapters:
) )
max_results = 3 max_results = 3
all_questions = await sync_to_async(list)(all_questions) all_questions = await sync_to_async(list)(all_questions) # type: ignore
if len(all_questions) < max_results: if len(all_questions) < max_results:
return all_questions return all_questions

6
src/khoj/routers/helpers.py
View File

@@ -267,7 +267,7 @@ async def text_to_image(message: str) -> Tuple[Optional[str], int]:
) )
image = response.data[0].b64_json image = response.data[0].b64_json
except openai.OpenAIError as e: except openai.OpenAIError as e:
logger.error(f"Image Generation failed with {e.http_status}: {e.error}") logger.error(f"Image Generation failed with {e}", exc_info=True)
status_code = 500 status_code = 500
return image, status_code return image, status_code
@@ -302,7 +302,7 @@ class ApiUserRateLimiter:
class ConversationCommandRateLimiter: class ConversationCommandRateLimiter:
def __init__(self, trial_rate_limit: int, subscribed_rate_limit: int): def __init__(self, trial_rate_limit: int, subscribed_rate_limit: int):
self.cache = defaultdict(lambda: defaultdict(list)) self.cache: Dict[str, Dict[str, List[float]]] = defaultdict(lambda: defaultdict(list))
self.trial_rate_limit = trial_rate_limit self.trial_rate_limit = trial_rate_limit
self.subscribed_rate_limit = subscribed_rate_limit self.subscribed_rate_limit = subscribed_rate_limit
self.restricted_commands = [ConversationCommand.Online, ConversationCommand.Image] self.restricted_commands = [ConversationCommand.Online, ConversationCommand.Image]
@@ -351,7 +351,7 @@ class ApiIndexedDataLimiter:
if state.billing_enabled is False: if state.billing_enabled is False:
return return
subscribed = has_required_scope(request, ["premium"]) subscribed = has_required_scope(request, ["premium"])
incoming_data_size_mb = 0 incoming_data_size_mb = 0.0
deletion_file_names = set() deletion_file_names = set()
if not request.user.is_authenticated: if not request.user.is_authenticated: