Update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
1 year ago · 61ddac3259
1 changed files with 2 additions and 2 deletions
--- a/eqa_search.py
+++ b/eqa_search.py
@ -92,7 +92,7 @@ def _get_llm_op(config):
    if config.llm_src.lower() == 'ernie':
        return ops.LLM.Ernie(api_key=config.ernie_api_key, secret_key=config.ernie_secret_key)

-    raise RuntimeError('Unknown llm source: [%s], only support \'openai\' and \'dolly\'' % (config.llm_src))
+    raise RuntimeError('Unknown llm source: [%s], only support openai, dolly and ernie' % (config.llm_src))


 def _get_prompt(config):