agentx-python 0.3.3__py3-none-any.whl → 0.4.1__py3-none-any.whl

agentx/agentx.py

@@ -5,6 +5,7 @@ import logging
 
 from agentx.util import get_headers
 from agentx.resources.agent import Agent
+from agentx.resources.workforce import Workforce
 
 
 class AgentX:
@@ -37,15 +38,29 @@ class AgentX:
         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}"
+            )

agentx/resources/agent.py

@@ -10,7 +10,7 @@ from .conversation import Conversation
 
 @dataclass
 class Agent(BaseModel):
-    id: str
+    id: str = Field(alias="_id")
     name: str
     avatar: Optional[str]
     createdAt: Optional[str]

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}"
+            )

agentx/version.py

@@ -1 +1 @@
-VERSION = "0.3.0"
+VERSION = "0.4.1"

{agentx_python-0.3.3.dist-info → agentx_python-0.4.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.3.3
-Summary: Offical Python SDK for AgentX (https://www.agentx.so/)
+Version: 0.4.1
+Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python-sdk
 Author: Robin Wang and AgentX Team
 Author-email: contact@agentx.so
@@ -13,12 +13,16 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: urllib3>=1.26.11
 Requires-Dist: certifi
+Requires-Dist: requests
+Requires-Dist: pydantic
+Requires-Dist: pydantic_core
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
@@ -42,6 +46,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
 
@@ -110,3 +116,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.

agentx_python-0.4.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+agentx/__init__.py,sha256=AiY83tsCtjSMIBVsvNuxOgSU4i-rCREQrqDrNmCoSd0,279
+agentx/agentx.py,sha256=KqOdK8xtqdBZh8jlQehzeUdKK8PQ1vcmU2P3iBF9UQc,2502
+agentx/util.py,sha256=grYAa8YTvoQSyuy4GvUOwq6SVBRr4DdrBlcTuZwQpzw,132
+agentx/version.py,sha256=Z4S0cUqfWFIxSvanoTDQh676x2hjV4pKRi-FEpPk-k4,18
+agentx/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentx/resources/agent.py,sha256=SCXuRudp145e-g3R9ohugiPLPgmQt5ARgkbdAoqtIFQ,1542
+agentx/resources/conversation.py,sha256=Xwy3IfFF6C7U7Xu6BNC32IOPqmaKNwCBTbN2w4FpI8w,4485
+agentx/resources/workforce.py,sha256=9yPrh3zrlbD_D8EruntT8dhQyTZDyxvocYS3VeP3LVw,4143
+agentx_python-0.4.1.dist-info/licenses/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
+agentx_python-0.4.1.dist-info/METADATA,sha256=HVnzmSzzTX1VNHQqzxuxKfkMImttFVxz9Lta14hLtjY,4518
+agentx_python-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agentx_python-0.4.1.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.4.1.dist-info/RECORD,,

{agentx_python-0.3.3.dist-info → agentx_python-0.4.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

agentx_python-0.3.3.dist-info/RECORD

@@ -1,12 +0,0 @@
-agentx/__init__.py,sha256=AiY83tsCtjSMIBVsvNuxOgSU4i-rCREQrqDrNmCoSd0,279
-agentx/agentx.py,sha256=kmpAgnvtLdA3ytn0TJuDmchOQZydjuD4NSP77K3tsYM,1859
-agentx/util.py,sha256=grYAa8YTvoQSyuy4GvUOwq6SVBRr4DdrBlcTuZwQpzw,132
-agentx/version.py,sha256=OgfvlOui_Z4FwAja2bTJgiUkMeuAXZ561wYewaEr6SQ,18
-agentx/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentx/resources/agent.py,sha256=SMu4H1UDL31Bub4hnaySoySeckgAchRSaAaHDpXa1L8,1521
-agentx/resources/conversation.py,sha256=Xwy3IfFF6C7U7Xu6BNC32IOPqmaKNwCBTbN2w4FpI8w,4485
-agentx_python-0.3.3.dist-info/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
-agentx_python-0.3.3.dist-info/METADATA,sha256=HErykm3XrfFIvbeXV_WBgCbIOpU0Dz3yW4YJp5EZu2Q,3008
-agentx_python-0.3.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-agentx_python-0.3.3.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
-agentx_python-0.3.3.dist-info/RECORD,,