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

agentx_python/__init__.py

@@ -1,7 +1,6 @@
 import logging
-import os
 
-from agentx_python.agent import Agent
+from agentx_python.agentx import AgentX
 from agentx_python.version import VERSION
 
 logging.basicConfig(
@@ -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_python.util import get_headers
+from agentx_python.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_python.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_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/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.2.1.dist-info/METADATA

@@ -0,0 +1,71 @@
+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

@@ -0,0 +1,12 @@
+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/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,,