feat: improve action picking

Signed-off-by: mudler <mudler@localai.io>
Add Github reviewer and improve reasoning
2025-04-11 20:43:48 +02:00 · 2025-04-10 23:36:44 +02:00
24 changed files with 334 additions and 1163 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -3,7 +3,7 @@ name: Run Go Tests
 on:
  push:
    branches:
-      - 'main'
+      - '**'
  pull_request:
    branches:
      - '**'
--- a/2
+++ b/2
@@ -3,7 +3,7 @@ IMAGE_NAME?=webui
 ROOT_DIR:=$(shell dirname $(realpath $(lastword $(MAKEFILE_LIST))))

 prepare-tests:
-	docker compose up -d --build
+	docker compose up -d

 cleanup-tests:
 	docker compose down
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 <p align="center">
-  <img src="./webui/react-ui/public/logo_1.png" alt="LocalAGI Logo" width="220"/>
+  <img src="https://github.com/user-attachments/assets/6958ffb3-31cf-441e-b99d-ce34ec6fc88f" alt="LocalAGI Logo" width="220"/>
 </p>

 <h3 align="center"><em>Your AI. Your Hardware. Your Rules.</em></h3>
@@ -45,129 +45,14 @@ LocalAGI ensures your data stays exactly where you want it—on your hardware. N
 git clone https://github.com/mudler/LocalAGI
 cd LocalAGI

-# CPU setup (default)
-docker compose up
+# CPU setup
+docker compose up -f docker-compose.yml

-# NVIDIA GPU setup
-docker compose -f docker-compose.nvidia.yaml up
-
-# Intel GPU setup (for Intel Arc and integrated GPUs)
-docker compose -f docker-compose.intel.yaml up
-
-# Start with a specific model (see available models in models.localai.io, or localai.io to use any model in huggingface)
-MODEL_NAME=gemma-3-12b-it docker compose up
-
-# NVIDIA GPU setup with custom multimodal and image models
-MODEL_NAME=gemma-3-12b-it \
-MULTIMODAL_MODEL=minicpm-v-2_6 \
-IMAGE_MODEL=flux.1-dev \
-docker compose -f docker-compose.nvidia.yaml up
+# GPU setup
+docker compose up -f docker-compose.gpu.yml
 ```

-Now you can access and manage your agents at [http://localhost:8080](http://localhost:8080)
-
-## 📚🆕 Local Stack Family
-
-🆕 LocalAI is now part of a comprehensive suite of AI tools designed to work together:
-
-<table>
-  <tr>
-    <td width="50%" valign="top">
-      <a href="https://github.com/mudler/LocalAI">
-        <img src="https://raw.githubusercontent.com/mudler/LocalAI/refs/heads/master/core/http/static/logo_horizontal.png" width="300" alt="LocalAI Logo">
-      </a>
-    </td>
-    <td width="50%" valign="top">
-      <h3><a href="https://github.com/mudler/LocalAI">LocalAI</a></h3>
-      <p>LocalAI is the free, Open Source OpenAI alternative. LocalAI act as a drop-in replacement REST API that's compatible with OpenAI API specifications for local AI inferencing. Does not require GPU.</p>
-    </td>
-  </tr>
-  <tr>
-    <td width="50%" valign="top">
-      <a href="https://github.com/mudler/LocalRecall">
-        <img src="https://raw.githubusercontent.com/mudler/LocalRecall/refs/heads/main/static/localrecall_horizontal.png" width="300" alt="LocalRecall Logo">
-      </a>
-    </td>
-    <td width="50%" valign="top">
-      <h3><a href="https://github.com/mudler/LocalRecall">LocalRecall</a></h3>
-      <p>A REST-ful API and knowledge base management system that provides persistent memory and storage capabilities for AI agents.</p>
-    </td>
-  </tr>
-</table>
-
-## 🖥️ Hardware Configurations
-
-LocalAGI supports multiple hardware configurations through Docker Compose profiles:
-
-### CPU (Default)
- No special configuration needed
- Runs on any system with Docker
- Best for testing and development
- Supports text models only
-
-### NVIDIA GPU
- Requires NVIDIA GPU and drivers
- Uses CUDA for acceleration
- Best for high-performance inference
- Supports text, multimodal, and image generation models
- Run with: `docker compose -f docker-compose.nvidia.yaml up`
- Default models:
-  - Text: `arcee-agent`
-  - Multimodal: `minicpm-v-2_6`
-  - Image: `flux.1-dev`
- Environment variables:
-  - `MODEL_NAME`: Text model to use
-  - `MULTIMODAL_MODEL`: Multimodal model to use
-  - `IMAGE_MODEL`: Image generation model to use
-  - `LOCALAI_SINGLE_ACTIVE_BACKEND`: Set to `true` to enable single active backend mode
-
-### Intel GPU
- Supports Intel Arc and integrated GPUs
- Uses SYCL for acceleration
- Best for Intel-based systems
- Supports text, multimodal, and image generation models
- Run with: `docker compose -f docker-compose.intel.yaml up`
- Default models:
-  - Text: `arcee-agent`
-  - Multimodal: `minicpm-v-2_6`
-  - Image: `sd-1.5-ggml`
- Environment variables:
-  - `MODEL_NAME`: Text model to use
-  - `MULTIMODAL_MODEL`: Multimodal model to use
-  - `IMAGE_MODEL`: Image generation model to use
-  - `LOCALAI_SINGLE_ACTIVE_BACKEND`: Set to `true` to enable single active backend mode
-
-## Customize models
-
-You can customize the models used by LocalAGI by setting environment variables when running docker-compose. For example:
-
-```bash
-# CPU with custom model
-MODEL_NAME=gemma-3-12b-it docker compose up
-
-# NVIDIA GPU with custom models
-MODEL_NAME=gemma-3-12b-it \
-MULTIMODAL_MODEL=minicpm-v-2_6 \
-IMAGE_MODEL=flux.1-dev \
-docker compose -f docker-compose.nvidia.yaml up
-
-# Intel GPU with custom models
-MODEL_NAME=gemma-3-12b-it \
-MULTIMODAL_MODEL=minicpm-v-2_6 \
-IMAGE_MODEL=sd-1.5-ggml \
-docker compose -f docker-compose.intel.yaml up
-```
-
-If no models are specified, it will use the defaults:
- Text model: `arcee-agent`
- Multimodal model: `minicpm-v-2_6`
- Image model: `flux.1-dev` (NVIDIA) or `sd-1.5-ggml` (Intel)
-
-Good (relatively small) models that have been tested are:
-
- `qwen_qwq-32b` (best in co-ordinating agents)
- `gemma-3-12b-it`
- `gemma-3-27b-it`
+Access your agents at `http://localhost:8080`

 ## 🏆 Why Choose LocalAGI?

@@ -213,8 +98,6 @@ Explore detailed documentation including:

 ### Environment Configuration

-LocalAGI supports environment configurations. Note that these environment variables needs to be specified in the localagi container in the docker-compose file to have effect.
-
 | Variable | What It Does |
 |----------|--------------|
 | `LOCALAGI_MODEL` | Your go-to model |
