diff --git a/api/apps/llm_app.py b/api/apps/llm_app.py index 10a1484ee..62ebaf30d 100644 --- a/api/apps/llm_app.py +++ b/api/apps/llm_app.py @@ -264,7 +264,7 @@ def add_llm(): try: with open(os.path.join(get_project_base_directory(), "web/src/assets/yay.jpg"), "rb") as f: m, tc = mdl.describe(f.read()) - if not tc: + if not m and not tc: raise Exception(m) except Exception as e: msg += f"\nFail to access model({mdl_nm})." + str(e) diff --git a/rag/llm/cv_model.py b/rag/llm/cv_model.py index 74127b3cb..4985cf013 100644 --- a/rag/llm/cv_model.py +++ b/rag/llm/cv_model.py @@ -162,6 +162,7 @@ class GptV4(Base): ) return res.choices[0].message.content.strip(), res.usage.total_tokens + class AzureGptV4(Base): def __init__(self, key, model_name, lang="Chinese", **kwargs): api_key = json.loads(key).get('api_key', '')