fix: limit elastic search memory usage

This commit is contained in:
青轩 2024-09-11 13:30:58 +08:00
parent a7d8430cc1
commit 40d138d41d
4 changed files with 8 additions and 15 deletions

View file

@ -11,9 +11,9 @@
# sudo docker build --network=host -t memoryscope .
# To run docker image:
# sudo docker run -it --rm --net=host memoryscope
# sudo docker run -it --rm --memory=4G --net=host memoryscope
# To run docker image with arguments (refer to memoryscope/core/config/arguments.py):
# sudo docker run -it --rm --net=host -e "OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" -e "language=en" -e "human_name=superman" -e "generation_backend=openai_generation" -e "generation_model=gpt-4o" -e "embedding_backend=openai_embedding" -e "embedding_model=text-embedding-3-small" -e "enable_ranker=False" memoryscope
# sudo docker run -it --rm --memory=4G --net=host -e "OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" -e "language=en" -e "human_name=superman" -e "generation_backend=openai_generation" -e "generation_model=gpt-4o" -e "embedding_backend=openai_embedding" -e "embedding_model=text-embedding-3-small" -e "enable_ranker=False" memoryscope
FROM python:3.11

View file

@ -9,14 +9,7 @@ services:
# OPENAI_API_KEY: "sk-0000000000"
volumes:
- ./memoryscope/core/config:/memory_scope_project/memoryscope/memoryscope/core/config
depends_on:
- elasticsearch
elasticsearch:
image: docker.elastic.co/elasticsearch/elasticsearch:8.13.2
ports:
- "9200:9200"
environment:
discovery.type: "single-node"
xpack.security.enabled: "false"
xpack.license.self_generated.type: "trial"
deploy:
resources:
limits:
memory: 4G

View file

@ -38,7 +38,7 @@
OPENAI_API_KEY: "sk-0000000000"
```
3. Run `docker-compose up` to build and launch the memory-scope cli interface.
3. Run `docker-compose run memory_scope_main` to build and launch the memory-scope cli interface.
## III. Install from PyPI

View file

@ -38,7 +38,7 @@
DASHSCOPE_API_KEY: "sk-0000000000"
```
3. 运行 `docker-compose up` 命令来构建并启动 MemoryScope CLI 界面。
3. 运行 `docker-compose run memory_scope_main` 命令来构建并启动 MemoryScope CLI 界面。
## 三、通过 PYPI 安装