agentx-python 0.3.2__tar.gz → 0.4.0__tar.gz

{agentx_python-0.3.2 → agentx_python-0.4.0}/PKG-INFO

@@ -1,7 +1,7 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.3.2
-Summary: Offical Python SDK for AgentX (https://www.agentx.so/)
+Version: 0.4.0
+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
@@ -19,11 +19,12 @@ Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-![Logo](agentx/images/LOGO.png)
+![Logo](https://agentx-resources.s3.us-west-1.amazonaws.com/AgentX-logo-387x60.png)
 
 [![PyPI version](https://img.shields.io/pypi/v/agentx-python)](https://pypi.org/project/agentx-python/)
 
@@ -42,6 +43,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 +113,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.3.2 → agentx_python-0.4.0}/README.md

@@ -1,4 +1,4 @@
-![Logo](agentx/images/LOGO.png)
+![Logo](https://agentx-resources.s3.us-west-1.amazonaws.com/AgentX-logo-387x60.png)
 
 [![PyPI version](https://img.shields.io/pypi/v/agentx-python)](https://pypi.org/project/agentx-python/)
 
@@ -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.

{agentx_python-0.3.2 → agentx_python-0.4.0}/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_python-0.3.2 → agentx_python-0.4.0}/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_python-0.4.0/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_python-0.4.0/agentx/version.py

@@ -0,0 +1 @@
+VERSION = "0.4.0"

{agentx_python-0.3.2 → agentx_python-0.4.0}/agentx_python.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.3.2
-Summary: Offical Python SDK for AgentX (https://www.agentx.so/)
+Version: 0.4.0
+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
@@ -19,11 +19,12 @@ Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-![Logo](agentx/images/LOGO.png)
+![Logo](https://agentx-resources.s3.us-west-1.amazonaws.com/AgentX-logo-387x60.png)
 
 [![PyPI version](https://img.shields.io/pypi/v/agentx-python)](https://pypi.org/project/agentx-python/)
 
@@ -42,6 +43,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 +113,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.3.2 → agentx_python-0.4.0}/agentx_python.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ agentx/version.py
 agentx/resources/__init__.py
 agentx/resources/agent.py
 agentx/resources/conversation.py
+agentx/resources/workforce.py
 agentx_python.egg-info/PKG-INFO
 agentx_python.egg-info/SOURCES.txt
 agentx_python.egg-info/dependency_links.txt

agentx_python-0.4.0/setup.py

@@ -0,0 +1,40 @@
+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=get_version(),
+    packages=find_packages(),
+    install_requires=[
+        "urllib3>=1.26.11",
+        "certifi",
+    ],
+    author="Robin Wang and AgentX Team",
+    author_email="contact@agentx.so",
+    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=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires=">=3.6",
+)

agentx_python-0.3.2/agentx/version.py

@@ -1 +0,0 @@
-VERSION = "0.3.0"

agentx_python-0.3.2/setup.py

@@ -1,23 +0,0 @@
-from setuptools import setup, find_packages
-
-setup(
-    name="agentx-python",
-    version="0.3.2",  # Update this version number each time you make a release
-    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(),
-    long_description_content_type="text/markdown",
-    url="https://github.com/AgentX-ai/AgentX-python-sdk",
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-    ],
-    python_requires=">=3.6",
-)