more release cleanup (#161)

* delete redundant component which merged upstream, TextMessageTermination

cleanup imports for _task_agent

update autogen to a version without the pre-built team bug

* try a new action

* oops

* oops again

* add needs

* download all artifacts

* oops

* set version for helm/others

* oops

* load saved containers

* so close

* I can smell it

* not sure

* no idea

* I give up

Author: EItanya

.github/workflows/ci.yaml

@@ -7,6 +7,8 @@ on:
 
 jobs:
   build:
+    env:
+      VERSION: v0.0.1-test
     strategy:
       matrix:
         image:
@@ -25,11 +27,39 @@ jobs:
       uses: docker/setup-buildx-action@v3
     - name: Run make build
       env:
-        DOCKER_BUILD_ARGS: "--platform linux/amd64,linux/arm64"
+        DOCKER_BUILD_ARGS: "--platform linux/amd64,linux/arm64 --output type=tar,dest=${{ runner.temp }}/${{ matrix.image }}.tar"
         DOCKER_BUILDER: "docker buildx"
       run: make build-${{ matrix.image }}
       working-directory: ./
+    - name: Upload artifact
+      uses: actions/upload-artifact@v4
+      with:
+        name: ${{matrix.image}}.tar
+        path: ${{ runner.temp }}
 
+  e2e-test:
+    env:
+      VERSION: v0.0.1-test
+    runs-on: ubuntu-latest
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v4
+
+    - name: Set up Helm
+      uses: azure/[email protected]
+      with:
+        version: v3.17.0
+
+    - name: Create k8s Kind Cluster
+      uses: helm/kind-action@v1
+      with:
+        cluster_name: kagent
+
+    - name: Install Kagent
+      run: make helm-install
+      env:
+        OPENAI_API_KEY: sk-fake-openai-api-key
+    
   python-test:
     runs-on: ubuntu-latest
     steps:

DEVELOPMENT.md

@@ -20,7 +20,7 @@ Running outside Kubernetes:
 uv sync --all-extras
 
 # Run the autogen backend
-uv run autogenstudio ui
+uv run kagent-engine serve
 ```
 
 If you get an error that looks like this:

go/cli/internal/cli/get.go

@@ -23,6 +23,12 @@ func GetAgentCmd(c *ishell.Context) {
 			c.Printf("Failed to get agents: %v\n", err)
 			return
 		}
+
+		if len(agentList) == 0 {
+			c.Println("No agents found")
+			return
+		}
+
 		if err := printTeams(agentList); err != nil {
 			c.Printf("Failed to print agents: %v\n", err)
 			return
@@ -52,6 +58,11 @@ func GetRunCmd(c *ishell.Context) {
 			return
 		}
 
+		if len(runList) == 0 {
+			c.Println("No runs found")
+			return
+		}
+
 		if err := printRuns(runList); err != nil {
 			c.Printf("Failed to print runs: %v\n", err)
 			return
@@ -80,6 +91,12 @@ func GetSessionCmd(c *ishell.Context) {
 			c.Printf("Failed to get sessions: %v\n", err)
 			return
 		}
+
+		if len(sessionList) == 0 {
+			c.Println("No sessions found")
+			return
+		}
+
 		if err := printSessions(sessionList); err != nil {
 			c.Printf("Failed to print sessions: %v\n", err)
 			return

go/controller/internal/autogen/autogen_api_translator.go

@@ -478,7 +478,7 @@ func translateTerminationCondition(terminationCondition v1alpha1.TerminationCond
 		}, nil
 	case terminationCondition.TextMessageTermination != nil:
 		return &api.Component{
-			Provider:      "kagent.terminations.TextMessageTermination",
+			Provider:      "autogen_agentchat.conditions.TextMessageTermination",
 			ComponentType: "termination",
 			Version:       makePtr(1),
 			//ComponentVersion: 1,

python/README.md

@@ -26,7 +26,7 @@ uv sync --all-extras
 ## Running the engine
 
 ```bash
-uv run autogenstudio ui
+uv run kagent-engine serve
 ```
 
 ## Testing

python/pyproject.toml

@@ -46,10 +46,10 @@ kagent-engine = "kagent.cli:run"
 tool_gen = "kagent.tools.utils.tool_gen:main"
 
 [tool.uv.sources]
-autogenstudio = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-studio", rev = "361a2b2ff1ee88d1db205913ada14d8d57734fdd" }
-autogen-agentchat = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-agentchat", rev = "361a2b2ff1ee88d1db205913ada14d8d57734fdd" }
-autogen-ext = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-ext", rev = "361a2b2ff1ee88d1db205913ada14d8d57734fdd" }
-autogen-core = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-core", rev = "361a2b2ff1ee88d1db205913ada14d8d57734fdd" }
+autogenstudio = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-studio", rev = "21acedd4a97579a2194514746039e2198fe95104" }
+autogen-agentchat = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-agentchat", rev = "21acedd4a97579a2194514746039e2198fe95104" }
+autogen-ext = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-ext", rev = "21acedd4a97579a2194514746039e2198fe95104" }
+autogen-core = { git = "https://github.com/kagent-dev/autogen.git", subdirectory = "python/packages/autogen-core", rev = "21acedd4a97579a2194514746039e2198fe95104" }
 kagent = { workspace = true }
 
 [tool.ruff]

python/src/kagent/agents/_task_agent.py

@@ -18,19 +18,14 @@
 from autogen_core.model_context import ChatCompletionContext, UnboundedChatCompletionContext
 from autogen_core.models import (
     AssistantMessage,
-    ChatCompletionClient,
-    CreateResult,
     FunctionExecutionResultMessage,
-    LLMMessage,
-    SystemMessage,
-    UserMessage,
 )
 from pydantic import BaseModel, Field
 from typing_extensions import Self
 
 
 class TaskAgentState(BaseState):
-    """State for a Society of Mind agent."""
+    """State for a Task agent."""
 
     inner_team_state: Mapping[str, Any] = Field(default_factory=dict)
     model_context_state: Mapping[str, Any] = Field(default_factory=dict)
@@ -65,7 +60,7 @@ class TaskAgent(BaseChatAgent, Component[TaskAgentConfig]):
     component_provider_override = "kagent.agents.TaskAgent"
 
     DEFAULT_DESCRIPTION = "An agent that uses an inner team of agents to generate responses."
-    """str: The default description for a SocietyOfMindAgent."""
+    """str: The default description for a TaskAgent."""
 
     def __init__(
         self,
@@ -155,9 +150,9 @@ async def save_state(self) -> Mapping[str, Any]:
         return state.model_dump()
 
     async def load_state(self, state: Mapping[str, Any]) -> None:
-        society_of_mind_state = TaskAgentState.model_validate(state)
-        await self._model_context.load_state(society_of_mind_state.model_context_state)
-        await self._team.load_state(society_of_mind_state.inner_team_state)
+        task_agent_state = TaskAgentState.model_validate(state)
+        await self._model_context.load_state(task_agent_state.model_context_state)
+        await self._team.load_state(task_agent_state.inner_team_state)
 
     def _to_config(self) -> TaskAgentConfig:
         return TaskAgentConfig(

python/src/kagent/cli.py

@@ -312,3 +312,7 @@ def serve(
 
 def run():
     app()
+
+
+if __name__ == "__main__":
+    run()

python/src/kagent/terminations/__init__.py

@@ -142,7 +142,7 @@ export const createInnerAssistantAgent = (formData: AgentFormData, modelClient:
  */
 export const createInnerTeam = (assistantAgent: Component<AssistantAgentConfig>, modelClient: Component<ModelConfig>): Component<RoundRobinGroupChatConfig> => {
   const textTermination: Component<TextMessageTerminationConfig> = {
-    provider: "kagent.terminations.TextMessageTermination",
+    provider: "autogen_agentchat.conditions.TextMessageTermination",
     component_type: "termination",
     version: 1,
     component_version: 1,