File size: 1,746 Bytes
b46e390
50a2d09
 
 
 
 
4013420
d36bd19
50a2d09
b46e390
d4d958d
50a2d09
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b46e390
50a2d09
b46e390
50a2d09
 
b46e390
50a2d09
b46e390
50a2d09
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
---
title: YourAI - LightRAG Powered
emoji: πŸ”₯
colorFrom: red
colorTo: yellow
sdk: docker
pinned: false
license: mit
app_port: 7860
---

# YourAI - LightRAG Powered

A sophisticated AI assistant powered by LightRAG (Retrieval-Augmented Generation) with conversation memory and custom AI creation capabilities.

## Features

πŸ”₯ **Fire Safety AI** - Specialized AI for fire safety regulations and codes
πŸ€– **General AI** - General-purpose conversational AI with memory
πŸ› οΈ **Custom AI Builder** - Create your own AI with uploaded knowledge
πŸ’¬ **Conversation Memory** - Maintains context within conversations
πŸ“š **Knowledge Management** - Advanced GraphRAG for better information retrieval

## How to Use

1. **Chat with Fire Safety AI** - Ask questions about fire safety regulations
2. **General Chat** - Have conversations with context awareness
3. **Create Custom AI** - Upload documents to create specialized AI assistants
4. **Manage Conversations** - View and manage your chat history

## Technology Stack

- **LightRAG**: Advanced retrieval-augmented generation
- **Cloudflare Workers AI**: LLM and embedding models
- **FastAPI**: High-performance API framework
- **NetworkX**: Graph storage for knowledge relationships
- **NanoVectorDB**: Vector storage for embeddings

## API Endpoints

- `POST /chat/fire-safety` - Chat with fire safety AI
- `POST /chat/general` - General conversation
- `POST /chat/custom/{ai_id}` - Chat with custom AI
- `POST /create-custom-ai` - Create custom AI
- `POST /upload` - Upload knowledge files
- `GET /conversations` - Get user conversations

## Models Used

- **LLM**: `@cf/meta/llama-3.2-3b-instruct`
- **Embedding**: `@cf/baai/bge-m3`

## License

MIT License - Feel free to use and modify!