agentx-python 0.1__py3-none-any.whl → 0.3.1__py3-none-any.whl

agentx/__init__.py

@@ -0,0 +1,13 @@
+import logging
+
+from agentx.agentx import AgentX
+from agentx.version import VERSION
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S %Z",
+)
+
+__all__ = ["AgentX"]
+__version__ = VERSION

agentx/agentx.py

@@ -0,0 +1,51 @@
+from typing import List
+import requests
+import os
+import logging
+
+from agentx.util import get_headers
+from agentx.resources.agent import Agent
+
+
+class AgentX:
+
+    def __init__(self, api_key: str = None):
+        self.api_key = api_key or os.getenv("AGENTX_API_KEY")
+        if self.api_key and not os.getenv("AGENTX_API_KEY"):
+            os.environ["AGENTX_API_KEY"] = self.api_key
+
+    def get_agent(self, id: str) -> Agent:
+        url = f"https://api.agentx.so/api/v1/access/agents/{id}"
+        # Make a GET request to the AgentX API
+        response = requests.get(url, headers=get_headers())
+        # Check if response was successful
+        if response.status_code == 200:
+            agent_res = response.json()
+            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"),
+            )
+        else:
+            raise Exception(f"Failed to retrieve agent: {response.reason}")
+
+    def list_agents(self) -> List[Agent]:
+        url = "https://api.agentx.so/api/v1/access/agents"
+        # Make a GET request to the AgentX API
+        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()
+            ]
+        else:
+            raise Exception(f"Failed to list agents: {response.reason}")

agentx/resources/agent.py

@@ -0,0 +1,46 @@
+from typing import Optional, List
+from pydantic import BaseModel, Field
+import requests
+import os
+import logging
+from dataclasses import dataclass
+from agentx.util import get_headers
+from .conversation import Conversation
+
+
+@dataclass
+class Agent(BaseModel):
+    id: str
+    name: str
+    avatar: Optional[str]
+    createdAt: Optional[str]
+    updatedAt: Optional[str]
+
+    def __init__(self, **data):
+        super().__init__(**data)
+
+    def get_conversation(self, id: str) -> Conversation:
+        list_of_conversations = self.list_conversations()
+        return next(
+            (conv for conv in list_of_conversations if conv.id == id),
+            Exception("404 - Conversation not found"),
+        )
+
+    def list_conversations(self) -> List[Conversation]:
+        url = f"https://api.agentx.so/api/v1/access/agents/{self.id}/conversations"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            return [
+                Conversation(
+                    agent_id=self.id,
+                    id=conv_res.get("_id"),
+                    title=conv_res.get("title"),
+                    users=conv_res.get("users"),
+                    agents=conv_res.get("bots"),
+                    createdAt=conv_res.get("createdAt"),
+                    updatedAt=conv_res.get("updatedAt"),
+                )
+                for conv_res in response.json()
+            ]
+        else:
+            raise Exception(f"Failed to retrieve agent details: {response.reason}")

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/util.py

@@ -0,0 +1,5 @@
+import os
+
+
+def get_headers(api_key: str = None):
+    return {"accept": "*/*", "x-api-key": api_key or os.getenv("AGENTX_API_KEY")}

agentx/version.py

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

agentx_python/__init__.py

@@ -1,8 +1,7 @@
 import logging
-import os
 
