agentx-python 0.2.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

{agentx_python → agentx}/__init__.py

@@ -1,7 +1,7 @@
 import logging
 
-from agentx_python.agentx import AgentX
-from agentx_python.version import VERSION
+from agentx.agentx import AgentX
+from agentx.version import VERSION
 
 logging.basicConfig(
     level=logging.INFO,

{agentx_python → agentx}/agentx.py

@@ -3,8 +3,8 @@ import requests
 import os
 import logging
 
-from agentx_python.util import get_headers
-from agentx_python.resources.agent import Agent
+from agentx.util import get_headers
+from agentx.resources.agent import Agent
 
 
 class AgentX:

{agentx_python → agentx}/resources/agent.py

@@ -4,7 +4,7 @@ import requests
 import os
 import logging
 from dataclasses import dataclass
-from agentx_python.util import get_headers
+from agentx.util import get_headers
 from .conversation import Conversation
 
 

agentx/resources/conversation.py

@@ -0,0 +1,128 @@
+import json
+import requests
+from typing import Optional, List, Any, Iterator
+from pydantic import BaseModel, Field
+from agentx.util import get_headers
+
+
+class ChatResponse(BaseModel):
+    text: str | None
+    cot: str | None
+    botId: str
+    reference: Optional[Any]
+    tasks: Optional[Any]
+
+
+class Message(BaseModel):
+    id: str = Field(alias="_id")
+    conversationId: str
+    role: str  # user or bot
+    botId: Optional[str] = Field(alias="bot", default=None)
+    userId: Optional[str] = Field(alias="user", default=None)
+    text: str | None
+    cot: str | None
+    createdAt: str
+    updatedAt: str
+
+    class Config:
+        populate_by_name = True
+        extra = "ignore"
+
+
+class Conversation(BaseModel):
+    agent_id: str
+    id: str = Field(alias="_id")
+    title: Optional[str] = Field(default=None)  # conversation customized title
+    users: List[str]
+    agents: List[str] = Field(alias="bots")
+    createdAt: Optional[str]
+    updatedAt: Optional[str]
+
+    class Config:
+        populate_by_name = True
+        extra = "ignore"
+
+    def __init__(self, **data):
+        super().__init__(**data)
+
+    def new_conversation(self) -> "Conversation":
+        url = f"https://api.agentx.so/api/v1/access/agents/{self.agent_id}/conversations/new"
+        response = requests.post(
+            url,
+            headers=get_headers(),
+            json={"type": "chat"},
+        )
+        if response.status_code == 200:
+            new_conv = response.json()
+            new_conv["agent_id"] = self.agent_id
+            return Conversation(**new_conv)
+        else:
+            raise Exception(
+                f"Failed to create new conversation: {response.status_code} - {response.reason}"
+            )
+
+    def list_messages(self) -> List[Message]:
+        url = f"https://api.agentx.so/api/v1/access/agents/{self.agent_id}/conversations/{self.id}"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            res = response.json()
+            if res.get("messages"):
+                message_list = []
+                for message in res["messages"]:
+                    if message.get("bot"):
+                        message["role"] = "bot"
+                    else:
+                        message["role"] = "user"
+                    message_list.append(Message(**message))
+                return message_list
+            else:
+                raise Exception("No messages found in the conversation.")
+        else:
+            raise Exception(
+                f"Failed to retrieve agent details: {response.status_code} - {response.reason}"
+            )
+
+    def chat(self, message: str, context: int = None):
+        url = f"https://api.agentx.so/api/v1/access/conversations/{self.id}/message"
+        response = requests.post(
+            url,
+            headers=get_headers(),
+            json={"message": message, "context": context},
+        )
+        return response.json()
+
+    def chat_stream(self, message: str, context: int = None) -> Iterator[ChatResponse]:
+        url = f"https://api.agentx.so/api/v1/access/conversations/{self.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")
+                    # yield chunk
+                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

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

agentx_python-0.3.2.dist-info/METADATA

