fixups

Author: mudler

docker-compose.yaml

@@ -63,7 +63,7 @@ services:
       dockerfile: Dockerfile.webui
     ports:
       - 8080:3000
-    image: quay.io/mudler/localagi:master
+    #image: quay.io/mudler/localagi:master
     environment:
       - LOCALAGI_MODEL=arcee-agent
       - LOCALAGI_LLM_API_URL=http://localai:8080

main.go

@@ -11,7 +11,7 @@ import (
 	"github.com/mudler/LocalAGI/webui"
 )
 
-var testModel = os.Getenv("LOCALAGI_MODEL")
+var baseModel = os.Getenv("LOCALAGI_MODEL")
 var multimodalModel = os.Getenv("LOCALAGI_MULTIMODAL_MODEL")
 var apiURL = os.Getenv("LOCALAGI_LLM_API_URL")
 var apiKey = os.Getenv("LOCALAGI_LLM_API_KEY")
@@ -24,11 +24,11 @@ var imageModel = os.Getenv("LOCALAGI_IMAGE_MODEL")
 var conversationDuration = os.Getenv("LOCALAGI_CONVERSATION_DURATION")
 
 func init() {
-	if testModel == "" {
-		testModel = "hermes-2-pro-mistral"
+	if baseModel == "" {
+		panic("LOCALAGI_MODEL not set")
 	}
 	if apiURL == "" {
-		apiURL = "http://192.168.68.113:8080"
+		panic("LOCALAGI_API_URL not set")
 	}
 	if timeout == "" {
 		timeout = "5m"
@@ -54,7 +54,7 @@ func main() {
 
 	// Create the agent pool
 	pool, err := state.NewAgentPool(
-		testModel,
+		baseModel,
 		multimodalModel,
 		imageModel,
 		apiURL,
@@ -78,7 +78,7 @@ func main() {
 		webui.WithApiKeys(apiKeys...),
 		webui.WithLLMAPIUrl(apiURL),
 		webui.WithLLMAPIKey(apiKey),
-		webui.WithLLMModel(testModel),
+		webui.WithLLMModel(baseModel),
 		webui.WithStateDir(stateDir),
 	)
 

tests/e2e/e2e_suite_test.go

@@ -13,15 +13,5 @@ func TestE2E(t *testing.T) {
 	RunSpecs(t, "E2E test suite")
 }
 
-var testModel = os.Getenv("LOCALAGI_MODEL")
 var apiURL = os.Getenv("LOCALAI_API_URL")
 var localagiURL = os.Getenv("LOCALAGI_API_URL")
-
-func init() {
-	if testModel == "" {
-		testModel = "hermes-2-pro-mistral"
-	}
-	if apiURL == "" {
-		apiURL = "http://192.168.68.113:8080"
-	}
-}