agenticmem 0.1.2.3__py3-none-any.whl → 0.1.3.7__py3-none-any.whl

agenticmem/__init__.py

@@ -22,6 +22,19 @@ from agenticmem_commons.api_schema.retriever_schema import (
     SearchInteractionResponse,
     SearchUserProfileResponse,
 )
+from agenticmem_commons.config_schema import (
+    StorageConfigTest,
+    StorageConfigLocal,
+    StorageConfigS3,
+    StorageConfigSupabase,
+    StorageConfig,
+    ProfileExtractorConfig,
+    FeedbackAggregatorConfig,
+    AgentFeedbackConfig,
+    AgentSuccessConfig,
+    ToolUseConfig,
+    Config,
+)
 
 debug = False
 log = None  # Set to either 'debug' or 'info', controls console logging
@@ -44,4 +57,15 @@ __all__ = [
     "SearchUserProfileRequest",
     "SearchInteractionResponse",
     "SearchUserProfileResponse",
+    "StorageConfigTest",
+    "StorageConfigLocal",
+    "StorageConfigS3",
+    "StorageConfigSupabase",
+    "StorageConfig",
+    "ProfileExtractorConfig",
+    "FeedbackAggregatorConfig",
+    "AgentFeedbackConfig",
+    "AgentSuccessConfig",
+    "ToolUseConfig",
+    "Config",
 ]

agenticmem/client.py

@@ -1,4 +1,4 @@
-from agenticmem_commons import IS_TEST_ENV
+import os
 import aiohttp
 from urllib.parse import urljoin
 import requests
@@ -12,8 +12,19 @@ from agenticmem_commons.api_schema.retriever_schema import (
     GetInteractionsResponse,
     GetUserProfilesRequest,
     GetUserProfilesResponse,
+    GetRawFeedbacksRequest,
+    GetRawFeedbacksResponse,
+    GetFeedbacksRequest,
+    GetFeedbacksResponse,
 )
 
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+IS_TEST_ENV = os.environ.get("IS_TEST_ENV", "false").strip() == "true"
+
 if IS_TEST_ENV:
     BACKEND_URL = "http://127.0.0.1:8000"  # Local server for testing
 else:
@@ -30,6 +41,7 @@ from agenticmem_commons.api_schema.service_schemas import (
     DeleteUserInteractionResponse,
 )
 from agenticmem_commons.api_schema.login_schema import Token
+from agenticmem_commons.config_schema import Config
 
 
 class AgenticMemClient:
@@ -276,3 +288,79 @@ class AgenticMemClient:
             json=request.model_dump(),
         )
         return GetUserProfilesResponse(**response)
+
+    async def set_config(self, config: Union[Config, dict]) -> dict:
+        """Set configuration for the organization.
+
+        Args:
+            config (Union[Config, dict]): The configuration to set
+
+        Returns:
+            dict: Response containing success status and message
+        """
+        if isinstance(config, dict):
+            config = Config(**config)
+        response = await self._make_async_request(
+            "POST",
+            "/api/set_config",
+            json=config.model_dump(),
+        )
+        return response
+
+    async def get_config(self) -> Config:
+        """Get configuration for the organization.
+
+        Returns:
+            Config: The current configuration
+        """
+        response = await self._make_async_request(
+            "GET",
+            "/api/get_config",
+        )
+        return Config(**response)
+
+    async def get_raw_feedbacks(
+        self,
+        request: Optional[Union[GetRawFeedbacksRequest, dict]] = None,
+    ) -> GetRawFeedbacksResponse:
+        """Get raw feedbacks.
+
+        Args:
+            request (Optional[Union[GetRawFeedbacksRequest, dict]]): The get request, defaults to empty request if None
+
+        Returns:
+            GetRawFeedbacksResponse: Response containing raw feedbacks
+        """
+        if request is None:
+            request = GetRawFeedbacksRequest()
+        elif isinstance(request, dict):
+            request = GetRawFeedbacksRequest(**request)
+        response = await self._make_async_request(
+            "POST",
+            "/api/get_raw_feedbacks",
+            json=request.model_dump(),
+        )
+        return GetRawFeedbacksResponse(**response)
+
+    async def get_feedbacks(
+        self,
+        request: Optional[Union[GetFeedbacksRequest, dict]] = None,
+    ) -> GetFeedbacksResponse:
+        """Get feedbacks.
+
+        Args:
+            request (Optional[Union[GetFeedbacksRequest, dict]]): The get request, defaults to empty request if None
+
+        Returns:
+            GetFeedbacksResponse: Response containing feedbacks
+        """
+        if request is None:
+            request = GetFeedbacksRequest()
+        elif isinstance(request, dict):
+            request = GetFeedbacksRequest(**request)
+        response = await self._make_async_request(
+            "POST",
+            "/api/get_feedbacks",
+            json=request.model_dump(),
+        )
+        return GetFeedbacksResponse(**response)

