bug fixes

BerriAI · Sep 16, 2023 · d2f8a7b · d2f8a7b
1 parent 122c993
commit d2f8a7b
Show file tree

Hide file tree

Showing 6 changed files with 3 additions and 3 deletions.
diff --git a/litellm/__pycache__/__init__.cpython-311.pyc b/litellm/__pycache__/__init__.cpython-311.pyc
diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc
diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py
@@ -349,7 +349,6 @@ def test_completion_nlp_cloud_streaming():
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
 
-
 #### Test Function calling + streaming ####
 
 def test_completion_openai_with_functions():

diff --git a/litellm/utils.py b/litellm/utils.py
@@ -1086,7 +1086,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None):
         elif model in litellm.ai21_models:
             custom_llm_provider = "ai21"
         ## together_ai 
-        elif model in litellm.together_ai_models or "togethercomputer":
+        elif model in litellm.together_ai_models:
             custom_llm_provider = "together_ai"
         ## aleph_alpha 
         elif model in litellm.aleph_alpha_models:
@@ -2435,6 +2435,7 @@ def handle_nlp_cloud_chunk(self, chunk):
         chunk = chunk.decode("utf-8")
         data_json = json.loads(chunk)
         try:
+            print(f"data json: {data_json}")
             return data_json["generated_text"]
         except:
             raise ValueError(f"Unable to parse response. Original response: {chunk}")

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.682"
+version = "0.1.683"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"