Merge pull request #298 from Fosowl/dev

Backend Containerization with Docker
This commit is contained in:
Martin
2025-06-06 18:02:55 +02:00
committed by GitHub
20 changed files with 451 additions and 284 deletions
+2 -3
View File
@@ -3,10 +3,9 @@ __pycache__/
*.py[cod] *.py[cod]
# Virtual environments # Virtual environments
venv/ agentic_seek_env/
.venv/ .agentic_seek_env/
# Environment variables (secrets)
.env .env
# Git metadata # Git metadata
+8
View File
@@ -1,4 +1,12 @@
SEARXNG_BASE_URL="http://127.0.0.1:8080" SEARXNG_BASE_URL="http://127.0.0.1:8080"
REDIS_BASE_URL="redis://redis:6379/0"
WORK_DIR="/Users/username/Documents/workspace_with_my_files"
OLLAMA_PORT="11434"
LM_STUDIO_PORT="1234"
CUSTOM_ADDITIONAL_LLM_PORT="11435"
OPENAI_API_KEY='xxxxx' OPENAI_API_KEY='xxxxx'
DEEPSEEK_API_KEY='xxxxx' DEEPSEEK_API_KEY='xxxxx'
OPENROUTER_API_KEY='xxxxx' OPENROUTER_API_KEY='xxxxx'
TOGETHER_API_KEY='xxxxx'
GOOGLE_API_KEY='xxxxx'
ANTHROPIC_API_KEY='xxxxx'
+3
View File
@@ -6,6 +6,8 @@
*.egg-info *.egg-info
cookies.json cookies.json
test_agent.py test_agent.py
searxng/uwsgi.ini.new
searxng/settings.yml.new
config.ini config.ini
.voices/ .voices/
experimental/ experimental/
@@ -19,6 +21,7 @@ agentic_seek_env/*
.env .env
*/.env */.env
dsk/ dsk/
chrome136/
### react ### ### react ###
.DS_* .DS_*
+64 -8
View File
@@ -1,9 +1,30 @@
FROM ubuntu:22.04
# Warning: doesn't work yet, backend is run on host machine for now
WORKDIR /app FROM --platform=linux/amd64 python:3.11-slim
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update -qq -y && \ # Install essential packages and Chrome dependencies
RUN apt-get update -y && apt-get install -y \
wget \
gnupg2 \
ca-certificates \
unzip \
xvfb \
libxss1 \
libappindicator1 \
fonts-liberation \
libnss3 \
libatk1.0-0 \
libatk-bridge2.0-0 \
libcups2 \
libdrm2 \
libxcomposite1 \
libxdamage1 \
libxrandr2 \
xdg-utils \
dbus \
&& rm -rf /var/lib/apt/lists/*
RUN apt-get update -y && \
apt-get install -y \ apt-get install -y \
gcc \ gcc \
g++ \ g++ \
@@ -24,23 +45,58 @@ apt-get install -y \
libgtk-4-1 \ libgtk-4-1 \
libnss3 \ libnss3 \
xdg-utils \ xdg-utils \
wget && \ wget \
&& rm -rf /var/lib/apt/lists/*
RUN apt-get update -y && \
apt-get install -y \
alsa-utils \
&& rm -rf /var/lib/apt/lists/*
ENV CHROME_TESTING_VERSION=134.0.6998.88
ENV DISPLAY=:99
WORKDIR /app
RUN set -eux; \
wget -qO /tmp/chrome.zip \
"https://storage.googleapis.com/chrome-for-testing-public/${CHROME_TESTING_VERSION}/linux64/chrome-linux64.zip"; \
unzip -q /tmp/chrome.zip -d /opt; \
rm /tmp/chrome.zip; \
ln -s /opt/chrome-linux64/chrome /usr/local/bin/google-chrome; \
ln -s /opt/chrome-linux64/chrome /usr/local/bin/chrome; \
mkdir -p /opt/chrome; \
ln -s /opt/chrome-linux64/chrome /opt/chrome/chrome; \
google-chrome --version
RUN set -eux; \
wget -qO /tmp/chromedriver.zip \
"https://storage.googleapis.com/chrome-for-testing-public/${CHROME_TESTING_VERSION}/linux64/chromedriver-linux64.zip"; \
unzip -q /tmp/chromedriver.zip -d /tmp; \
mv /tmp/chromedriver-linux64/chromedriver /usr/local/bin; \
rm /tmp/chromedriver.zip; \
chmod +x /usr/local/bin/chromedriver; \
chromedriver --version
RUN chmod +x /opt/chrome/chrome RUN chmod +x /opt/chrome/chrome
# Install dependencies
RUN pip3 install --upgrade pip setuptools wheel
COPY requirements.txt . COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt RUN pip install --no-cache-dir -r requirements.txt
RUN mkdir -p /opt/workspace
RUN mkdir -p /tmp && chmod 1777 /tmp
# Copy application code # Copy application code
COPY api.py . COPY api.py .
COPY sources/ ./sources/ COPY sources/ ./sources/
COPY prompts/ ./prompts/ COPY prompts/ ./prompts/
COPY crx/ crx/ COPY crx/ crx/
COPY llm_router/ llm_router/ COPY llm_router/ llm_router/
COPY .env .
COPY config.ini . COPY config.ini .
# Expose port
EXPOSE 8000 EXPOSE 8000
# Run the application # Run the application
+88 -107
View File
@@ -22,7 +22,7 @@
* 📋 Plans & Executes Complex Tasks - From trip planning to complex projects — it can split big tasks into steps and get things done using multiple AI agents. * 📋 Plans & Executes Complex Tasks - From trip planning to complex projects — it can split big tasks into steps and get things done using multiple AI agents.
* 🎙️ Voice-Enabled - Clean, fast, futuristic voice and speech to text allowing you to talk to it like it's your personal AI from a sci-fi movie * 🎙️ Voice-Enabled - Clean, fast, futuristic voice and speech to text allowing you to talk to it like it's your personal AI from a sci-fi movie. (In progress)
### **Demo** ### **Demo**
@@ -32,19 +32,17 @@ https://github.com/user-attachments/assets/b8ca60e9-7b3b-4533-840e-08f9ac426316
Disclaimer: This demo, including all the files that appear (e.g: CV_candidates.zip), are entirely fictional. We are not a corporation, we seek open-source contributors not candidates. Disclaimer: This demo, including all the files that appear (e.g: CV_candidates.zip), are entirely fictional. We are not a corporation, we seek open-source contributors not candidates.
> 🛠⚠️ **Active Work in Progress** Please note that Code/Bash is not dockerized yet but will be soon (see docker_deployement branch) - Do not deploy over network or production. > 🛠⚠️ **Active Work in Progress**
> 🙏 This project started as a side-project with zero roadmap and zero funding. It's grown way beyond what I expected by ending in GitHub Trending. Contributions, feedback, and patience are deeply appreciated. > 🙏 This project started as a side-project and has zero roadmap and zero funding. It's grown way beyond what I expected by ending in GitHub Trending. Contributions, feedback, and patience are deeply appreciated.
## Installation ## Prerequisites
Make sure you have chrome driver, docker and python3.10 installed. Make sure you have chrome driver, docker and python3.10 installed.
We highly advice you use exactly python3.10 for the setup. Dependencies error might happen otherwise.
For issues related to chrome driver, see the **Chromedriver** section. For issues related to chrome driver, see the **Chromedriver** section.
### 1️⃣ **Clone the repository and setup** ### 1. **Clone the repository and setup**
```sh ```sh
git clone https://github.com/Fosowl/agenticSeek.git git clone https://github.com/Fosowl/agenticSeek.git
@@ -52,82 +50,57 @@ cd agenticSeek
mv .env.example .env mv .env.example .env
``` ```
## Step 2: Install UV Package Manager ### 2. Change the .env file content
### For Linux/macOS:
```bash
curl -LsSf https://astral.sh/uv/install.sh | sh
```
### For Windows:
```powershell
powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
```
## Step 3: Create Virtual Environment
```bash
uv venv
```
### 3️⃣ **Install package**
Ensure Python, Docker and docker compose, and Google chrome are installed.
We recommand Python 3.10.0.
**Automatic Installation (Recommanded):**
For Linux/Macos:
```sh
./install.sh
```
For windows:
```sh ```sh
./install.bat SEARXNG_BASE_URL="http://127.0.0.1:8080"
REDIS_BASE_URL="redis://redis:6379/0"
WORK_DIR="/Users/mlg/Documents/workspace_for_ai"
OLLAMA_PORT="11434"
LM_STUDIO_PORT="1234"
CUSTOM_ADDITIONAL_LLM_PORT="11435"
OPENAI_API_KEY='optional'
DEEPSEEK_API_KEY='optional'
OPENROUTER_API_KEY='optional'
TOGETHER_API_KEY='optional'
GOOGLE_API_KEY='optional'
ANTHROPIC_API_KEY='optional'
``` ```
**Manually:** **API Key are totally optional for user who choose to run LLM locally. Which is the primary purpose of this project. Leave empty if you have sufficient hardware**
**Note: For any OS, ensure the ChromeDriver you install matches your installed Chrome version. Run `google-chrome --version`. See known issues if you have chrome >135** The following environment variables configure your application's connections and API keys.
- *Linux*: Update the `.env` file with your own values as needed:
Update Package List: `sudo apt update` - **SEARXNG_BASE_URL**: Leave unchanged
- **REDIS_BASE_URL**: Leave unchanged
- **WORK_DIR**: Path to your working directory on your local machine. AgenticSeek will be able to read and interact with these files.
- **OLLAMA_PORT**: Port number for the Ollama service.
- **LM_STUDIO_PORT**: Port number for the LM Studio service.
- **CUSTOM_ADDITIONAL_LLM_PORT**: Port for any additional custom LLM service.
Install Dependencies: `sudo apt install -y alsa-utils portaudio19-dev python3-pyaudio libgtk-3-dev libnotify-dev libgconf-2-4 libnss3 libxss1` All API key environment variables below are **optional**. You only need to provide them if you plan to use external APIs instead of running LLMs locally.
Install ChromeDriver matching your Chrome browser version: ### 3. **Start Docker**
`sudo apt install -y chromium-chromedriver`
Install requirements: `pip3 install -r requirements.txt` Make sure Docker is installed and running on your system. You can start Docker using the following commands:
- *Macos*: - **On Linux/macOS:**
Open a terminal and run:
```sh
sudo systemctl start docker
```
Or launch Docker Desktop from your applications menu if installed.
Update brew : `brew update` - **On Windows:**
Start Docker Desktop from the Start menu.
Install chromedriver : `brew install --cask chromedriver` You can verify Docker is running by executing:
```sh
Install portaudio: `brew install portaudio` docker info
```
Upgrade pip : `python3 -m pip install --upgrade pip` If you see information about your Docker installation, it is running correctly.
Upgrade wheel : : `pip3 install --upgrade setuptools wheel`
Install requirements: `pip3 install -r requirements.txt`
- *Windows*:
Install pyreadline3 `pip install pyreadline3`
Install portaudio manually (e.g., via vcpkg or prebuilt binaries) and then run: `pip install pyaudio`
Download and install chromedriver manually from: https://sites.google.com/chromium.org/driver/getting-started
Place chromedriver in a directory included in your PATH.
Install requirements: `pip3 install -r requirements.txt`
--- ---
@@ -149,7 +122,7 @@ See below for a list of local supported provider.
**Update the config.ini** **Update the config.ini**
Change the config.ini file to set the provider_name to a supported provider and provider_model to a LLM supported by your provider. We recommand reasoning model such as *Qwen* or *Deepseek*. Change the config.ini file to set the provider_name to a supported provider and provider_model to a LLM supported by your provider. We recommend reasoning model such as *Qwen* or *Deepseek*.
See the **FAQ** at the end of the README for required hardware. See the **FAQ** at the end of the README for required hardware.
@@ -162,19 +135,23 @@ provider_server_address = 127.0.0.1:11434
agent_name = Jarvis # name of your AI agent_name = Jarvis # name of your AI
recover_last_session = True # whenever to recover the previous session recover_last_session = True # whenever to recover the previous session
save_session = True # whenever to remember the current session save_session = True # whenever to remember the current session
speak = True # text to speech speak = False # text to speech
listen = False # Speech to text, only for CLI listen = False # Speech to text, only for CLI, experimental
work_dir = /Users/mlg/Documents/workspace # The workspace for AgenticSeek.
jarvis_personality = False # Whenever to use a more "Jarvis" like personality (experimental) jarvis_personality = False # Whenever to use a more "Jarvis" like personality (experimental)
languages = en zh # The list of languages, Text to speech will default to the first language on the list languages = en zh # The list of languages, Text to speech will default to the first language on the list
[BROWSER] [BROWSER]
headless_browser = True # Whenever to use headless browser, recommanded only if you use web interface. headless_browser = True # leave unchanged unless using CLI on host.
stealth_mode = True # Use undetected selenium to reduce browser detection stealth_mode = True # Use undetected selenium to reduce browser detection
``` ```
Warning: Do *NOT* set provider_name to `openai` if using LM-studio for running LLMs. Set it to `lm-studio`. **Warning**:
Note: Some provider (eg: lm-studio) require you to have `http://` in front of the IP. For example `http://127.0.0.1:1234` - The `config.ini` file format does not support comments.
Do not copy and paste the example configuration directly, as comments will cause errors. Instead, manually modify the `config.ini` file with your desired settings, excluding any comments.
- Do *NOT* set provider_name to `openai` if using LM-studio for running LLMs. Set it to `lm-studio`.
- Some provider (eg: lm-studio) require you to have `http://` in front of the IP. For example `http://127.0.0.1:1234`
**List of local providers** **List of local providers**
@@ -196,6 +173,8 @@ Next step: [Start services and run AgenticSeek](#Start-services-and-Run)
## Setup to run with an API ## Setup to run with an API
**Running with an API is optional, see above to run locally.**
Set the desired provider in the `config.ini`. See below for a list of API providers. Set the desired provider in the `config.ini`. See below for a list of API providers.
```sh ```sh
@@ -221,9 +200,7 @@ Example: export `TOGETHER_API_KEY="xxxxx"`
| togetherAI | No | Use together AI API (non-private) | | togetherAI | No | Use together AI API (non-private) |
| google | No | Use google gemini API (non-private) | | google | No | Use google gemini API (non-private) |
*We advice against using gpt-4o or other closedAI models*, performance are poor for web browsing and task planning. Please note that coding/bash might fail with gemini, it seems to ignore our prompt for format to respect, which are optimized for deepseek r1. Model such are gpt-4o seem to perform poorly with our prompt as well.
Please also note that coding/bash might fail with gemini, it seem to ignore our prompt for format to respect, which are optimized for deepseek r1.
Next step: [Start services and run AgenticSeek](#Start-services-and-Run) Next step: [Start services and run AgenticSeek](#Start-services-and-Run)
@@ -235,44 +212,44 @@ Next step: [Start services and run AgenticSeek](#Start-services-and-Run)
## Start services and Run ## Start services and Run
Activate your python env if needed.
```sh
source .venv/bin/activate
```
Start required services. This will start all services from the docker-compose.yml, including: Start required services. This will start all services from the docker-compose.yml, including:
- searxng - searxng
- redis (required by searxng) - redis (required by searxng)
- frontend - frontend
- backend (if using `full`)
```sh
sudo ./start_services.sh full # MacOS
start ./start_services.cmd full # Window
```
**Warning:** This step will download and load all Docker images, which may take up to 30 minutes. After starting the services, please wait until the backend service is fully running (you should see backend: <info> in the log) before sending any messages. The backend services may take longer to start than others.
Go to `http://localhost:3000/` and you should see the web interface.
**Optional:** Run with the CLI interface:
To run with CLI interface you would have to install package on host:
```sh
./install.sh
./install.bat # windows
```
Start services:
```sh ```sh
sudo ./start_services.sh # MacOS sudo ./start_services.sh # MacOS
start ./start_services.cmd # Window start ./start_services.cmd # Window
``` ```
**Options 1:** Run with the CLI interface. Then run : `python3 cli.py`
```sh
python3 cli.py
```
We advice you set `headless_browser` to False in the config.ini for CLI mode.
**Options 2:** Run with the Web interface.
Start the backend.
```sh
python3 api.py
```
Go to `http://localhost:3000/` and you should see the web interface.
--- ---
## Usage ## Usage
Make sure the services are up and running with `./start_services.sh` and run the AgenticSeek with `python3 cli.py` for CLI mode or `python3 api.py` then go to `localhost:3000` for web interface. Make sure the services are up and running with `./start_services.sh full` and go to `localhost:3000` for web interface.
You can also use speech to text by setting `listen = True` in the config. Only for CLI mode. You can also use speech to text by setting `listen = True` in the config. Only for CLI mode.
@@ -368,6 +345,8 @@ Next step: [Start services and run AgenticSeek](#Start-services-and-Run)
## Speech to Text ## Speech to Text
Warning: speech to text only work in CLI mode at the moment.
Please note that currently speech to text only work in english. Please note that currently speech to text only work in english.
The speech-to-text functionality is disabled by default. To enable it, set the listen option to True in the config.ini file: The speech-to-text functionality is disabled by default. To enable it, set the listen option to True in the config.ini file:
@@ -407,7 +386,6 @@ recover_last_session = False
save_session = False save_session = False
speak = False speak = False
listen = False listen = False
work_dir = /Users/mlg/Documents/ai_folder
jarvis_personality = False jarvis_personality = False
languages = en zh languages = en zh
[BROWSER] [BROWSER]
@@ -435,8 +413,6 @@ stealth_mode = False
- listen -> listen to voice input (True) or not (False). - listen -> listen to voice input (True) or not (False).
- work_dir -> Folder the AI will have access to. eg: /Users/user/Documents/.
- jarvis_personality -> Uses a JARVIS-like personality (True) or not (False). This simply change the prompt file. - jarvis_personality -> Uses a JARVIS-like personality (True) or not (False). This simply change the prompt file.
- languages -> The list of supported language, needed for the llm router to work properly, avoid putting too many or too similar languages. - languages -> The list of supported language, needed for the llm router to work properly, avoid putting too many or too similar languages.
@@ -550,7 +526,7 @@ Yes with Ollama, lm-studio or server providers, all speech to text, LLM and text
**Q: Why should I use AgenticSeek when I have Manus?** **Q: Why should I use AgenticSeek when I have Manus?**
This started as Side-Project we did out of interest about AI agents. Whats special about it is that we want to use local model and avoid APIs. This started as Side-Project we did out of interest about AI agents. Whats special about it is that we want to use local model and avoid APIs.
We draw inspiration from Jarvis and Friday (Iron man movies) to make it "cool" but for functionnality we take more inspiration from Manus, because that's what people want in the first place: a local manus alternative. We draw inspiration from Jarvis and Friday (Iron man movies) to make it "cool" but for functionality we take more inspiration from Manus, because that's what people want in the first place: a local manus alternative.
Unlike Manus, AgenticSeek prioritizes independence from external systems, giving you more control, privacy and avoid api cost. Unlike Manus, AgenticSeek prioritizes independence from external systems, giving you more control, privacy and avoid api cost.
## Contribute ## Contribute
@@ -568,3 +544,8 @@ Were looking for developers to improve AgenticSeek! Check out open issues or
> [antoineVIVIES](https://github.com/antoineVIVIES) | Taipei Time > [antoineVIVIES](https://github.com/antoineVIVIES) | Taipei Time
> [steveh8758](https://github.com/steveh8758) | Taipei Time > [steveh8758](https://github.com/steveh8758) | Taipei Time
## Special Thanks:
> [tcsenpai](https://github.com/tcsenpai) and [plitc](https://github.com/plitc) For helping with backend dockerization
+10 -1
View File
@@ -22,6 +22,10 @@ from sources.utility import pretty_print
from sources.logger import Logger from sources.logger import Logger
from sources.schemas import QueryRequest, QueryResponse from sources.schemas import QueryRequest, QueryResponse
from dotenv import load_dotenv
load_dotenv()
from celery import Celery from celery import Celery
@@ -34,7 +38,7 @@ config.read('config.ini')
api.add_middleware( api.add_middleware(
CORSMiddleware, CORSMiddleware,
allow_origins=["http://localhost", "http://localhost:3000"], allow_origins=["*"],
allow_credentials=True, allow_credentials=True,
allow_methods=["*"], allow_methods=["*"],
allow_headers=["*"], allow_headers=["*"],
@@ -247,4 +251,9 @@ async def process_query(request: QueryRequest):
interaction.save_session() interaction.save_session()
if __name__ == "__main__": if __name__ == "__main__":
envport = os.getenv("BACKEND_PORT")
if envport:
port = int(envport)
else:
port = 8000
uvicorn.run(api, host="0.0.0.0", port=8000) uvicorn.run(api, host="0.0.0.0", port=8000)
+1 -2
View File
@@ -3,12 +3,11 @@ is_local = True
provider_name = ollama provider_name = ollama
provider_model = deepseek-r1:14b provider_model = deepseek-r1:14b
provider_server_address = 127.0.0.1:11434 provider_server_address = 127.0.0.1:11434
agent_name = Name_of_your_AI agent_name = Jarvis
recover_last_session = False recover_last_session = False
save_session = False save_session = False
speak = False speak = False
listen = False listen = False
work_dir = /Users/mlg/Documents/workspace_for_agenticseek
jarvis_personality = False jarvis_personality = False
languages = en languages = en
[BROWSER] [BROWSER]
+34 -32
View File
@@ -3,6 +3,7 @@ version: '3'
services: services:
redis: redis:
container_name: redis container_name: redis
profiles: ["core", "full"]
image: docker.io/valkey/valkey:8-alpine image: docker.io/valkey/valkey:8-alpine
command: valkey-server --save 30 1 --loglevel warning command: valkey-server --save 30 1 --loglevel warning
restart: unless-stopped restart: unless-stopped
@@ -24,6 +25,7 @@ services:
searxng: searxng:
container_name: searxng container_name: searxng
profiles: ["core", "full"]
image: docker.io/searxng/searxng:latest image: docker.io/searxng/searxng:latest
restart: unless-stopped restart: unless-stopped
ports: ports:
@@ -31,8 +33,8 @@ services:
volumes: volumes:
- ./searxng:/etc/searxng:rw,z - ./searxng:/etc/searxng:rw,z
environment: environment:
- SEARXNG_BASE_URL=http://localhost:8080/ - SEARXNG_BASE_URL=${SEARXNG_BASE_URL:-http://localhost:8080/}
- SEARXNG_SECRET_KEY=$(openssl rand -hex 32) - SEARXNG_SECRET_KEY=${SEARXNG_SECRET_KEY}
- UWSGI_WORKERS=4 - UWSGI_WORKERS=4
- UWSGI_THREADS=4 - UWSGI_THREADS=4
cap_add: cap_add:
@@ -51,6 +53,7 @@ services:
frontend: frontend:
container_name: frontend container_name: frontend
profiles: ["core", "full"]
build: build:
context: ./frontend context: ./frontend
dockerfile: Dockerfile.frontend dockerfile: Dockerfile.frontend
@@ -62,39 +65,38 @@ services:
environment: environment:
- NODE_ENV=development - NODE_ENV=development
- CHOKIDAR_USEPOLLING=true - CHOKIDAR_USEPOLLING=true
- BACKEND_URL=http://backend:8000 - REACT_APP_BACKEND_URL=http://0.0.0.0:${BACKEND_PORT:-8000}
networks: networks:
- agentic-seek-net - agentic-seek-net
# NOTE: backend service is not working yet due to issue with chromedriver on docker. backend:
# Therefore backend is run on host machine. container_name: backend
# Open to pull requests to fix this. profiles: ["backend", "full"]
build:
#backend: context: .
# container_name: backend dockerfile: Dockerfile.backend
# build: ports:
# context: ./ - ${BACKEND_PORT:-7777}:${BACKEND_PORT:-7777}
# dockerfile: Dockerfile.backend - ${OLLAMA_PORT:-11434}:${OLLAMA_PORT:-11434}
# stdin_open: true - ${LM_STUDIO_PORT:-1234}:${LM_STUDIO_PORT:-1234}
# tty: true - ${CUSTOM_ADDITIONAL_LLM_PORT:-8000}:${CUSTOM_ADDITIONAL_LLM_PORT:-8000}
# shm_size: 8g volumes:
# ports: - ./:/app
# - "8000:8000" - ${WORK_DIR:-.}:/opt/workspace
# volumes: command: python3 api.py
# - ./:/app environment:
# environment: - SEARXNG_URL=${SEARXNG_BASE_URL:-http://searxng:8080}
# - NODE_ENV=development - REDIS_URL=${REDIS_BASE_URL:-redis://redis:6379/0}
# - REDIS_URL=redis://redis:6379/0 - WORK_DIR=/opt/workspace
# - SEARXNG_URL=http://searxng:8080 - OPENAI_API_KEY=${OPENAI_API_KEY}
# - OLLAMA_URL=http://localhost:11434 - DEEPSEEK_API_KEY=${DEEPSEEK_API_KEY}
# - LM_STUDIO_URL=http://localhost:1234 - OPENROUTER_API_KEY=${OPENROUTER_API_KEY}
# extra_hosts: - TOGETHER_API_KEY=${TOGETHER_API_KEY}
# - "host.docker.internal:host-gateway" - GOOGLE_API_KEY=${GOOGLE_API_KEY}
# depends_on: - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
# - redis - HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
# - searxng - DSK_DEEPSEEK_API_KEY=${DSK_DEEPSEEK_API_KEY}
# networks: network_mode: "host"
# - agentic-seek-net
volumes: volumes:
redis-data: redis-data:
+7 -5
View File
@@ -4,6 +4,8 @@ import axios from 'axios';
import './App.css'; import './App.css';
import { colors } from './colors'; import { colors } from './colors';
const BACKEND_URL = process.env.BACKEND_PORT || 'http://0.0.0.0:8000';
function App() { function App() {
const [query, setQuery] = useState(''); const [query, setQuery] = useState('');
const [messages, setMessages] = useState([]); const [messages, setMessages] = useState([]);
@@ -27,7 +29,7 @@ function App() {
const checkHealth = async () => { const checkHealth = async () => {
try { try {
await axios.get('http://127.0.0.1:8000/health'); await axios.get(`${BACKEND_URL}/health`);
setIsOnline(true); setIsOnline(true);
console.log('System is online'); console.log('System is online');
} catch { } catch {
@@ -39,7 +41,7 @@ function App() {
const fetchScreenshot = async () => { const fetchScreenshot = async () => {
try { try {
const timestamp = new Date().getTime(); const timestamp = new Date().getTime();
const res = await axios.get(`http://127.0.0.1:8000/screenshots/updated_screen.png?timestamp=${timestamp}`, { const res = await axios.get(`${BACKEND_URL}/screenshots/updated_screen.png?timestamp=${timestamp}`, {
responseType: 'blob' responseType: 'blob'
}); });
console.log('Screenshot fetched successfully'); console.log('Screenshot fetched successfully');
@@ -90,7 +92,7 @@ function App() {
const fetchLatestAnswer = async () => { const fetchLatestAnswer = async () => {
try { try {
const res = await axios.get('http://127.0.0.1:8000/latest_answer'); const res = await axios.get(`${BACKEND_URL}/latest_answer`);
const data = res.data; const data = res.data;
updateData(data); updateData(data);
@@ -141,7 +143,7 @@ function App() {
setIsLoading(false); setIsLoading(false);
setError(null); setError(null);
try { try {
const res = await axios.get('http://127.0.0.1:8000/stop'); const res = await axios.get(`${BACKEND_URL}/stop`);
setStatus("Requesting stop..."); setStatus("Requesting stop...");
} catch (err) { } catch (err) {
console.error('Error stopping the agent:', err); console.error('Error stopping the agent:', err);
@@ -162,7 +164,7 @@ function App() {
try { try {
console.log('Sending query:', query); console.log('Sending query:', query);
setQuery('waiting for response...'); setQuery('waiting for response...');
const res = await axios.post('http://127.0.0.1:8000/query', { const res = await axios.post(`${BACKEND_URL}/query`, {
query, query,
tts_enabled: false tts_enabled: false
}); });
+1 -1
View File
@@ -17,7 +17,6 @@ playsound3>=1.0.0
soundfile>=0.13.1 soundfile>=0.13.1
transformers>=4.46.3 transformers>=4.46.3
torch>=2.4.1 torch>=2.4.1
python-dotenv>=1.0.0
ollama>=0.4.7 ollama>=0.4.7
scipy>=1.9.3 scipy>=1.9.3
soundfile>=0.13.1 soundfile>=0.13.1
@@ -41,6 +40,7 @@ fake_useragent>=2.1.0
selenium_stealth>=1.0.6 selenium_stealth>=1.0.6
undetected-chromedriver>=3.5.5 undetected-chromedriver>=3.5.5
sentencepiece>=0.2.0 sentencepiece>=0.2.0
together>=1.5.0
tqdm>4 tqdm>4
openai openai
sniffio sniffio
+3 -3
View File
@@ -5,12 +5,12 @@ gid = searxng
# Number of workers (usually CPU count) # Number of workers (usually CPU count)
# default value: %k (= number of CPU core, see Dockerfile) # default value: %k (= number of CPU core, see Dockerfile)
workers = 1 workers = 4
# Number of threads per worker # Number of threads per worker
# default value: 4 (see Dockerfile) # default value: 4 (see Dockerfile)
enable-threads = true enable-threads = 4
threads = 1 threads = 4
# The right granted on the created socket # The right granted on the created socket
chmod-socket = 666 chmod-socket = 666
+14 -7
View File
@@ -41,7 +41,7 @@ class BrowserAgent(Agent):
self.memory = Memory(self.load_prompt(prompt_path), self.memory = Memory(self.load_prompt(prompt_path),
recover_last_session=False, # session recovery in handled by the interaction class recover_last_session=False, # session recovery in handled by the interaction class
memory_compression=False, memory_compression=False,
model_provider=provider.get_model_name()) model_provider=provider.get_model_name() if provider else None)
def get_today_date(self) -> str: def get_today_date(self) -> str:
"""Get the date""" """Get the date"""
@@ -77,14 +77,14 @@ class BrowserAgent(Agent):
def get_unvisited_links(self) -> List[str]: def get_unvisited_links(self) -> List[str]:
return "\n".join([f"[{i}] {link}" for i, link in enumerate(self.navigable_links) if link not in self.search_history]) return "\n".join([f"[{i}] {link}" for i, link in enumerate(self.navigable_links) if link not in self.search_history])
def make_newsearch_prompt(self, user_prompt: str, search_result: dict) -> str: def make_newsearch_prompt(self, prompt: str, search_result: dict) -> str:
search_choice = self.stringify_search_results(search_result) search_choice = self.stringify_search_results(search_result)
self.logger.info(f"Search results: {search_choice}") self.logger.info(f"Search results: {search_choice}")
return f""" return f"""
Based on the search result: Based on the search result:
{search_choice} {search_choice}
Your goal is to find accurate and complete information to satisfy the users request. Your goal is to find accurate and complete information to satisfy the users request.
User request: {user_prompt} User request: {prompt}
To proceed, choose a relevant link from the search results. Announce your choice by saying: "I will navigate to <link>" To proceed, choose a relevant link from the search results. Announce your choice by saying: "I will navigate to <link>"
Do not explain your choice. Do not explain your choice.
""" """
@@ -235,13 +235,17 @@ class BrowserAgent(Agent):
return links return links
def select_link(self, links: List[str]) -> str | None: def select_link(self, links: List[str]) -> str | None:
"""
Select the first unvisited link that is not the current page.
Preference is given to links not in search_history.
"""
for lk in links: for lk in links:
if lk == self.current_page: if lk == self.current_page or lk in self.search_history:
self.logger.info(f"Already visited {lk}. Skipping.") self.logger.info(f"Skipping already visited or current link: {lk}")
continue continue
self.logger.info(f"Selected link: {lk}") self.logger.info(f"Selected link: {lk}")
return lk return lk
self.logger.warning("No link selected.") self.logger.warning("No suitable link selected.")
return None return None
def get_page_text(self, limit_to_model_ctx = False) -> str: def get_page_text(self, limit_to_model_ctx = False) -> str:
@@ -396,7 +400,10 @@ class BrowserAgent(Agent):
if (link == None and len(extracted_form) < 3) or Action.GO_BACK.value in answer or link in self.search_history: if (link == None and len(extracted_form) < 3) or Action.GO_BACK.value in answer or link in self.search_history:
pretty_print(f"Going back to results. Still {len(unvisited)}", color="status") pretty_print(f"Going back to results. Still {len(unvisited)}", color="status")
self.status_message = "Going back to search results..." self.status_message = "Going back to search results..."
prompt = self.make_newsearch_prompt(user_prompt, unvisited) request_prompt = user_prompt
if link is None:
request_prompt += f"\nYou previously choosen:\n{self.last_answer} but the website is unavailable. Consider other options."
prompt = self.make_newsearch_prompt(request_prompt, unvisited)
self.search_history.append(link) self.search_history.append(link)
self.current_page = link self.current_page = link
continue continue
+26 -8
View File
@@ -19,6 +19,7 @@ import time
import random import random
import os import os
import shutil import shutil
import uuid
import tempfile import tempfile
import markdownify import markdownify
import sys import sys
@@ -42,7 +43,14 @@ def get_chrome_path() -> str:
paths = ["/Applications/Google Chrome.app/Contents/MacOS/Google Chrome", paths = ["/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
"/Applications/Google Chrome Beta.app/Contents/MacOS/Google Chrome Beta"] "/Applications/Google Chrome Beta.app/Contents/MacOS/Google Chrome Beta"]
else: # Linux else: # Linux
paths = ["/usr/bin/google-chrome", "/usr/bin/chromium-browser", "/usr/bin/chromium", "/opt/chrome/chrome", "/usr/local/bin/chrome"] paths = ["/usr/bin/google-chrome",
"/opt/chrome/chrome",
"/usr/bin/chromium-browser",
"/usr/bin/chromium",
"/usr/local/bin/chrome",
"/opt/google/chrome/chrome-headless-shell",
#"/app/chrome_bundle/chrome136/chrome-linux64"
]
for path in paths: for path in paths:
if os.path.exists(path) and os.access(path, os.X_OK): if os.path.exists(path) and os.access(path, os.X_OK):
@@ -75,6 +83,7 @@ def install_chromedriver() -> str:
chromedriver_path = shutil.which("chromedriver") chromedriver_path = shutil.which("chromedriver")
if not chromedriver_path: if not chromedriver_path:
try: try:
print("ChromeDriver not found, attempting to install automatically...")
chromedriver_path = chromedriver_autoinstaller.install() chromedriver_path = chromedriver_autoinstaller.install()
except Exception as e: except Exception as e:
raise FileNotFoundError( raise FileNotFoundError(
@@ -120,17 +129,28 @@ def create_driver(headless=False, stealth_mode=True, crx_path="./crx/nopecha.crx
chrome_options.binary_location = chrome_path chrome_options.binary_location = chrome_path
if headless: if headless:
chrome_options.add_argument("--headless") #chrome_options.add_argument("--headless")
chrome_options.add_argument("--headless=new")
chrome_options.add_argument("--disable-gpu") chrome_options.add_argument("--disable-gpu")
chrome_options.add_argument("--disable-webgl") chrome_options.add_argument("--disable-webgl")
user_data_dir = tempfile.mkdtemp() user_data_dir = tempfile.mkdtemp()
user_agent = get_random_user_agent() user_agent = get_random_user_agent()
width, height = (1920, 1080) width, height = (1920, 1080)
chrome_options.add_argument(f"--user-data-dir={user_data_dir}") user_data_dir = tempfile.mkdtemp(prefix="chrome_profile_")
chrome_options.add_argument(f"--accept-lang={lang}-{lang.upper()},{lang};q=0.9")
chrome_options.add_argument("--timezone=Europe/Paris")
chrome_options.add_argument("--no-sandbox") chrome_options.add_argument("--no-sandbox")
chrome_options.add_argument("--disable-dev-shm-usage") chrome_options.add_argument('--disable-dev-shm-usage')
profile_dir = f"/tmp/chrome_profile_{uuid.uuid4().hex[:8]}"
chrome_options.add_argument(f'--user-data-dir={profile_dir}')
chrome_options.add_argument(f"--accept-lang={lang}-{lang.upper()},{lang};q=0.9")
chrome_options.add_argument("--disable-extensions")
chrome_options.add_argument("--disable-background-timer-throttling")
chrome_options.add_argument("--timezone=Europe/Paris")
chrome_options.add_argument('--remote-debugging-port=9222')
chrome_options.add_argument('--disable-background-timer-throttling')
chrome_options.add_argument('--disable-backgrounding-occluded-windows')
chrome_options.add_argument('--disable-renderer-backgrounding')
chrome_options.add_argument('--disable-features=TranslateUI')
chrome_options.add_argument('--disable-ipc-flooding-protection')
chrome_options.add_argument("--mute-audio") chrome_options.add_argument("--mute-audio")
chrome_options.add_argument("--disable-notifications") chrome_options.add_argument("--disable-notifications")
chrome_options.add_argument("--autoplay-policy=user-gesture-required") chrome_options.add_argument("--autoplay-policy=user-gesture-required")
@@ -698,8 +718,6 @@ if __name__ == "__main__":
input("press enter to continue") input("press enter to continue")
print("AntiCaptcha / Form Test") print("AntiCaptcha / Form Test")
browser.go_to("https://www.google.com/recaptcha/api2/demo")
time.sleep(50)
browser.go_to("https://bot.sannysoft.com") browser.go_to("https://bot.sannysoft.com")
time.sleep(5) time.sleep(5)
#txt = browser.get_text() #txt = browser.get_text()
+3 -43
View File
@@ -1,8 +1,6 @@
from typing import List, Tuple, Type, Dict from typing import List, Tuple, Type, Dict
import re import re
import langid import langid
import nltk
from nltk.sentiment.vader import SentimentIntensityAnalyzer
from transformers import MarianMTModel, MarianTokenizer from transformers import MarianMTModel, MarianTokenizer
from sources.utility import pretty_print, animate_thinking from sources.utility import pretty_print, animate_thinking
@@ -16,7 +14,6 @@ class LanguageUtility:
args: args:
supported_language: list of languages for translation, determine which Helsinki-NLP model to load supported_language: list of languages for translation, determine which Helsinki-NLP model to load
""" """
self.sid = None
self.translators_tokenizer = None self.translators_tokenizer = None
self.translators_model = None self.translators_model = None
self.logger = Logger("language.log") self.logger = Logger("language.log")
@@ -25,11 +22,6 @@ class LanguageUtility:
def load_model(self) -> None: def load_model(self) -> None:
animate_thinking("Loading language utility...", color="status") animate_thinking("Loading language utility...", color="status")
try:
nltk.data.find('vader_lexicon')
except LookupError:
nltk.download('vader_lexicon')
self.sid = SentimentIntensityAnalyzer()
self.translators_tokenizer = {lang: MarianTokenizer.from_pretrained(f"Helsinki-NLP/opus-mt-{lang}-en") for lang in self.supported_language if lang != "en"} self.translators_tokenizer = {lang: MarianTokenizer.from_pretrained(f"Helsinki-NLP/opus-mt-{lang}-en") for lang in self.supported_language if lang != "en"}
self.translators_model = {lang: MarianMTModel.from_pretrained(f"Helsinki-NLP/opus-mt-{lang}-en") for lang in self.supported_language if lang != "en"} self.translators_model = {lang: MarianMTModel.from_pretrained(f"Helsinki-NLP/opus-mt-{lang}-en") for lang in self.supported_language if lang != "en"}
@@ -65,49 +57,17 @@ class LanguageUtility:
translation = model.generate(**inputs) translation = model.generate(**inputs)
return tokenizer.decode(translation[0], skip_special_tokens=True) return tokenizer.decode(translation[0], skip_special_tokens=True)
def detect_emotion(self, text: str) -> str:
"""
Detect the dominant emotion in the given text
Args:
text: string to analyze
Returns: string of the dominant emotion
"""
try:
scores = self.sid.polarity_scores(text)
emotions = {
'Happy': max(scores['pos'], 0),
'Angry': 0,
'Sad': max(scores['neg'], 0),
'Fear': 0,
'Surprise': 0
}
if scores['compound'] < -0.5:
emotions['Angry'] = abs(scores['compound']) * 0.5
emotions['Fear'] = abs(scores['compound']) * 0.5
elif scores['compound'] > 0.5:
emotions['Happy'] = scores['compound']
emotions['Surprise'] = scores['compound'] * 0.5
dominant_emotion = max(emotions, key=emotions.get)
if emotions[dominant_emotion] == 0:
return 'Neutral'
self.logger.info(f"Emotion: {dominant_emotion} for text: {text}")
return dominant_emotion
except Exception as e:
raise e
def analyze(self, text): def analyze(self, text):
""" """
Combined analysis of language and emotion Combined analysis of language and emotion
Args: Args:
text: string to analyze text: string to analyze
Returns: dictionary with language and emotion results Returns: dictionary with language related information
""" """
try: try:
language = self.detect_language(text) language = self.detect_language(text)
emotions = self.detect_emotion(text)
return { return {
"language": language, "language": language
"emotions": emotions
} }
except Exception as e: except Exception as e:
raise e raise e
@@ -125,4 +85,4 @@ if __name__ == "__main__":
pretty_print(f"Language: {detector.detect_language(text)}", color="status") pretty_print(f"Language: {detector.detect_language(text)}", color="status")
result = detector.analyze(text) result = detector.analyze(text)
trans = detector.translate(text, result['language']) trans = detector.translate(text, result['language'])
pretty_print(f"Translation: {trans} - from: {result['language']} - Emotion: {result['emotions']}") pretty_print(f"Translation: {trans} - from: {result['language']}")
+34 -2
View File
@@ -3,11 +3,18 @@ from typing import List, Tuple, Type, Dict
import queue import queue
import threading import threading
import numpy as np import numpy as np
import torch
import time import time
from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
IMPORT_FOUND = True
try:
import torch
import librosa import librosa
import pyaudio import pyaudio
from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
except ImportError:
print(Fore.RED + "Speech To Text disabled." + Fore.RESET)
IMPORT_FOUND = False
audio_queue = queue.Queue() audio_queue = queue.Queue()
done = False done = False
@@ -23,6 +30,9 @@ class AudioRecorder:
self.chunk = chunk self.chunk = chunk
self.record_seconds = record_seconds self.record_seconds = record_seconds
self.verbose = verbose self.verbose = verbose
self.thread = None
self.audio = None
if IMPORT_FOUND:
self.audio = pyaudio.PyAudio() self.audio = pyaudio.PyAudio()
self.thread = threading.Thread(target=self._record, daemon=True) self.thread = threading.Thread(target=self._record, daemon=True)
@@ -30,6 +40,8 @@ class AudioRecorder:
""" """
Record audio from the microphone and add it to the audio queue. Record audio from the microphone and add it to the audio queue.
""" """
if not IMPORT_FOUND:
return
stream = self.audio.open(format=self.format, channels=self.channels, rate=self.rate, stream = self.audio.open(format=self.format, channels=self.channels, rate=self.rate,
input=True, frames_per_buffer=self.chunk) input=True, frames_per_buffer=self.chunk)
if self.verbose: if self.verbose:
@@ -58,10 +70,14 @@ class AudioRecorder:
def start(self) -> None: def start(self) -> None:
"""Start the recording thread.""" """Start the recording thread."""
if not IMPORT_FOUND:
return
self.thread.start() self.thread.start()
def join(self) -> None: def join(self) -> None:
"""Wait for the recording thread to finish.""" """Wait for the recording thread to finish."""
if not IMPORT_FOUND:
return
self.thread.join() self.thread.join()
class Transcript: class Transcript:
@@ -69,6 +85,9 @@ class Transcript:
Transcript is a class that transcribes audio from the audio queue and adds it to the transcript. Transcript is a class that transcribes audio from the audio queue and adds it to the transcript.
""" """
def __init__(self): def __init__(self):
if not IMPORT_FOUND:
print(Fore.RED + "Transcript: Speech to Text is disabled." + Fore.RESET)
return
self.last_read = None self.last_read = None
device = self.get_device() device = self.get_device()
torch_dtype = torch.float16 if device == "cuda" else torch.float32 torch_dtype = torch.float16 if device == "cuda" else torch.float32
@@ -91,6 +110,8 @@ class Transcript:
) )
def get_device(self) -> str: def get_device(self) -> str:
if not IMPORT_FOUND:
return "cpu"
if torch.backends.mps.is_available(): if torch.backends.mps.is_available():
return "mps" return "mps"
if torch.cuda.is_available(): if torch.cuda.is_available():
@@ -108,6 +129,8 @@ class Transcript:
def transcript_job(self, audio_data: np.ndarray, sample_rate: int = 16000) -> str: def transcript_job(self, audio_data: np.ndarray, sample_rate: int = 16000) -> str:
"""Transcribe the audio data.""" """Transcribe the audio data."""
if not IMPORT_FOUND:
return ""
if audio_data.dtype != np.float32: if audio_data.dtype != np.float32:
audio_data = audio_data.astype(np.float32) / np.iinfo(audio_data.dtype).max audio_data = audio_data.astype(np.float32) / np.iinfo(audio_data.dtype).max
if len(audio_data.shape) > 1: if len(audio_data.shape) > 1:
@@ -122,6 +145,9 @@ class AudioTranscriber:
AudioTranscriber is a class that transcribes audio from the audio queue and adds it to the transcript. AudioTranscriber is a class that transcribes audio from the audio queue and adds it to the transcript.
""" """
def __init__(self, ai_name: str, verbose: bool = False): def __init__(self, ai_name: str, verbose: bool = False):
if not IMPORT_FOUND:
print(Fore.RED + "AudioTranscriber: Speech to Text is disabled." + Fore.RESET)
return
self.verbose = verbose self.verbose = verbose
self.ai_name = ai_name self.ai_name = ai_name
self.transcriptor = Transcript() self.transcriptor = Transcript()
@@ -152,6 +178,8 @@ class AudioTranscriber:
""" """
Transcribe the audio data using AI stt model. Transcribe the audio data using AI stt model.
""" """
if not IMPORT_FOUND:
return
global done global done
if self.verbose: if self.verbose:
print(Fore.BLUE + "AudioTranscriber: Started processing..." + Fore.RESET) print(Fore.BLUE + "AudioTranscriber: Started processing..." + Fore.RESET)
@@ -185,9 +213,13 @@ class AudioTranscriber:
def start(self): def start(self):
"""Start the transcription thread.""" """Start the transcription thread."""
if not IMPORT_FOUND:
return
self.thread.start() self.thread.start()
def join(self): def join(self):
if not IMPORT_FOUND:
return
"""Wait for the transcription thread to finish.""" """Wait for the transcription thread to finish."""
self.thread.join() self.thread.join()
+8 -2
View File
@@ -5,9 +5,14 @@ import subprocess
from sys import modules from sys import modules
from typing import List, Tuple, Type, Dict from typing import List, Tuple, Type, Dict
IMPORT_FOUND = True
try:
from kokoro import KPipeline from kokoro import KPipeline
from IPython.display import display, Audio from IPython.display import display, Audio
import soundfile as sf import soundfile as sf
except ImportError:
print("Speech synthesis disabled. Please install the kokoro package.")
IMPORT_FOUND = False
if __name__ == "__main__": if __name__ == "__main__":
from utility import pretty_print, animate_thinking from utility import pretty_print, animate_thinking
@@ -33,7 +38,7 @@ class Speech():
} }
self.pipeline = None self.pipeline = None
self.language = language self.language = language
if enable: if enable and IMPORT_FOUND:
self.pipeline = KPipeline(lang_code=self.lang_map[language]) self.pipeline = KPipeline(lang_code=self.lang_map[language])
self.voice = self.voice_map[language][voice_idx] self.voice = self.voice_map[language][voice_idx]
self.speed = 1.2 self.speed = 1.2
@@ -57,7 +62,7 @@ class Speech():
sentence (str): The text to convert to speech. Will be pre-processed. sentence (str): The text to convert to speech. Will be pre-processed.
voice_idx (int, optional): Index of the voice to use from the voice map. voice_idx (int, optional): Index of the voice to use from the voice map.
""" """
if not self.pipeline: if not self.pipeline or not IMPORT_FOUND:
return return
if voice_idx >= len(self.voice_map[self.language]): if voice_idx >= len(self.voice_map[self.language]):
pretty_print("Invalid voice number, using default voice", color="error") pretty_print("Invalid voice number, using default voice", color="error")
@@ -159,6 +164,7 @@ class Speech():
if __name__ == "__main__": if __name__ == "__main__":
# TODO add info message for cn2an, jieba chinese related import # TODO add info message for cn2an, jieba chinese related import
IMPORT_FOUND = False
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
speech = Speech() speech = Speech()
tosay_en = """ tosay_en = """
+12 -17
View File
@@ -50,19 +50,15 @@ class Tools():
def set_allow_language_exec_bash(value: bool) -> None: def set_allow_language_exec_bash(value: bool) -> None:
self.allow_language_exec_bash = value self.allow_language_exec_bash = value
def check_config_dir_validity(self): def safe_get_work_dir_path(self):
"""Check if the config directory is valid.""" path = None
path = self.config['MAIN']['work_dir'] path = os.getenv('WORK_DIR', path)
if path == "": if path is None or path == "":
print("WARNING: Work directory not set in config.ini") path = self.config['MAIN']['work_dir'] if 'MAIN' in self.config and 'work_dir' in self.config['MAIN'] else None
return False if path is None or path == "":
if path.lower() == "none": print("No work directory specified, using default.")
print("WARNING: Work directory set to none in config.ini") path = self.create_work_dir()
return False return path
if not os.path.exists(path):
print(f"WARNING: Work directory {path} does not exist")
return False
return True
def config_exists(self): def config_exists(self):
"""Check if the config file exists.""" """Check if the config file exists."""
@@ -73,11 +69,10 @@ class Tools():
default_path = os.path.dirname(os.getcwd()) default_path = os.path.dirname(os.getcwd())
if self.config_exists(): if self.config_exists():
self.config.read('./config.ini') self.config.read('./config.ini')
config_path = self.config['MAIN']['work_dir'] workdir_path = self.safe_get_work_dir_path()
dir_path = default_path if not self.check_config_dir_validity() else config_path
else: else:
dir_path = default_path workdir_path = default_path
return dir_path return workdir_path
@abstractmethod @abstractmethod
def execute(self, blocks:[str], safety:bool) -> str: def execute(self, blocks:[str], safety:bool) -> str:
+31 -6
View File
@@ -1,10 +1,35 @@
@echo off @echo off
docker-compose up if "%1"=="full" (
if %ERRORLEVEL% neq 0 ( echo Starting full deployment...
echo Error: Failed to start containers. Check Docker logs with 'docker compose logs'. ) else (
echo Possible fixes: Ensure Docker Desktop is running or check if port 8080 is free. echo Starting partial deployment... (backend run on host), use "full" to run all services in containers
exit /b 1
) )
timeout /t 10 /nobreak >nul where openssl >nul 2>&1
if %ERRORLEVEL% == 0 (
for /f %%i in ('openssl rand -hex 32') do set SEARXNG_SECRET_KEY=%%i
) else (
where python3 >nul 2>&1
if %ERRORLEVEL% == 0 (
for /f %%i in ('python3 -c "import secrets; print(secrets.token_hex(32))"') do set SEARXNG_SECRET_KEY=%%i
) else (
echo Error: Neither openssl nor python is available to generate a secret key.
exit /b 1
)
)
REM Stop all containers
echo Stopping containers...
docker stop $(docker ps -aq) >nul 2>&1
REM Generate secret key
for /f %%i in ('powershell -command "[System.Web.Security.Membership]::GeneratePassword(64,0)"') do set SEARXNG_SECRET_KEY=%%i
if "%1"=="full" (
docker compose up -d backend
timeout /t 5 /nobreak >nul
docker compose --profile full up
) else (
docker compose --profile core up
)
+70 -5
View File
@@ -1,12 +1,35 @@
#!/bin/bash #!/bin/bash
source .env
command_exists() { command_exists() {
command -v "$1" &> /dev/null command -v "$1" &> /dev/null
} }
if [ -z "$WORK_DIR" ]; then
echo "Error: WORK_DIR environment variable is not set. Please set it in your .env file."
exit 1
fi
# if [[ "$OSTYPE" == "darwin"* ]]; then
# Check if Docker is installed é running dir_size_bytes=$(du -s -b "$WORK_DIR" 2>/dev/null | awk '{print $1}')
# else
dir_size_bytes=$(du -s --bytes "$WORK_DIR" 2>/dev/null | awk '{print $1}')
fi
max_size_bytes=$((2 * 1024 * 1024 * 1024))
echo "Mounting $WORK_DIR ($dir_size_bytes bytes) to docker."
if [ "$dir_size_bytes" -gt "$max_size_bytes" ]; then
echo "Error: WORK_DIR ($WORK_DIR) contains more than 2GB of data ($(du -sh "$WORK_DIR" 2>/dev/null | awk '{print $1}'))."
exit 1
fi
if [ "$1" = "full" ]; then
echo "Starting full deployment with backend and all services..."
else
echo "Starting core deployment with frontend and search services only... use ./start_services.sh full to start backend as well"
fi
if ! command_exists docker; then if ! command_exists docker; then
echo "Error: Docker is not installed. Please install Docker first." echo "Error: Docker is not installed. Please install Docker first."
@@ -60,15 +83,57 @@ if [ ! -f "docker-compose.yml" ]; then
exit 1 exit 1
fi fi
# start docker compose for searxng, redis, frontend services # Stop all running containers to ensure a clean state
echo "Warning: stopping all docker containers (t-4 seconds)..." echo "Warning: stopping all docker containers (t-4 seconds)..."
sleep 4 sleep 4
docker stop $(docker ps -a -q) docker stop $(docker ps -a -q)
echo "All containers stopped" echo "All containers stopped"
if ! $COMPOSE_CMD up; then # export searxng secret key (cross-platform)
if command -v openssl &> /dev/null; then
export SEARXNG_SECRET_KEY=$(openssl rand -hex 32)
else
# Fallback: use Python if openssl is not available
if command -v python3 &> /dev/null; then
export SEARXNG_SECRET_KEY=$(python3 -c "import secrets; print(secrets.token_hex(32))")
else
echo "Error: Neither openssl nor python is available to generate a secret key."
exit 1
fi
fi
if [ "$1" = "full" ]; then
# First start backend and wait for it to be healthy
echo "Full docker deployement. Starting backend service..."
if ! $COMPOSE_CMD up -d backend; then
echo "Error: Failed to start backend container."
exit 1
fi
# Wait for backend to be healthy (check if it's running and not restarting)
echo "Waiting for backend to be ready..."
for i in {1..30}; do
if [ "$(docker inspect -f '{{.State.Running}}' backend)" = "true" ] && \
[ "$(docker inspect -f '{{.State.Restarting}}' backend)" = "false" ]; then
echo "backend is ready!"
break
fi
if [ $i -eq 30 ]; then
echo "Error: backend failed to start properly after 30 seconds"
$COMPOSE_CMD logs backend
exit 1
fi
sleep 1
done
if ! $COMPOSE_CMD --profile full up; then
echo "Error: Failed to start containers. Check Docker logs with '$COMPOSE_CMD logs'." echo "Error: Failed to start containers. Check Docker logs with '$COMPOSE_CMD logs'."
echo "Possible fixes: Run with sudo or ensure port 8080 is free." echo "Possible fixes: Run with sudo or ensure port 8080 is free."
exit 1 exit 1
fi fi
else
if ! $COMPOSE_CMD --profile core up; then
echo "Error: Failed to start containers. Check Docker logs with '$COMPOSE_CMD logs'."
echo "Possible fixes: Run with sudo or ensure port 8080 is free."
exit 1
fi
fi
sleep 10 sleep 10
+1 -1
View File
@@ -23,7 +23,7 @@ class TestBrowserAgentParsing(unittest.TestCase):
"https://thriveonai.com/15-ai-startups-in-japan-to-take-note-of", "https://thriveonai.com/15-ai-startups-in-japan-to-take-note-of",
"www.google.com", "www.google.com",
"https://test.org/about?page=1", "https://test.org/about?page=1",
"https://weatherstack.com/documentation", "https://weatherstack.com/documentation"
] ]
result = self.agent.extract_links(test_text) result = self.agent.extract_links(test_text)
self.assertEqual(result, expected) self.assertEqual(result, expected)