mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-08-04 05:50:00 +00:00
97d7554438
Some checks failed
CodeQL Advanced / Analyze (actions) (push) Has been cancelled
CodeQL Advanced / Analyze (javascript-typescript) (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Pre-commit / pre-commit (push) Has been cancelled
Test / Run Web + Local Brain Smoke (push) Has been cancelled
Test / Run Frontend Guardrails (push) Has been cancelled
Test / Run Python Tests (push) Has been cancelled
Co-authored-by: Douglas <douglas.ym.lai@gmail.com> Co-authored-by: Douglas Lai <115660088+Douglasymlai@users.noreply.github.com> Co-authored-by: Cursor <cursoragent@cursor.com> Co-authored-by: Tao Sun <168447269+fengju0213@users.noreply.github.com> Co-authored-by: Weijie Bai <happy.regina.bai@gmail.com>
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# 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.
|
|
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
|
|
import asyncio
|
|
|
|
from app.service.task import TaskLock
|
|
from app.utils.agent_memory import (
|
|
build_agent_memory_snapshot,
|
|
build_memory_context,
|
|
record_agent_memory_snapshot,
|
|
serialize_agent_memory,
|
|
)
|
|
|
|
|
|
class FakeMemory:
|
|
def get_context(self):
|
|
return (
|
|
[
|
|
{
|
|
"role": "user",
|
|
"content": "Inspect the repository",
|
|
"tool_calls": [],
|
|
},
|
|
{
|
|
"role": "assistant",
|
|
"content": "I will inspect it.",
|
|
"tool_calls": [
|
|
{
|
|
"id": "call_1",
|
|
"function": {
|
|
"name": "shell_exec",
|
|
"arguments": '{"cmd": "pwd"}',
|
|
},
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"role": "tool",
|
|
"content": "/tmp/project",
|
|
"tool_call_id": "call_1",
|
|
},
|
|
],
|
|
0,
|
|
)
|
|
|
|
|
|
class FakeAgent:
|
|
agent_name = "single_agent"
|
|
agent_id = "agent_1"
|
|
memory = FakeMemory()
|
|
|
|
|
|
def test_serialize_agent_memory_includes_tool_calls_and_tool_results():
|
|
messages = serialize_agent_memory(FakeAgent())
|
|
|
|
assert messages[1]["tool_calls"][0]["function"]["name"] == "shell_exec"
|
|
assert messages[1]["tool_calls"][0]["function"]["arguments"] == {
|
|
"cmd": "pwd"
|
|
}
|
|
assert messages[2]["tool_call_id"] == "call_1"
|
|
|
|
|
|
def test_record_agent_memory_snapshot_on_task_lock():
|
|
task_lock = TaskLock("project_1", asyncio.Queue(), {})
|
|
|
|
snapshot = record_agent_memory_snapshot(
|
|
task_lock,
|
|
FakeAgent(),
|
|
scope="single_agent",
|
|
task_id="task_1",
|
|
task_content="Inspect",
|
|
task_result="Done",
|
|
)
|
|
|
|
assert snapshot is not None
|
|
assert len(task_lock.agent_memory_history) == 1
|
|
assert task_lock.agent_memory_history[0]["agent_name"] == "single_agent"
|
|
|
|
|
|
def test_build_memory_context_formats_recent_snapshots():
|
|
task_lock = TaskLock("project_1", asyncio.Queue(), {})
|
|
snapshot = build_agent_memory_snapshot(
|
|
FakeAgent(),
|
|
scope="single_agent",
|
|
task_id="task_1",
|
|
task_content="Inspect",
|
|
task_result="Done",
|
|
)
|
|
task_lock.add_agent_memory_snapshot(snapshot)
|
|
|
|
context = build_memory_context(task_lock)
|
|
|
|
assert "Serialized Agent Memory" in context
|
|
assert "single_agent" in context
|
|
assert "assistant tool_calls: shell_exec" in context
|