-from agentx_python.agent import Agent
-from agentx_python.version import VERSION
+from agentx.agentx import AgentX
+from agentx.version import VERSION
 
 logging.basicConfig(
     level=logging.INFO,
@@ -10,10 +9,5 @@ logging.basicConfig(
     datefmt="%Y-%m-%d %H:%M:%S %Z",
 )
 
-
-api_key = os.environ.get("OPENAI_API_KEY")
-
-Agent = Agent()
-
+__all__ = ["AgentX"]
 __version__ = VERSION
-__all__ = ["api_key", "Agent"]

agentx_python/agentx.py

@@ -0,0 +1,51 @@
+from typing import List
+import requests
+import os
+import logging
+
+from agentx.util import get_headers
+from agentx.resources.agent import Agent
+
+
+class AgentX:
+
+    def __init__(self, api_key: str = None):
+        self.api_key = api_key or os.getenv("AGENTX_API_KEY")
+        if self.api_key and not os.getenv("AGENTX_API_KEY"):
+            os.environ["AGENTX_API_KEY"] = self.api_key
+
+    def get_agent(self, id: str) -> Agent:
+        url = f"https://api.agentx.so/api/v1/access/agents/{id}"
+        # Make a GET request to the AgentX API
+        response = requests.get(url, headers=get_headers())
+        # Check if response was successful
+        if response.status_code == 200:
+            agent_res = response.json()
+            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"),
+            )
+        else:
+            raise Exception(f"Failed to retrieve agent: {response.reason}")
+
+    def list_agents(self) -> List[Agent]:
+        url = "https://api.agentx.so/api/v1/access/agents"
+        # Make a GET request to the AgentX API
+        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()
+            ]
+        else:
+            raise Exception(f"Failed to list agents: {response.reason}")

agentx_python/resources/agent.py

@@ -0,0 +1,46 @@
+from typing import Optional, List
+from pydantic import BaseModel, Field
+import requests
+import os
+import logging
+from dataclasses import dataclass
+from agentx.util import get_headers
+from .conversation import Conversation
+
+
+@dataclass
+class Agent(BaseModel):
+    id: str
+    name: str
+    avatar: Optional[str]
+    createdAt: Optional[str]
+    updatedAt: Optional[str]
+
+    def __init__(self, **data):
+        super().__init__(**data)
+
+    def get_conversation(self, id: str) -> Conversation:
+        list_of_conversations = self.list_conversations()
+        return next(
+            (conv for conv in list_of_conversations if conv.id == id),
+            Exception("404 - Conversation not found"),
+        )
+
+    def list_conversations(self) -> List[Conversation]:
+        url = f"https://api.agentx.so/api/v1/access/agents/{self.id}/conversations"
+        response = requests.get(url, headers=get_headers())
+        if response.status_code == 200:
+            return [
+                Conversation(
+                    agent_id=self.id,
+                    id=conv_res.get("_id"),
+                    title=conv_res.get("title"),
+                    users=conv_res.get("users"),
+                    agents=conv_res.get("bots"),
+                    createdAt=conv_res.get("createdAt"),
+                    updatedAt=conv_res.get("updatedAt"),
+                )
+                for conv_res in response.json()
+            ]
+        else:
+            raise Exception(f"Failed to retrieve agent details: {response.reason}")

agentx_python/resources/conversation.py

@@ -0,0 +1,63 @@
+from typing import Optional, List
+from pydantic import BaseModel, Field
+import os
+import requests
+from agentx.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/util.py

@@ -0,0 +1,5 @@
+import os
+
+
+def get_headers(api_key: str = None):
+    return {"accept": "*/*", "x-api-key": api_key or os.getenv("AGENTX_API_KEY")}

agentx_python-0.3.1.dist-info/METADATA

