feat: enhance gemini models (#11497)

2024-12-17 12:05:13 +08:00
parent 56cfdce453
commit 74fdc16bd1
23 changed files with 138 additions and 113 deletions
--- a/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py
+++ b/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py
@@ -326,6 +326,7 @@ def test_fetch_prompt_messages__basic(faker, llm_node, model_config):
                    tenant_id="test",
                    type=FileType.IMAGE,
                    filename="test1.jpg",
+                    extension=".jpg",
                    transfer_method=FileTransferMethod.REMOTE_URL,
                    remote_url=fake_remote_url,
                )
@@ -395,6 +396,7 @@ def test_fetch_prompt_messages__basic(faker, llm_node, model_config):
                    tenant_id="test",
                    type=FileType.IMAGE,
                    filename="test1.jpg",
+                    extension=".jpg",
                    transfer_method=FileTransferMethod.REMOTE_URL,
                    remote_url=fake_remote_url,
                )