3
3
4
4
5
5
class GlobalOptions :
6
- def __init__ (self , apiKey : str = "" , baseURL : str = "" , defaultModel : str = "" , env : Mapping [str , str ] = None ):
6
+ def __init__ (self ,
7
+ apiKey : str = "" , baseURL : str = "" , defaultModelProvider : str = "" , defaultModel : str = "" ,
8
+ env : Mapping [str , str ] = None ):
7
9
self .APIKey = apiKey
8
10
self .BaseURL = baseURL
9
11
self .DefaultModel = defaultModel
12
+ self .DefaultModelProvider = defaultModelProvider
10
13
self .Env = env
11
14
12
15
def toEnv (self ):
@@ -19,6 +22,8 @@ def toEnv(self):
19
22
self .Env ["OPENAI_BASE_URL" ] = self .BaseURL
20
23
if self .DefaultModel != "" :
21
24
self .Env ["GPTSCRIPT_SDKSERVER_DEFAULT_MODEL" ] = self .DefaultModel
25
+ if self .DefaultModelProvider != "" :
26
+ self .Env ["GPTSCRIPT_SDKSERVER_DEFAULT_MODEL_PROVIDER" ] = self .DefaultModelProvider
22
27
23
28
24
29
class Options (GlobalOptions ):
@@ -36,9 +41,10 @@ def __init__(self,
36
41
forceSequential : bool = False ,
37
42
apiKey : str = "" ,
38
43
baseURL : str = "" ,
44
+ defaultModelProvider : str = "" ,
39
45
defaultModel : str = ""
40
46
):
41
- super ().__init__ (apiKey , baseURL , defaultModel )
47
+ super ().__init__ (apiKey , baseURL , defaultModelProvider , defaultModel )
42
48
self .input = input
43
49
self .disableCache = disableCache
44
50
self .subTool = subTool
0 commit comments