sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 18

Use standard per minute rate limits across user types

Browse source
This commit is contained in:
Debanjum 2024-11-01 16:15:09 -07:00
parent ac21b10dd5
commit 9c7b36dc69
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/khoj/routers/api_chat.py
View file

@ -167,7 +167,7 @@ async def text_to_speech(
common: CommonQueryParams, common: CommonQueryParams,
text: str, text: str,
rate_limiter_per_minute=Depends( rate_limiter_per_minute=Depends(
ApiUserRateLimiter(requests=10, subscribed_requests=60, window=60, slug="chat_minute") ApiUserRateLimiter(requests=30, subscribed_requests=30, window=60, slug="chat_minute")
), ),
rate_limiter_per_day=Depends( rate_limiter_per_day=Depends(
ApiUserRateLimiter(requests=100, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day") ApiUserRateLimiter(requests=100, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day")
@ -555,7 +555,7 @@ async def chat(
common: CommonQueryParams, common: CommonQueryParams,
body: ChatRequestBody, body: ChatRequestBody,
rate_limiter_per_minute=Depends( rate_limiter_per_minute=Depends(
ApiUserRateLimiter(requests=5, subscribed_requests=20, window=60, slug="chat_minute") ApiUserRateLimiter(requests=20, subscribed_requests=20, window=60, slug="chat_minute")
), ),
rate_limiter_per_day=Depends( rate_limiter_per_day=Depends(
ApiUserRateLimiter(requests=100, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day") ApiUserRateLimiter(requests=100, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day")

3
src/khoj/routers/research.py
View file

@ -1,12 +1,11 @@
import json import json
import logging import logging
from datetime import datetime from datetime import datetime
from typing import Any, Callable, Dict, List, Optional from typing import Callable, Dict, List, Optional
import yaml import yaml
from fastapi import Request from fastapi import Request
from khoj.database.adapters import ConversationAdapters, EntryAdapters
from khoj.database.models import Agent, KhojUser from khoj.database.models import Agent, KhojUser
from khoj.processor.conversation import prompts from khoj.processor.conversation import prompts
from khoj.processor.conversation.utils import ( from khoj.processor.conversation.utils import (