modify post_args test to new json_payload

deepset-ai · Jan 3, 2024 · 1bf42a4 · 1bf42a4
1 parent 57d0685
commit 1bf42a4
Showing 1 changed file with 7 additions and 10 deletions.
diff --git 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