Skip to content

Commit fa05612

Browse files
authored
Update docker config to include default assistant persona and update logging config (#6)
Initialize log and prevent joining thread for async consumer support Update Dockerfile to set default config path
1 parent dd777cf commit fa05612

File tree

3 files changed

+24
-10
lines changed

3 files changed

+24
-10
lines changed

Dockerfile

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,12 @@ FROM python:3.9-slim
33
LABEL vendor=neon.ai \
44
ai.neon.name="neon-llm-gemini"
55

6-
ENV OVOS_CONFIG_BASE_FOLDER neon
7-
ENV OVOS_CONFIG_FILENAME diana.yaml
8-
ENV XDG_CONFIG_HOME /config
6+
ENV OVOS_CONFIG_BASE_FOLDER=neon
7+
ENV OVOS_CONFIG_FILENAME=diana.yaml
8+
ENV OVOS_DEFAULT_CONFIG=/opt/neon/diana.yaml
9+
ENV XDG_CONFIG_HOME=/config
10+
ENV CHATBOT_VERSION=v2
11+
912
COPY docker_overlay/ /
1013

1114
WORKDIR /app

docker_overlay/etc/neon/diana.yaml renamed to docker_overlay/opt/neon/diana.yaml

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,15 @@ logs:
55
- pika
66
warning:
77
- filelock
8-
info: []
8+
- watchdog
9+
- httpcore
10+
info:
11+
- openai
12+
- asyncio
13+
- matplotlib
914
debug: []
1015
MQ:
11-
server: api.neon.ai
16+
server: neon-rabbitmq
1217
port: 5672
1318
users:
1419
mq_handler:
@@ -19,4 +24,8 @@ LLM_GEMINI:
1924
role: "You are trying to give a short answer in less than 40 words."
2025
context_depth: 3
2126
max_tokens: 100
22-
num_parallel_processes: 2
27+
num_parallel_processes: 2
28+
llm_bots:
29+
gemini:
30+
- description: You are trying to give a short answer in less than 40 words.
31+
name: assistant

neon_llm_gemini/__main__.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,16 @@
2525
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2626

2727
from neon_llm_gemini.rmq import GeminiMQ
28+
from neon_utils.log_utils import init_log
2829

2930

3031
def main():
32+
init_log(log_name="gemini")
33+
3134
# Run RabbitMQ
32-
geminiMQ = GeminiMQ()
33-
geminiMQ.run(run_sync=False, run_consumers=True,
34-
daemonize_consumers=True)
35-
geminiMQ.observer_thread.join()
35+
gemini_mq_service = GeminiMQ()
36+
gemini_mq_service.run(run_sync=False, run_consumers=True,
37+
daemonize_consumers=True)
3638

3739

3840
if __name__ == "__main__":

0 commit comments

Comments
 (0)