feat: refine

wgzesg-bd · wgzesg-bd · commit 26f32823f5de · 2025-06-13T14:34:35.000+08:00
diff --git a/arkitect/core/component/agent/base_agent.py b/arkitect/core/component/agent/base_agent.py
@@ -69,7 +69,6 @@ async def _astream(self, state: State, **kwargs: Any) -> AsyncIterable[BaseEvent
         pass
 
     async def astream(self, state: State, **kwargs: Any) -> AsyncIterable[BaseEvent]:
-
         if self.pre_agent_call_hook:
             async for event in self.pre_agent_call_hook.pre_agent_call(state):
                 yield event
diff --git a/arkitect/core/component/agent/default_agent.py b/arkitect/core/component/agent/default_agent.py
@@ -17,14 +17,13 @@
 from pydantic import BaseModel
 
 from arkitect.core.component.agent import BaseAgent
-from arkitect.core.component.llm_event_stream.llm_event_stream import LLMEventStream
-
 from arkitect.core.component.llm_event_stream.hooks import (
     PostLLMCallHook,
     PostToolCallHook,
     PreLLMCallHook,
     PreToolCallHook,
 )
+from arkitect.core.component.llm_event_stream.llm_event_stream import LLMEventStream
 from arkitect.core.component.llm_event_stream.model import State
 from arkitect.types.responses.event import BaseEvent
 
diff --git a/arkitect/core/component/checkpoint/checkpoint.py b/arkitect/core/component/checkpoint/checkpoint.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import uuid
+
 from pydantic import BaseModel, ConfigDict, Field
 
 from arkitect.core.component.llm_event_stream.model import State
diff --git a/arkitect/core/component/llm_event_stream/chat_completion.py b/arkitect/core/component/llm_event_stream/chat_completion.py
@@ -23,7 +23,7 @@
 )
 
 from arkitect.core.component.tool.tool_pool import ToolPool
-from arkitect.types.llm.model import Message, ArkChatParameters
+from arkitect.types.llm.model import ArkChatParameters, Message
 from arkitect.types.responses.event import BaseEvent, MessageEvent, StateUpdateEvent
 
 from .model import State
diff --git a/arkitect/core/component/llm_event_stream/hooks.py b/arkitect/core/component/llm_event_stream/hooks.py
@@ -14,15 +14,12 @@
 import abc
 from typing import Any, AsyncIterable, Optional, Union
 
-
 from arkitect.types.responses.event import (
     BaseEvent,
 )
 
 from .model import ContextInterruption, State
 
-from .model import State
-
 
 class HookInterruptException(Exception):
     def __init__(
diff --git a/arkitect/core/component/llm_event_stream/llm_event_stream.py b/arkitect/core/component/llm_event_stream/llm_event_stream.py
@@ -21,21 +21,20 @@
     List,
     Optional,
 )
-from arkitect.core.component.tool.utils import (
-    convert_to_chat_completion_content_part_param,
-)
+
 from volcenginesdkarkruntime import AsyncArk
 from volcenginesdkarkruntime.types.chat import (
     ChatCompletionMessageParam,
 )
-from volcenginesdkarkruntime.types.context import CreateContextResponse
 
 from arkitect.core.client import default_ark_client
 
 # from arkitect.core.component.agent.base_agent import BaseAgent
-
 from arkitect.core.component.tool.mcp_client import MCPClient
 from arkitect.core.component.tool.tool_pool import ToolPool, build_tool_pool
+from arkitect.core.component.tool.utils import (
+    convert_to_chat_completion_content_part_param,
+)
 from arkitect.types.llm.model import (
     ArkChatParameters,
     ArkContextParameters,
@@ -47,15 +46,16 @@
     ToolCallEvent,
     ToolCompletedEvent,
 )
+
+from .chat_completion import _AsyncChat
+from .context_completion import _AsyncContext
 from .hooks import (
     HookInterruptException,
     PostLLMCallHook,
     PostToolCallHook,
     PreLLMCallHook,
     PreToolCallHook,
 )
-from .chat_completion import _AsyncChat
-from .context_completion import _AsyncContext
 from .model import ContextInterruption, State
 
 
diff --git a/arkitect/core/component/llm_event_stream/model.py b/arkitect/core/component/llm_event_stream/model.py
@@ -16,12 +16,10 @@
 
 from pydantic import BaseModel, Field
 
-from arkitect.types.llm.model import ArkChatParameters, ArkContextParameters
 from arkitect.types.responses.event import StateUpdateEvent
 
 
 class State(BaseModel):
-
     details: dict = {}
     events: List[StateUpdateEvent] = Field(default_factory=list)
 
