Skip to content

Commit 6f51e7b

Browse files
authored
Feature model initialization (#7)
* Enhance devcontainer setup and improve RAG functionality - Updated devcontainer configuration to include multiple docker-compose files. - Added new docker-compose.devcontainer.yml for devcontainer overriddes - Refined environment variables for text and image collections in .env file. - Improved prompt generation in rag.py for better user interaction. - Streamlined chat.sh for continuous question and image identification modes. - Adjusted docker-compose.yml to use model_cache for caching. * Refactor prompts for clarity and implement lazy loading for models in MemoryAlphaRAG
1 parent 6b05a44 commit 6f51e7b

File tree

6 files changed

+299
-247
lines changed

6 files changed

+299
-247
lines changed

.devcontainer/devcontainer.json

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,14 @@
11
{
22
"name": "Memoryalpha RAG API",
3-
"dockerComposeFile": "../docker-compose.yml",
3+
"dockerComposeFile": [
4+
"../docker-compose.yml",
5+
"docker-compose.devcontainer.yml"
6+
],
47
"service": "lcars",
58
"overrideCommand": true,
69
"remoteUser": "vscode",
710
"postAttachCommand": "/workspace/memoryalpha-rag-api/wait-for-ollama.sh",
11+
"postCreateCommand": "sudo chown -R vscode:users ${containerWorkspaceFolder}/.cache",
812
"features": {
913
"ghcr.io/nils-geistmann/devcontainers-features/create-remote-user:0": {
1014
"passwordLessSudo": true
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
version: "3.9"
2+
3+
services:
4+
lcars:
5+
build:
6+
context: .
7+
dockerfile: Dockerfile
8+
user: vscode
9+
volumes:
10+
- ./.cache:/home/vscode/.cache # Override default cache location for devcontainer only

.env

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,5 @@ DEFAULT_IMAGE_MODEL="qwen2.5vl:3b"
33

44
OLLAMA_URL="http://ollama:11434"
55
DB_PATH="/data/enmemoryalpha_db"
6-
COLLECTION_NAME="memoryalpha"
6+
TEXT_COLLECTION_NAME="memoryalpha_text"
7+
IMAGE_COLLECTION_NAME="memoryalpha_images"

0 commit comments

Comments
 (0)