mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-03 05:29:12 +00:00
Rename initialize_{search,processor,server} to configure_{search,procesor,server}
- Search is being reconfigured multiple times in /regenerate and n/reload. More appropriate name is configure_ rather than initialize_ for it - Standardize name of methods under configure.py
This commit is contained in:
@@ -1,6 +1,3 @@
|
||||
# Standard Packages
|
||||
import sys
|
||||
|
||||
# External Packages
|
||||
import torch
|
||||
|
||||
@@ -16,7 +13,7 @@ from src.utils.helpers import get_absolute_path
|
||||
from src.utils.rawconfig import FullConfig
|
||||
|
||||
|
||||
def initialize_server(cmd_args):
|
||||
def configure_server(cmd_args):
|
||||
# Load config from CLI
|
||||
args = cli(cmd_args)
|
||||
|
||||
@@ -30,15 +27,15 @@ def initialize_server(cmd_args):
|
||||
state.verbose = args.verbose
|
||||
|
||||
# Initialize the search model from Config
|
||||
state.model = initialize_search(state.model, args.config, args.regenerate, device=state.device, verbose=state.verbose)
|
||||
state.model = configure_search(state.model, args.config, args.regenerate, device=state.device, verbose=state.verbose)
|
||||
|
||||
# Initialize Processor from Config
|
||||
state.processor_config = initialize_processor(args.config, verbose=state.verbose)
|
||||
state.processor_config = configure_processor(args.config, verbose=state.verbose)
|
||||
|
||||
return args.host, args.port, args.socket
|
||||
|
||||
|
||||
def initialize_search(model: SearchModels, config: FullConfig, regenerate: bool, t: SearchType = None, device=torch.device("cpu"), verbose: int = 0):
|
||||
def configure_search(model: SearchModels, config: FullConfig, regenerate: bool, t: SearchType = None, device=torch.device("cpu"), verbose: int = 0):
|
||||
# Initialize Org Notes Search
|
||||
if (t == SearchType.Org or t == None) and config.content_type.org:
|
||||
# Extract Entries, Generate Notes Embeddings
|
||||
@@ -67,7 +64,7 @@ def initialize_search(model: SearchModels, config: FullConfig, regenerate: bool,
|
||||
return model
|
||||
|
||||
|
||||
def initialize_processor(config: FullConfig, verbose: int):
|
||||
def configure_processor(config: FullConfig, verbose: int):
|
||||
if not config.processor:
|
||||
return
|
||||
|
||||
|
||||
@@ -9,7 +9,7 @@ from fastapi.staticfiles import StaticFiles
|
||||
from PyQt6 import QtCore, QtGui, QtWidgets
|
||||
|
||||
# Internal Packages
|
||||
from src.configure import initialize_server
|
||||
from src.configure import configure_server
|
||||
from src.router import router
|
||||
from src.utils import constants
|
||||
|
||||
@@ -22,7 +22,7 @@ app.include_router(router)
|
||||
|
||||
def run():
|
||||
# Setup Application Server
|
||||
host, port, socket = initialize_server(sys.argv[1:])
|
||||
host, port, socket = configure_server(sys.argv[1:])
|
||||
|
||||
# Setup GUI
|
||||
gui = QtWidgets.QApplication([])
|
||||
|
||||
@@ -12,7 +12,7 @@ from fastapi.responses import HTMLResponse, FileResponse
|
||||
from fastapi.templating import Jinja2Templates
|
||||
|
||||
# Internal Packages
|
||||
from src.configure import initialize_search
|
||||
from src.configure import configure_search
|
||||
from src.search_type import image_search, text_search
|
||||
from src.processor.conversation.gpt import converse, extract_search_type, message_to_log, message_to_prompt, understand, summarize
|
||||
from src.search_filter.explicit_filter import ExplicitFilter
|
||||
@@ -127,13 +127,13 @@ def search(q: str, n: Optional[int] = 5, t: Optional[SearchType] = None, r: Opti
|
||||
|
||||
@router.get('/reload')
|
||||
def reload(t: Optional[SearchType] = None):
|
||||
state.model = initialize_search(state.model, state.config, regenerate=False, t=t, device=state.device)
|
||||
state.model = configure_search(state.model, state.config, regenerate=False, t=t, device=state.device)
|
||||
return {'status': 'ok', 'message': 'reload completed'}
|
||||
|
||||
|
||||
@router.get('/regenerate')
|
||||
def regenerate(t: Optional[SearchType] = None):
|
||||
state.model = initialize_search(state.model, state.config, regenerate=True, t=t, device=state.device)
|
||||
state.model = configure_search(state.model, state.config, regenerate=True, t=t, device=state.device)
|
||||
return {'status': 'ok', 'message': 'regeneration completed'}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user