diff --git a/arkitect/core/component/memory/__init__.py b/arkitect/core/component/memory/__init__.py
@@ -18,7 +18,6 @@
     InMemoryMemoryServiceSingleton,
 )
 
-
 __all__ = [
     "BaseMemoryService",
     "InMemoryMemoryService",
diff --git a/arkitect/core/component/memory/mem0_memory_service.py b/arkitect/core/component/memory/mem0_memory_service.py
@@ -74,7 +74,6 @@
 
 
 class Mem0MemoryService(BaseMemoryService):
-
     def __init__(self, config: Mem0Config = default_ark_config) -> None:
         self.mem0_config = config if config else Mem0Config()
         self._llm = AsyncArk()
diff --git a/arkitect/core/component/runner/runner.py b/arkitect/core/component/runner/runner.py
@@ -30,7 +30,7 @@
 from arkitect.core.component.checkpoint.checkpoint import Checkpoint
 from arkitect.core.component.llm_event_stream.model import State
 from arkitect.core.component.memory.base_memory_service import BaseMemoryService
-from arkitect.core.component.runner.config import RunnerConfig, MemoryUpdateSetting
+from arkitect.core.component.runner.config import MemoryUpdateSetting, RunnerConfig
 from arkitect.telemetry.logger import ERROR
 from arkitect.types.llm.model import Message
 from arkitect.types.responses.event import BaseEvent, StateUpdateEvent
diff --git a/arkitect/core/component/tool/mcp_client.py b/arkitect/core/component/tool/mcp_client.py
@@ -11,15 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import sys
 import asyncio
 import datetime
-from datetime import timedelta
 import logging
+import sys
 from contextlib import AsyncExitStack
+from datetime import timedelta
 from typing import Any, Dict, TextIO
 
-
 from arkitect.core.component.tool.utils import (
     mcp_to_chat_completion_tool,
 )
@@ -36,7 +35,6 @@
 from mcp.client.streamable_http import streamablehttp_client
 from mcp.types import CallToolResult
 
-
 logger = logging.getLogger(__name__)
 
 
diff --git a/arkitect/core/component/tool/tool_pool.py b/arkitect/core/component/tool/tool_pool.py
@@ -14,7 +14,6 @@
 
 from typing import Any, Callable, Dict
 
-
 from arkitect.core.component.tool.mcp_client import MCPClient
 from arkitect.core.component.tool.utils import (
     find_duplicate_tools,
diff --git a/arkitect/types/llm/model.py b/arkitect/types/llm/model.py
@@ -57,7 +57,6 @@
     TTLTypes,
 )
 
-
 from arkitect.core.errors import InvalidParameter, MissingParameter
 from arkitect.core.runtime import Request, Response
 from arkitect.utils.func_convert import schema_for_function
diff --git a/arkitect/types/responses/event.py b/arkitect/types/responses/event.py
@@ -22,16 +22,17 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import Any, Optional
 import time
+from typing import Any, Optional
+
 from pydantic import BaseModel, Field
 
 from arkitect.core.errors.exceptions import APIException
 from arkitect.types.llm.model import (
     ActionDetail,
     ArkChatCompletionChunk,
-    Message,
     BotUsage,
+    Message,
     ToolDetail,
 )
 
@@ -98,7 +99,6 @@ class InternalServiceError(ErrorEvent):
 
 
 class MessageEvent(BaseEvent, ArkChatCompletionChunk):
-
     def to_chunk(self) -> ArkChatCompletionChunk:
         return self
 
diff --git a/arkitect/types/responses/utils.py b/arkitect/types/responses/utils.py
@@ -23,19 +23,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import volcenginesdkarkruntime.types.chat.chat_completion_chunk as completion_chunk
 
-from arkitect.core.errors.exceptions import APIException
-from arkitect.types.llm.model import (
-    ActionDetail,
-    ArkChatCompletionChunk,
-    BotUsage,
-    ToolDetail,
-)
 from arkitect.types.responses.event import (
     BaseEvent,
-    ErrorEvent,
-    MessageEvent,
     ToolCallEvent,
     ToolChunk,
     ToolCompletedEvent,
diff --git a/examples/memory_service/agent_with_memory/main.py b/examples/memory_service/agent_with_memory/main.py
@@ -13,11 +13,11 @@
 import os
 import re
 from typing import AsyncIterable
+
 from mem0.configs.base import MemoryConfig as Mem0Config
 from mem0.embeddings.configs import EmbedderConfig
 from mem0.llms.configs import LlmConfig
 from mem0.vector_stores.configs import VectorStoreConfig
-from arkitect.core.component.runner.config import RunnerConfig, MemoryUpdateSetting
 from tools import get_commute_duration, get_instructions, web_search
 
 from arkitect.core.component.agent import DefaultAgent
@@ -28,6 +28,7 @@
     Mem0MemoryServiceSingleton,
 )
 from arkitect.core.component.runner import Runner
+from arkitect.core.component.runner.config import MemoryUpdateSetting, RunnerConfig
 from arkitect.launcher.local.serve import launch_serve
 from arkitect.telemetry.trace import task
 from arkitect.types.llm.model import ArkChatCompletionChunk, ArkChatRequest, ArkMessage
diff --git a/examples/memory_service/client/housing_agent_client.py b/examples/memory_service/client/housing_agent_client.py
@@ -1,6 +1,6 @@
 import asyncio
-import os
 import json
+import os
 from pathlib import Path
 
 from openai import AsyncOpenAI
diff --git a/examples/memory_service/responses_with_memory/main.py b/examples/memory_service/responses_with_memory/main.py
@@ -4,9 +4,23 @@
 from typing import AsyncIterable
 
 import volcenginesdkarkruntime.types.chat.chat_completion_chunk as completion_chunk
+from mem0.configs.base import MemoryConfig as Mem0Config
+from mem0.embeddings.configs import EmbedderConfig
+from mem0.llms.configs import LlmConfig
+from mem0.vector_stores.configs import VectorStoreConfig
+from openai import OpenAI
+from openai.types.responses.response_stream_event import (
+    ResponseCompletedEvent,
+    ResponseStreamEvent,
+    ResponseTextDeltaEvent,
+)
+from responses_client import ResponsesClientWithLongTermMemory
+
 from arkitect.core.component.memory import (
     Mem0MemoryService as MemoryService,
-)  # InMemoryMemoryServiceSingleton,; InMemoryMemoryService as MemoryService,
+)
+
+# InMemoryMemoryServiceSingleton,; InMemoryMemoryService as MemoryService,
 from arkitect.core.component.memory import Mem0MemoryServiceSingleton
 
 # from arkitect.core.component.memory import (
