26 lines
719 B
Python
26 lines
719 B
Python
|
import instructor
|
||
|
|
import openai
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_has_embedding():
|
||
|
|
oai = openai.OpenAI()
|
||
|
|
client = instructor.from_openai(oai)
|
||
|
|
|
||
|
|
embedding = client.embeddings.create(
|
||
|
|
input="Hello world", model="text-embedding-3-small"
|
||
|
|
)
|
||
|
|
assert embedding is not None, "The 'embeddings' attribute is None."
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_has_embedding_async():
|
||
|
|
oai = openai.AsyncOpenAI()
|
||
|
|
client = instructor.from_openai(oai)
|
||
|
|
|
||
|
|
# Check if the 'embeddings' attribute can be accessed through the client
|
||
|
|
embedding = await client.embeddings.create(
|
||
|
|
input="Hello world", model="text-embedding-3-small"
|
||
|
|
)
|
||
|
|
assert embedding is not None, "The 'embeddings' attribute is None."
|