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, }