alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixups

Browse Source
This commit is contained in:
Ettore Di Giacinto
2025-04-09 21:25:44 +02:00
parent e128cde613
commit 2f19feff5e
3 changed files with 7 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yaml
View File

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

12
main.go
View File

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

10
tests/e2e/e2e_suite_test.go
View File

@@ -13,15 +13,5 @@ func TestE2E(t *testing.T) {
RunSpecs(t, "E2E test suite") RunSpecs(t, "E2E test suite")
} }
var testModel = os.Getenv("LOCALAGI_MODEL")
var apiURL = os.Getenv("LOCALAI_API_URL") var apiURL = os.Getenv("LOCALAI_API_URL")
var localagiURL = os.Getenv("LOCALAGI_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"
}
}