mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-23 16:35:59 +00:00
feat(events): human_message content uses OpenAI user message format
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -78,7 +78,7 @@ async def run_agent(
|
|||||||
run_id=run_id,
|
run_id=run_id,
|
||||||
event_type="human_message",
|
event_type="human_message",
|
||||||
category="message",
|
category="message",
|
||||||
content=user_input,
|
content={"role": "user", "content": user_input},
|
||||||
metadata=msg_metadata or None,
|
metadata=msg_metadata or None,
|
||||||
)
|
)
|
||||||
journal.set_first_human_message(user_input)
|
journal.set_first_human_message(user_input)
|
||||||
|
|||||||
@@ -373,7 +373,7 @@ class TestDbBackedLifecycle:
|
|||||||
run_id = record.run_id
|
run_id = record.run_id
|
||||||
|
|
||||||
# Write human_message
|
# Write human_message
|
||||||
await event_store.put(thread_id="t1", run_id=run_id, event_type="human_message", category="message", content="Hello DB")
|
await event_store.put(thread_id="t1", run_id=run_id, event_type="human_message", category="message", content={"role": "user", "content": "Hello DB"})
|
||||||
|
|
||||||
# Simulate journal
|
# Simulate journal
|
||||||
journal = RunJournal(run_id, "t1", event_store, flush_threshold=100)
|
journal = RunJournal(run_id, "t1", event_store, flush_threshold=100)
|
||||||
@@ -554,3 +554,21 @@ class TestDictContent:
|
|||||||
assert len(record["content"]) <= 100
|
assert len(record["content"]) <= 100
|
||||||
|
|
||||||
await close_engine()
|
await close_engine()
|
||||||
|
|
||||||
|
|
||||||
|
class TestOpenAIHumanMessage:
|
||||||
|
@pytest.mark.anyio
|
||||||
|
async def test_human_message_openai_format(self):
|
||||||
|
store = MemoryRunEventStore()
|
||||||
|
await store.put(
|
||||||
|
thread_id="t1",
|
||||||
|
run_id="r1",
|
||||||
|
event_type="human_message",
|
||||||
|
category="message",
|
||||||
|
content={"role": "user", "content": "What is AI?"},
|
||||||
|
metadata={"message_id": "msg_001"},
|
||||||
|
)
|
||||||
|
messages = await store.list_messages("t1")
|
||||||
|
assert len(messages) == 1
|
||||||
|
assert messages[0]["content"] == {"role": "user", "content": "What is AI?"}
|
||||||
|
assert messages[0]["content"]["role"] == "user"
|
||||||
|
|||||||
Reference in New Issue
Block a user