refactor: uixt MCP Server

This commit is contained in:
lilong.129
2025-05-24 23:06:08 +08:00
parent 0a68701faa
commit 02611d3d5a
7 changed files with 612 additions and 447 deletions

View File

@@ -5,10 +5,8 @@ import (
_ "image/gif"
_ "image/png"
"github.com/httprunner/httprunner/v5/uixt/ai"
"github.com/httprunner/httprunner/v5/uixt/option"
"github.com/httprunner/httprunner/v5/uixt/types"
"github.com/rs/zerolog/log"
)
var (
@@ -89,36 +87,3 @@ type IDriver interface {
StartCaptureLog(identifier ...string) error
StopCaptureLog() (result interface{}, err error)
}
func NewXTDriver(driver IDriver, opts ...option.AIServiceOption) (*XTDriver, error) {
driverExt := &XTDriver{
IDriver: driver,
}
services := option.NewAIServiceOptions(opts...)
var err error
if services.CVService != "" {
driverExt.CVService, err = ai.NewCVService(services.CVService)
if err != nil {
log.Error().Err(err).Msg("init vedem image service failed")
return nil, err
}
}
if services.LLMService != "" {
driverExt.LLMService, err = ai.NewLLMService(services.LLMService)
if err != nil {
log.Error().Err(err).Msg("init llm service failed")
return nil, err
}
}
return driverExt, nil
}
// XTDriver = IDriver + AI
type XTDriver struct {
IDriver
CVService ai.ICVService // OCR/CV
LLMService ai.ILLMService // LLM
}