--- a/core/action/goal.go
+++ b/core/action/goal.go
@@ -10,11 +10,12 @@ import (
 // NewGoal creates a new intention action
 // The inention action is special as it tries to identify
 // a tool to use and a reasoning over to use it
-func NewGoal() *GoalAction {
-	return &GoalAction{}
+func NewGoal(s ...string) *GoalAction {
+	return &GoalAction{tools: s}
 }

 type GoalAction struct {
+	tools []string
 }
 type GoalResponse struct {
 	Goal     string `json:"goal"`
--- a/core/action/plan.go
+++ b/core/action/plan.go
@@ -41,7 +41,7 @@ func (a *PlanAction) Plannable() bool {
 func (a *PlanAction) Definition() types.ActionDefinition {
 	return types.ActionDefinition{
 		Name:        PlanActionName,
-		Description: "Use it for situations that involves doing more actions in sequence.",
+		Description: "Use this tool for solving complex tasks that involves calling more tools in sequence.",
 		Properties: map[string]jsonschema.Definition{
 			"subtasks": {
 				Type:        jsonschema.Array,
--- a/core/agent/actions.go
+++ b/core/agent/actions.go
@@ -24,27 +24,15 @@ type decisionResult struct {
 func (a *Agent) decision(
 	ctx context.Context,
 	conversation []openai.ChatCompletionMessage,
-	tools []openai.Tool, toolchoice string, maxRetries int) (*decisionResult, error) {
-
-	var choice *openai.ToolChoice
-
-	if toolchoice != "" {
-		choice = &openai.ToolChoice{
-			Type:     openai.ToolTypeFunction,
-			Function: openai.ToolFunction{Name: toolchoice},
-		}
-	}
+	tools []openai.Tool, toolchoice any, maxRetries int) (*decisionResult, error) {

 	var lastErr error
 	for attempts := 0; attempts < maxRetries; attempts++ {
 		decision := openai.ChatCompletionRequest{
-			Model:    a.options.LLMAPI.Model,
-			Messages: conversation,
-			Tools:    tools,
-		}
-
-		if choice != nil {
-			decision.ToolChoice = *choice
+			Model:      a.options.LLMAPI.Model,
+			Messages:   conversation,
+			Tools:      tools,
+			ToolChoice: toolchoice,
 		}

 		resp, err := a.client.CreateChatCompletion(ctx, decision)
@@ -54,9 +42,6 @@ func (a *Agent) decision(
 			continue
 		}

-		jsonResp, _ := json.Marshal(resp)
-		xlog.Debug("Decision response", "response", string(jsonResp))
-
 		if len(resp.Choices) != 1 {
 			lastErr = fmt.Errorf("no choices: %d", len(resp.Choices))
 			xlog.Warn("Attempt to make a decision failed", "attempt", attempts+1, "error", lastErr)
@@ -204,7 +189,10 @@ func (a *Agent) generateParameters(ctx context.Context, pickTemplate string, act
 		result, attemptErr = a.decision(ctx,
 			cc,
 			a.availableActions().ToTools(),
-			act.Definition().Name.String(),
+			openai.ToolChoice{
+				Type:     openai.ToolTypeFunction,
+				Function: openai.ToolFunction{Name: act.Definition().Name.String()},
+			},
 			maxAttempts,
 		)
 		if attemptErr == nil && result.actionParams != nil {
@@ -265,7 +253,6 @@ func (a *Agent) handlePlanning(ctx context.Context, job *types.Job, chosenAction

 		params, err := a.generateParameters(ctx, pickTemplate, subTaskAction, conv, subTaskReasoning, maxRetries)
 		if err != nil {
-			xlog.Error("error generating action's parameters", "error", err)
 			return conv, fmt.Errorf("error generating action's parameters: %w", err)

 		}
@@ -295,7 +282,6 @@ func (a *Agent) handlePlanning(ctx context.Context, job *types.Job, chosenAction

 		result, err := a.runAction(ctx, subTaskAction, actionParams)
 		if err != nil {
-			xlog.Error("error running action", "error", err)
 			return conv, fmt.Errorf("error running action: %w", err)
 		}

@@ -381,9 +367,7 @@ func (a *Agent) prepareHUD() (promptHUD *PromptHUD) {
 func (a *Agent) pickAction(ctx context.Context, templ string, messages []openai.ChatCompletionMessage, maxRetries int) (types.Action, types.ActionParams, string, error) {
 	c := messages

-	xlog.Debug("[pickAction] picking action starts", "messages", messages)
-
-	// Identify the goal of this conversation
+	xlog.Debug("[pickAction] picking action", "messages", messages)

 	if !a.options.forceReasoning {
 		xlog.Debug("not forcing reasoning")
@@ -392,7 +376,7 @@ func (a *Agent) pickAction(ctx context.Context, templ string, messages []openai.
 		thought, err := a.decision(ctx,
 			messages,
 			a.availableActions().ToTools(),
-			"",
+			nil,
 			maxRetries)
 		if err != nil {
 			return nil, nil, "", err
@@ -431,83 +415,120 @@ func (a *Agent) pickAction(ctx context.Context, templ string, messages []openai.
 		}, c...)
 	}

-	thought, err := a.decision(ctx,
-		c,
-		types.Actions{action.NewReasoning()}.ToTools(),
-		action.NewReasoning().Definition().Name.String(), maxRetries)
-	if err != nil {
-		return nil, nil, "", err
-	}
-	originalReasoning := ""
-	response := &action.ReasoningResponse{}
-	if thought.actionParams != nil {
-		if err := thought.actionParams.Unmarshal(response); err != nil {
-			return nil, nil, "", err
-		}
-		originalReasoning = response.Reasoning
-	}
-	if thought.message != "" {
-		originalReasoning = thought.message
-	}
-
-	xlog.Debug("[pickAction] picking action", "messages", c)
-	// thought, err := a.askLLM(ctx,
-	// 	c,
-
-	actionsID := []string{"reply"}
+	actionsID := []string{}
 	for _, m := range a.availableActions() {
 		actionsID = append(actionsID, m.Definition().Name.String())
 	}

-	xlog.Debug("[pickAction] actionsID", "actionsID", actionsID)
+	// thoughtPromptStringBuilder := strings.Builder{}
+	// thoughtPromptStringBuilder.WriteString("You have to pick an action based on the conversation and the prompt. Describe the full reasoning process for your choice. Here is a list of actions: ")
+	// for _, m := range a.availableActions() {
+	// 	thoughtPromptStringBuilder.WriteString(
+	// 		m.Definition().Name.String() + ": " + m.Definition().Description + "\n",
+	// 	)
+	// }

-	intentionsTools := action.NewIntention(actionsID...)
-	// TODO: FORCE to select ana ction here
-	// NOTE: we do not give the full conversation here to pick the action
-	// to avoid hallucinations
+	// thoughtPromptStringBuilder.WriteString("To not use any action, respond with 'none'")
+
+	//thoughtPromptStringBuilder.WriteString("\n\nConversation: " + Messages(c).RemoveIf(func(msg openai.ChatCompletionMessage) bool {
+	//	return msg.Role == "system"
+	//}).String())
+
+	//thoughtPrompt := thoughtPromptStringBuilder.String()
+
+	//thoughtConv := []openai.ChatCompletionMessage{}
+
+	thought, err := a.askLLM(ctx,
+		c,
+		maxRetries,
+	)
+	if err != nil {
+		return nil, nil, "", err
+	}
+	originalReasoning := thought.Content
+
+	// From the thought, get the action call
+	// Get all the available actions IDs
+
+	// by grammar, let's decide if we have achieved the goal
+	//  1. analyze response and check if  goal is achieved

-	// Extract an action
 	params, err := a.decision(ctx,
-		append(c, openai.ChatCompletionMessage{
-			Role:    "system",
-			Content: "Pick the relevant action given the following reasoning: " + originalReasoning,
-		}),
-		types.Actions{intentionsTools}.ToTools(),
-		intentionsTools.Definition().Name.String(), maxRetries)
+		[]openai.ChatCompletionMessage{
+			{
+				Role:    "system",
+				Content: "Extract an action to perform from the following reasoning: ",
+			},
+			{
+				Role:    "user",
+				Content: originalReasoning,
+			}},
+		types.Actions{action.NewGoal()}.ToTools(),
+		action.NewGoal().Definition().Name, maxRetries)
 	if err != nil {
 		return nil, nil, "", fmt.Errorf("failed to get the action tool parameters: %v", err)
 	}

-	if params.actionParams == nil {
-		xlog.Debug("[pickAction] no action params found")
-		return nil, nil, params.message, nil
-	}
-
-	actionChoice := action.IntentResponse{}
-	err = params.actionParams.Unmarshal(&actionChoice)
+	goalResponse := action.GoalResponse{}
+	err = params.actionParams.Unmarshal(&goalResponse)
 	if err != nil {
 		return nil, nil, "", err
 	}

-	if actionChoice.Tool == "" || actionChoice.Tool == "reply" {
-		xlog.Debug("[pickAction] no action found, replying")
+	if goalResponse.Achieved {
+		xlog.Debug("[pickAction] goal achieved", "goal", goalResponse.Goal)
 		return nil, nil, "", nil
 	}

-	chosenAction := a.availableActions().Find(actionChoice.Tool)
+	// if the goal is not achieved, pick an action
+	xlog.Debug("[pickAction] goal not achieved", "goal", goalResponse.Goal)

-	xlog.Debug("[pickAction] chosenAction", "chosenAction", chosenAction, "actionName", actionChoice.Tool)
+	xlog.Debug("[pickAction] thought", "conv", c, "originalReasoning", originalReasoning)

-	// // Let's double check if the action is correct by asking the LLM to judge it
+	// TODO: FORCE to select ana ction here
+	// NOTE: we do not give the full conversation here to pick the action
+	// to avoid hallucinations
+	params, err = a.decision(ctx,
+		[]openai.ChatCompletionMessage{
+			{
+				Role:    "system",
+				Content: "Extract an action to perform from the following reasoning: ",
+			},
+			{
+				Role:    "user",
+				Content: originalReasoning,
+			}},
+		a.availableActions().ToTools(),
+		nil, maxRetries)
+	if err != nil {
+		return nil, nil, "", fmt.Errorf("failed to get the action tool parameters: %v", err)
+	}

-	// if chosenAction!= nil {
-	// 	promptString:= "Given the following goal and thoughts, is the action correct? \n\n"
-	// 	promptString+= fmt.Sprintf("Goal: %s\n", goalResponse.Goal)
-	// 	promptString+= fmt.Sprintf("Thoughts: %s\n", originalReasoning)
-	// 	promptString+= fmt.Sprintf("Action: %s\n", chosenAction.Definition().Name.String())
-	// 	promptString+= fmt.Sprintf("Action description: %s\n", chosenAction.Definition().Description)
-	// 	promptString+= fmt.Sprintf("Action parameters: %s\n", params.actionParams)
+	chosenAction := a.availableActions().Find(params.actioName)

+	// xlog.Debug("[pickAction] params", "params", params)
+
+	// if params.actionParams == nil {
+	// 	return nil, nil, params.message, nil
+	// }
+
+	// xlog.Debug("[pickAction] actionChoice", "actionChoice", params.actionParams, "message", params.message)
+
+	// actionChoice := action.IntentResponse{}
+
+	// err = params.actionParams.Unmarshal(&actionChoice)
+	// if err != nil {
+	// 	return nil, nil, "", err
+	// }
+
+	// if actionChoice.Tool == "" || actionChoice.Tool == "none" {
+	// 	return nil, nil, "", nil
+	// }
+
+	// // Find the action
+	// chosenAction := a.availableActions().Find(actionChoice.Tool)
+	// if chosenAction == nil {
+	// 	return nil, nil, "", fmt.Errorf("no action found for intent:" + actionChoice.Tool)
 	// }

 	return chosenAction, nil, originalReasoning, nil
--- a/core/agent/agent.go
+++ b/core/agent/agent.go
@@ -249,7 +249,7 @@ func (a *Agent) runAction(ctx context.Context, chosenAction types.Action, params
 		}
 	}

-	xlog.Info("[runAction] Running action", "action", chosenAction.Definition().Name, "agent", a.Character.Name, "params", params.String())
+	xlog.Info("Running action", "action", chosenAction.Definition().Name, "agent", a.Character.Name)

 	if chosenAction.Definition().Name.Is(action.StateActionName) {
 		// We need to store the result in the state
@@ -270,8 +270,6 @@ func (a *Agent) runAction(ctx context.Context, chosenAction types.Action, params
 		}
 	}

-	xlog.Debug("[runAction] Action result", "action", chosenAction.Definition().Name, "params", params.String(), "result", result.Result)
-	
 	return result, nil
 }

@@ -605,13 +603,7 @@ func (a *Agent) consumeJob(job *types.Job, role string) {
 	var err error
 	conv, err = a.handlePlanning(job.GetContext(), job, chosenAction, actionParams, reasoning, pickTemplate, conv)
 	if err != nil {
-		xlog.Error("error handling planning", "error", err)
-		//job.Result.Conversation = conv
-		//job.Result.SetResponse(msg.Content)
-		a.reply(job, role, append(conv, openai.ChatCompletionMessage{
-			Role:    "assistant",
-			Content: fmt.Sprintf("Error handling planning: %v", err),
-		}), actionParams, chosenAction, reasoning)
+		job.Result.Finish(fmt.Errorf("error running action: %w", err))
 		return
 	}

@@ -697,6 +689,26 @@ func (a *Agent) consumeJob(job *types.Job, role string) {
 		job.SetNextAction(&followingAction, &followingParams, reasoning)
 		a.consumeJob(job, role)
 		return
+	} else if followingAction == nil {
+		xlog.Info("Not following another action", "agent", a.Character.Name)
+
+		if !a.options.forceReasoning {
+			xlog.Info("Finish conversation with reasoning", "reasoning", reasoning, "agent", a.Character.Name)
+
+			msg := openai.ChatCompletionMessage{
+				Role:    "assistant",
+				Content: reasoning,
+			}
+
+			conv = append(conv, msg)
+			job.Result.SetResponse(msg.Content)
+			job.Result.Conversation = conv
+			job.Result.AddFinalizer(func(conv []openai.ChatCompletionMessage) {
+				a.saveCurrentConversation(conv)
+			})
+			job.Result.Finish(nil)
+			return
+		}
 	}

 	a.reply(job, role, conv, actionParams, chosenAction, reasoning)
--- a/core/agent/agent_test.go
+++ b/core/agent/agent_test.go
@@ -126,8 +126,6 @@ var _ = Describe("Agent test", func() {
 			agent, err := New(
 				WithLLMAPIURL(apiURL),
 				WithModel(testModel),
-				EnableForceReasoning,
-				WithTimeout("10m"),
 				WithLoopDetectionSteps(3),
 				//	WithRandomIdentity(),
 				WithActions(&TestAction{response: map[string]string{
@@ -176,7 +174,7 @@ var _ = Describe("Agent test", func() {
 			agent, err := New(
 				WithLLMAPIURL(apiURL),
 				WithModel(testModel),
-				WithTimeout("10m"),
+
 				//	WithRandomIdentity(),
 				WithActions(&TestAction{response: map[string]string{
 					"boston": testActionResult,
@@ -201,7 +199,6 @@ var _ = Describe("Agent test", func() {
 			agent, err := New(
 				WithLLMAPIURL(apiURL),
 				WithModel(testModel),
-				WithTimeout("10m"),
 				EnableHUD,
 				//	EnableStandaloneJob,
 				//	WithRandomIdentity(),
@@ -238,7 +235,7 @@ var _ = Describe("Agent test", func() {
 			defer agent.Stop()

 			result := agent.Ask(
-				types.WithText("Thoroughly plan a trip to San Francisco from Venice, Italy; check flight times, visa requirements and whether electrical items are allowed in cabin luggage."),
+				types.WithText("plan a trip to San Francisco from Venice, Italy"),
 			)
 			Expect(len(result.State)).To(BeNumerically(">", 1))

@@ -260,7 +257,6 @@ var _ = Describe("Agent test", func() {
 				WithLLMAPIURL(apiURL),
 				WithModel(testModel),
 				WithLLMAPIKey(apiKeyURL),
-				WithTimeout("10m"),
 				WithNewConversationSubscriber(func(m openai.ChatCompletionMessage) {
 					mu.Lock()
 					message = m
@@ -275,7 +271,7 @@ var _ = Describe("Agent test", func() {
 				EnableStandaloneJob,
 				EnableHUD,
 				WithPeriodicRuns("1s"),
-				WithPermanentGoal("use the new_conversation tool to initiate a conversation with the user"),
+				WithPermanentGoal("use the new_conversation tool"),
 				//	EnableStandaloneJob,
 				//	WithRandomIdentity(),
 			)
--- a/core/agent/templates.go
+++ b/core/agent/templates.go
@@ -115,7 +115,7 @@ Available Tools:
 const reSelfEvalTemplate = pickSelfTemplate

 const pickActionTemplate = hudTemplate + `
-Your only task is to analyze the conversation and determine a goal and the best tool to use, or just a final response if we have fullfilled the goal.
+Your only task is to analyze the situation and determine a goal and the best tool to use, or just a final response if we have fullfilled the goal.

 Guidelines:
 1. Review the current state, what was done already and context
--- a/docker-compose.gpu.intel.yaml
+++ b/docker-compose.gpu.intel.yaml
@@ -0,0 +1,75 @@
+services:
+  localai:
+    # See https://localai.io/basics/container/#standard-container-images for
+    # a list of available container images (or build your own with the provided Dockerfile)
+    # Available images with CUDA, ROCm, SYCL, Vulkan
+    # Image list (quay.io): https://quay.io/repository/go-skynet/local-ai?tab=tags
+    # Image list (dockerhub): https://hub.docker.com/r/localai/localai
+    image: localai/localai:master-sycl-f32-ffmpeg-core
+    command: 
+    # - rombo-org_rombo-llm-v3.0-qwen-32b # minimum suggested model
+    - arcee-agent # (smaller)
+    - granite-embedding-107m-multilingual
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8080/readyz"]
+      interval: 60s
+      timeout: 10m
+      retries: 120
+    ports:
+    - 8081:8080
+    environment:
+      - DEBUG=true
+      #- LOCALAI_API_KEY=sk-1234567890
+    volumes:
+      - ./volumes/models:/build/models:cached
+      - ./volumes/images:/tmp/generated/images
+    devices:
+      # On a system with integrated GPU and an Arc 770, this is the Arc 770
+      - /dev/dri/card1
+      - /dev/dri/renderD129
+
+  localrecall:
+    image: quay.io/mudler/localrecall:main
+    ports:
+      - 8080
+    environment:
+      - COLLECTION_DB_PATH=/db
+      - EMBEDDING_MODEL=granite-embedding-107m-multilingual
+      - FILE_ASSETS=/assets
+      - OPENAI_API_KEY=sk-1234567890
+      - OPENAI_BASE_URL=http://localai:8080
+    volumes:
+      - ./volumes/localrag/db:/db
+      - ./volumes/localrag/assets/:/assets
+
+  localrecall-healthcheck:
+    depends_on:
+      localrecall:
+        condition: service_started
+    image: busybox
+    command: ["sh", "-c", "until wget -q -O - http://localrecall:8080 > /dev/null 2>&1; do echo 'Waiting for localrecall...'; sleep 1; done; echo 'localrecall is up!'"]
+
+  localagi:
+    depends_on:
+      localai:
+        condition: service_healthy
+      localrecall-healthcheck:
+        condition: service_completed_successfully
+    build:
+      context: .
+      dockerfile: Dockerfile.webui
+    ports:
+      - 8080:3000
+    image: quay.io/mudler/localagi:master
+    environment:
+      - LOCALAGI_MODEL=arcee-agent
+      - LOCALAGI_LLM_API_URL=http://localai:8080
+      #- LOCALAGI_LLM_API_KEY=sk-1234567890
+      - LOCALAGI_LOCALRAG_URL=http://localrecall:8080
+      - LOCALAGI_STATE_DIR=/pool
+      - LOCALAGI_TIMEOUT=5m
+      - LOCALAGI_ENABLE_CONVERSATIONS_LOGGING=false
+    extra_hosts:
+      - "host.docker.internal:host-gateway"
+    volumes:
+      - ./volumes/localagi/:/pool
--- a/docker-compose.gpu.yaml
+++ b/docker-compose.gpu.yaml
@@ -0,0 +1,85 @@
+services:
+  localai:
+    # See https://localai.io/basics/container/#standard-container-images for
+    # a list of available container images (or build your own with the provided Dockerfile)
+    # Available images with CUDA, ROCm, SYCL, Vulkan
+    # Image list (quay.io): https://quay.io/repository/go-skynet/local-ai?tab=tags
+    # Image list (dockerhub): https://hub.docker.com/r/localai/localai
+    image: localai/localai:master-gpu-nvidia-cuda-12
+    command: 
+    - mlabonne_gemma-3-27b-it-abliterated
+    - qwen_qwq-32b
+    # Other good alternative options:
+    # - rombo-org_rombo-llm-v3.0-qwen-32b # minimum suggested model
+    # - arcee-agent
+    - granite-embedding-107m-multilingual
+    - flux.1-dev
+    - minicpm-v-2_6
+    environment:
+      # Enable if you have a single GPU which don't fit all the models
+      - LOCALAI_SINGLE_ACTIVE_BACKEND=true
+      - DEBUG=true
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8080/readyz"]
+      interval: 10s
+      timeout: 20m
+      retries: 20
+    ports:
+    - 8081:8080
+    volumes:
+      - ./volumes/models:/build/models:cached
+      - ./volumes/images:/tmp/generated/images
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
+  localrecall:
+    image: quay.io/mudler/localrecall:main
+    ports:
+      - 8080
+    environment:
+      - COLLECTION_DB_PATH=/db
+      - EMBEDDING_MODEL=granite-embedding-107m-multilingual
+      - FILE_ASSETS=/assets
+      - OPENAI_API_KEY=sk-1234567890
+      - OPENAI_BASE_URL=http://localai:8080
+    volumes:
+      - ./volumes/localrag/db:/db
+      - ./volumes/localrag/assets/:/assets
+
+  localrecall-healthcheck:
+    depends_on:
+      localrecall:
+        condition: service_started
+    image: busybox
+    command: ["sh", "-c", "until wget -q -O - http://localrecall:8080 > /dev/null 2>&1; do echo 'Waiting for localrecall...'; sleep 1; done; echo 'localrecall is up!'"]
+
+  localagi:
+    depends_on:
+      localai:
+        condition: service_healthy
+      localrecall-healthcheck:
+        condition: service_completed_successfully
+    build:
+      context: .
+      dockerfile: Dockerfile.webui
+    ports:
+      - 8080:3000
+    image: quay.io/mudler/localagi:master
+    environment:
+      - LOCALAGI_MODEL=qwen_qwq-32b
+      - LOCALAGI_LLM_API_URL=http://localai:8080
+      #- LOCALAGI_LLM_API_KEY=sk-1234567890
+      - LOCALAGI_LOCALRAG_URL=http://localrecall:8080
+      - LOCALAGI_STATE_DIR=/pool
+      - LOCALAGI_TIMEOUT=5m
+      - LOCALAGI_ENABLE_CONVERSATIONS_LOGGING=false
+      - LOCALAGI_MULTIMODAL_MODEL=minicpm-v-2_6
+      - LOCALAGI_IMAGE_MODEL=flux.1-dev
+    extra_hosts:
+      - "host.docker.internal:host-gateway"
+    volumes:
+      - ./volumes/localagi/:/pool
--- a/docker-compose.intel.yaml
+++ b/docker-compose.intel.yaml
@@ -1,33 +0,0 @@
-services:
-  localai:
-    extends:
-      file: docker-compose.yaml
-      service: localai
-    environment:
-      - LOCALAI_SINGLE_ACTIVE_BACKEND=true
-      - DEBUG=true
-    image: localai/localai:master-sycl-f32-ffmpeg-core
-    devices:
-      # On a system with integrated GPU and an Arc 770, this is the Arc 770
-      - /dev/dri/card1
-      - /dev/dri/renderD129
-    command: 
-    - ${MODEL_NAME:-arcee-agent}
-    - ${MULTIMODAL_MODEL:-minicpm-v-2_6}
-    - ${IMAGE_MODEL:-sd-1.5-ggml}
-    - granite-embedding-107m-multilingual
-
-  localrecall:
-    extends:
-      file: docker-compose.yaml
-      service: localrecall
-
-  localrecall-healthcheck:
-    extends:
-      file: docker-compose.yaml
-      service: localrecall-healthcheck
-
-  localagi:
-    extends:
-      file: docker-compose.yaml
-      service: localagi
--- a/docker-compose.nvidia.yaml
+++ b/docker-compose.nvidia.yaml
@@ -1,31 +0,0 @@
-services:
-  localai:
-    extends:
-      file: docker-compose.yaml
-      service: localai
-    environment:
-      - LOCALAI_SINGLE_ACTIVE_BACKEND=true
-      - DEBUG=true
-    image: localai/localai:master-sycl-f32-ffmpeg-core
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: 1
-              capabilities: [gpu]
-
-  localrecall:
-    extends:
-      file: docker-compose.yaml
-      service: localrecall
-
-  localrecall-healthcheck:
-    extends:
-      file: docker-compose.yaml
-      service: localrecall-healthcheck
-
-  localagi:
-    extends:
-      file: docker-compose.yaml
-      service: localagi
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -7,9 +7,7 @@ services:
    # Image list (dockerhub): https://hub.docker.com/r/localai/localai
    image: localai/localai:master-ffmpeg-core
    command: 
-    - ${MODEL_NAME:-arcee-agent}
-    - ${MULTIMODAL_MODEL:-minicpm-v-2_6}
-    - ${IMAGE_MODEL:-flux.1-dev}
+    - arcee-agent # (smaller)
    - granite-embedding-107m-multilingual
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8080/readyz"]
@@ -25,6 +23,14 @@ services:
      - ./volumes/models:/build/models:cached
      - ./volumes/images:/tmp/generated/images

+    # decomment the following piece if running with Nvidia GPUs
+    # deploy:
+    #   resources:
+    #     reservations:
+    #       devices:
+    #         - driver: nvidia
+    #           count: 1
+    #           capabilities: [gpu]
  localrecall:
    image: quay.io/mudler/localrecall:main
    ports:
@@ -59,9 +65,7 @@ services:
      - 8080:3000
    #image: quay.io/mudler/localagi:master
    environment:
-      - LOCALAGI_MODEL=${MODEL_NAME:-arcee-agent}
-      - LOCALAGI_MULTIMODAL_MODEL=${MULTIMODAL_MODEL:-minicpm-v-2_6}
-      - LOCALAGI_IMAGE_MODEL=${IMAGE_MODEL:-sd-1.5-ggml}
+      - LOCALAGI_MODEL=arcee-agent
      - LOCALAGI_LLM_API_URL=http://localai:8080
      #- LOCALAGI_LLM_API_KEY=sk-1234567890
      - LOCALAGI_LOCALRAG_URL=http://localrecall:8080
@@ -71,4 +75,4 @@ services:
    extra_hosts:
      - "host.docker.internal:host-gateway"
    volumes:
-      - ./volumes/localagi/:/pool
+      - ./volumes/localagi/:/pool
--- a/services/actions.go
+++ b/services/actions.go
@@ -29,8 +29,6 @@ const (
 	ActionGithubPRReader                 = "github-pr-reader"
 	ActionGithubPRCommenter              = "github-pr-commenter"
 	ActionGithubPRReviewer               = "github-pr-reviewer"
-	ActionGithubPRCreator                = "github-pr-creator"
-	ActionGithubGetAllContent            = "github-get-all-repository-content"
 	ActionGithubREADME                   = "github-readme"
 	ActionScraper                        = "scraper"
 	ActionWikipedia                      = "wikipedia"
@@ -51,14 +49,12 @@ var AvailableActions = []string{
 	ActionGithubIssueCloser,
 	ActionGithubIssueSearcher,
 	ActionGithubRepositoryGet,
-	ActionGithubGetAllContent,
 	ActionGithubRepositoryCreateOrUpdate,
 	ActionGithubIssueReader,
 	ActionGithubIssueCommenter,
 	ActionGithubPRReader,
 	ActionGithubPRCommenter,
 	ActionGithubPRReviewer,
-	ActionGithubPRCreator,
 	ActionGithubREADME,
 	ActionScraper,
 	ActionBrowse,
@@ -122,10 +118,6 @@ func Action(name, agentName string, config map[string]string, pool *state.AgentP
 		a = actions.NewGithubPRCommenter(config)
 	case ActionGithubPRReviewer:
 		a = actions.NewGithubPRReviewer(config)
-	case ActionGithubPRCreator:
-		a = actions.NewGithubPRCreator(config)
-	case ActionGithubGetAllContent:
-		a = actions.NewGithubRepositoryGetAllContent(config)
 	case ActionGithubIssueCommenter:
 		a = actions.NewGithubIssueCommenter(config)
 	case ActionGithubRepositoryGet:
@@ -209,11 +201,6 @@ func ActionsConfigMeta() []config.FieldGroup {
 			Label:  "GitHub Repository Get Content",
 			Fields: actions.GithubRepositoryGetContentConfigMeta(),
 		},
-		{
-			Name:   "github-get-all-repository-content",
-			Label:  "GitHub Get All Repository Content",
-			Fields: actions.GithubRepositoryGetAllContentConfigMeta(),
-		},
 		{
 			Name:   "github-repository-create-or-update-content",
 			Label:  "GitHub Repository Create/Update Content",
@@ -239,11 +226,6 @@ func ActionsConfigMeta() []config.FieldGroup {
 			Label:  "GitHub PR Reviewer",
 			Fields: actions.GithubPRReviewerConfigMeta(),
 		},
-		{
-			Name:   "github-pr-creator",
-			Label:  "GitHub PR Creator",
-			Fields: actions.GithubPRCreatorConfigMeta(),
-		},
 		{
 			Name:   "twitter-post",
 			Label:  "Twitter Post",
--- a/services/actions/githubprcreator.go
+++ b/services/actions/githubprcreator.go
@@ -1,315 +0,0 @@
-package actions
-
-import (
-	"context"
-	"fmt"
-
-	"github.com/google/go-github/v69/github"
-	"github.com/mudler/LocalAGI/core/types"
-	"github.com/mudler/LocalAGI/pkg/config"
-	"github.com/sashabaranov/go-openai/jsonschema"
-)
-
-type GithubPRCreator struct {
-	token, repository, owner, customActionName, defaultBranch string
-	client                                                    *github.Client
-}
-
-func NewGithubPRCreator(config map[string]string) *GithubPRCreator {
-	client := github.NewClient(nil).WithAuthToken(config["token"])
-
-	return &GithubPRCreator{
-		client:           client,
-		token:            config["token"],
-		repository:       config["repository"],
-		owner:            config["owner"],
-		customActionName: config["customActionName"],
-		defaultBranch:    config["defaultBranch"],
-	}
-}
-
-func (g *GithubPRCreator) createOrUpdateBranch(ctx context.Context, branchName string) error {
-	// Get the latest commit SHA from the default branch
-	ref, _, err := g.client.Git.GetRef(ctx, g.owner, g.repository, "refs/heads/"+g.defaultBranch)
-	if err != nil {
-		return fmt.Errorf("failed to get reference: %w", err)
-	}
-
-	// Try to get the branch if it exists
-	_, resp, err := g.client.Git.GetRef(ctx, g.owner, g.repository, "refs/heads/"+branchName)
-	if err != nil {
-		// If branch doesn't exist, create it
-		if resp != nil && resp.StatusCode == 404 {
-			newRef := &github.Reference{
-				Ref:    github.String("refs/heads/" + branchName),
-				Object: &github.GitObject{SHA: ref.Object.SHA},
-			}
-			_, _, err = g.client.Git.CreateRef(ctx, g.owner, g.repository, newRef)
-			if err != nil {
-				return fmt.Errorf("failed to create branch: %w", err)
-			}
-			return nil
-		}
-		return fmt.Errorf("failed to check branch existence: %w", err)
-	}
-
-	// Branch exists, update it to the latest commit
-	updateRef := &github.Reference{
-		Ref:    github.String("refs/heads/" + branchName),
-		Object: &github.GitObject{SHA: ref.Object.SHA},
-	}
-	_, _, err = g.client.Git.UpdateRef(ctx, g.owner, g.repository, updateRef, true)
-	if err != nil {
-		return fmt.Errorf("failed to update branch: %w", err)
-	}
-
-	return nil
-}
-
-func (g *GithubPRCreator) createOrUpdateFile(ctx context.Context, branch string, filePath string, content string, message string) error {
-	// Get the current file content if it exists
-	var sha *string
-	fileContent, _, _, err := g.client.Repositories.GetContents(ctx, g.owner, g.repository, filePath, &github.RepositoryContentGetOptions{
-		Ref: branch,
-	})
-	if err == nil && fileContent != nil {
-		sha = fileContent.SHA
-	}
-
-	// Create or update the file
-	_, _, err = g.client.Repositories.CreateFile(ctx, g.owner, g.repository, filePath, &github.RepositoryContentFileOptions{
-		Message: &message,
-		Content: []byte(content),
-		Branch:  &branch,
-		SHA:     sha,
-	})
-	if err != nil {
-		return fmt.Errorf("failed to create/update file: %w", err)
-	}
-
-	return nil
-}
-
-func (g *GithubPRCreator) Run(ctx context.Context, params types.ActionParams) (types.ActionResult, error) {
-	result := struct {
-		Repository string `json:"repository"`
-		Owner      string `json:"owner"`
-		Branch     string `json:"branch"`
-		Title      string `json:"title"`
-		Body       string `json:"body"`
-		BaseBranch string `json:"base_branch"`
-		Files      []struct {
-			Path    string `json:"path"`
-			Content string `json:"content"`
-		} `json:"files"`
-	}{}
-	err := params.Unmarshal(&result)
-	if err != nil {
-		return types.ActionResult{}, fmt.Errorf("failed to unmarshal params: %w", err)
-	}
-
-	if g.repository != "" && g.owner != "" {
-		result.Repository = g.repository
-		result.Owner = g.owner
-	}
-
-	if result.BaseBranch == "" {
-		result.BaseBranch = g.defaultBranch
-	}
-
-	// Create or update branch
-	err = g.createOrUpdateBranch(ctx, result.Branch)
-	if err != nil {
-		return types.ActionResult{}, fmt.Errorf("failed to create/update branch: %w", err)
-	}
-
-	// Create or update files
-	for _, file := range result.Files {
-		err = g.createOrUpdateFile(ctx, result.Branch, file.Path, file.Content, fmt.Sprintf("Update %s", file.Path))
-		if err != nil {
-			return types.ActionResult{}, fmt.Errorf("failed to update file %s: %w", file.Path, err)
-		}
-	}
-
-	// Check if PR already exists for this branch
-	prs, _, err := g.client.PullRequests.List(ctx, result.Owner, result.Repository, &github.PullRequestListOptions{
-		State: "open",
-		Head:  fmt.Sprintf("%s:%s", result.Owner, result.Branch),
-	})
-	if err != nil {
-		return types.ActionResult{}, fmt.Errorf("failed to list pull requests: %w", err)
-	}
-
-	if len(prs) > 0 {
-		// Update existing PR
-		pr := prs[0]
-		update := &github.PullRequest{
-			Title: &result.Title,
-			Body:  &result.Body,
-		}
-		updatedPR, _, err := g.client.PullRequests.Edit(ctx, result.Owner, result.Repository, pr.GetNumber(), update)
-		if err != nil {
-			return types.ActionResult{}, fmt.Errorf("failed to update pull request: %w", err)
-		}
-		return types.ActionResult{
-			Result: fmt.Sprintf("Updated pull request #%d: %s", updatedPR.GetNumber(), updatedPR.GetHTMLURL()),
-		}, nil
-	}
-
-	// Create new pull request
-	newPR := &github.NewPullRequest{
-		Title: &result.Title,
-		Body:  &result.Body,
-		Head:  &result.Branch,
-		Base:  &result.BaseBranch,
-	}
-
-	createdPR, _, err := g.client.PullRequests.Create(ctx, result.Owner, result.Repository, newPR)
-	if err != nil {
-		return types.ActionResult{}, fmt.Errorf("failed to create pull request: %w", err)
-	}
-
-	return types.ActionResult{
-		Result: fmt.Sprintf("Created pull request #%d: %s", createdPR.GetNumber(), createdPR.GetHTMLURL()),
-	}, nil
-}
-
-func (g *GithubPRCreator) Definition() types.ActionDefinition {
-	actionName := "create_github_pr"
-	if g.customActionName != "" {
-		actionName = g.customActionName
-	}
-	description := "Create a GitHub pull request with file changes"
-	if g.repository != "" && g.owner != "" && g.defaultBranch != "" {
-		return types.ActionDefinition{
-			Name:        types.ActionDefinitionName(actionName),
-			Description: description,
-			Properties: map[string]jsonschema.Definition{
-				"branch": {
-					Type:        jsonschema.String,
-					Description: "The name of the new branch to create",
-				},
-				"title": {
-					Type:        jsonschema.String,
-					Description: "The title of the pull request",
-				},
-				"body": {
-					Type:        jsonschema.String,
-					Description: "The body/description of the pull request",
-				},
-				"files": {
-					Type: jsonschema.Array,
-					Items: &jsonschema.Definition{
-						Type: jsonschema.Object,
-						Properties: map[string]jsonschema.Definition{
-							"path": {
-								Type:        jsonschema.String,
-								Description: "The path of the file to create/update",
-							},
-							"content": {
-								Type:        jsonschema.String,
-								Description: "The content of the file",
-							},
-						},
-						Required: []string{"path", "content"},
-					},
-					Description: "Array of files to create or update",
-				},
-			},
-			Required: []string{"branch", "title", "files"},
-		}
-	}
-	return types.ActionDefinition{
-		Name:        types.ActionDefinitionName(actionName),
-		Description: description,
-		Properties: map[string]jsonschema.Definition{
-			"branch": {
-				Type:        jsonschema.String,
-				Description: "The name of the new branch to create",
-			},
-			"title": {
-				Type:        jsonschema.String,
-				Description: "The title of the pull request",
-			},
-			"body": {
-				Type:        jsonschema.String,
-				Description: "The body/description of the pull request",
-			},
-			"base_branch": {
-				Type:        jsonschema.String,
-				Description: "The base branch to merge into (defaults to configured default branch)",
-			},
-			"files": {
-				Type: jsonschema.Array,
-				Items: &jsonschema.Definition{
-					Type: jsonschema.Object,
-					Properties: map[string]jsonschema.Definition{
-						"path": {
-							Type:        jsonschema.String,
-							Description: "The path of the file to create/update",
-						},
-						"content": {
-							Type:        jsonschema.String,
-							Description: "The content of the file",
-						},
-					},
-					Required: []string{"path", "content"},
-				},
-				Description: "Array of files to create or update",
-			},
-			"repository": {
-				Type:        jsonschema.String,
-				Description: "The repository to create the pull request in",
-			},
-			"owner": {
-				Type:        jsonschema.String,
-				Description: "The owner of the repository",
-			},
-		},
-		Required: []string{"branch", "title", "files", "repository", "owner"},
-	}
-}
-
-func (a *GithubPRCreator) Plannable() bool {
-	return true
-}
-
-// GithubPRCreatorConfigMeta returns the metadata for GitHub PR Creator action configuration fields
-func GithubPRCreatorConfigMeta() []config.Field {
-	return []config.Field{
-		{
-			Name:     "token",
-			Label:    "GitHub Token",
-			Type:     config.FieldTypeText,
-			Required: true,
-			HelpText: "GitHub API token with repository access",
-		},
-		{
-			Name:     "repository",
-			Label:    "Repository",
-			Type:     config.FieldTypeText,
-			Required: false,
-			HelpText: "GitHub repository name",
-		},
-		{
-			Name:     "owner",
-			Label:    "Owner",
-			Type:     config.FieldTypeText,
-			Required: false,
-			HelpText: "GitHub repository owner",
-		},
-		{
-			Name:     "customActionName",
-			Label:    "Custom Action Name",
-			Type:     config.FieldTypeText,
-			HelpText: "Custom name for this action",
-		},
-		{
-			Name:     "defaultBranch",
-			Label:    "Default Branch",
-			Type:     config.FieldTypeText,
-			Required: false,
-			HelpText: "Default branch to create PRs against (defaults to main)",
-		},
-	}
-}
--- a/services/actions/githubprcreator_test.go
+++ b/services/actions/githubprcreator_test.go
@@ -1,118 +0,0 @@
-package actions_test
-
-import (
-	"context"
-	"os"
-
-	"github.com/mudler/LocalAGI/services/actions"
-	. "github.com/onsi/ginkgo/v2"
-	. "github.com/onsi/gomega"
-)
-
-var _ = Describe("GithubPRCreator", func() {
-	var (
-		action *actions.GithubPRCreator
-		ctx    context.Context
-	)
-
-	BeforeEach(func() {
-		ctx = context.Background()
-
-		// Check for required environment variables
-		token := os.Getenv("GITHUB_TOKEN")
-		repo := os.Getenv("TEST_REPOSITORY")
-		owner := os.Getenv("TEST_OWNER")
-
-		// Skip tests if any required environment variable is missing
-		if token == "" || repo == "" || owner == "" {
-			Skip("Skipping GitHub PR creator tests: required environment variables not set")
-		}
-
-		config := map[string]string{
-			"token":            token,
-			"repository":       repo,
-			"owner":            owner,
-			"customActionName": "test_create_pr",
-			"defaultBranch":    "main",
-		}
-
-		action = actions.NewGithubPRCreator(config)
-	})
-
-	Describe("Creating pull requests", func() {
-		It("should successfully create a pull request with file changes", func() {
-			params := map[string]interface{}{
-				"branch":      "test-branch",
-				"title":       "Test PR",
-				"body":        "This is a test pull request",
-				"base_branch": "main",
-				"files": []map[string]interface{}{
-					{
-						"path":    "test.txt",
-						"content": "This is a test file",
-					},
-				},
-			}
-
-			result, err := action.Run(ctx, params)
-			Expect(err).NotTo(HaveOccurred())
-			Expect(result.Result).To(ContainSubstring("pull request #"))
-		})
-
-		It("should handle missing required fields", func() {
-			params := map[string]interface{}{
-				"title": "Test PR",
-				"body":  "This is a test pull request",
-			}
-
-			_, err := action.Run(ctx, params)
-			Expect(err).To(HaveOccurred())
-		})
-	})
-
-	Describe("Action Definition", func() {
-		It("should return correct action definition", func() {
-			def := action.Definition()
-			Expect(def.Name.String()).To(Equal("test_create_pr"))
-			Expect(def.Description).To(ContainSubstring("Create a GitHub pull request with file changes"))
-			Expect(def.Properties).To(HaveKey("branch"))
-			Expect(def.Properties).To(HaveKey("title"))
-			Expect(def.Properties).To(HaveKey("files"))
-		})
-
-		It("should handle custom action name", func() {
-			config := map[string]string{
-				"token":            "test-token",
-				"customActionName": "custom_action_name",
-			}
-			action := actions.NewGithubPRCreator(config)
-			def := action.Definition()
-			Expect(def.Name.String()).To(Equal("custom_action_name"))
-		})
-	})
-
-	Describe("Configuration", func() {
-		It("should handle missing repository and owner in config", func() {
-			config := map[string]string{
-				"token": "test-token",
-			}
-			action := actions.NewGithubPRCreator(config)
-			def := action.Definition()
-			Expect(def.Properties).To(HaveKey("repository"))
-			Expect(def.Properties).To(HaveKey("owner"))
-		})
-
-		It("should handle provided repository and owner in config", func() {
-			config := map[string]string{
-				"token":         "test-token",
-				"repository":    "test-repo",
-				"defaultBranch": "main",
-				"owner":         "test-owner",
-			}
-			action := actions.NewGithubPRCreator(config)
-			def := action.Definition()
-			Expect(def.Properties).NotTo(HaveKey("repository"))
-			Expect(def.Properties).NotTo(HaveKey("owner"))
-		})
-	})
-})
--- a/services/actions/githubprreviewer.go
+++ b/services/actions/githubprreviewer.go
@@ -128,13 +128,11 @@ func (g *GithubPRReviewer) Run(ctx context.Context, params types.ActionParams) (
 	}

 	actionResult := fmt.Sprintf(
-		"Pull request https://github.com/%s/%s/pull/%d reviewed successfully with status: %s, comments: %v, message: %s",
+		"Pull request https://github.com/%s/%s/pull/%d reviewed successfully with status: %s",
 		result.Owner,
 		result.Repository,
 		result.PRNumber,
 		strings.ToLower(result.ReviewAction),
-		result.Comments,
-		result.ReviewComment,
 	)

 	return types.ActionResult{Result: actionResult}, nil
--- a/services/actions/githubprreviewer_test.go
+++ b/services/actions/githubprreviewer_test.go
@@ -1,103 +0,0 @@
-package actions_test
-
-import (
-	"context"
-	"os"
-
-	"github.com/mudler/LocalAGI/core/types"
-	"github.com/mudler/LocalAGI/services/actions"
-	. "github.com/onsi/ginkgo/v2"
-	. "github.com/onsi/gomega"
-)
-
-var _ = Describe("GithubPRReviewer", func() {
-	var (
-		reviewer *actions.GithubPRReviewer
-		ctx      context.Context
-	)
-
-	BeforeEach(func() {
-		ctx = context.Background()
-
-		// Check for required environment variables
-		token := os.Getenv("GITHUB_TOKEN")
-		repo := os.Getenv("TEST_REPOSITORY")
-		owner := os.Getenv("TEST_OWNER")
-		prNumber := os.Getenv("TEST_PR_NUMBER")
-
-		// Skip tests if any required environment variable is missing
-		if token == "" || repo == "" || owner == "" || prNumber == "" {
-			Skip("Skipping GitHub PR reviewer tests: required environment variables not set")
-		}
-
-		config := map[string]string{
-			"token":            token,
-			"repository":       repo,
-			"owner":            owner,
-			"customActionName": "test_review_github_pr",
-		}
-
-		reviewer = actions.NewGithubPRReviewer(config)
-	})
-
-	Describe("Reviewing a PR", func() {
-		It("should successfully submit a review with comments", func() {
-			prNumber := os.Getenv("TEST_PR_NUMBER")
-			Expect(prNumber).NotTo(BeEmpty())
-
-			params := types.ActionParams{
-				"pr_number":      prNumber,
-				"review_comment": "Test review comment from integration test",
-				"review_action":  "COMMENT",
-				"comments": []map[string]interface{}{
-					{
-						"file":    "README.md",
-						"line":    1,
-						"comment": "Test line comment from integration test",
-					},
-				},
-			}
-
-			result, err := reviewer.Run(ctx, params)
-			Expect(err).NotTo(HaveOccurred())
-			Expect(result.Result).To(ContainSubstring("reviewed successfully"))
-		})
-
-		It("should handle invalid PR number", func() {
-			params := types.ActionParams{
-				"pr_number":      999999,
-				"review_comment": "Test review comment",
-				"review_action":  "COMMENT",
-			}
-
-			result, err := reviewer.Run(ctx, params)
-			Expect(err).To(HaveOccurred())
-			Expect(result.Result).To(ContainSubstring("not found"))
-		})
-
-		It("should handle invalid review action", func() {
-			prNumber := os.Getenv("TEST_PR_NUMBER")
-			Expect(prNumber).NotTo(BeEmpty())
-
-			params := types.ActionParams{
-				"pr_number":      prNumber,
-				"review_comment": "Test review comment",
-				"review_action":  "INVALID_ACTION",
-			}
-
-			_, err := reviewer.Run(ctx, params)
-			Expect(err).To(HaveOccurred())
-		})
-	})
-
-	Describe("Action Definition", func() {
-		It("should return correct action definition", func() {
-			def := reviewer.Definition()
-			Expect(def.Name).To(Equal(types.ActionDefinitionName("test_review_github_pr")))
-			Expect(def.Description).To(ContainSubstring("Review a GitHub pull request"))
-			Expect(def.Properties).To(HaveKey("pr_number"))
-			Expect(def.Properties).To(HaveKey("review_action"))
-			Expect(def.Properties).To(HaveKey("comments"))
-		})
-	})
-})
--- a/services/actions/githubrepositorygetallcontent.go
+++ b/services/actions/githubrepositorygetallcontent.go
@@ -1,174 +0,0 @@
-package actions
-
-import (
-	"context"
-	"fmt"
-	"strings"
-
-	"github.com/google/go-github/v69/github"
-	"github.com/mudler/LocalAGI/core/types"
-	"github.com/mudler/LocalAGI/pkg/config"
-	"github.com/sashabaranov/go-openai/jsonschema"
-)
-
-type GithubRepositoryGetAllContent struct {
-	token, repository, owner, customActionName string
-	client                                     *github.Client
-}
-
-func NewGithubRepositoryGetAllContent(config map[string]string) *GithubRepositoryGetAllContent {
-	client := github.NewClient(nil).WithAuthToken(config["token"])
-
-	return &GithubRepositoryGetAllContent{
-		client:           client,
-		token:            config["token"],
-		repository:       config["repository"],
-		owner:            config["owner"],
-		customActionName: config["customActionName"],
-	}
-}
-
-func (g *GithubRepositoryGetAllContent) getContentRecursively(ctx context.Context, path string) (string, error) {
-	var result strings.Builder
-
-	// Get content at the current path
-	_, directoryContent, _, err := g.client.Repositories.GetContents(ctx, g.owner, g.repository, path, nil)
-	if err != nil {
-		return "", fmt.Errorf("error getting content at path %s: %w", path, err)
-	}
-
-	// Process each item in the directory
-	for _, item := range directoryContent {
-		if item.GetType() == "dir" {
-			// Recursively get content for subdirectories
-			subContent, err := g.getContentRecursively(ctx, item.GetPath())
-			if err != nil {
-				return "", err
-			}
-			result.WriteString(subContent)
-		} else if item.GetType() == "file" {
-			// Get file content
-			fileContent, _, _, err := g.client.Repositories.GetContents(ctx, g.owner, g.repository, item.GetPath(), nil)
-			if err != nil {
-				return "", fmt.Errorf("error getting file content for %s: %w", item.GetPath(), err)
-			}
-
-			content, err := fileContent.GetContent()
-			if err != nil {
-				return "", fmt.Errorf("error decoding content for %s: %w", item.GetPath(), err)
-			}
-
-			// Add file content to result with clear markers
-			result.WriteString(fmt.Sprintf("\n--- START FILE: %s ---\n", item.GetPath()))
-			result.WriteString(content)
-			result.WriteString(fmt.Sprintf("\n--- END FILE: %s ---\n", item.GetPath()))
-		}
-	}
-
-	return result.String(), nil
-}
-
-func (g *GithubRepositoryGetAllContent) Run(ctx context.Context, params types.ActionParams) (types.ActionResult, error) {
-	result := struct {
-		Repository string `json:"repository"`
-		Owner      string `json:"owner"`
-		Path       string `json:"path,omitempty"`
-	}{}
-	err := params.Unmarshal(&result)
-	if err != nil {
-		return types.ActionResult{}, fmt.Errorf("failed to unmarshal params: %w", err)
-	}
-
-	if g.repository != "" && g.owner != "" {
-		result.Repository = g.repository
-		result.Owner = g.owner
-	}
-
-	// Start from root if no path specified
-	if result.Path == "" {
-		result.Path = "."
-	}
-
-	content, err := g.getContentRecursively(ctx, result.Path)
-	if err != nil {
-		return types.ActionResult{}, err
-	}
-
-	return types.ActionResult{Result: content}, nil
-}
-
-func (g *GithubRepositoryGetAllContent) Definition() types.ActionDefinition {
-	actionName := "get_all_github_repository_content"
-	if g.customActionName != "" {
-		actionName = g.customActionName
-	}
-	description := "Get all content of a GitHub repository recursively"
-	if g.repository != "" && g.owner != "" {
-		return types.ActionDefinition{
-			Name:        types.ActionDefinitionName(actionName),
-			Description: description,
-			Properties: map[string]jsonschema.Definition{
-				"path": {
-					Type:        jsonschema.String,
-					Description: "Optional path to start from (defaults to repository root)",
-				},
-			},
-		}
-	}
-	return types.ActionDefinition{
-		Name:        types.ActionDefinitionName(actionName),
-		Description: description,
-		Properties: map[string]jsonschema.Definition{
-			"path": {
-				Type:        jsonschema.String,
-				Description: "Optional path to start from (defaults to repository root)",
-			},
-			"repository": {
-				Type:        jsonschema.String,
-				Description: "The repository to get content from",
-			},
-			"owner": {
-				Type:        jsonschema.String,
-				Description: "The owner of the repository",
-			},
-		},
-		Required: []string{"repository", "owner"},
-	}
-}
-
-func (a *GithubRepositoryGetAllContent) Plannable() bool {
-	return true
-}
-
-// GithubRepositoryGetAllContentConfigMeta returns the metadata for GitHub Repository Get All Content action configuration fields
-func GithubRepositoryGetAllContentConfigMeta() []config.Field {
-	return []config.Field{
-		{
-			Name:     "token",
-			Label:    "GitHub Token",
-			Type:     config.FieldTypeText,
-			Required: true,
-			HelpText: "GitHub API token with repository access",
-		},
-		{
-			Name:     "repository",
-			Label:    "Repository",
-			Type:     config.FieldTypeText,
-			Required: false,
-			HelpText: "GitHub repository name",
-		},
-		{
-			Name:     "owner",
-			Label:    "Owner",
-			Type:     config.FieldTypeText,
-			Required: false,
-			HelpText: "GitHub repository owner",
-		},
-		{
-			Name:     "customActionName",
-			Label:    "Custom Action Name",
-			Type:     config.FieldTypeText,
-			HelpText: "Custom name for this action",
-		},
-	}
-}
--- a/services/actions/githubrepositorygetallcontent_test.go
+++ b/services/actions/githubrepositorygetallcontent_test.go
@@ -1,114 +0,0 @@
-package actions_test
-
-import (
-	"context"
-	"os"
-	"strings"
-
-	"github.com/mudler/LocalAGI/services/actions"
-	. "github.com/onsi/ginkgo/v2"
-	. "github.com/onsi/gomega"
-)
-
-var _ = Describe("GithubRepositoryGetAllContent", func() {
-	var (
-		action *actions.GithubRepositoryGetAllContent
-		ctx    context.Context
-	)
-
-	BeforeEach(func() {
-		ctx = context.Background()
-
-		// Check for required environment variables
-		token := os.Getenv("GITHUB_TOKEN")
-		repo := os.Getenv("TEST_REPOSITORY")
-		owner := os.Getenv("TEST_OWNER")
-
-		// Skip tests if any required environment variable is missing
-		if token == "" || repo == "" || owner == "" {
-			Skip("Skipping GitHub repository get all content tests: required environment variables not set")
-		}
-
-		config := map[string]string{
-			"token":            token,
-			"repository":       repo,
-			"owner":            owner,
-			"customActionName": "test_get_all_content",
-		}
-
-		action = actions.NewGithubRepositoryGetAllContent(config)
-	})
-
-	Describe("Getting repository content", func() {
-		It("should successfully get content from root directory with proper file markers", func() {
-			params := map[string]interface{}{
-				"path": ".",
-			}
-
-			result, err := action.Run(ctx, params)
-			Expect(err).NotTo(HaveOccurred())
-			Expect(result.Result).NotTo(BeEmpty())
-
-			// Verify file markers
-			Expect(result.Result).To(ContainSubstring("--- START FILE:"))
-			Expect(result.Result).To(ContainSubstring("--- END FILE:"))
-
-			// Verify markers are properly paired
-			startCount := strings.Count(result.Result, "--- START FILE:")
-			endCount := strings.Count(result.Result, "--- END FILE:")
-			Expect(startCount).To(Equal(endCount), "Number of start and end markers should match")
-		})
-
-		It("should handle non-existent path", func() {
-			params := map[string]interface{}{
-				"path": "non-existent-path",
-			}
-
-			_, err := action.Run(ctx, params)
-			Expect(err).To(HaveOccurred())
-		})
-	})
-
-	Describe("Action Definition", func() {
-		It("should return correct action definition", func() {
-			def := action.Definition()
-			Expect(def.Name.String()).To(Equal("test_get_all_content"))
-			Expect(def.Description).To(ContainSubstring("Get all content of a GitHub repository recursively"))
-			Expect(def.Properties).To(HaveKey("path"))
-		})
-
-		It("should handle custom action name", func() {
-			config := map[string]string{
-				"token":            "test-token",
-				"customActionName": "custom_action_name",
-			}
-			action := actions.NewGithubRepositoryGetAllContent(config)
-			def := action.Definition()
-			Expect(def.Name.String()).To(Equal("custom_action_name"))
-		})
-	})
-
-	Describe("Configuration", func() {
-		It("should handle missing repository and owner in config", func() {
-			config := map[string]string{
-				"token": "test-token",
-			}
-			action := actions.NewGithubRepositoryGetAllContent(config)
-			def := action.Definition()
-			Expect(def.Properties).To(HaveKey("repository"))
-			Expect(def.Properties).To(HaveKey("owner"))
-		})
-
-		It("should handle provided repository and owner in config", func() {
-			config := map[string]string{
-				"token":      "test-token",
-				"repository": "test-repo",
-				"owner":      "test-owner",
-			}
-			action := actions.NewGithubRepositoryGetAllContent(config)
-			def := action.Definition()
-			Expect(def.Properties).NotTo(HaveKey("repository"))
-			Expect(def.Properties).NotTo(HaveKey("owner"))
-		})
-	})
-})
--- a/webui/react-ui/src/pages/AgentStatus.jsx
+++ b/webui/react-ui/src/pages/AgentStatus.jsx
@@ -1,12 +1,13 @@
 import { useState, useEffect } from 'react';
-import { useParams, Link } from 'react-router-dom';
+import { useParams, Link, useNavigate } from 'react-router-dom';

 function AgentStatus() {
  const { name } = useParams();
+  const navigate = useNavigate();
  const [statusData, setStatusData] = useState(null);
  const [loading, setLoading] = useState(true);
  const [error, setError] = useState(null);
-  const [_eventSource, setEventSource] = useState(null);
+  const [eventSource, setEventSource] = useState(null);
  const [liveUpdates, setLiveUpdates] = useState([]);

  // Update document title
@@ -48,7 +49,7 @@ function AgentStatus() {
        const data = JSON.parse(event.data);
        setLiveUpdates(prev => [data, ...prev.slice(0, 19)]); // Keep last 20 updates
      } catch (err) {
-        setLiveUpdates(prev => [event.data, ...prev.slice(0, 19)]);
+        console.error('Error parsing SSE data:', err);
      }
    });

@@ -128,9 +129,23 @@ function AgentStatus() {
                  <h2 className="text-sm font-semibold mb-2">Agent Action:</h2>
                  <div className="status-details">
                    <div className="status-row">
-                      <span className="status-label">{index}</span>
-                      <span className="status-value">{formatValue(item)}</span>
+                      <span className="status-label">Result:</span>
+                      <span className="status-value">{formatValue(item.Result)}</span>
                    </div>
+                    <div className="status-row">
+                      <span className="status-label">Action:</span>
+                      <span className="status-value">{formatValue(item.Action)}</span>
+                    </div>
+                    <div className="status-row">
+                      <span className="status-label">Parameters:</span>
+                      <span className="status-value pre-wrap">{formatValue(item.Params)}</span>
+                    </div>
+                    {item.Reasoning && (
+                      <div className="status-row">
+                        <span className="status-label">Reasoning:</span>
+                        <span className="status-value reasoning">{formatValue(item.Reasoning)}</span>
+                      </div>
+                    )}
                  </div>
                </div>
              </div>
--- a/webui/react-ui/vite.config.js
+++ b/webui/react-ui/vite.config.js
@@ -30,7 +30,6 @@ export default defineConfig(({ mode }) => {
        '/status': backendUrl,
        '/action': backendUrl,
        '/actions': backendUrl,
-        '/avatars': backendUrl
      }
    }
  }
--- a/webui/routes.go
+++ b/webui/routes.go
@@ -4,7 +4,6 @@ import (
 	"crypto/subtle"
 	"embed"
 	"errors"
-	"fmt"
 	"math/rand"
 	"net/http"
 	"path/filepath"
@@ -239,20 +238,9 @@ func (app *App) registerRoutes(pool *state.AgentPool, webapp *fiber.App) {
 			history = &state.Status{ActionResults: []types.ActionState{}}
 		}

-		entries := []string{}
-		for _, h := range Reverse(history.Results()) {
-			entries = append(entries, fmt.Sprintf(
-				"Result: %v Action: %v Params: %v Reasoning: %v", 
-				h.Result,
-				h.Action.Definition().Name, 
-				h.Params, 
-				h.Reasoning,
-			))
-		}
-
 		return c.JSON(fiber.Map{
 			"Name":    c.Params("name"),
-			"History": entries,
+			"History": Reverse(history.Results()),
 		})
 	})
Author	SHA1	Message	Date
mudler	e4271b4d2f	feat: improve action picking Signed-off-by: mudler <mudler@localai.io>	2025-04-11 20:43:48 +02:00
Ettore Di Giacinto	9dad2b0ba4	Add Github reviewer and improve reasoning	2025-04-10 23:36:44 +02:00