hugging-chat-api

English | 简体中文

Unofficial HuggingChat Python API, extensible for chatbots etc.

Note

Some recent versions may no longer be fully backward compatible to some extent, a good idea is to review this README or issues promptly after any problem arise. Custom parameters(temprature, max_token, etc) is no longer supported

Recently new updates:

• Assistant(Image Generator, etc)
• Web search
• Memorize context
• Supports for changing LLMs

Installation

pip3 install hugchat

Usage

API

The following are all common usages of this repo, You may not necessarily use all of them, You can add or delete some as needed :)

from hugchat import hugchat
from hugchat.login import Login

# Log in to huggingface and grant authorization to huggingchat
EMAIL = "your email"
PASSWD = "your password"
cookie_path_dir = "./cookies/" # NOTE: trailing slash (/) is required to avoid errors
sign = Login(EMAIL, PASSWD)
cookies = sign.login(cookie_dir_path=cookie_path_dir, save_cookies=True)

# Create your ChatBot
chatbot = hugchat.ChatBot(cookies=cookies.get_dict())  # or cookie_path="usercookies/<email>.json"

# Non stream response
query_result = chatbot.chat("Hi!")
print(query_result) # or query_result.text or query_result["text"]

# Stream response
for resp in chatbot.query(
    "Hello",
    stream=True
):
    print(resp)

# Web search (new feature)
query_result = chatbot.query("Hi!", web_search=True)
print(query_result)
for source in query_result.web_search_sources:
    print(source.link)
    print(source.title)
    print(source.hostname)

# Create a new conversation
chatbot.new_conversation(switch_to = True) # switch to the new conversation

# Get conversations on the server that are not from the current session (all your conversations in huggingchat)
conversation_list = chatbot.get_remote_conversations(replace_conversation_list=True)
# Get conversation list(local)
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

# Get information about the current conversation
info = chatbot.get_conversation_info()
print(info.id, info.title, info.model, info.system_prompt, info.history)

# Assistant
assistant = chatbot.search_assistant(assistant_name="ChatGpt") # assistant name list in https://huggingface.co/chat/assistants
assistant_list = chatbot.get_assistant_list_by_page(page=0)
chatbot.new_conversation(assistant=assistant, switch_to=True) # create a new conversation with assistant

# [DANGER] Delete all the conversations for the logged in user
chatbot.delete_all_conversations()

The query() function receives these parameters:

• text: Required[str].
• retry_count: Optional[int]. Number of retries for requesting huggingchat. Default is 5
• web_search : Optional[bool]. Whether to use online search.

CLI

version 0.0.5.2 or newer

Simply run the following command in your terminal to start the CLI mode

python -m hugchat.cli

CLI params:

• -u <your huggingface email> : Provide account email to login.
• -p : Force request password to login, ignores saved cookies.
• -s : Enable streaming mode output in CLI.
• -c : Continue previous conversation in CLI ".

Commands in cli mode:

• /new : Create and switch to a new conversation.

• /ids : Shows a list of all ID numbers and ID strings in current session.

• /switch : Shows a list of all conversations' info in current session. Then you can choose one to switch to.

• /switch all : Shows a list of all conversations' info in your account. Then you can choose one to switch to. (not recommended if your account has a lot of conversations)

• /del <index> : Deletes the conversation linked with the index passed. Will not delete active session.

• /delete-all : Deletes all the conversations for the logged in user.

• /clear : Clear the terminal.

• /llm : Get available models you can switch to.

• /llm <index> : Switches model to given model index based on /llm.

• /share : Toggles settings for sharing data with model author. On by default.

• /exit : Closes CLI environment.

• /stream : Toggles streaming the response.

• /web : Toggles web search.

• /web-hint : Toggles display web search hint.

• AI is an area of active research with known problems such as biased generation and misinformation. Do not use this application for high-stakes decisions or advice.

• Server resources are precious, it is not recommended to request this API in a high frequency. (Hugging Face's CTO🤗 just liked the suggestion)

Disclaimers

This is not an official Hugging Face product. This is a personal project and is not affiliated with Hugging Face in any way. Don't sue us.

Star History