agenticmem-0.1.3.7.dist-info/METADATA

@@ -0,0 +1,245 @@
+Metadata-Version: 2.3
+Name: agenticmem
+Version: 0.1.3.7
+Summary: A Python client for the AgenticMem API
+License: MIT
+Author: AgenticMem Team
+Requires-Python: >=3.10,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: agenticmem-commons (==0.1.3.7)
+Requires-Dist: aiohttp (>=3.12.9,<4.0.0)
+Requires-Dist: griffe (==0.48.0)
+Requires-Dist: mkdocs-with-pdf (>=0.9.3,<0.10.0)
+Requires-Dist: mkdocstrings[python] (>=0.18.0)
+Requires-Dist: pydantic (>=2.0.0,<3.0.0)
+Requires-Dist: python-dateutil (>=2.8.0,<3.0.0)
+Requires-Dist: requests (>=2.25.0,<3.0.0)
+Description-Content-Type: text/markdown
+
+# AgenticMem Python Client
+
+A Python client library for interacting with the AgenticMem API. This client provides easy-to-use interfaces for managing user interactions, profiles, configuration, and agent feedback.
+
+## Installation
+
+```bash
+pip install agenticmem
+```
+
+## Quick Start
+
+```python
+import agenticmem
+from agenticmem import (
+    InteractionRequest,
+    SearchUserProfileRequest,
+    SearchInteractionRequest,
+    UserActionType
+)
+from datetime import datetime
+
+# Initialize the client
+client = agenticmem.AgenticMemClient(url_endpoint="http://127.0.0.1:8081/")
+
+# Login with email/password (async)
+token = await client.login("user@example.com", "password")
+client.api_key = token.api_key
+
+# Publish a user interaction (async)
+response = await client.publish_interaction(
+    request_id="your_request_id",
+    user_id="your_user_id",
+    interaction_requests=[
+        InteractionRequest(
+            role="User",
+            content="Hey, this is John. How can I help you today?",
+            user_action=UserActionType.NONE,
+            user_action_description="",
+            interacted_image_url=""
+        )
+    ],
+    source="conversation"
+)
+print(f"Published interaction: {response.success} - {response.message}")
+
+# Search user profiles (async)
+profiles = await client.search_profiles(
+    request=SearchUserProfileRequest(
+        user_id="your_user_id",
+        query="john",
+        threshold=0.7
+    )
+)
+for profile in profiles.user_profiles:
+    print(f"Profile {profile.profile_id}: {profile.profile_content}")
+
+# Search interactions (async)
+interactions = await client.search_interactions(
+    request=SearchInteractionRequest(
+        user_id="your_user_id",
+        query="name is john"
+    )
+)
+for interaction in interactions.interactions:
+    print(f"Interaction {interaction.interaction_id}: {interaction.content}")
+
+# Get all interactions (async)
+all_interactions = await client.get_interactions(
+    request={"user_id": "your_user_id", "top_k": 10}
+)
+for interaction in all_interactions.interactions:
+    print(f"Interaction: {interaction.content}")
+
+# Get all profiles (async)
+all_profiles = await client.get_profiles(
+    request={"user_id": "your_user_id", "top_k": 10}
+)
+for profile in all_profiles.user_profiles:
+    print(f"Profile: {profile.profile_content}")
+
+# Get profile change log (async)
+change_log = await client.get_profile_change_log()
+print(f"Profile changes: {len(change_log.profile_change_logs)} logs")
+```
+
+## Features
+
+### Authentication
+- Email/password login (async)
+- API key authentication
+- Custom endpoint configuration
+
+### User Interaction Management
+- Publish user interactions (text, images, actions)
+- Search interactions with semantic queries
+- Get direct list of interactions
+- Delete specific interactions
+- Support for various interaction types (text, visual, click events)
+
+### User Profile Management
+- Search user profiles with semantic queries
+- Get direct list of user profiles
+- Delete specific profiles or profiles matching search queries
+- View profile change log history
+- Filter by source and custom features
+
+### Configuration Management
+- Get and set AgenticMem configuration
+- Configure profile extractors
+- Configure agent feedback systems
+- Storage configuration (Supabase, S3, Local)
+
+### Agent Feedback System
+- Get raw feedbacks from user interactions
+- Get aggregated feedbacks
+- Automatic feedback generation based on conversations
+- Configurable feedback thresholds
+
+## API Response Types
+
+All API methods return strongly-typed responses:
+
+### Authentication
+- `login()` returns `Token`
+
+### Interaction Management
+- `publish_interaction()` returns `PublishUserInteractionResponse`
+- `search_interactions()` returns `SearchInteractionResponse`
+- `get_interactions()` returns `GetInteractionsResponse`
+- `delete_interaction()` returns `DeleteUserInteractionResponse`
+
+### Profile Management
+- `search_profiles()` returns `SearchUserProfileResponse`
+- `get_profiles()` returns `GetUserProfilesResponse`
+- `delete_profile()` returns `DeleteUserProfileResponse`
+- `get_profile_change_log()` returns `ProfileChangeLogResponse`
+
+### Configuration Management
+- `get_config()` returns `Config`
+- `set_config()` returns `dict`
+
+### Feedback Management
+- `get_raw_feedbacks()` returns `GetRawFeedbacksResponse`
+- `get_feedbacks()` returns `GetFeedbacksResponse`
+
+## Advanced Usage Examples
+
+### Visual Interactions with Images
+
+```python
+import base64
+
+# Function to encode the image
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
+# Publish image interaction
+base64_image = encode_image("./path/to/image.png")
+await client.publish_interaction(
+    request_id="your_request_id",
+    user_id="your_user_id",
+    interaction_requests=[
+        InteractionRequest(
+            content="I like this",
+            image_encoding=base64_image,
+        )
+    ]
+)
+```
+
+### Configuration Management
+
+```python
+from agenticmem import ProfileExtractorConfig, AgentFeedbackConfig, FeedbackAggregatorConfig
+
+# Get current configuration
+config = await client.get_config()
+print(f"Current config: {config}")
+
+# Update configuration
+feedback_config = AgentFeedbackConfig(
+    feedback_name="sales_feedback",
+    feedback_definition_prompt="Extract actionable feedback for sales representatives",
+    feedback_aggregator_config=FeedbackAggregatorConfig()
+)
+
+config.agent_feedback_configs = [feedback_config]
+await client.set_config(config)
+```
+
+### Feedback Management
+
+```python
+# Get raw feedbacks
+raw_feedbacks = await client.get_raw_feedbacks()
+for feedback in raw_feedbacks.raw_feedbacks:
+    print(f"Feedback: {feedback.feedback_content}")
+
+# Get aggregated feedbacks
+feedbacks = await client.get_feedbacks()
+for feedback in feedbacks.feedbacks:
+    print(f"Aggregated feedback: {feedback}")
+```
+
+## Documentation
+
+For detailed documentation, please visit the `docs/` directory in this repository.
+
+## Important Notes
+
+- All client methods are **async** and require `await`
+- Image interactions support base64 encoded images
+- The client supports flexible request types (dict or typed objects)
+- Configuration changes affect the entire organization
+- Feedback generation is automatic based on configured prompts
+
+## License
+
+MIT License
+

