Archives/talk-codebase
2
0
Fork 0
mirror of https://github.com/rsaryev/talk-codebase synced 2024-11-10 07:10:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

added try clause to handle imports

Browse Source
This commit is contained in:
fastdaima 2024-07-26 13:26:38 +05:30
parent a4ad57a058
commit 98445b9b35
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
talk_codebase/config.py
View File

@ -1,12 +1,21 @@
import os
openai_flag = True
try:
import openai
except:
openai_flag = False
import gpt4all
import openai
import questionary
import yaml
from talk_codebase.consts import MODEL_TYPES
config_path = os.path.join(os.path.expanduser("~"), ".talk_codebase_config.yaml")
@ -142,13 +151,16 @@ def remove_model_type():
def configure_model_type(config):
if config.get("model_type"):
return
choices = [{"name": "Local", "value": MODEL_TYPES["LOCAL"]}]
if openai_flag: choices.append(
{"name": "OpenAI", "value": MODEL_TYPES["OPENAI"]})
model_type = questionary.select(
"🤖 Select model type:",
choices=[
{"name": "Local", "value": MODEL_TYPES["LOCAL"]},
{"name": "OpenAI", "value": MODEL_TYPES["OPENAI"]},
]
choices=choices
).ask()
config["model_type"] = model_type
save_config(config)