agenticmem 0.1.1__py3-none-any.whl → 0.1.2.1__py3-none-any.whl

agenticmem/client.py

@@ -1,6 +1,7 @@
-import requests
+import aiohttp
 from urllib.parse import urljoin
 import os
+import requests
 from typing import Optional, Union
 from agenticmem_commons.api_schema.retriever_schema import (
     SearchInteractionRequest,
@@ -13,7 +14,7 @@ from agenticmem_commons.api_schema.retriever_schema import (
     GetUserProfilesResponse,
 )
 
-if os.environ.get("TEST_ENV", "False") == "True":
+if os.environ.get("TEST_ENV", True):
     BACKEND_URL = "http://127.0.0.1:8000"  # Local server for testing
 else:
     BACKEND_URL = "http://agenticmem-test.us-west-2.elasticbeanstalk.com:8081"  # Elastic Beanstalk server url
@@ -34,16 +35,38 @@ from agenticmem_commons.api_schema.login_schema import Token
 class AgenticMemClient:
     """Client for interacting with the AgenticMem API."""
 
-    def __init__(self, api_key: str = ""):
+    def __init__(self, api_key: str = "", url_endpoint: str = ""):
         """Initialize the AgenticMem client.
 
         Args:
             api_key (str): Your API key for authentication
         """
         self.api_key = api_key
-        self.base_url = BACKEND_URL
+        self.base_url = url_endpoint if url_endpoint else BACKEND_URL
         self.session = requests.Session()
 
+    async def _make_async_request(
+        self, method: str, endpoint: str, headers: Optional[dict] = None, **kwargs
+    ):
+        """Make an async HTTP request to the API."""
+        url = urljoin(self.base_url, endpoint)
+
+        headers = headers or {}
+        if self.api_key:
+            headers.update(
+                {
+                    "Authorization": f"Bearer {self.api_key}",
+                    "Content-Type": "application/json",
+                }
+            )
+
+        async with aiohttp.ClientSession() as async_session:
+            response = await async_session.request(
+                method, url, headers=headers, **kwargs
+            )
+            response.raise_for_status()
+            return await response.json()
+
     def _make_request(
         self, method: str, endpoint: str, headers: Optional[dict] = None, **kwargs
     ):
@@ -72,17 +95,9 @@ class AgenticMemClient:
         response.raise_for_status()
         return response.json()
 
-    def login(self, email: str, password: str) -> Token:
-        """Login to the AgenticMem API.
-
-        Args:
-            email (str): The user's email
-            password (str): The user's password
-
-        Returns:
-            Token: Response containing success status and message
-        """
-        response = self._make_request(
+    async def login(self, email: str, password: str) -> Token:
+        """Async login to the AgenticMem API."""
+        response = await self._make_async_request(
             "POST",
             "/token",
             data={"username": email, "password": password},
@@ -93,11 +108,13 @@ class AgenticMemClient:
         )
         return Token(**response)
 
-    def publish_interaction(
+    async def publish_interaction(
         self,
         request_id: str,
         user_id: str,
         interaction_requests: list[Union[InteractionRequest, dict]],
+        source: str = "",
+        agent_version: str = "",
     ) -> PublishUserInteractionResponse:
         """Publish user interactions.
 
@@ -105,7 +122,7 @@ class AgenticMemClient:
             request_id (str): The request ID
             user_id (str): The user ID
             interaction_requests (List[InteractionRequest]): List of interaction requests
-
+            source (str, optional): The source of the interaction
         Returns:
             PublishUserInteractionResponse: Response containing success status and message
         """
@@ -121,15 +138,17 @@ class AgenticMemClient:
             request_id=request_id,
             user_id=user_id,
             interaction_requests=interaction_requests,
+            source=source,
+            agent_version=agent_version,
         )
-        response = self._make_request(
+        response = await self._make_async_request(
             "POST",
             "/api/publish_interaction",
-            data=request.model_dump_json(),
+            json=request.model_dump(),
         )
         return PublishUserInteractionResponse(**response)
 
-    def search_interactions(
+    async def search_interactions(
         self,
         request: Union[SearchInteractionRequest, dict],
     ) -> SearchInteractionResponse:
@@ -143,14 +162,14 @@ class AgenticMemClient:
         """
         if isinstance(request, dict):
             request = SearchInteractionRequest(**request)
-        response = self._make_request(
+        response = await self._make_async_request(
             "POST",
             "/api/search_interactions",
-            data=request.model_dump_json(),
+            json=request.model_dump(),
         )
         return SearchInteractionResponse(**response)
 
-    def search_profiles(
+    async def search_profiles(
         self,
         request: Union[SearchUserProfileRequest, dict],
     ) -> SearchUserProfileResponse:
@@ -164,12 +183,12 @@ class AgenticMemClient:
         """
         if isinstance(request, dict):
             request = SearchUserProfileRequest(**request)
-        response = self._make_request(
-            "POST", "/api/search_profiles", data=request.model_dump_json()
+        response = await self._make_async_request(
+            "POST", "/api/search_profiles", json=request.model_dump()
         )
         return SearchUserProfileResponse(**response)
 
-    def delete_profile(
+    async def delete_profile(
         self, user_id: str, profile_id: str = "", search_query: str = ""
     ) -> DeleteUserProfileResponse:
         """Delete user profiles.
@@ -187,12 +206,12 @@ class AgenticMemClient:
             profile_id=profile_id,
             search_query=search_query,
         )
-        response = self._make_request(
-            "DELETE", "/api/delete_profile", data=request.model_dump_json()
+        response = await self._make_async_request(
+            "DELETE", "/api/delete_profile", json=request.model_dump()
         )
         return DeleteUserProfileResponse(**response)
 
-    def delete_interaction(
+    async def delete_interaction(
         self, user_id: str, interaction_id: str
     ) -> DeleteUserInteractionResponse:
         """Delete a user interaction.
@@ -207,16 +226,16 @@ class AgenticMemClient:
         request = DeleteUserInteractionRequest(
             user_id=user_id, interaction_id=interaction_id
         )
-        response = self._make_request(
-            "DELETE", "/api/delete_interaction", data=request.model_dump_json()
+        response = await self._make_async_request(
+            "DELETE", "/api/delete_interaction", json=request.model_dump()
         )
         return DeleteUserInteractionResponse(**response)
 
-    def get_profile_change_log(self) -> ProfileChangeLogResponse:
-        response = self._make_request("GET", "/api/profile_change_log")
+    async def get_profile_change_log(self) -> ProfileChangeLogResponse:
+        response = await self._make_async_request("GET", "/api/profile_change_log")
         return ProfileChangeLogResponse(**response)
 
-    def get_interactions(
+    async def get_interactions(
         self,
         request: Union[GetInteractionsRequest, dict],
     ) -> GetInteractionsResponse:
@@ -230,14 +249,14 @@ class AgenticMemClient:
         """
         if isinstance(request, dict):
             request = GetInteractionsRequest(**request)
-        response = self._make_request(
+        response = await self._make_async_request(
             "POST",
             "/api/get_interactions",
-            data=request.model_dump_json(),
+            json=request.model_dump(),
         )
         return GetInteractionsResponse(**response)
 
-    def get_profiles(
+    async def get_profiles(
         self,
         request: Union[GetUserProfilesRequest, dict],
     ) -> GetUserProfilesResponse:
@@ -251,9 +270,9 @@ class AgenticMemClient:
         """
         if isinstance(request, dict):
             request = GetUserProfilesRequest(**request)
-        response = self._make_request(
+        response = await self._make_async_request(
             "POST",
             "/api/get_profiles",
-            data=request.model_dump_json(),
+            json=request.model_dump(),
         )
         return GetUserProfilesResponse(**response)

{agenticmem-0.1.1.dist-info → agenticmem-0.1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: agenticmem
-Version: 0.1.1
+Version: 0.1.2.1
 Summary: A Python client for the AgenticMem API
 License: MIT
 Author: AgenticMem Team
@@ -9,7 +9,10 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: agenticmem-commons (==0.1.1)
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: agenticmem-commons (==0.1.2)
+Requires-Dist: aiohttp (>=3.12.9,<4.0.0)
 Requires-Dist: griffe (==0.48.0)
 Requires-Dist: mkdocstrings[python] (>=0.18.0)
 Requires-Dist: pydantic (>=2.0.0,<3.0.0)
@@ -33,7 +36,7 @@ pip install agenticmem
 from agenticmem import AgenticMemClient
 from agenticmem_commons.api_schema.service_schemas import InteractionRequest
 from agenticmem_commons.api_schema.retriever_schema import (
-    SearchInteractionRequest, 
+    SearchInteractionRequest,
     SearchUserProfileRequest,
     GetInteractionsRequest,
     GetUserProfilesRequest
@@ -48,8 +51,8 @@ token = client.login(email="user@example.com", password="password123")
 
 # Publish a user interaction
 interaction = InteractionRequest(
-    timestamp=int(datetime.utcnow().timestamp()),
-    text_interaction="User clicked on product X",
+    created_at=int(datetime.utcnow().timestamp()),
+    content="User clicked on product X",
     user_action="click",
     user_action_description="Clicked on product details button"
 )
@@ -87,7 +90,7 @@ interactions_request = SearchInteractionRequest(
 )
 interactions = client.search_interactions(interactions_request)
 for interaction in interactions.interactions:
-    print(f"Interaction {interaction.interaction_id}: {interaction.text_interaction}")
+    print(f"Interaction {interaction.interaction_id}: {interaction.content}")
 
 # Get interactions directly
 interactions_request = GetInteractionsRequest(

agenticmem-0.1.2.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+agenticmem/__init__.py,sha256=zjj8VIn65S5NzJZV64IUG8G1p0J-9SYBJJMSof8mDU8,1229
+agenticmem/client.py,sha256=zTXWz3FeW68511r8mMt-kF-OhAYePLv02tswoL1oNpM,9412
+agenticmem/client_utils.py,sha256=kWTWlyO7s768v38ef41nRDq87Qmb60NX2vmZTVlCB6g,297
+agenticmem-0.1.2.1.dist-info/METADATA,sha256=M7QTy_AWaMTT5Tvgz-uTks-k73qcTGAJBQqtbpd7Gd0,4459
+agenticmem-0.1.2.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+agenticmem-0.1.2.1.dist-info/RECORD,,

{agenticmem-0.1.1.dist-info → agenticmem-0.1.2.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.7.0
+Generator: poetry-core 2.1.2
 Root-Is-Purelib: true
 Tag: py3-none-any

agenticmem-0.1.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-agenticmem/__init__.py,sha256=zjj8VIn65S5NzJZV64IUG8G1p0J-9SYBJJMSof8mDU8,1229
-agenticmem/client.py,sha256=_ZRjGFsxHnFdrCoQJEu8ofioxZYvI1vSlRCsMK8-Z5g,8480
-agenticmem/client_utils.py,sha256=kWTWlyO7s768v38ef41nRDq87Qmb60NX2vmZTVlCB6g,297
-agenticmem-0.1.1.dist-info/METADATA,sha256=ccLBT23K7aLygLLscXgSJxu2VY7FzCgFijAW1V0CdAI,4332
-agenticmem-0.1.1.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-agenticmem-0.1.1.dist-info/RECORD,,