Merge branch 'master' into master

README.md

@@ -77,6 +77,12 @@ conversation_list = chatbot.get_conversation_list()
 # Get the available models (not hardcore)
 models = chatbot.get_available_llm_models()
 
+# Get image link.
+# Work only for model "CohereForAI/c4ai-command-r-plus"
+chat_result = chatbot.chat("Draw a cat.")
+print(chat_result.get_final_text())
+print(chat_result.get_image_link())
+
 # Switch model with given index
 chatbot.switch_llm(0) # Switch to the first model
 chatbot.switch_llm(1) # Switch to the second model

src/hugchat/hugchat.py

@@ -711,6 +711,7 @@ def _stream_query(
             resp.encoding = 'utf-8'
 
             if resp.status_code != 200:
+
                 retry_count -= 1
                 if retry_count <= 0:
                     raise exceptions.ChatError(
@@ -725,6 +726,12 @@ def _stream_query(
                     if obj.__contains__("type"):
                         _type = obj["type"]
 
+                        if _type == "file":
+                            _sha = obj["sha"]
+                            _image_link = f"{self.hf_base_url}/chat/conversation/{conversation}/output/{_sha}"
+                            yield {"type": _type, "image_link": _image_link}
+                            continue
+
                         if _type == "finalAnswer":
                             final_answer = obj
                             break_flag = True