Fixing sending version

jamesrochabrun · Jan 23, 2025 · c4cf758 · c4cf758
1 parent e44cbdb
commit c4cf758
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
    }