Update.
This commit is contained in:
parent
5788e02128
commit
22d646e8ae
17
test_openai_api.py
Normal file
17
test_openai_api.py
Normal file
@ -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)
|
Loading…
Reference in New Issue
Block a user