updated ImageFrame and added URLImageFrame and UserImageFrame

2024-04-08 23:10:10 -07:00
parent 96950ca6df
commit 319b8e7816
16 changed files with 97 additions and 30 deletions
--- a/tests/test_aggregators.py
+++ b/tests/test_aggregators.py
@@ -54,13 +54,13 @@ class TestDailyFrameAggregators(unittest.IsolatedAsyncioTestCase):
            TextFrame("Hello, "),
            TextFrame("world."),
            AudioFrame(b"hello"),
-            ImageFrame("image", b"image"),
+            ImageFrame(b"image", (0, 0)),
            AudioFrame(b"world"),
            LLMResponseEndFrame(),
        ]

        expected_output_frames = [
-            ImageFrame("image", b"image"),
+            ImageFrame(b"image", (0, 0)),
            LLMResponseStartFrame(),
            TextFrame("Hello, "),
            TextFrame("world."),