@@ -18,17 +32,6 @@
 from arkitect.launcher.local.serve import launch_serve
 from arkitect.telemetry.trace import task
 from arkitect.types.llm.model import ArkChatCompletionChunk, ArkChatRequest
-from mem0.configs.base import MemoryConfig as Mem0Config
-from mem0.embeddings.configs import EmbedderConfig
-from mem0.llms.configs import LlmConfig
-from mem0.vector_stores.configs import VectorStoreConfig
-from openai import OpenAI
-from openai.types.responses.response_stream_event import (
-    ResponseCompletedEvent,
-    ResponseStreamEvent,
-    ResponseTextDeltaEvent,
-)
-from responses_client import ResponsesClientWithLongTermMemory
 
 DEFAULT_EMBEDDING_MODEL = "doubao-embedding-text-240715"
 DEFAULT_LLM_MODEL = "doubao-1-5-vision-pro-32k-250115"
@@ -96,7 +99,6 @@ def convert_chunk(chunk: ResponseStreamEvent) -> ArkChatCompletionChunk | None:
 
 @task(distributed=False)
 async def main(request: ArkChatRequest) -> AsyncIterable[ArkChatCompletionChunk]:
-
     mem_service: MemoryService = Mem0MemoryServiceSingleton.get_instance_sync(
         default_ark_config
     )
diff --git a/examples/memory_service/responses_with_memory/responses_client.py b/examples/memory_service/responses_with_memory/responses_client.py
@@ -1,14 +1,13 @@
-from enum import Enum
 import re
+from enum import Enum
 
 from openai import NOT_GIVEN, OpenAI
 from openai.types.responses.response_input_param import ResponseInputParam
 from openai.types.responses.response_stream_event import ResponseCompletedEvent
 
-
 from arkitect.core.component.memory.base_memory_service import BaseMemoryService
-from arkitect.types.llm.model import ArkMessage
 from arkitect.telemetry.logger import INFO
+from arkitect.types.llm.model import ArkMessage
 
 
 class MemoryUpdateBehaviour(str, Enum):

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`)`
`24`	`24`
`25`	`25`	`from arkitect.core.component.tool.tool_pool import ToolPool`
`26`		`-from arkitect.types.llm.model import Message, ArkChatParameters`
	`26`	`+from arkitect.types.llm.model import ArkChatParameters, Message`
`27`	`27`	`from arkitect.types.responses.event import BaseEvent, MessageEvent, StateUpdateEvent`
`28`	`28`
`29`	`29`	`from .model import State`
Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,6 @@`
`18`	`18`	`InMemoryMemoryServiceSingleton,`
`19`	`19`	`)`
`20`	`20`
`21`		`-`
`22`	`21`	`__all__ = [`
`23`	`22`	`"BaseMemoryService",`
`24`	`23`	`"InMemoryMemoryService",`