feat(arkitect): responses

Author: wgzesg-bd

arkitect/core/component/memory/base_memory_service.py

@@ -40,7 +40,7 @@ def content(self) -> str:
 
 class BaseMemoryService(ABC):
     @abstractmethod
-    async def add_or_update_memory(
+    async def update_memory(
         self,
         user_id: str,
         new_messages: list[ArkMessage | dict | Response | ChatCompletionMessage],

arkitect/core/component/memory/in_memory_memory_service.py

@@ -26,7 +26,7 @@
     Memory,
     SearchMemoryResponse,
 )
-from arkitect.core.component.memory.utils import format_ark_message_as_string
+from arkitect.core.component.memory.utils import format_message_as_string
 from arkitect.types.llm.model import ArkMessage
 from arkitect.utils.common import Singleton
 
@@ -53,7 +53,7 @@ def __init__(
         self._llm = AsyncArk()
 
     @override
-    async def add_or_update_memory(
+    async def update_memory(
         self,
         user_id: str,
         new_messages: list[ArkMessage | dict | Response | ChatCompletionMessage],
@@ -86,7 +86,7 @@ async def search_memory(
         memories = self.memory[user_id]
         results = "用户过去的交互记录\n\n"
         for memory in memories:
-            content = format_ark_message_as_string(memory)
+            content = format_message_as_string(memory)
             results += content
         summary = await self._llm.chat.completions.create(
             model=self.default_search_model,

arkitect/core/component/memory/mem0_memory_service.py

@@ -34,7 +34,7 @@
     Memory,
     SearchMemoryResponse,
 )
-from arkitect.core.component.memory.utils import format_ark_message_as_dict
+from arkitect.core.component.memory.utils import format_message_as_dict
 from arkitect.telemetry.logger import ERROR, INFO
 from arkitect.types.llm.model import ArkMessage
 from arkitect.utils.common import Singleton
@@ -75,7 +75,7 @@ def __init__(self, config: Mem0Config = default_ark_config) -> None:
         self._task_queue: asyncio.Queue = asyncio.Queue()
 
     @override
-    async def add_or_update_memory(
+    async def update_memory(
         self,
         user_id: str,
         new_messages: list[ArkMessage | dict | Response | ChatCompletionMessage],
@@ -84,7 +84,7 @@ async def add_or_update_memory(
     ) -> None:
         conversation = []
         for item in new_messages:
-            conversation.append(format_ark_message_as_dict(item))
+            conversation.append(format_message_as_dict(item))
         if blocking:
             await self._add_memory(conversation, user_id)
         else:

arkitect/core/component/memory/utils.py

@@ -30,7 +30,7 @@ def _ark_message_to_string(messages: list[ArkMessage | dict]) -> str:
     return content
 
 
-def format_ark_message_as_string(
+def format_message_as_string(
     message: ArkMessage | dict | Response | ChatCompletionMessage,
 ) -> str:
     if isinstance(message, ArkMessage):
@@ -45,7 +45,7 @@ def format_ark_message_as_string(
         raise ValueError("Invalid message type")
 
 
-def format_ark_message_as_dict(
+def format_message_as_dict(
     message: ArkMessage | dict | Response | ChatCompletionMessage,
 ) -> dict:
     if isinstance(message, ArkMessage):

examples/memory_service/responses_with_memory/responses_client.py

@@ -5,12 +5,10 @@
 from openai.types.responses.response_input_param import ResponseInputParam
 from openai.types.responses.response_stream_event import ResponseCompletedEvent
 
-# from arkitect.core.component.memory import (
-#     Mem0MemoryService as MemoryService,
-# )  # InMemoryMemoryServiceSingleton,; InMemoryMemoryService as MemoryService,
-# from arkitect.core.component.memory import Mem0MemoryServiceSingleton
+
 from arkitect.core.component.memory.base_memory_service import BaseMemoryService
 from arkitect.types.llm.model import ArkMessage
+from arkitect.telemetry.logger import INFO
 
 
 class MemoryUpdateBehaviour(str, Enum):
@@ -77,7 +75,7 @@ async def chat(
                     previous_response_id if previous_response_id else NOT_GIVEN
                 ),
                 tools=[{"type": "web_search_preview"}] if use_tool else None,
-                instructions=await self.get_instructions(user_id=user_id),
+                instructions=await self.get_user_profile(user_id=user_id),
             )
             print("\n--- Streaming Response Chunk Details ---")
             last_chunk = None
@@ -93,7 +91,7 @@ async def chat(
 
         if memory_update_behavior != MemoryUpdateBehaviour.NO_AUTO_UPDATE:
             messages.append(last_chunk.response)
-            await self.memory_service.add_or_update_memory(
+            await self.memory_service.update_memory(
                 user_id=user_id,
                 new_messages=messages,
                 blocking=(
@@ -103,11 +101,11 @@ async def chat(
                 ),
             )
         else:
-            print("No memory update")
+            INFO("No memory update")
         print("--- End of Streaming Response ---")
         return
 
-    async def get_instructions(self, user_id: str) -> str:
+    async def get_user_profile(self, user_id: str) -> str:
         memory = await self.memory_service.search_memory(
             user_id, query="Details of room preferences for this user."
         )