references/kit
1
0
Fork 0
mirror of https://github.com/mark3labs/kit.git synced 2026-06-14 03:30:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

skip model name validation with custom provider URL

Browse Source
This commit is contained in:
Ed Zynda
2025-06-19 15:14:21 +03:00
parent d7f4c7a61c
commit 33df79adb6
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/models/providers.go
+7 -5
View File
@@ -48,8 +48,8 @@ func CreateProvider(ctx context.Context, config *ProviderConfig) (model.ToolCall
// Get the global registry for validation
registry := GetGlobalRegistry()
// Validate the model exists (skip for ollama as it's not in models.dev)
if provider != "ollama" {
// Validate the model exists (skip for ollama as it's not in models.dev, and skip when using custom provider URL)
if provider != "ollama" && config.ProviderURL == "" {
modelInfo, err := registry.ValidateModel(provider, modelName)
if err != nil {
// Provide helpful suggestions
@@ -162,11 +162,13 @@ func createOpenAIProvider(ctx context.Context, config *ProviderConfig, modelName
openaiConfig.BaseURL = config.ProviderURL
}
// Check if this is a reasoning model to handle beta limitations
// Check if this is a reasoning model to handle beta limitations (skip validation if using custom URL)
registry := GetGlobalRegistry()
isReasoningModel := false
if modelInfo, err := registry.ValidateModel("openai", modelName); err == nil && modelInfo.Reasoning {
isReasoningModel = true
if config.ProviderURL == "" {
if modelInfo, err := registry.ValidateModel("openai", modelName); err == nil && modelInfo.Reasoning {
isReasoningModel = true
}
}
if config.MaxTokens > 0 {