Skip to content

Commit 0584724

Browse files
Add HTTP Referer request header to Vercel AI Gateway provider (#2369)
1 parent 3e1f634 commit 0584724

File tree

1 file changed

+8
-2
lines changed
  • pydantic_ai_slim/pydantic_ai/providers

1 file changed

+8
-2
lines changed

pydantic_ai_slim/pydantic_ai/providers/vercel.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,16 @@ def __init__(
9898
'or pass the API key via `VercelProvider(api_key=...)` to use the Vercel provider.'
9999
)
100100

101+
default_headers = {'http-referer': 'https://ai.pydantic.dev/', 'x-title': 'pydantic-ai'}
102+
101103
if openai_client is not None:
102104
self._client = openai_client
103105
elif http_client is not None:
104-
self._client = AsyncOpenAI(base_url=self.base_url, api_key=api_key, http_client=http_client)
106+
self._client = AsyncOpenAI(
107+
base_url=self.base_url, api_key=api_key, http_client=http_client, default_headers=default_headers
108+
)
105109
else:
106110
http_client = cached_async_http_client(provider='vercel')
107-
self._client = AsyncOpenAI(base_url=self.base_url, api_key=api_key, http_client=http_client)
111+
self._client = AsyncOpenAI(
112+
base_url=self.base_url, api_key=api_key, http_client=http_client, default_headers=default_headers
113+
)

0 commit comments

Comments
 (0)