Skip to content

Commit e1778ea

Browse files
committed
fix: Pass fake api_key to the internal ai model
1 parent e8e3379 commit e1778ea

1 file changed

Lines changed: 2 additions & 21 deletions

File tree

tests/ai_test_model.py

Lines changed: 2 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,4 @@
1-
import collections.abc
2-
from typing import override
3-
41
import httpx
5-
from httpx import Auth, Request, Response
62
from pydantic import BaseModel
73

84
from splunklib.ai import OpenAIModel
@@ -37,20 +33,6 @@ async def create_model(s: TestLLMSettings) -> PredefinedModel:
3733
raise Exception("unreachable")
3834

3935

40-
class _InternalAIAuth(Auth):
41-
token: str
42-
43-
def __init__(self, token: str) -> None:
44-
self.token = token
45-
46-
@override
47-
def auth_flow(
48-
self, request: Request
49-
) -> collections.abc.Generator[Request, Response, None]:
50-
request.headers["api-key"] = self.token
51-
yield request
52-
53-
5436
class _TokenResponse(BaseModel):
5537
access_token: str
5638

@@ -79,14 +61,13 @@ async def _buildInternalAIModel(
7961

8062
token = _TokenResponse.model_validate_json(response.text).access_token
8163

82-
auth_handler = _InternalAIAuth(token)
8364
model = "gpt-5-nano"
8465

8566
return OpenAIModel(
8667
model=model,
8768
base_url=f"{base_url}/{model}",
88-
api_key="", # unused
69+
api_key="test-api-key", # unused
8970
extra_body={"user": f'{{"appkey":"{app_key}"}}'},
90-
httpx_client=httpx.AsyncClient(auth=auth_handler),
71+
httpx_client=httpx.AsyncClient(headers={"api-key": token}),
9172
temperature=0.0,
9273
)

0 commit comments

Comments
 (0)