Update base.py (#14405)

run-llama · Jun 26, 2024 · 39cdf82 · 39cdf82
1 parent 621d24b
commit 39cdf82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama-index-integrations/llms/llama-index-llms-yi/llama_index/llms/yi/base.py b/llama-index-integrations/llms/llama-index-llms-yi/llama_index/llms/yi/base.py
@@ -51,7 +51,7 @@ class Yi(OpenAI):
         # get api key from: https://platform.01.ai/
         llm = Yi(model="yi-large", api_key="YOUR_API_KEY")
 
-        response = llm.chat("Hi, who are you?")
+        response = llm.complete("Hi, who are you?")
         print(response)
         ```
     """