diff --git a/test_openai_api.py b/test_openai_api.py new file mode 100644 index 0000000..be09f7e --- /dev/null +++ b/test_openai_api.py @@ -0,0 +1,17 @@ +from openai import OpenAI + +client = OpenAI( + base_url = 'https://ollama.molodetz.nl/v1', + api_key='ollama', # required, but unused +) + +response = client.chat.completions.create( + model="llama2", + messages=[ + {"role": "system", "content": "You are a helpful assistant."}, + {"role": "user", "content": "Who won the world series in 2020?"}, + {"role": "assistant", "content": "The LA Dodgers won in 2020."}, + {"role": "user", "content": "Where was it played?"} + ] +) +print(response.choices[0].message.content)