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

Use typed entry source instead of raw str to map source to conf in api.py

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-11-08 17:57:15 -08:00
parent 8178004e6d
commit c043eb54ae
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/khoj/routers/api.py
View File

@@ -50,6 +50,7 @@ from database.models import (
LocalPdfConfig,
LocalPlaintextConfig,
KhojUser,
Entry as DbEntry,
GithubConfig,
NotionConfig,
)
@@ -60,12 +61,12 @@ api = APIRouter()
logger = logging.getLogger(__name__)
def map_config_to_object(content_source: str):
if content_source == "github":
def map_config_to_object(content_source: DbEntry.EntrySource):
if content_source == DbEntry.EntrySource.GITHUB:
return GithubConfig
if content_source == "notion":
if content_source == DbEntry.EntrySource.GITHUB:
return NotionConfig
if content_source == "computer":
if content_source == DbEntry.EntrySource.COMPUTER:
return "Computer"