Awesome
Table of Contents
- What is AilingBot
- Features
- Quick Start
- <g-emoji class="g-emoji" alias="book" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f4d6.png">📖</g-emoji>User Guide
- <g-emoji class="g-emoji" alias="computer" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f4bb.png">💻</g-emoji>Development Guide
- <g-emoji class="g-emoji" alias="thinking" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f914.png">🤔</g-emoji>Frequently Asked Questions
- <g-emoji class="g-emoji" alias="dart" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f3af.png">🎯</g-emoji>Roadmap
What is AilingBot
AilingBot is an open-source engineering development framework and an all-in-one solution for integrating AI models into IM chatbots. With AilingBot, you can:
- ☕ Code-free usage: Quickly integrate existing AI large-scale models into mainstream IM chatbots (such as WeChat Work, Feishu, DingTalk, Slack etc.) to interact with AI models through IM chatbots and complete business requirements. Currently, AilingBot has built-in capabilities for multi-turn dialogue and document knowledge Q&A, and more capabilities will be added in the future.
- 🛠️Secondary development: AilingBot provides a clear engineering architecture, interface definition, and necessary basic components. You do not need to develop the engineering framework for large-scale model services from scratch. You only need to implement your Chat Policy and complete end-to-end AI model empowerment to IM chatbots through simple configurations. It also supports expanding to your own end (such as your own IM, web application, or mobile application) by developing your own channel.
Features
- 💯 Open source & Free: Completely open source and free.
- 📦 Ready to use: No need for development, with pre-installed capabilities to integrate with existing mainstream IM and AI models.
- 🔗 LangChain Friendly: Easy to integrate with LangChain.
- 🧩 Modular: The project is organized in a modular way, with modules dependent on each other through abstract protocols. Modules of the same type can be implemented by implementing the protocol, allowing for plug-and-play.
- 💻 Extensible: AilingBot can be extended to new usage scenarios and capabilities. For example, integrating with new IMs, new AI models, or customizing your own chat policy.
- 🔥 High performance: AilingBot uses a coroutine-based asynchronous mode to improve system concurrency performance. At the same time, system concurrency processing capabilities can be further improved through multi-processes.
- 🔌 API Integration: AilingBot provides a set of clear API interfaces for easy integration and collaboration with other systems and processes.
🚀 Quick Start
Start an AI chatbot in 5 minutes
Below is a guide on how to quickly start an AI chatbot based on the command-line interface using AilingBot. The effect is shown in the following figure:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/command-line-screenshot.png" alt="Command-line chatbot"/> </p>💡 First, you need to have an OpenAI API key. If you don't have one, refer to relevant materials on the Internet to obtain it.
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -it --rm \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
ailingbot poetry run ailingbot chat
Using PIP
Installation
pip install ailingbot
Generate Configuration File
ailingbot init --silence --overwrite
This will create a file called settings.toml
in the current directory, which is the configuration file for AilingBot.
Next, modify the necessary configurations. To start the bot, only one configuration is needed. Find the following
section in settings.toml
:
[policy.llm]
_type = "openai"
model_name = "gpt-3.5-turbo"
openai_api_key = ""
temperature = 0
Change the value of openai_api_key
to your actual OpenAI API key.
Start the Chatbot
Start the chatbot with the following command:
ailingbot chat
Start API Service
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -it --rm \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
-p 8080:8080 \
ailingbot poetry run ailingbot api
Using PIP
Installation
pip install ailingbot
Generate Configuration File
Same as starting the command line bot.
Start the Service
Start the bot using the following command:
ailingbot api
Now, enter http://localhost:8080/docs
in your browser to see the API documentation. (If it is not a local start,
please enter http://{your public IP}:8080/docs
)
Here is an example request:
curl -X 'POST' \
'http://localhost:8080/chat/' \
-H 'accept: application/json' \
-H 'Content-Type: application/json' \
-d '{
"text": "你好"
}'
And the response:
{
"type": "text",
"conversation_id": "default_conversation",
"uuid": "afb35218-2978-404a-ab39-72a9db6f303b",
"ack_uuid": "3f09933c-e577-49a5-8f56-fa328daa136f",
"receiver_id": "anonymous",
"scope": "user",
"meta": {},
"echo": {},
"text": "你好!很高兴和你聊天。有什么我可以帮助你的吗?",
"reason": null,
"suggestion": null
}
Integrating with WeChat Work
Here's a guide on how to quickly integrate the chatbot with WeChat Work.
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -d \
-e AILINGBOT_POLICY__NAME=conversation \
-e AILINGBOT_POLICY__HISTORY_SIZE=5 \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
-e AILINGBOT_CHANNEL__NAME=wechatwork \
-e AILINGBOT_CHANNEL__CORPID={your WeChat Work corpid} \
-e AILINGBOT_CHANNEL__CORPSECRET={your WeChat Work corpsecret} \
-e AILINGBOT_CHANNEL__AGENTID={your WeChat Work agentid} \
-e AILINGBOT_CHANNEL__TOKEN={your WeChat Work webhook token} \
-e AILINGBOT_CHANNEL__AES_KEY={your WeChat Work webhook aes_key} \
-p 8080:8080 \
ailingbot poetry run ailingbot serve
Using PIP
Installation
pip install ailingbot
Generate Configuration File
ailingbot init --silence --overwrite
Modify Configuration File
Open settings.toml
, and fill in the following section with your WeChat Work robot's real information:
[channel]
name = "wechatwork"
corpid = "" # Fill in with real information
corpsecret = "" # Fill in with real information
agentid = 0 # Fill in with real information
token = "" # Fill in with real information
aes_key = "" # Fill in with real information
In the llm
section, fill in your OpenAI API Key:
[policy.llm]
_type = "openai"
model_name = "gpt-3.5-turbo"
openai_api_key = "" # Fill in with your real OpenAI API Key here
temperature = 0
Start the Service
ailingbot serve
Finally, we need to go to the WeChat Work admin console to configure the webhook address so that WeChat Work knows to
forward the received user messages to our webhook.
The webhook URL is: http(s)://your_public_IP:8080/webhook/wechatwork/event/
After completing the above configuration, you can find the chatbot in WeChat Work and start chatting:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/wechatwork-screenshot.png" alt="WeChat Work chatbot" width="300"/> </p>Integrating with Feishu
Here's a guide on how to quickly integrate the chatbot with Feishu and enable a new conversation policy: uploading documents and performing knowledge-based question answering on them.
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -d \
-e AILINGBOT_POLICY__NAME=document_qa \
-e AILINGBOT_POLICY__CHUNK_SIZE=1000 \
-e AILINGBOT_POLICY__CHUNK_OVERLAP=0 \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
-e AILINGBOT_POLICY__LLM__MODEL_NAME=gpt-3.5-turbo-16k \
-e AILINGBOT_CHANNEL__NAME=feishu \
-e AILINGBOT_CHANNEL__APP_ID={your Feishu app id} \
-e AILINGBOT_CHANNEL__APP_SECRET={your Feishu app secret} \
-e AILINGBOT_CHANNEL__VERIFICATION_TOKEN={your Feishu webhook verification token} \
-p 8080:8080 \
ailingbot poetry run ailingbot serve
Using PIP
Installation
pip install ailingbot
Generate Configuration File
ailingbot init --silence --overwrite
Modify Configuration File
Open settings.toml
, and change the channel
section to the following, filling in your Feishu robot's real
information:
[channel]
name = "feishu"
app_id = "" # Fill in with real information
app_secret = "" # Fill in with real information
verification_token = "" # Fill in with real information
Replace the policy
section with the following document QA policy:
[policy]
name = "document_qa"
chunk_size = 1000
chunk_overlap = 5
Finally, it is recommended to use the 16k model when using the document QA policy. Therefore,
change policy.llm.model_name
to the following configuration:
[policy.llm]
_type = "openai"
model_name = "gpt-3.5-turbo-16k" # Change to gpt-3.5-turbo-16k
openai_api_key = "" # Fill in with real information
temperature = 0
Start the Service
ailingbot serve
Finally, we need to go to the Feishu admin console to configure the webhook address.
The webhook URL for Feishu is: http(s)://your_public_IP:8080/webhook/feishu/event/
After completing the above configuration, you can find the chatbot in Feishu and start chatting:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/feishu-screenshot.png" alt="Feishu chatbot" width="1000"/> </p>Integrating with DingTalk
Here's a guide on how to quickly integrate the chatbot with DingTalk.
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -d \
-e AILINGBOT_POLICY__NAME=conversation \
-e AILINGBOT_POLICY__HISTORY_SIZE=5 \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
-e AILINGBOT_CHANNEL__NAME=dingtalk \
-e AILINGBOT_CHANNEL__APP_KEY={your DingTalk app key} \
-e AILINGBOT_CHANNEL__APP_SECRET={your DingTalk app secret} \
-e AILINGBOT_CHANNEL__ROBOT_CODE={your DingTalk robot code} \
-p 8080:8080 \
ailingbot poetry run ailingbot serve
Using PIP
Installation
pip install ailingbot
Generate Configuration File
ailingbot init --silence --overwrite
Modify Configuration File
Open settings.toml
, and change the channel
section to the following, filling in your DingTalk robot's real
information:
[channel]
name = "dingtalk"
app_key = "" # Fill in with real information
app_secret = "" # Fill in with real information
robot_code = "" # Fill in with real information
Start the Service
ailingbot serve
Finally, we need to go to the DingTalk admin console to configure the webhook address.
The webhook URL for DingTalk is: http(s)://your_public_IP:8080/webhook/dingtalk/event/
After completing the above configuration, you can find the chatbot in DingTalk and start chatting:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/dingtalk-screenshot.png" alt="DingTalk chatbot" /> </p>Integrating with Slack
Here's a guide on how to quickly integrate the chatbot with Slack and enable a new conversation policy: uploading documents and performing knowledge-based question answering on them.
Using Docker
git clone https://github.com/ericzhang-cn/ailingbot.git ailingbot
cd ailingbot
docker build -t ailingbot .
docker run -d \
-e AILINGBOT_POLICY__NAME=document_qa \
-e AILINGBOT_POLICY__CHUNK_SIZE=1000 \
-e AILINGBOT_POLICY__CHUNK_OVERLAP=0 \
-e AILINGBOT_POLICY__LLM__OPENAI_API_KEY={your OpenAI API key} \
-e AILINGBOT_POLICY__LLM__MODEL_NAME=gpt-3.5-turbo-16k \
-e AILINGBOT_CHANNEL__NAME=slack \
-e AILINGBOT_CHANNEL__VERIFICATION_TOKEN={your Slack App webhook verification token} \
-e AILINGBOT_CHANNEL__OAUTH_TOKEN={your Slack App oauth token} \
-p 8080:8080 \
ailingbot poetry run ailingbot serve
Using PIP
Installation
pip install ailingbot
Generate Configuration File
ailingbot init --silence --overwrite
Modify Configuration File
Open settings.toml
, and change the channel
section to the following, filling in your Slack robot's real information:
[channel]
name = "slack"
verification_token = "" # Fill in with real information
oauth_token = "" # Fill in with real information
Replace the policy
section with the following document QA policy:
[policy]
name = "document_qa"
chunk_size = 1000
chunk_overlap = 5
Finally, it is recommended to use the 16k model when using the document QA policy. Therefore,
change policy.llm.model_name
to the following configuration:
[policy.llm]
_type = "openai"
model_name = "gpt-3.5-turbo-16k" # Change to gpt-3.5-turbo-16k
openai_api_key = "" # Fill in with real information
temperature = 0
Start the Service
ailingbot serve
Finally, we need to go to the Slack admin console to configure the webhook address.
The webhook URL for Slack is: http(s)://your_public_IP:8080/webhook/slack/event/
After completing the above configuration, you can find the chatbot in Slack and start chatting:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/slack-screenshot.png" alt="Slack chatbot" width="1000"/> </p>📖User Guide
Main Process
The main processing flow of AilingBot is as follows:
<p align="center"> <img src="https://raw.githubusercontent.com/ericzhang-cn/ailingbot/main/img/flow.png" alt="Main Process" width="500"/> </p>- First, the user sends a message to the IM bot.
- If a webhook is configured, the instant messaging tool will forward the request sent to the bot to the webhook service address.
- The webhook service processes the original IM message and converts it into AilingBot's internal message format, which is then sent to ChatBot.
- ChatBot processes the request and forms a response message based on the configured chat policy. During this process, ChatBot may perform operations such as requesting a large language model, accessing a vector database, or calling an external API to complete the request processing.
- ChatBot sends the response message to the IM Agent. The IM Agent is responsible for converting the AilingBot internal response message format into a specific IM format and calling the IM open capability API to send the response message.
- The IM bot displays the message to the user, completing the entire processing process.
Main Concepts
- IM bot: A capability built into most instant messaging tools that allows administrators to create a bot and process user messages through a program.
- Channel: A channel represents different terminals, which can be an IM or a custom terminal (such as the web).
- Webhook: An HTTP(S) service used to receive user messages forwarded by IM bots. Different channels have their own specifications for webhooks, so each channel requires its own webhook implementation.
- IM Agent: Used to call IM open capability APIs. Different IM open capability APIs are different, so each channel requires a corresponding agent implementation.
- ChatBot: The core component used to receive and respond to user messages.
- Chat Policy: Defines how to respond to users and is called by ChatBot. A chat policy specifically defines the robot's abilities, such as chitchat or knowledge Q&A.
- LLM: Large language model, such as OpenAI's ChatGPT and open ChatGLM, are all different large language models. The large language model is a key component for implementing AI capabilities.
Configuration
Configuration Methods
AilingBot can be configured in two ways:
- Using configuration files: AilingBot reads
settings.toml
in the current directory as the configuration file in TOML format. Please refer to the following section for specific configuration items. - Using environment variables: AilingBot also reads configuration items in environment variables. Please refer to the following section for a list of environment variables.
💡 Both configuration files and environment variables can be used together. If a configuration item exists in both, the environment variable takes precedence.
Configuration Mapping
All configurations have the following mappings between TOML keys and environment variables:
- All environment variables start with
AILINGBOT_
. - Double underscores
__
are used as separators between levels. - Underscores in configuration keys are preserved in environment variables.
- Case-insensitive.
For example:
- The corresponding environment variable of
some_conf
isAILINGBOT_SOME_CONF
. - The corresponding environment variable of
some_conf.conf_1
isAILINGBOT_SOME_CONF__CONF_1
. - The corresponding environment variable of
some_conf.conf_1.subconf
isAILINGBOT_SOME_CONF__CONF_1__SUBCONF
.
Configuration Items
General
Configuration Item | Description | TOML | Environment Variable |
---|---|---|---|
Language | Language code (Reference: http://www.lingoes.net/en/translator/langcode.htm) | lang | AILINGBOT_LANG |
Timezone | Timezone code (Reference: https://en.wikipedia.org/wiki/List_of_tz_database_time_zones) | tz | AILINGBOT_TZ |
Policy Name | Predefined policy name or complete policy class path | policy.name | AILINGBOT_POLICY__NAME |
Channel Name | Predefined channel name | channel.name | AILINGBOT_CHANNEL__NAME |
Webhook Path | Complete class path of non-predefined channel webhook | channel.webhook_name | AILINGBOT_CHANNEL__WEBHOOK_NAME |
Agent Path | Complete class path of non-predefined channel agent | channel.agent_name | AILINGBOT_CHANNEL__AGENT_NAME |
Uvicorn Config | All uvicorn configurations (Reference: uvicorn settings). These configurations will be passed to uvicorn | uvicorn.* | AILINGBOT_UVICORN__* |
Configuration example:
lang = "zh_CN"
tz = "Asia/Shanghai"
[policy]
name = "conversation"
# More policy configurations
[channel]
name = "wechatwork"
# More channel configurations
[uvicorn]
host = "0.0.0.0"
port = 8080
Built-in Policy Configuration
conversation
Conversation uses LangChain's Conversation as the policy, which enables direct interaction with LLM and has a conversation history context, enabling multi-turn conversations.
Configuration Item | Description | TOML | Environment Variable |
---|---|---|---|
History Size | Indicates how many rounds of historical conversations to keep | policy.history_size | AILINGBOT_POLICY__HISTORY_SIZE |
Configuration example:
# Use the conversation policy and keep 5 rounds of historical conversations
[policy]
name = "conversation"
history_size = 5
document_qa
Document_qa uses LangChain's Stuff as the policy. Users can upload a document and then ask questions based on the document content.
Configuration Item | Description | TOML | Environment Variable |
---|---|---|---|
Chunk Size | Corresponds to LangChain Splitter's chunk_size | policy.chunk_size | AILINGBOT_POLICY__CHUNK_SIZE |
Chunk Overlap | Corresponds to LangChain Splitter's chunk_overlap | policy.chunk_overlap | AILINGBOT_POLICY__CHUNK_OVERLAP |
Configuration example:
# Use the document_qa policy, with chunk_size and chunk_overlap set to 1000 and 0, respectively
[policy]
name = "document_qa"
chunk_size = 1000
chunk_overlap = 0
Model Configuration
The model configuration is consistent with LangChain. The following is an example.
OpenAI
[policy.llm]
_type = "openai" # Corresponding environment variable: AILINGBOT_POLICY__LLM___TYPE
model_name = "gpt-3.5-turbo" # Corresponding environment variable: AILINGBOT_POLICY__LLM__MODEL_NAME
openai_api_key = "sk-pd*****************************aAb" # Corresponding environment variable: AILINGBOT_POLICY__LLM__OPENAI_API_KEY
Command Line Tools
Initialize Configuration File (init)
Usage
The init
command generates a configuration file settings.toml
in the current directory. By default, the user will be
prompted interactively. You can use the --silence
option to generate the configuration file directly using default
settings.
Usage: ailingbot init [OPTIONS]
Initialize the AilingBot environment.
Options:
--silence Without asking the user.
--overwrite Overwrite existing file if a file with the same name already
exists.
--help Show this message and exit.
Options
Option | Description | Type | Remarks |
---|---|---|---|
--silence | Generate the default configuration directly without asking the user. | Flag | |
--overwrite | Allow overwriting the settings.toml file in the current directory. | Flag |
View Current Configuration (config)
The config
command reads the current environment configuration (including the configuration file and environment
variables) and merges them.
Usage
Usage: ailingbot config [OPTIONS]
Show current configuration information.
Options:
-k, --config-key TEXT Configuration key.
--help Show this message and exit.
Options
Option | Description | Type | Remarks |
---|---|---|---|
-k, --config-key | Configuration key | String | If not passed, the complete configuration information will be displayed. |
Start Command Line Bot (chat)
The chat
command starts an interactive command-line bot for testing the current chat policy.
Usage
Usage: ailingbot chat [OPTIONS]
Start an interactive bot conversation environment.
Options:
--debug Enable debug mode.
--help Show this message and exit.
Options
Option | Description | Type | Remarks |
---|---|---|---|
--debug | Enable debug mode | Flag | The debug mode will output more information, such as the prompt. |
Start Webhook Service (serve)
The serve
command starts a Webhook HTTP server for interacting with specific IM.
Usage
Usage: ailingbot serve [OPTIONS]
Run webhook server to receive events.
Options:
--log-level [TRACE|DEBUG|INFO|SUCCESS|WARNING|ERROR|CRITICAL]
The minimum severity level from which logged
messages should be sent to(read from
environment variable AILINGBOT_LOG_LEVEL if
is not passed into). [default: TRACE]
--log-file TEXT STDOUT, STDERR, or file path(read from
environment variable AILINGBOT_LOG_FILE if
is not passed into). [default: STDERR]
--help Show this message and exit.
Options
Option | Description | Type | Remarks |
---|---|---|---|
--log-level | The minimum severity level from which logged messages should be sent to. | String | By default, all log levels will be displayed (TRACE). |
--log-file | The location where logs are output. | String | By default, logs will be output to standard error (STDERR). |
Start API Service (api)
The api
command starts the API HTTP server.
Usage
Usage: ailingbot api [OPTIONS]
Run endpoint server.
Options:
--log-level [TRACE|DEBUG|INFO|SUCCESS|WARNING|ERROR|CRITICAL]
The minimum severity level from which logged
messages should be sent to(read from
environment variable AILINGBOT_LOG_LEVEL if
is not passed into). [default: TRACE]
--log-file TEXT STDOUT, STDERR, or file path(read from
environment variable AILINGBOT_LOG_FILE if
is not passed into). [default: STDERR]
--help Show this message and exit.
Options
Option | Description | Type | Remarks |
---|---|---|---|
--log-level | Display log level, which will display logs at this level and above | String | By default, all levels are displayed (TRACE) |
--log-file | Log output location | String | By default, logs are printed to standard error (STDERR) |
🔌API
TBD
💻Development Guide
Development Guidelines
TBD
Developing Chat Policy
TBD
Developing Channel
TBD
🤔Frequently Asked Questions
- Due to the fact that WeChat Work does not support uploading file event callbacks, the built-in
document_qa
policy cannot be used for WeChat Work. - The webhook of each IM requires a public IP. If you do not have one, you can consider testing locally through the " intranet penetration" solution. Please refer to online resources for specific methods.
- We expect the chat policy to be stateless, and the state should be stored externally. However, in specific
implementations, the policy may still have local states (such as storing conversation history in local variables).
Therefore, when uvicorn has multiple worker processes, these local states cannot be shared because each process has a
separate chat policy instance, and a request from the same user may be responded to by different workers, leading to
unexpected behavior. To avoid this, please ensure that at least one of the following two conditions is met:
- Chat policy does not use local states.
- Only one uvicorn worker is started.
🎯Roadmap
- Provide complete usage and developer documentation.
- Support more channels.
- WeChat Work
- Feishu
- DingTalk
- Slack
- Support more request message types.
- Text request
- Image request
- File request
- Support more response message types.
- Text response
- Image response
- File response
- Markdown response
- Table response
- Develop more out-of-the-box chat policies.
- Multi-round conversation policy
- Document question and answer policy
- Database question and answer policy
- Online search question and answer policy
- Support calling standalone chat policy services through HTTP.
- Abstract basic components
- Large language model
- Knowledge base
- Tools
- Support local model deployment.
- ChatGLM-6B
- Support API.
- Web management dashboard.
- Provide deployment capability based on Docker containers.
- Enhance the observability and controllability of the system.
- Complete test cases.