fixed an issue in the tests

new_test.py

@@ -0,0 +1,13 @@
+import os
+
+from unify import Unify
+
+unify = Unify(
+    # This is the default and optional to include.
+    api_key=os.environ.get("UNIFY_KEY"),
+)
+response = unify.generate(
+    messages="Hello Llama! Who was Isaac Newton?",
+    model="llama-chat",
+    provider="anyscale",
+)

unify/tests.py

@@ -29,7 +29,7 @@ def test_incorrect_model_name_raises_internal_server_error(self) -> None:
         unify = Unify(self.valid_api_key)
         # Provide incorrect model name to generate function
         with self.assertRaises(InternalServerError):
-            unify.generate("hello", "llama-chat")
+            unify.generate(messages="hello", model="llama-chat")
 
     def test_generate_returns_string_when_stream_false(self) -> None:
         # Instantiate Unify with a valid API key
@@ -72,7 +72,7 @@ async def test_incorrect_model_name_raises_internal_server_error(self) -> None:
         async_unify = AsyncUnify(api_key=self.valid_api_key)
         # Provide incorrect model name to generate function
         with self.assertRaises(InternalServerError):
-            await async_unify.generate("hello", "llama-chat")
+            await async_unify.generate(messages="hello", model="llama-chat")
 
     async def test_generate_returns_string_when_stream_false(self) -> None:
         # Instantiate AsyncUnify with a valid API key