klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +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, LocalPdfConfig,
LocalPlaintextConfig, LocalPlaintextConfig,
KhojUser, KhojUser,
Entry as DbEntry,
GithubConfig, GithubConfig,
NotionConfig, NotionConfig,
) )
@@ -60,12 +61,12 @@ api = APIRouter()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def map_config_to_object(content_source: str): def map_config_to_object(content_source: DbEntry.EntrySource):
if content_source == "github": if content_source == DbEntry.EntrySource.GITHUB:
return GithubConfig return GithubConfig
if content_source == "notion": if content_source == DbEntry.EntrySource.GITHUB:
return NotionConfig return NotionConfig
if content_source == "computer": if content_source == DbEntry.EntrySource.COMPUTER:
return "Computer" return "Computer"