GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-02 14:29:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: set LLMService/CVService for case runner

Browse Source
This commit is contained in:
lilong.129
2025-04-28 19:45:07 +08:00
parent 7fa4155390
commit 4d7dc466f3
5 changed files with 31 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
runner.go
View File

@@ -418,6 +418,17 @@ func (r *CaseRunner) parseConfig() (parsedConfig *TConfig, err error) {
}
r.parametersIterator = parametersIterator
// ai options
aiOpts := []ai.AIServiceOption{}
if parsedConfig.LLMService != "" {
aiOpts = append(aiOpts, ai.WithLLMService(parsedConfig.LLMService))
}
if parsedConfig.CVService == "" {
// default to vedem
parsedConfig.CVService = ai.CVServiceTypeVEDEM
}
aiOpts = append(aiOpts, ai.WithCVService(parsedConfig.CVService))
// parse android devices config
for _, androidDeviceOptions := range parsedConfig.Android {
err := r.parseDeviceConfig(androidDeviceOptions, parsedConfig.Variables)
@@ -435,7 +446,7 @@ func (r *CaseRunner) parseConfig() (parsedConfig *TConfig, err error) {
return nil, errors.Wrap(err, "init android driver failed")
}
driverExt := uixt.NewXTDriver(driver, ai.WithCVService(ai.CVServiceTypeVEDEM))
driverExt := uixt.NewXTDriver(driver, aiOpts...)
r.uixtDrivers[androidDeviceOptions.SerialNumber] = driverExt
}
// parse iOS devices config
@@ -455,7 +466,7 @@ func (r *CaseRunner) parseConfig() (parsedConfig *TConfig, err error) {
return nil, errors.Wrap(err, "init ios driver failed")
}
driverExt := uixt.NewXTDriver(driver, ai.WithCVService(ai.CVServiceTypeVEDEM))
driverExt := uixt.NewXTDriver(driver, aiOpts...)
r.uixtDrivers[iosDeviceOptions.UDID] = driverExt
}
// parse harmony devices config
@@ -475,7 +486,7 @@ func (r *CaseRunner) parseConfig() (parsedConfig *TConfig, err error) {
return nil, errors.Wrap(err, "init harmony driver failed")
}
driverExt := uixt.NewXTDriver(driver, ai.WithCVService(ai.CVServiceTypeVEDEM))
driverExt := uixt.NewXTDriver(driver, aiOpts...)
r.uixtDrivers[harmonyDeviceOptions.ConnectKey] = driverExt
}