diff --git a/gptcli/assistant.py b/gptcli/assistant.py index 6adac67..07de4b1 100644 --- a/gptcli/assistant.py +++ b/gptcli/assistant.py @@ -56,7 +56,7 @@ class AssistantConfig(TypedDict, total=False): def get_completion_provider(model: str) -> CompletionProvider: - if model.startswith("gpt"): + if model.startswith("gpt") or model.startswith("ft:gpt"): return OpenAICompletionProvider() elif model.startswith("claude"): return AnthropicCompletionProvider()