Hugging Face's logo

Spaces:
GoutamSachdev
/
agent_backend
Sleeping

App Files Community
agent_backend
File size: 4,680 Bytes
9144705
 
 
 
 
 
 
 
 
 
 
 
338e04d
9144705
338e04d
9144705
338e04d
9144705
338e04d
9144705
338e04d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73dcc95
 
 
338e04d
73dcc95
338e04d
 
 
 
9144705
 
 
 
 
 
 
 
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
---
title: ChatKit Backend
emoji: 🤖
colorFrom: blue
colorTo: indigo
sdk: docker
pinned: false
app_port: 7860
---

# ChatKit Python Backend

This FastAPI service implements an advanced multi-agent support system using the **OpenAI Agents SDK** and **OpenAI ChatKit**. It provides specialized support agents for different platforms (Kimi, DeepSeek, Google) with integrated RAG and persistent memory.

## 🏗️ Architecture Overview

The system is built on a modular, agentic architecture designed for high-performance customer support.

### Component Diagram

```mermaid
graph TD
    Client[ChatKit Frontend] <--> API[FastAPI Orchestrator]
    
    subgraph "Agent Layer"
        API <--> KimiAgent[Kimi Agent]
        API <--> DeepSeekAgent[DeepSeek Agent]
        API <--> GoogleAgent[Google Agent]
        API <--> Summ[Summarizer Agent]
    end
    
    subgraph "Tools & Intelligence"
        KimiAgent & DeepSeekAgent & GoogleAgent --> RAG[LlamaIndex RAG Tool]
        KimiAgent & DeepSeekAgent & GoogleAgent --> Facts[Fact Recording Tool]
        RAG --> Chroma[ChromaDB Vector Store]
        RAG --> WebsiteData[Website Knowledge Base]
    end
    
    subgraph "Persistence"
        API --> SQLiteThreads[SQLite: Chat Threads]
        API --> SQLiteState[SQLite: User State]
    end
    
    subgraph "External Providers"
        KimiAgent --> Groq[Groq API]
        DeepSeekAgent --> OpenRouter[OpenRouter API]
        GoogleAgent --> OR_Gemini[OpenRouter/Gemini]
    end
```

### Key Components

1.  **FastAPI Orchestrator**: Handles request routing, SSE streaming, and handoffs between Agents and ChatKit.
2.  **OpenAI Agents SDK**: Provides the logic for agent loops, tool calling, and handoffs.
3.  **ChatKit Server**: Manages the ChatKit protocol, ensuring real-time UI updates (widgets, thoughts, tool results).
4.  **Vector RAG Engine**: Uses LlamaIndex and ChromaDB to query school-specific services from scraped website data.
5.  **Multi-Modal**: Integrated Groq Whisper for audio transcriptions.

---

## 🤖 Model Configuration

The system utilizes a specialized mix of state-of-the-art models to balance performance, cost, and reasoning capabilities.

| Agent / Service | Model Name | Primary Provider | API Class |
| :--- | :--- | :--- | :--- |
| **Kimi Agent** | `moonshotai/kimi-k2-instruct-0905` | **Groq** | `OpenAIResponsesModel` |
| **DeepSeek Agent** | `deepseek/deepseek-chat` | **OpenRouter** | `OpenAIChatCompletionsModel` |
| **Google Agent** | `google/gemini-2.5-flash` | **OpenRouter** | `OpenAIChatCompletionsModel` |
| **Summarizer Agent** | `meta-llama/llama-4-scout-17b-16e-instruct` | **Groq** | `OpenAIResponsesModel` |
| **Audio Transcription** | `whisper-large-v3-turbo` | **Groq** | Native Deepgram/Whisper |

---

## 📋 Assumptions & Limitations

### Assumptions
-   **API Availability**: The system assumes stable connections to Groq and OpenRouter.
-   **Static Knowledge**: The RAG system assumes the vector database is pre-built from the school's website data (accessible in the `website/` folder).
-   **Single Instance**: Currently architected for single-instance deployment (SQLite persistence).

### Limitations
-   **Responses API Compatibility**: Only Groq natively supports the full `OpenAIResponsesModel` required for advanced thread state. OpenRouter models use `OpenAIChatCompletionsModel` with a custom persistence bridge.
-   **Concurrency**: SQLite is configured in WAL mode, but extremely high concurrent traffic would require a transition to PostgreSQL.
-   **Context Limits**: While the Summarizer Agent mitigates context bloat, extremely complex multi-turn sessions still rely on the provider's context window (e.g., 128k for DeepSeek/Gemini).

---

## 💰 Cost Estimation (Rough Calculation)

*Calculations based on 1,000 user queries with an average of 2,000 tokens per turn (1,500 input / 500 output).*

| Model (Provider) | Input Cost (per 1k) | Output Cost (per 1k) | **Estimated Total / 1,000 Queries** |
| :--- | :--- | :--- | :--- |
| **DeepSeek Chat** (OpenRouter) | $0.021 | $0.035 | **$0.056** |
| **Gemini 2.5 Flash** (OpenRouter) | $0.150 | $0.200 | **$0.350** |
| **Kimi** (Groq) | Free Tier / $0 | Free Tier / $0 | **$0.000** |

*Note: Costs are based on current OpenRouter pricing (Jan 2025) and Groq's high-speed free tier. DeepSeek remains the most cost-effective provider for reasoning-intensive tasks.*

---

## 🚀 Getting Started

To enable the realtime assistant you need to install both the ChatKit Python package and the OpenAI SDK, then provide an `OPENAI_API_KEY` environment variable.

```bash
uv sync
export OPENAI_API_KEY=sk-proj-...
uv run uvicorn app.main:app --reload
```