sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add migration script for getting the new offline model

Browse source
This commit is contained in:
sabaimran 2023-08-01 09:25:05 -07:00
parent 90efc2ea7a
commit 3a5d93d673
2 changed files with 21 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/khoj/migrations/migrate_offline_model.py Normal file
View file

@ -0,0 +1,19 @@
import os
from khoj.utils.yaml import load_config_from_file, save_config_to_file
def migrate_offline_model(args):
raw_config = load_config_from_file(args.config_file)
version = raw_config.get("version")
if version == "0.10.0" or version == None:
# If the user has downloaded the offline model, remove it from the cache.
offline_model_path = os.path.expanduser("~/.cache/gpt4all/llama-2-7b-chat.ggmlv3.q4_K_S.bin")
if os.path.exists(offline_model_path):
os.remove(offline_model_path)
raw_config["version"] = args.version_no
save_config_to_file(raw_config, args.config_file)
return args

3
src/khoj/utils/cli.py
View file

@ -8,6 +8,7 @@ from khoj.utils.helpers import resolve_absolute_path
from khoj.utils.yaml import parse_config_from_file
from khoj.migrations.migrate_version import migrate_config_to_version
from khoj.migrations.migrate_processor_config_openai import migrate_processor_conversation_schema
from khoj.migrations.migrate_offline_model import migrate_offline_model
def cli(args=None):
@ -55,7 +56,7 @@ def cli(args=None):
def run_migrations(args):
migrations = [migrate_config_to_version, migrate_processor_conversation_schema]
migrations = [migrate_config_to_version, migrate_processor_conversation_schema, migrate_offline_model]
for migration in migrations:
args = migration(args)
return args