From c4cf75888b704c485a01d036dd6741c8b1ae8d2e Mon Sep 17 00:00:00 2001 From: jamesrochabrun Date: Wed, 22 Jan 2025 23:38:44 -0800 Subject: [PATCH] Fixing sending version --- .../SwiftOpenAIExample/LocalChatDemo/LocalChatDemoView.swift | 4 ++-- Sources/OpenAI/LocalModelService/LocalModelService.swift | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Examples/SwiftOpenAIExample/SwiftOpenAIExample/LocalChatDemo/LocalChatDemoView.swift b/Examples/SwiftOpenAIExample/SwiftOpenAIExample/LocalChatDemo/LocalChatDemoView.swift index c207629..bae7a95 100644 --- a/Examples/SwiftOpenAIExample/SwiftOpenAIExample/LocalChatDemo/LocalChatDemoView.swift +++ b/Examples/SwiftOpenAIExample/SwiftOpenAIExample/LocalChatDemo/LocalChatDemoView.swift @@ -99,8 +99,8 @@ struct LocalChatDemoView: View { messages: [.init( role: .user, content: content)], - // Make sure you run `ollama pull llama3` in your terminal to download this model. - model: .custom("llama3")) + // Make sure you run `ollama pull llama3.1` in your terminal to download this model. + model: .custom("llama3.1")) switch selectedSegment { case .chatCompletion: try await chatProvider.startChat(parameters: parameters) diff --git a/Sources/OpenAI/LocalModelService/LocalModelService.swift b/Sources/OpenAI/LocalModelService/LocalModelService.swift index 3eced27..6df11da 100644 --- a/Sources/OpenAI/LocalModelService/LocalModelService.swift +++ b/Sources/OpenAI/LocalModelService/LocalModelService.swift @@ -30,7 +30,7 @@ struct LocalModelService: OpenAIService { self.session = URLSession(configuration: configuration) self.decoder = decoder self.apiKey = apiKey - self.openAIEnvironment = OpenAIEnvironment(baseURL: baseURL, proxyPath: proxyPath, version: overrideVersion) + self.openAIEnvironment = OpenAIEnvironment(baseURL: baseURL, proxyPath: proxyPath, version: overrideVersion ?? "v1") self.debugEnabled = debugEnabled }