fix bad test

This commit is contained in:
ishaan-jaff 2023-07-31 17:02:35 -07:00
parent 929b6959a0
commit 8e6d344db2
2 changed files with 2 additions and 3 deletions

View file

@ -10,7 +10,6 @@ parent_dir = os.path.join(current_dir, '..')
# Add the parent directory to the system path
sys.path.append(parent_dir)
import main
from main import embedding, completion
main.success_callback = ["posthog"]
@ -20,7 +19,7 @@ main.set_verbose = True
user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}]
model_val = None
model_val = "krrish is a model"
# test on empty
try:
response = completion(model=model_val, messages=messages)