feat(fine tuned models) enable access to fine tuned OpenAI models whi…

…ch all start with ft:gpt...
kharvd · Jan 31, 2024 · 082f5ae · 082f5ae
1 parent 9ce23b0
commit 082f5ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()