33
44
55class 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 ):
79 self .APIKey = apiKey
810 self .BaseURL = baseURL
911 self .DefaultModel = defaultModel
12+ self .DefaultModelProvider = defaultModelProvider
1013 self .Env = env
1114
1215 def toEnv (self ):
@@ -19,6 +22,8 @@ def toEnv(self):
1922 self .Env ["OPENAI_BASE_URL" ] = self .BaseURL
2023 if self .DefaultModel != "" :
2124 self .Env ["GPTSCRIPT_SDKSERVER_DEFAULT_MODEL" ] = self .DefaultModel
25+ if self .DefaultModelProvider != "" :
26+ self .Env ["GPTSCRIPT_SDKSERVER_DEFAULT_MODEL_PROVIDER" ] = self .DefaultModelProvider
2227
2328
2429class Options (GlobalOptions ):
@@ -36,9 +41,10 @@ def __init__(self,
3641 forceSequential : bool = False ,
3742 apiKey : str = "" ,
3843 baseURL : str = "" ,
44+ defaultModelProvider : str = "" ,
3945 defaultModel : str = ""
4046 ):
41- super ().__init__ (apiKey , baseURL , defaultModel )
47+ super ().__init__ (apiKey , baseURL , defaultModelProvider , defaultModel )
4248 self .input = input
4349 self .disableCache = disableCache
4450 self .subTool = subTool
0 commit comments