remove trust remote code option

litellm/main.py

@@ -274,7 +274,7 @@ def completion(
             top_k=top_k,
             task=task,
             remove_input=remove_input,
-            return_full_text=return_full_text
+            return_full_text=return_full_text,
         )
         # For logging - save the values of the litellm-specific params passed in
         litellm_params = get_litellm_params(

litellm/tests/test_completion.py

@@ -200,6 +200,22 @@ def hf_test_completion():
 
 # hf_test_completion()
 
+def hf_test_completion_task_none():
+    try:
+        litellm.set_verbose=True
+        user_message = "My name is Merve and my favorite"
+        messages = [{ "content": user_message,"role": "user"}]
+        model = "huggingface/cerebras/btlm-3b-8k-base"
+        response = completion(
+            model=model, 
+            messages=messages,
+            task=None,
+        )
+        # Add any assertions here to check the response
+        print(response)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+hf_test_completion_task_none()
 
 # this should throw an exception, to trigger https://logs.litellm.ai/
 # def hf_test_error_logs():

litellm/utils.py

@@ -907,7 +907,7 @@ def get_optional_params(  # use the openai defaults
     custom_llm_provider="",
     top_k=40,
     return_full_text=False,
-    task=None
+    task=None,
 ):
     optional_params = {}
     if model in litellm.anthropic_models:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.780"
+version = "0.1.782"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"