File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1- import collections .abc
2- from typing import override
3-
41import httpx
5- from httpx import Auth , Request , Response
62from pydantic import BaseModel
73
84from 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-
5436class _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 )
You can’t perform that action at this time.
0 commit comments