@@ -0,0 +1,112 @@
+Metadata-Version: 2.2
+Name: agentx-python
+Version: 0.3.2
+Summary: Offical 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
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: urllib3>=1.26.11
+Requires-Dist: certifi
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+![Logo](agentx/images/LOGO.png)
+
+[![PyPI version](https://img.shields.io/pypi/v/agentx-python)](https://pypi.org/project/agentx-python/)
+
+---
+
+## Fast way to build AI Agents and create agent workforce
+
+The official AgentX Python SDK for [AgentX](https://www.agentx.so/)
+
+Why build AI agent with AgentX?
+
+- Simplicity, Agent - Conversation - Message structure.
+- Include chain-of-thoughts.
+- Choose from most open and closed sourced LLM vendors.
+- Built-in Voice(ASR, TTS), Image Gen, Document, CSV/excel tool, OCR, etc.
+- Support all running MCP (model context protocol).
+- Support RAG with built-in re-rank.
+- Multi-agent workforce orchestration.
+
+## Installation
+
+```bash
+pip install --upgrade agentx-python
+```
+
+## Usage
+
+Provide an `api_key` inline or set `AGENTX_API_KEY` as an environment variable.
+You can get an API key from https://app.agentx.so
+
+### Agent
+
+```python
+from agentx import AgentX
+
+client = AgentX(api_key="<your api key here>")
+
+# Get the list of agents you have
+print(client.list_agents())
+```
+
+### Conversation
+
+Each Conversation has `agents` and `users` tied to it.
+
+```python
+# get agent
+my_agent = client.get_agent(id="<agent id here>")
+
+# Get the list of conversation from this agent
+existing_conversations = my_agent.list_conversations()
+print(existing_conversations)
+
+# Get the list of history messages from a conversation
+last_conversation = existing_conversations[-1]
+msgs = last_conversation.list_messages()
+print(msgs)
+```
+
+### Chat
+
+A `chat` needs to happen in the conversation. You can do `stream` response too, default `False`.
+
+```python
+a_conversation = my_agent.get_conversation(id="<conversation id here>")
+
+response = a_conversation.chat_stream("Hello, what is your name?")
+for chunk in response:
+    print(chunk)
+```
+
+output looks like:
+
+```
+text=None cot='The user is greeting and asking for my ' botId='xxx'
+text=None cot='name, which are casual, straightforward questions.' botId='xxx'
+text=None cot=' I can answer these directly' botId='xxx'
+text='Hello' cot=None botId='xxx'
+text='!' cot=None botId='xxx'
+text=' I' cot=None botId='xxx'
+text=' am' cot=None botId='xxx'
+text=' AgentX' cot=None botId='xxx'
+text=None cot=None botId='xxx'
+```
+
+\*`cot` stands for chain-of-thoughts

agentx_python-0.3.2.dist-info/RECORD

@@ -0,0 +1,12 @@
+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.2.dist-info/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
+agentx_python-0.3.2.dist-info/METADATA,sha256=47m1FfWDCtAvT8frfeKYapKOBc-OHX_GSQxjhdMjDto,2956
+agentx_python-0.3.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agentx_python-0.3.2.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.3.2.dist-info/RECORD,,

{agentx_python-0.2.1.dist-info → agentx_python-0.3.2.dist-info}/WHEEL

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

agentx_python-0.3.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+agentx

agentx_python/resources/conversation.py

@@ -1,63 +0,0 @@
-from typing import Optional, List
-from pydantic import BaseModel, Field
-import os
-import requests
-from agentx_python.util import get_headers
-
-
-class Conversation(BaseModel):
-    agent_id: str
-    id: str
-    title: Optional[str] = Field(default=None)  # conversation customized title
-    users: List[str]
-    agents: List[str]
-    createdAt: Optional[str]
-    updatedAt: Optional[str]
-
-    def __init__(self, **data):
-        super().__init__(**data)
-
-    def generate_conversation_id(self):
-        return "generate new_conversation_id"
-
-    def list_messages(self):
-        url = f"https://api.agentx.so/api/v1/access/agents/{self.agent_id}/conversations/{self.id}"
-        response = requests.get(url, headers=get_headers())
-        if response.status_code == 200:
-            return response.json()
-        else:
-            raise Exception(
-                f"Failed to retrieve agent details: {response.status_code} - {response.reason}"
-            )
-
-    def chat(self, message: str, stream: bool = False, context: int = None):
-        if stream:
-            return self._chat_stream(message, context)
-        else:
-            url = f"https://api.agentx.so/api/v1/access/conversations/{self.id}/message"
-            response = requests.post(
-                url,
-                headers=get_headers(),
-                json={"message": message, "context": context},
-            )
-            return response.json()
-
-    def _chat_stream(self, message: str, context: int = None):
-        url = f"https://api.agentx.so/api/v1/access/conversations/{self.id}/messagesse"
-        response = requests.post(
-            url, headers=get_headers(), json={"message": message, "context": context}
-        )
-        if response.status_code == 200:
-            buf = b""
-            for chunk in response.iter_content():
-                buf += chunk
-                try:
-                    chunk = buf.decode("utf-8")
-                    yield chunk
-                except UnicodeDecodeError:
-                    continue
-                buf = b""
-        else:
-            raise Exception(
-                f"Failed to send message: {response.status_code} - {response.reason}"
-            )

agentx_python/version.py

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

agentx_python-0.2.1.dist-info/METADATA

@@ -1,71 +0,0 @@
-Metadata-Version: 2.1
-Name: agentx-python
-Version: 0.2.1
-Summary: Offical 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
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: urllib3 >=1.26.11
-Requires-Dist: certifi
-
-# AgentX Python SDK API library
-
-[![PyPI version](https://img.shields.io/pypi/v/agentx-python)](https://pypi.org/project/agentx-python/)
-
-The AgentX Python SDK provides a convenient way to access to your Agent programmatically.
-This is a python SDK for AgentX (https://www.agentx.so/)
-
-## Installation
-
-```bash
-pip install --upgrade agentx-python
-```
-
-## Usage
-
-Provide an `api_key` inline or set `AGENTX_API_KEY` as an environment variable.
-You can get an API key from https://app.agentx.so
-
-### Agent
-
-```
-from agentx_python import AgentX
-
-client = AgentX(api_key="<your api key here>")
-
-# Get the list of agents you have
-print(client.list_agents())
-```
-
-### Conversation
-
-Each Conversation has `agents` and `users` tied to it.
-
-```
-# get agent
-my_agent = client.get_agent(id="<agent id here>")
-
-# Get the list of conversation from this agent
-print(my_agent.list_conversations())
-```
-
-### Chat
-
-A `chat` needs to happen in the conversation. You can do `stream` response too, default `False`.
-
-```
-a_conversation = my_agent.get_conversation(id="<conversation id here>")
-
-response = a_conversation.chat("Hello, what is your name?", stream=True)
-for chunk in response:
-    print(chunk, end="")
-
-# output:
-# My name is Rosita. I'm an AI assistant created by AgentX. It's nice to meet you! How can I help you today?
-```

agentx_python-0.2.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-agentx_python/__init__.py,sha256=5VThp1fX_kBYM2afhhiTQZxcbg_kBeDiPExAH88ZPT4,293
-agentx_python/agentx.py,sha256=D4R8cAam0LF8_PuHmAN7FXaRmrMTKsdzPdSs0HJkf6I,1873
-agentx_python/util.py,sha256=grYAa8YTvoQSyuy4GvUOwq6SVBRr4DdrBlcTuZwQpzw,132
-agentx_python/version.py,sha256=CpXi3jGlx23RvRyU7iytOMZrnspdWw4yofS8lpP1AJU,18
-agentx_python/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentx_python/resources/agent.py,sha256=wpsiiYp3ZjYaonq9wL2KDrUuCnFHvrxS9xfQgaKlLrk,1528
-agentx_python/resources/conversation.py,sha256=zpGOpSJsRTtHbeK_pizOtuGsYxpvLOUCkb-zThB0-cs,2186
-agentx_python-0.2.1.dist-info/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
-agentx_python-0.2.1.dist-info/METADATA,sha256=oJwDiWb1rt6LXftOI5fLuswBUANA9irYFFkL7jWMP1U,1851
-agentx_python-0.2.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-agentx_python-0.2.1.dist-info/top_level.txt,sha256=YYLKtZsKSxhcVZ9ijHXupbsRJ7lhEjETXLwAB7lG2Y8,14
-agentx_python-0.2.1.dist-info/RECORD,,

agentx_python-0.2.1.dist-info/top_level.txt

@@ -1 +0,0 @@
-agentx_python