tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-22 22:24:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) add support for returning logprobs

Browse Source
This commit is contained in:
ishaan-jaff
2023-10-31 15:33:45 -07:00
parent 1ada3a13d4
commit 85692398c1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_completion.py
+3 -3
View File
@@ -56,7 +56,7 @@ def test_completion_claude():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_claude()
test_completion_claude()
# def test_completion_oobabooga():
# try:
@@ -400,9 +400,9 @@ def test_completion_openai_prompt():
def test_completion_openai_prompt_array():
try:
litellm.set_verbose=True
litellm.set_verbose=False
response = text_completion(
model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10
model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10, logprobs=10
)
print(response)
response_str = response["choices"][0]["text"]