diff --git a/test_stream_agent.py b/test_stream_agent.py deleted file mode 100644 index f3d2cb8..0000000 --- a/test_stream_agent.py +++ /dev/null @@ -1,24 +0,0 @@ -from langchain_core.messages import AIMessage - -from stream_agent import format_message - - -def test_format_message_with_content(): - message = AIMessage(content="hello") - - assert format_message(message) == "hello" - - -def test_format_message_with_tool_call(): - message = AIMessage( - content="", - tool_calls=[ - { - "name": "get_demo_price", - "args": {"product": "молоко", "city": "Казань"}, - "id": "call-1", - } - ], - ) - - assert "get_demo_price" in format_message(message)