agenticmem-0.1.3.7.dist-info/RECORD

@@ -0,0 +1,6 @@
+agenticmem/__init__.py,sha256=2AOIWHjfEbGNBa8QAlfqWoMbKzK4T_ixXv9-EyRfbnE,1804
+agenticmem/client.py,sha256=Rpmz_4UTXcO3ckOsLQZBBSkXrX0YO5YKjepUNnkheOE,12137
+agenticmem/client_utils.py,sha256=kWTWlyO7s768v38ef41nRDq87Qmb60NX2vmZTVlCB6g,297
+agenticmem-0.1.3.7.dist-info/METADATA,sha256=epRBEcQ-7msg7y-SfPQMIDTASJrDZJbn4aAqg8nRng0,7165
+agenticmem-0.1.3.7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+agenticmem-0.1.3.7.dist-info/RECORD,,

agenticmem-0.1.2.3.dist-info/METADATA

@@ -1,145 +0,0 @@
-Metadata-Version: 2.3
-Name: agenticmem
-Version: 0.1.2.3
-Summary: A Python client for the AgenticMem API
-License: MIT
-Author: AgenticMem Team
-Requires-Python: >=3.10,<4.0
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: agenticmem-commons (==0.1.3.4)
-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)
-Requires-Dist: python-dateutil (>=2.8.0,<3.0.0)
-Requires-Dist: requests (>=2.25.0,<3.0.0)
-Description-Content-Type: text/markdown
-
-# AgenticMem Python Client
-
-A Python client library for interacting with the AgenticMem API. This client provides easy-to-use interfaces for managing user interactions and profiles.
-
-## Installation
-
-```bash
-pip install agenticmem
-```
-
-## Quick Start
-
-```python
-from agenticmem import AgenticMemClient
-from agenticmem_commons.api_schema.service_schemas import InteractionRequest
-from agenticmem_commons.api_schema.retriever_schema import (
-    SearchInteractionRequest,
-    SearchUserProfileRequest,
-    GetInteractionsRequest,
-    GetUserProfilesRequest
-)
-from datetime import datetime
-
-# Initialize the client
-client = AgenticMemClient(api_key="your_api_key")
-
-# Optional: Login with email/password
-token = client.login(email="user@example.com", password="password123")
-
-# Publish a user interaction
-interaction = InteractionRequest(
-    created_at=int(datetime.utcnow().timestamp()),
-    content="User clicked on product X",
-    user_action="click",
-    user_action_description="Clicked on product details button"
-)
-
-response = client.publish_interaction(
-    user_id="user123",
-    request_id="req456",
-    interaction_requests=[interaction]
-)
-print(f"Published interaction: {response.success} - {response.message}")
-
-# Search user profiles
-profiles_request = SearchUserProfileRequest(
-    user_id="user123",
-    search_query="recent interactions",
-    top_k=5
-)
-profiles = client.search_profiles(profiles_request)
-for profile in profiles.profiles:
-    print(f"Profile {profile.profile_id}: {profile.profile_content}")
-
-# Get user profiles directly
-profiles_request = GetUserProfilesRequest(
-    user_id="user123"
-)
-profiles = client.get_profiles(profiles_request)
-for profile in profiles.profiles:
-    print(f"Profile: {profile}")
-
-# Search interactions
-interactions_request = SearchInteractionRequest(
-    user_id="user123",
-    start_time=int(datetime(2024, 1, 1).timestamp()),
-    end_time=int(datetime.utcnow().timestamp())
-)
-interactions = client.search_interactions(interactions_request)
-for interaction in interactions.interactions:
-    print(f"Interaction {interaction.interaction_id}: {interaction.content}")
-
-# Get interactions directly
-interactions_request = GetInteractionsRequest(
-    user_id="user123"
-)
-interactions = client.get_interactions(interactions_request)
-for interaction in interactions.interactions:
-    print(f"Interaction: {interaction}")
-
-# Get profile change log
-change_log = client.get_profile_change_log()
-print(f"Profile changes: {change_log}")
-```
-
-## Features
-
-- Authentication
-  - API key authentication
-  - Email/password login
-- User interaction management
-  - Publish user interactions
-  - Delete specific interactions
-  - Search interactions with time range and filters
-  - Get direct list of interactions
-- User profile management
-  - Search user profiles with customizable queries
-  - Get direct list of user profiles
-  - Delete specific profiles or profiles matching a search query
-  - View profile change log history
-
-## API Response Types
-
-All API methods return strongly-typed responses:
-
-- `login()` returns `Token`
-- `publish_interaction()` returns `PublishUserInteractionResponse`
-- `search_interactions()` returns `SearchInteractionResponse`
-- `get_interactions()` returns `GetInteractionsResponse`
-- `search_profiles()` returns `SearchUserProfileResponse`
-- `get_profiles()` returns `GetUserProfilesResponse`
-- `delete_profile()` returns `DeleteUserProfileResponse`
-- `delete_interaction()` returns `DeleteUserInteractionResponse`
-- `get_profile_change_log()` returns `ProfileChangeLogResponse`
-
-## Documentation
-
-For detailed documentation, please visit [docs link].
-
-## License
-
-MIT License
-

agenticmem-0.1.2.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-agenticmem/__init__.py,sha256=zjj8VIn65S5NzJZV64IUG8G1p0J-9SYBJJMSof8mDU8,1229
-agenticmem/client.py,sha256=lCRxsRZYfz29Kz44n-ebaFigq7pWhSS5LTI48_c_-vM,9424
-agenticmem/client_utils.py,sha256=kWTWlyO7s768v38ef41nRDq87Qmb60NX2vmZTVlCB6g,297
-agenticmem-0.1.2.3.dist-info/METADATA,sha256=WMXUX_uTYEJfJX30LyOmVptfmy8_2DV-TJ0vMoOHD3c,4461
-agenticmem-0.1.2.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-agenticmem-0.1.2.3.dist-info/RECORD,,