llmcall

type Agent struct {
	Name string
	Mode string
	// Tier is the model pool this agent routes on. Empty means high.
	Tier adaptive.ModelTier
}

type ClientFactory interface {
	Client(ctx context.Context, model Model, choice *adaptive.RouteChoice, router *adaptive.AdaptiveModelRouter) (StreamClient, error)
}

type ClientFactoryFunc func(context.Context, Model, *adaptive.RouteChoice, *adaptive.AdaptiveModelRouter) (StreamClient, error)

type Model struct {
	ProviderID string
	ID         string
	APIID      string
	Params     orclient.RequestParams
}

type ModelResolver interface {
	GetModel(ctx context.Context, providerID, modelID string) (Model, error)
}

type ModelResolverFunc func(context.Context, string, string) (Model, error)

type OpenRouterClientFactory func(ctx context.Context, model Model) (*orclient.Client, error)

type ResolvedCall struct {
	Model      Model
	Choice     *adaptive.RouteChoice
	Candidates []adaptive.ModelCandidate
	Params     orclient.RequestParams
}

type Service struct {
	Models         ModelResolver
	Clients        ClientFactory
	Router         *adaptive.AdaptiveModelRouter
	DisableRouting bool
}

type Stream interface {
	Next() (orclient.StreamPart, error)
	Close() error
}

type StreamClient interface {
	DoStream(ctx context.Context, params orclient.RequestParams) (Stream, error)
}

type StreamInput struct {
	UserID          string
	SessionID       string
	ParentSessionID string
	Model           Model
	Agent           Agent
	System          []string
	Messages        []msgmodel.ModelMessage
	Tools           []orclient.Tool
	DisabledTools   map[string]bool
	UserTools       map[string]bool
	ToolChoice      *orclient.ToolChoice
}