ChromaDB_Chatbot_Public - GitLite

KO | EN

gitlite — search

#python #react #typescript #golang #rust #deep-learning #nodejs #java #windows #awesome #kubernetes #security #pytorch #open-source

ChromaDB_Chatbot_Public

(★ 143)

Public version of my ChromaDB chatbot that keeps track of user profile and historical topics

File Explorer

Download Latest Version (.zip)

- convo_1685975115.7847478.yaml
- chat_1685974321.74513_user.txt
- placeholder.txt
- log_1685974847.0820637_update.txt
.gitignore
chat.py
chromadb_peek.py
key_openai.txt
LICENSE
README.md
requirements.txt
system_default.txt
system_instantiate_new_kb.txt
system_reflective_journaling.txt
system_split_kb.txt
system_update_existing_kb.txt
system_update_user_profile.txt
user_profile.txt

# Installation Guide

1. Get the code

git clone https://github.com/daveshap/ChromaDB_Chatbot_Public

Downloads the entire project code from GitHub to your computer.

cd ChromaDB_Chatbot_Public

Moves into the project folder you just downloaded.

2. Python

Easy Recommended

Prerequisites

Git Needed to download the project code from GitHub.
Python 3 On Windows, be sure to check 'Add Python to PATH' during installation.

pip install -r requirements.txt

Installs the Python libraries listed in requirements.txt (or similar).

python <실행할 파일명>.py # README에서 정확한 실행 파일명을 확인하세요

Runs the Python script (or module).

✅ If it runs without errors and prints output in the terminal, it worked.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# ChromaDB Chatbot

Public version of my ChromaDB chatbot that keeps track of user profile and historical topics. Should be mostly ready to go right out of the box.

## Setup

1. Install chromadb and openai (in `requirements.txt` file)
2. Update `user_profile.txt` file with your initial information
3. Update `key_openai.txt` with your OpenAI API key

## Usage

- Main chat client: `python chat.py`
- Take a look in your KB: `python chromadb_peek.py`

## Code Explanation

This Python script serves as the implementation of a chatbot that leverages the OpenAI's GPT-4 model. It additionally integrates the chatbot with a persistent knowledge base using the ChromaDB library. Here's an overview of how the different parts of the script function:

1. **Utility Functions**: The script starts with several utility functions to handle file operations and to interact with OpenAI's API. These include functions for saving and opening files, and a function to run the chatbot, managing retries in case of exceptions.
2. **Main Application**: The script's main operation is contained within a continuous loop (`while True:`), enabling continuous interaction with the user. This loop does the following:
   - **Instantiates the ChromaDB client** for persistent storage and knowledge base management.
   - **Initiates the chatbot** by loading OpenAI's API key and preparing a conversation list.
   - **Captures user input** and adds it to the conversation list. The input is also logged in a separate file for record-keeping.
   - **Searches the knowledge base** for relevant content based on the current conversation and updates the chatbot's default system message accordingly.
   - **Generates a response** from the chatbot based on the conversation so far, which includes the updated default system message and the user's input.
   - **Updates the user profile** based on the user's recent messages, using the chatbot's response as the updated profile.
   - **Updates the knowledge base** with the most recent conversation, either adding a new entry or updating an existing entry. If an existing entry becomes too long, it's split into two separate entries.

The script logs all interactions with the OpenAI API and updates to the knowledge base, providing a record of the chatbot's operations and aiding in debugging and optimization efforts. The use of the ChromaDB library allows for scalable storage and retrieval of the chatbot's knowledge base, accommodating a growing number of conversations and data points.

But seriously just look at the code, it's pretty straight forward.

## Contributing

You're welcome to submit a pull request to make mild changes or fix bugs. Any substantial refactors will be rejected. If you want to take this work and modify it, please just work on your own fork. This repo will eventually be made a public readonly archive.