mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
fix bad test
This commit is contained in:
parent
929b6959a0
commit
8e6d344db2
2 changed files with 2 additions and 3 deletions
|
@ -10,7 +10,6 @@ parent_dir = os.path.join(current_dir, '..')
|
||||||
# Add the parent directory to the system path
|
# Add the parent directory to the system path
|
||||||
sys.path.append(parent_dir)
|
sys.path.append(parent_dir)
|
||||||
|
|
||||||
|
|
||||||
import main
|
import main
|
||||||
from main import embedding, completion
|
from main import embedding, completion
|
||||||
main.success_callback = ["posthog"]
|
main.success_callback = ["posthog"]
|
||||||
|
@ -20,7 +19,7 @@ main.set_verbose = True
|
||||||
|
|
||||||
user_message = "Hello, how are you?"
|
user_message = "Hello, how are you?"
|
||||||
messages = [{ "content": user_message,"role": "user"}]
|
messages = [{ "content": user_message,"role": "user"}]
|
||||||
model_val = None
|
model_val = "krrish is a model"
|
||||||
# test on empty
|
# test on empty
|
||||||
try:
|
try:
|
||||||
response = completion(model=model_val, messages=messages)
|
response = completion(model=model_val, messages=messages)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue