tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-19 08:18:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

(fix) testing text_completion

Browse Source
This commit is contained in:
ishaan-jaff
2023-11-03 16:39:06 -07:00
parent 382e31d7b7
commit e29b2e8ce4
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_text_completion.py
+4 -2
View File
@@ -24,9 +24,11 @@ def test_completion_openai_prompt():
response = text_completion(
model="gpt-3.5-turbo", prompt="What's the weather in SF?"
)
print(response)
response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_prompt()
def test_completion_openai_prompt_array():
@@ -61,10 +63,10 @@ def test_completion_hf_prompt_array():
model="huggingface/mistralai/Mistral-7B-v0.1",
prompt=token_prompt, # token prompt is a 2d list
)
print("\n\n response\n\n")
print("\n\n response")
print(response)
# response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_hf_prompt_array()
# test_completion_hf_prompt_array()