From 1bf42a47bd253097888839916479f8c1f955abf0 Mon Sep 17 00:00:00 2001 From: Alistair Rogers Date: Wed, 3 Jan 2024 15:24:04 +0000 Subject: [PATCH] modify post_args test to new json_payload --- integrations/ollama/tests/test_generator.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/integrations/ollama/tests/test_generator.py b/integrations/ollama/tests/test_generator.py index 46e880d93..61e9b69b6 100644 --- a/integrations/ollama/tests/test_generator.py +++ b/integrations/ollama/tests/test_generator.py @@ -75,16 +75,13 @@ def test__json_payload(self, configuration, prompt): observed = component._json_payload(prompt=prompt) expected = { - "url": "https://localhost:11434/api/generate", - "json": { - "prompt": prompt, - "model": configuration["model_name"], - "stream": False, - "system": configuration["system_prompt"], - "raw": configuration["raw"], - "template": configuration["template"], - "options": {}, - }, + "prompt": prompt, + "model": configuration["model_name"], + "stream": False, + "system": configuration["system_prompt"], + "raw": configuration["raw"], + "template": configuration["template"], + "options": {}, } assert observed == expected