@@ -0,0 +1,80 @@
+Metadata-Version: 2.2
+Name: agentx-python
+Version: 0.3.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
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# 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
+
+```python
+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.
+
+```python
+# 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`.
+
+```python
+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.3.1.dist-info/RECORD

@@ -0,0 +1,19 @@
+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/__init__.py,sha256=AiY83tsCtjSMIBVsvNuxOgSU4i-rCREQrqDrNmCoSd0,279
+agentx_python/agentx.py,sha256=kmpAgnvtLdA3ytn0TJuDmchOQZydjuD4NSP77K3tsYM,1859
+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=SMu4H1UDL31Bub4hnaySoySeckgAchRSaAaHDpXa1L8,1521
+agentx_python/resources/conversation.py,sha256=MR0ZBkpSWxUdWoN5HklaoP79zKxp-uzrnX5koX3m504,2179
+agentx_python-0.3.1.dist-info/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
+agentx_python-0.3.1.dist-info/METADATA,sha256=YHCO7IJsO9XajdldjKJtWvgIwVhRRj8z3ffISqzarIs,2065
+agentx_python-0.3.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agentx_python-0.3.1.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.3.1.dist-info/RECORD,,

{agentx_python-0.1.dist-info → agentx_python-0.3.1.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.1.dist-info/top_level.txt

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

agentx_python/agent.py

@@ -1,55 +0,0 @@
-import requests
-import os
-import logging
-import agentx_python as agentx
-from functools import wraps
-
-
-def api_key_check_decorator(method):
-    @wraps(method)
-    def wrapper(self, *args, **kwargs):
-        self.middleware()
-        return method(self, *args, **kwargs)
-
-    return wrapper
-
-
-class Agent:
-
-    def middleware(self):
-        self.api_key = os.getenv("AGENTX_API_KEY") or agentx.api_key
-        if not self.api_key:
-            raise Exception(
-                "No API key provided. You can set your API key in code using 'agentx.api_key = <API-KEY>', or you can set the environment variable AGENTX_API_KEY=<API-KEY>). You can generate API keys in the AgentX https://app.agentx.so."
-            )
-
-    @api_key_check_decorator
-    def get_agent(self, id: str):
-        url = f"https://api.agentx.so/api/v1/access/agents/{id}"
-        headers = {"accept": "*/*", "x-api-key": self.api_key}
-
-        # Make a GET request to the AgentX API
-        response = requests.get(url, headers=headers)
-        # Check if response was successful
-        if response.status_code == 200:
-            return response.json()
-        else:
-            logging.info(
-                response.json()
-            )  # Print text content of response (or process it as needed)
-            raise Exception(
-                f"Failed to retrieve agent details: {response.status_code} - {response.reason}"
-            )
-
-    @api_key_check_decorator
-    def list_agents(self):
-        url = "https://api.agentx.so/api/v1/access/agents"
-        headers = {"accept": "*/*", "x-api-key": self.api_key}
-
-        # Make a GET request to the AgentX API
-        response = requests.get(url, headers=headers)
-        # Check if response was successful
-        if response.status_code == 200:
-            return response.json()
-        else:
-            logging.info(response.json())

agentx_python-0.1.dist-info/METADATA

@@ -1,32 +0,0 @@
-Metadata-Version: 2.1
-Name: agentx-python
-Version: 0.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
-
-This is a python SDK for AgentX (https://www.agentx.so/)
-
-## Installation
-
-```bash
-pip install --upgrade agentx-python
-```
-
-## Usage
-
-```
-from agentx_python import agent
-print(agent.get_agent("your-agent-id-here"))
-```

agentx_python-0.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-agentx_python/__init__.py,sha256=eocHL3Wv7lxa7us0uYQ4yBDtbYrF38hYOsb4lS69Jk8,373
-agentx_python/agent.py,sha256=FEA_gue4qfbm6WhrX9XZAGdx7cz7HP-clIVP6oqmB9Q,1862
-agentx_python/version.py,sha256=CpXi3jGlx23RvRyU7iytOMZrnspdWw4yofS8lpP1AJU,18
-agentx_python-0.1.dist-info/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
-agentx_python-0.1.dist-info/METADATA,sha256=RFKPnRAs4zy11ZKe2tkxDXXAAU73Ur3eUi25zknsI_k,772
-agentx_python-0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-agentx_python-0.1.dist-info/top_level.txt,sha256=YYLKtZsKSxhcVZ9ijHXupbsRJ7lhEjETXLwAB7lG2Y8,14
-agentx_python-0.1.dist-info/RECORD,,

agentx_python-0.1.dist-info/top_level.txt

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