diff --git a/src/core/graphbase.py b/src/core/graphbase.py index 178d6ef5..75bf1419 100644 --- a/src/core/graphbase.py +++ b/src/core/graphbase.py @@ -172,7 +172,8 @@ class GraphDatabase: def read_triples(file_path): with open(file_path, 'r', encoding='utf-8') as file: for line in file: - yield json.loads(line.strip()) + if line.strip(): + yield json.loads(line.strip()) triples = list(read_triples(file_path)) @@ -455,4 +456,4 @@ class GraphDatabase: if __name__ == "__main__": - pass \ No newline at end of file + pass diff --git a/src/models/chat_model.py b/src/models/chat_model.py index 3a671e21..e3a83d1e 100644 --- a/src/models/chat_model.py +++ b/src/models/chat_model.py @@ -61,7 +61,7 @@ class OpenModel(OpenAIBase): class CustomModel(OpenAIBase): def __init__(self, model_info): model_name = model_info["name"] - api_key = model_info.get("api_key", "custom_model") + api_key = model_info.get("api_key") or "custom_model" base_url = get_docker_safe_url(model_info["api_base"]) logger.info(f"> Custom model: {model_name}, base_url: {base_url}") diff --git a/web/src/views/SettingView.vue b/web/src/views/SettingView.vue index fe19385c..a08a4fe1 100644 --- a/web/src/views/SettingView.vue +++ b/web/src/views/SettingView.vue @@ -106,7 +106,7 @@ @click="handleChange('model_provider', 'custom'); handleChange('model_name', item.custom_id)" >