Skip to content

Commit 6d9e2a5

Browse files
committed
fix typing
1 parent a133796 commit 6d9e2a5

File tree

1 file changed

+4
-2
lines changed
  • pydantic_ai_slim/pydantic_ai/embeddings

1 file changed

+4
-2
lines changed

pydantic_ai_slim/pydantic_ai/embeddings/openai.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,10 @@
99
from . import OpenAIEmbeddingsCompatibleProvider
1010

1111
try:
12-
from openai import NOT_GIVEN, AsyncOpenAI
12+
from openai import AsyncOpenAI
1313
from openai.types import EmbeddingModel as LatestOpenAIEmbeddingModelNames
14+
15+
from pydantic_ai.models.openai import OMIT
1416
except ImportError as _import_error:
1517
raise ImportError(
1618
'Please install `openai` to use the OpenAI embeddings model, '
@@ -93,7 +95,7 @@ async def _embed(self, documents: Sequence[str], settings: OpenAIEmbeddingSettin
9395
response = await self._client.embeddings.create(
9496
input=documents, # pyright: ignore[reportArgumentType] # Sequence[str] not compatible with SequenceNotStr[str] :/
9597
model=self.model_name,
96-
dimensions=settings.get('dimensions') or NOT_GIVEN,
98+
dimensions=settings.get('dimensions') or OMIT,
9799
extra_headers=settings.get('extra_headers'),
98100
extra_body=settings.get('extra_body'),
99101
)

0 commit comments

Comments
 (0)