Skip to content

Commit 42aee95

Browse files
committed
Apply code review suggestions
1 parent 44dec59 commit 42aee95

File tree

5 files changed

+19
-12
lines changed

5 files changed

+19
-12
lines changed

Core/Sources/CodeCompletionService/API/AnthropicService.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,11 @@ public actor AnthropicService {
3535
case claude35Sonnet = "claude-3-5-sonnet-latest"
3636
case claude35Haiku = "claude-3-5-haiku-latest"
3737

38-
var maxToken: Int {
38+
public var maxToken: Int {
3939
switch self {
40-
case .claude3Opus: return 4096
41-
case .claude35Sonnet: return 4096
42-
case .claude35Haiku: return 4096
40+
case .claude3Opus: return 200_000
41+
case .claude35Sonnet: return 200_000
42+
case .claude35Haiku: return 200_000
4343
}
4444
}
4545
}

Core/Sources/CodeCompletionService/CodeCompletionService.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ public struct CodeCompletionService {
159159
)
160160
try Task.checkCancellation()
161161
return result
162-
case .anthropic:
162+
case .claude:
163163
let service = AnthropicService(
164164
url: model.endpoint,
165165
modelName: model.info.modelName,

Core/Sources/Fundamental/Models/ChatModel.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ public struct ChatModel: Codable, Equatable, Identifiable {
2323
case openAICompatible
2424
case googleAI
2525
case ollama
26-
case anthropic
26+
case claude
2727

2828
case unknown
2929
}
@@ -111,7 +111,7 @@ public struct ChatModel: Codable, Equatable, Identifiable {
111111
let baseURL = info.baseURL
112112
if baseURL.isEmpty { return "http://localhost:11434/api/chat" }
113113
return "\(baseURL)/api/chat"
114-
case .anthropic:
114+
case .claude:
115115
let baseURL = info.baseURL
116116
if baseURL.isEmpty { return "https://api.anthropic.com/v1/messages" }
117117
return "\(baseURL)/v1/messages"

CustomSuggestionService/ChatModelManagement/ChatModelEdit.swift

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,13 @@ struct ChatModelEdit {
100100
state.suggestedMaxTokens = nil
101101
}
102102
return .none
103+
case .claude:
104+
if let knownModel = AnthropicService.Models(rawValue: state.modelName) {
105+
state.suggestedMaxTokens = knownModel.maxToken
106+
} else {
107+
state.suggestedMaxTokens = nil
108+
}
109+
return .none
103110
default:
104111
state.suggestedMaxTokens = nil
105112
return .none

CustomSuggestionService/ChatModelManagement/ChatModelEditView.swift

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@ struct ChatModelEditView: View {
2727
googleAI
2828
case .ollama:
2929
ollama
30-
case .anthropic:
31-
anthropic
30+
case .claude:
31+
claude
3232
case .unknown:
3333
EmptyView()
3434
}
@@ -83,8 +83,8 @@ struct ChatModelEditView: View {
8383
Text("Google Generative AI").tag(format)
8484
case .ollama:
8585
Text("Ollama").tag(format)
86-
case .anthropic:
87-
Text("Anthropic").tag(format)
86+
case .claude:
87+
Text("Claude").tag(format)
8888
case .unknown:
8989
EmptyView()
9090
}
@@ -295,7 +295,7 @@ struct ChatModelEditView: View {
295295
}
296296

297297
@ViewBuilder
298-
var anthropic: some View {
298+
var claude: some View {
299299
baseURLTextField(prompt: Text("https://api.anthropic.com")) {
300300
Text("/v1/messages")
301301
}

0 commit comments

Comments
 (0)