diff --git a/Dockerfile b/Dockerfile index 067fa96..2a6fdb5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,9 +3,12 @@ FROM python:3.9-slim LABEL vendor=neon.ai \ ai.neon.name="neon-llm-gemini" -ENV OVOS_CONFIG_BASE_FOLDER neon -ENV OVOS_CONFIG_FILENAME diana.yaml -ENV XDG_CONFIG_HOME /config +ENV OVOS_CONFIG_BASE_FOLDER=neon +ENV OVOS_CONFIG_FILENAME=diana.yaml +ENV OVOS_DEFAULT_CONFIG=/opt/neon/diana.yaml +ENV XDG_CONFIG_HOME=/config +ENV CHATBOT_VERSION=v2 + COPY docker_overlay/ / WORKDIR /app diff --git a/docker_overlay/etc/neon/diana.yaml b/docker_overlay/opt/neon/diana.yaml similarity index 56% rename from docker_overlay/etc/neon/diana.yaml rename to docker_overlay/opt/neon/diana.yaml index c65890f..6abb49b 100644 --- a/docker_overlay/etc/neon/diana.yaml +++ b/docker_overlay/opt/neon/diana.yaml @@ -5,10 +5,15 @@ logs: - pika warning: - filelock - info: [] + - watchdog + - httpcore + info: + - openai + - asyncio + - matplotlib debug: [] MQ: - server: api.neon.ai + server: neon-rabbitmq port: 5672 users: mq_handler: @@ -19,4 +24,8 @@ LLM_GEMINI: role: "You are trying to give a short answer in less than 40 words." context_depth: 3 max_tokens: 100 - num_parallel_processes: 2 \ No newline at end of file + num_parallel_processes: 2 +llm_bots: + gemini: + - description: You are trying to give a short answer in less than 40 words. + name: assistant \ No newline at end of file diff --git a/neon_llm_gemini/__main__.py b/neon_llm_gemini/__main__.py index 393bfb8..8388800 100644 --- a/neon_llm_gemini/__main__.py +++ b/neon_llm_gemini/__main__.py @@ -25,14 +25,16 @@ # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. from neon_llm_gemini.rmq import GeminiMQ +from neon_utils.log_utils import init_log def main(): + init_log(log_name="gemini") + # Run RabbitMQ - geminiMQ = GeminiMQ() - geminiMQ.run(run_sync=False, run_consumers=True, - daemonize_consumers=True) - geminiMQ.observer_thread.join() + gemini_mq_service = GeminiMQ() + gemini_mq_service.run(run_sync=False, run_consumers=True, + daemonize_consumers=True) if __name__ == "__main__":