참고소스 수정본
This commit is contained in:
25
참고/instructor-main/tests/llm/test_openai/test_attr.py
Normal file
25
참고/instructor-main/tests/llm/test_openai/test_attr.py
Normal file
@@ -0,0 +1,25 @@
|
||||
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."
|
||||
Reference in New Issue
Block a user