From 71a68dd56d75ebb90693593f3c18e02caaebc411 Mon Sep 17 00:00:00 2001 From: adamdottv <2363879+adamdottv@users.noreply.github.com> Date: Tue, 6 May 2025 10:19:07 -0500 Subject: feat: add qwen3 models --- internal/llm/provider/provider.go | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'internal/llm/provider') diff --git a/internal/llm/provider/provider.go b/internal/llm/provider/provider.go index cad11eeb3..cb1a0fd59 100644 --- a/internal/llm/provider/provider.go +++ b/internal/llm/provider/provider.go @@ -55,6 +55,8 @@ type Provider interface { StreamResponse(ctx context.Context, messages []message.Message, tools []tools.BaseTool) <-chan ProviderEvent Model() models.Model + + MaxTokens() int64 } type providerClientOptions struct { @@ -168,6 +170,10 @@ func (p *baseProvider[C]) Model() models.Model { return p.options.model } +func (p *baseProvider[C]) MaxTokens() int64 { + return p.options.maxTokens +} + func (p *baseProvider[C]) StreamResponse(ctx context.Context, messages []message.Message, tools []tools.BaseTool) <-chan ProviderEvent { messages = p.cleanMessages(messages) return p.client.stream(ctx, messages, tools) -- cgit v1.2.3