workforce added

RobinXL · RobinXL · commit 95dd7076df90 · 2025-06-02T19:00:48.000-07:00
diff --git a/README.md b/README.md
@@ -17,6 +17,8 @@ Why build AI agent with AgentX?
 - Support all running MCP (model context protocol).
 - Support RAG with built-in re-rank.
 - Multi-agent workforce orchestration.
+- Multiple agents working together with a designated manager agent.
+- Cross vendor LLM orchestration.
 
 ## Installation
 
@@ -85,3 +87,50 @@ text=None cot=None botId='xxx'
 ```
 
 \*`cot` stands for chain-of-thoughts
+
+### Workforce
+
+A Workforce (team) consists of multiple agents working together with a designated manager agent.
+
+```python
+from agentx import AgentX
+
+client = AgentX(api_key="<your api key here>")
+
+# Get the list of workforces/teams you have
+workforces = client.list_workforces()
+print(workforces)
+
+# Get a specific workforce
+workforce = workforces[0]  # or any specific workforce
+print(f"Workforce: {workforce.name}")
+print(f"Manager: {workforce.manager.name}")
+print(f"Agents: {[agent.name for agent in workforce.agents]}")
+```
+
+#### Workforce Conversations
+
+```python
+# Create a new conversation with the workforce
+conversation = workforce.new_conversation()
+
+# List all existing conversations for the workforce
+conversations = workforce.list_conversations()
+print(conversations)
+```
+
+#### Chat with Workforce
+
+Chat with the entire workforce team and get streaming responses from all agents.
+
+```python
+# Stream chat with the workforce
+response = workforce.chat_stream(conversation.id, "How can you help me with this project?")
+for chunk in response:
+    if chunk.text:
+        print(chunk.text, end="")
+    if chunk.cot:
+        print(f" [COT: {chunk.cot}]")
+```
+
+The workforce chat allows you to leverage multiple specialized agents working together to provide comprehensive responses to your queries.
diff --git a/agentx/agentx.py b/agentx/agentx.py
@@ -5,6 +5,7 @@
 
 from agentx.util import get_headers
 from agentx.resources.agent import Agent
+from agentx.resources.workforce import Workforce
 
 
 class AgentX:
@@ -37,15 +38,29 @@ def list_agents(self) -> List[Agent]:
         response = requests.get(url, headers=get_headers())
         # Check if response was successful
         if response.status_code == 200:
-            return [
-                Agent(
-                    id=agent_res.get("_id"),
-                    name=agent_res.get("name"),
-                    avatar=agent_res.get("avatar"),
-                    createdAt=agent_res.get("createdAt"),
-                    updatedAt=agent_res.get("updatedAt"),
-                )
-                for agent_res in response.json()
-            ]
+            return [Agent(**agent) for agent in response.json()]
         else:
             raise Exception(f"Failed to list agents: {response.reason}")
+
+    @staticmethod
+    def list_workforces() -> List["Workforce"]:
+        """List all workforces/teams."""
+        url = "https://api.agentx.so/api/v1/access/teams"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            return [Workforce(**workforce) for workforce in response.json()]
+        else:
+            raise Exception(
+                f"Failed to list workforces: {response.status_code} - {response.reason}"
+            )
+
+    def get_profile(self):
+        """Get the current user's profile information."""
+        url = "https://api.agentx.so/api/v1/access/getProfile"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception(
+                f"Failed to get profile: {response.status_code} - {response.reason}"
+            )
diff --git a/agentx/resources/agent.py b/agentx/resources/agent.py
@@ -10,7 +10,7 @@
 
 @dataclass
 class Agent(BaseModel):
-    id: str
+    id: str = Field(alias="_id")
     name: str
     avatar: Optional[str]
     createdAt: Optional[str]
diff --git a/agentx/resources/workforce.py b/agentx/resources/workforce.py
@@ -0,0 +1,119 @@
+from typing import Optional, List, Dict, Any, Iterator
+from pydantic import BaseModel, Field
+import requests
+import os
+import json
+import logging
+from agentx.util import get_headers
+from agentx.resources.agent import Agent
+from agentx.resources.conversation import Conversation, ChatResponse
+
+
+class User(BaseModel):
+    id: str = Field(alias="_id")
+    name: str
+    email: str
+    deleted: bool
+    createdAt: str
+    updatedAt: str
+    avatar: str
+    status: int
+    customer: str
+    resetPwdToken: Optional[str] = None
+    defaultWorkspace: str
+    workspaces: List[str]
+
+    class Config:
+        populate_by_name = True
+        extra = "ignore"
+
+
+class Workforce(BaseModel):
+    id: str = Field(alias="_id")
+    agents: List[Agent]
+    name: str
+    image: str
+    description: str
+    manager: Agent
+    creator: User
+    context: int
+    references: bool
+    workspace: str
+    createdAt: str
+    updatedAt: str
+
+    class Config:
+        populate_by_name = True
+        extra = "ignore"
+
+    def new_conversation(self) -> Conversation:
+        """Create a new conversation for this workforce."""
+        url = f"https://api.agentx.so/api/v1/access/teams/{self.id}/conversations/new"
+        response = requests.post(
+            url,
+            headers=get_headers(),
+            json={"type": "chat"},
+        )
+        if response.status_code == 200:
+            conv_data = response.json()
+            # Set the agent_id to the manager's ID since this is a workforce conversation
+            conv_data["agent_id"] = self.manager.id
+            return Conversation(**conv_data)
+        else:
+            raise Exception(
+                f"Failed to create new conversation: {response.status_code} - {response.reason}"
+            )
+
+    def list_conversations(self) -> List[Conversation]:
+        """List all conversations for this workforce."""
+        url = f"https://api.agentx.so/api/v1/access/teams/{self.id}/conversations"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            conversations = []
+            for conv_data in response.json():
+                # Set the agent_id to the manager's ID since this is a workforce conversation
+                conv_data["agent_id"] = self.manager.id
+                conversations.append(Conversation(**conv_data))
+            return conversations
+        else:
+            raise Exception(
+                f"Failed to list conversations: {response.status_code} - {response.reason}"
+            )
+
+    def chat_stream(
+        self, conversation_id: str, message: str, context: int = -1
+    ) -> Iterator[ChatResponse]:
+        """Send a message to a team conversation and stream the response."""
+        url = f"https://api.agentx.so/api/v1/access/teams/conversations/{conversation_id}/jsonmessagesse"
+        response = requests.post(
+            url, headers=get_headers(), json={"message": message, "context": context}
+        )
+        result = ""
+        if response.status_code == 200:
+            buf = b""
+            for chunk in response.iter_content():
+                buf += chunk
+                try:
+                    chunk = buf.decode("utf-8")
+                except UnicodeDecodeError:
+                    continue
+                result += chunk
+                buf = b""
+                try:
+                    if result.count("{") == result.count("}"):
+                        catch_json = json.loads(result)
+                        if catch_json:
+                            result = ""
+                            yield ChatResponse(
+                                text=catch_json.get("text"),
+                                cot=catch_json.get("cot"),
+                                botId=catch_json.get("botId"),
+                                reference=catch_json.get("reference"),
+                                tasks=catch_json.get("tasks"),
+                            )
+                except json.JSONDecodeError:
+                    continue
+        else:
+            raise Exception(
+                f"Failed to send message: {response.status_code} - {response.reason}"
+            )
diff --git a/agentx/version.py b/agentx/version.py
@@ -1 +1 @@
-VERSION = "0.3.0"
+VERSION = "0.4.0"
diff --git a/setup.py b/setup.py
@@ -1,17 +1,34 @@
 from setuptools import setup, find_packages
 
+
+def get_version():
+    """Read version from version.py file"""
+    version_file = "agentx/version.py"
+    with open(version_file, "r", encoding="utf-8") as f:
+        for line in f:
+            if line.startswith("VERSION"):
+                return line.split("=")[1].strip().strip('"').strip("'")
+    raise RuntimeError("Unable to find version string.")
+
+
+def get_long_description():
+    """Read README.md file"""
+    with open("README.md", "r", encoding="utf-8") as f:
+        return f.read()
+
+
 setup(
     name="agentx-python",
-    version="0.3.3",  # Update this version number each time you make a release
+    version=get_version(),
     packages=find_packages(),
     install_requires=[
         "urllib3>=1.26.11",
         "certifi",
     ],
     author="Robin Wang and AgentX Team",
     author_email="contact@agentx.so",
-    description="Offical Python SDK for AgentX (https://www.agentx.so/)",
-    long_description=open("README.md").read(),
+    description="Official Python SDK for AgentX (https://www.agentx.so/)",
+    long_description=get_long_description(),
     long_description_content_type="text/markdown",
     url="https://github.com/AgentX-ai/AgentX-python-sdk",
     classifiers=[
