agentx-python 0.2.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,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.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/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_python/resources/conversation.py

@@ -2,7 +2,7 @@ from typing import Optional, List
 from pydantic import BaseModel, Field
 import os
 import requests
-from agentx_python.util import get_headers
+from agentx.util import get_headers
 
 
 class Conversation(BaseModel):

{agentx_python-0.2.1.dist-info → agentx_python-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: agentx-python
-Version: 0.2.1
+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
@@ -11,8 +11,17 @@ 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: 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
 
@@ -34,7 +43,7 @@ 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>")
@@ -47,7 +56,7 @@ print(client.list_agents())
 
 Each Conversation has `agents` and `users` tied to it.
 
-```
+```python
 # get agent
 my_agent = client.get_agent(id="<agent id here>")
 
@@ -59,7 +68,7 @@ print(my_agent.list_conversations())
 
 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)

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.2.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-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