(test) local ollama_chat

This commit is contained in:
ishaan-jaff 2023-12-26 20:11:14 +05:30
parent 3f6e6e7f55
commit db90ccd19f

View file

@ -85,7 +85,7 @@
# print(e) # print(e)
# asyncio.run(test_async_ollama()) # # asyncio.run(test_async_ollama())
# def test_completion_ollama(): # def test_completion_ollama():
@ -130,7 +130,7 @@
# } # }
# ] # ]
# response = completion( # response = completion(
# model="ollama/mistral", # model="ollama_chat/mistral",
# messages=messages, # messages=messages,
# functions=functions, # functions=functions,
# max_tokens=200, # max_tokens=200,
@ -143,7 +143,7 @@
# pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# # test_completion_ollama_function_calling() # test_completion_ollama_function_calling()
# async def async_test_completion_ollama_function_calling(): # async def async_test_completion_ollama_function_calling():