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

Fix migration script error when openai not available (#530)

Browse Source
This commit is contained in:
sabaimran
2023-11-02 11:28:08 -07:00
committed by GitHub
parent fe6720fa06
commit fd11b78552
1 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/khoj/migrations/migrate_server_pg.py
View File

@@ -99,9 +99,18 @@ def migrate_server_pg(args):
model_type=ChatModelOptions.ModelType.OFFLINE, model_type=ChatModelOptions.ModelType.OFFLINE,
) )
if "openai" in raw_config["processor"]["conversation"]: if (
"openai" in raw_config["processor"]["conversation"]
and raw_config["processor"]["conversation"]["openai"]
):
openai = raw_config["processor"]["conversation"]["openai"] openai = raw_config["processor"]["conversation"]["openai"]
if openai.get("api-key") is None:
logger.error("OpenAI API Key is not set. Will not be migrating OpenAI config.")
else:
if openai.get("chat-model") is None:
openai["chat-model"] = "gpt-3.5-turbo"
OpenAIProcessorConversationConfig.objects.create( OpenAIProcessorConversationConfig.objects.create(
api_key=openai.get("api-key"), api_key=openai.get("api-key"),
) )