Refactor Dockerfile to streamline dependencies by removing unnecessary installations and update requirements.txt to exclude rocksdb-py. Enhance AI message handling by limiting history to the last 12 messages and adjust maximum tool calling rounds from 25 to 12 in ai.py.

Author: robin-rpr

Dockerfile

@@ -1,23 +1,9 @@
 FROM python:3.11.3
 
-# System dependencies
+# Install Kafka
 RUN apt-get update && apt-get install -y \
     build-essential \
-    clang \
-    libclang-dev \
-    netcat
-
-# Install Kafka
-RUN apt-get install -y librdkafka-dev
-
-# Install RocksDB
-RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && \
-    . "$HOME/.cargo/env"
-ENV PATH="/root/.cargo/bin:${PATH}"
-RUN apt-get install -y software-properties-common gnupg && \
-    apt-key adv --keyserver keyserver.ubuntu.com --recv-keys 4F4EA0AAE5267A6C && \
-    apt-get update && \
-    apt-get install -y librocksdb-dev
+    librdkafka-dev
 
 # Install Java
 RUN apt-get update && \

requirements.txt

@@ -17,7 +17,6 @@ asyncpg==0.29.0
 beautifulsoup4==4.13.3
 Flask-Limiter
 confluent-kafka==2.5.3
-rocksdb-py==0.0.5
 scapy==2.6.1
 SQLAlchemy==2.0.36
 psycopg2-binary==2.9.10

templates/pages/ai.html

@@ -243,6 +243,11 @@
     append('user', text)
     history.push({ role: 'user', content: text })
 
+    // Keep only the last 12 messages.
+    if (history.length > 12) {
+        history = history.slice(-12)
+    }
+    
     const selectedModel = modelSelect.value
     const message = {
         messages: history,

utils/ai.py

@@ -69,9 +69,10 @@ def chat(messages: List[Dict[str, str]], model: str = "openrouter/auto") -> Gene
 
     conversation = [system_message] + list(messages)
     tools = _build_tools_schema()
+    max_rounds = 12
 
     # Maximum rounds of tool calling
-    for round_num in range(25):
+    for round_num in range(max_rounds):
         try:
             # Create chat completion with tools
             response = client.chat.completions.create(