mirror of
https://github.com/httprunner/httprunner.git
synced 2026-06-04 23:39:33 +08:00
81a92ae155
- Add comprehensive documentation for the new Query functionality - Update interface method names from Call to Plan for consistency - Add OpenAI GPT-4O model support documentation - Include detailed usage examples for basic and custom schema queries - Add configuration examples for multiple model services - Document new features like ResetHistory, Usage statistics, and automatic type conversion - Expand advanced features section with custom output format examples - Update all code examples to reflect the latest API changes The documentation now reflects the current state of the AI module with all three core capabilities: - Planning (renamed from Call) - Assertion - Query (new feature) All examples and configurations are updated to match the latest implementation.
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package ai
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/cloudwego/eino/schema"
|
|
"github.com/httprunner/httprunner/v5/uixt/option"
|
|
)
|
|
|
|
// ILLMService 定义了 LLM 服务接口，包括规划、断言和查询功能
|
|
type ILLMService interface {
|
|
Plan(ctx context.Context, opts *PlanningOptions) (*PlanningResult, error)
|
|
Assert(ctx context.Context, opts *AssertOptions) (*AssertionResult, error)
|
|
Query(ctx context.Context, opts *QueryOptions) (*QueryResult, error)
|
|
// RegisterTools registers tools for function calling
|
|
RegisterTools(tools []*schema.ToolInfo) error
|
|
}
|
|
|
|
func NewLLMService(modelType option.LLMServiceType) (ILLMService, error) {
|
|
modelConfig, err := GetModelConfig(modelType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
planner, err := NewPlanner(context.Background(), modelConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
asserter, err := NewAsserter(context.Background(), modelConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
querier, err := NewQuerier(context.Background(), modelConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &combinedLLMService{
|
|
planner: planner,
|
|
asserter: asserter,
|
|
querier: querier,
|
|
}, nil
|
|
}
|
|
|
|
// combinedLLMService 实现了 ILLMService 接口，组合了规划、断言和查询功能
|
|
// ⭐️支持采用不同的模型服务进行规划、断言和查询
|
|
type combinedLLMService struct {
|
|
planner IPlanner // 提供规划功能
|
|
asserter IAsserter // 提供断言功能
|
|
querier IQuerier // 提供查询功能
|
|
}
|
|
|
|
// Plan 执行规划功能
|
|
func (c *combinedLLMService) Plan(ctx context.Context, opts *PlanningOptions) (*PlanningResult, error) {
|
|
return c.planner.Plan(ctx, opts)
|
|
}
|
|
|
|
// Assert 执行断言功能
|
|
func (c *combinedLLMService) Assert(ctx context.Context, opts *AssertOptions) (*AssertionResult, error) {
|
|
return c.asserter.Assert(ctx, opts)
|
|
}
|
|
|
|
// Query 执行查询功能
|
|
func (c *combinedLLMService) Query(ctx context.Context, opts *QueryOptions) (*QueryResult, error) {
|
|
return c.querier.Query(ctx, opts)
|
|
}
|
|
|
|
// RegisterTools registers tools for function calling
|
|
func (c *combinedLLMService) RegisterTools(tools []*schema.ToolInfo) error {
|
|
// Only register tools to planner since asserter and querier don't need tools
|
|
if planner, ok := c.planner.(*Planner); ok {
|
|
return planner.RegisterTools(tools)
|
|
}
|
|
return nil
|
|
}
|