From 8963bbc86ddfa85fcbd8c3b217f4c56a0a8598ad Mon Sep 17 00:00:00 2001 From: Mario Candela Date: Thu, 20 Feb 2025 22:41:28 +0100 Subject: [PATCH] =?UTF-8?q?Fix:=20mapping=20LLMModel=20for=20SSH=20inline,?= =?UTF-8?q?=20removed=20old=20comments=20on=20docker-c=E2=80=A6=20(#168)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fix mapping LLMModel for SSH inline, removed old comments on docker-compose.yml --- docker-compose.yml | 5 ++--- protocols/strategies/ssh.go | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 30b7006..30b234b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -3,17 +3,16 @@ version: "3.9" services: beelzebub: build: . - #network_mode: host # Not work on Mac OS container_name: beelzebub restart: always - ports: # Remove me, if you use configuration network_mode: host + ports: - "22:22" - "2222:2222" - "8080:8080" - "8081:8081" - "80:80" - "3306:3306" - - "2112:2112" # Prometheus openmetrics + - "2112:2112" #Prometheus Open Metrics environment: RABBITMQ_URI: ${RABBITMQ_URI} volumes: diff --git a/protocols/strategies/ssh.go b/protocols/strategies/ssh.go index 63a916a..6cb7af2 100644 --- a/protocols/strategies/ssh.go +++ b/protocols/strategies/ssh.go @@ -57,7 +57,7 @@ func (sshStrategy *SSHStrategy) Init(beelzebubServiceConfiguration parser.Beelze OpenAIKey: beelzebubServiceConfiguration.Plugin.OpenAISecretKey, Protocol: tracer.SSH, Host: beelzebubServiceConfiguration.Plugin.Host, - Model: beelzebubServiceConfiguration.Plugin.LLMProvider, + Model: beelzebubServiceConfiguration.Plugin.LLMModel, Provider: llmProvider, CustomPrompt: beelzebubServiceConfiguration.Plugin.Prompt, }