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