From 13ff65a8fe66fa4bed9058dce3691f866f656e7c Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 28 Sep 2023 17:57:48 -0700
Subject: [PATCH] remove junk from completion

---
 litellm/main.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/litellm/main.py b/litellm/main.py
index 7cc9f26e1..08c4758de 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -213,8 +213,6 @@ def completion(
     litellm_logging_obj = kwargs.get('litellm_logging_obj', None)
     id = kwargs.get('id', None)
     metadata = kwargs.get('metadata', None)
-    task = kwargs.get('task', "text-generation-inference")
-    return_full_text = kwargs.get('return_full_text', False)
     request_timeout = kwargs.get('request_timeout', 0)
     fallbacks = kwargs.get('fallbacks', [])
     ######## end of unpacking kwargs ###########
@@ -267,9 +265,9 @@ def completion(
             model=model,
             custom_llm_provider=custom_llm_provider,
             top_k=kwargs.get('top_k', 40),
-            task=task,
+            task=kwargs.get('task', "text-generation-inference"),
             remove_input=kwargs.get('remove_input', True),
-            return_full_text=return_full_text,
+            return_full_text=kwargs.get('return_full_text', False),
         )
         # For logging - save the values of the litellm-specific params passed in
         litellm_params = get_litellm_params(