From 0e805ea83bc80d9444307fa1f50e1ae22ee07699 Mon Sep 17 00:00:00 2001
From: ykim-isabel <77634465+ykim-isabel@users.noreply.github.com>
Date: Sun, 4 Feb 2024 18:55:44 +0100
Subject: [PATCH] =?UTF-8?q?feat(fine=20tuned=20models)=20enable=20access?=
 =?UTF-8?q?=20to=20fine=20tuned=20OpenAI=20models=20whi=E2=80=A6=20(#66)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Allow the use of fine-tuned OpenAI gpt 3.5 Turbo models which start with
ft:gpt, this is especially useful when trying to compare the stock
models against fine-tuned models quickly.
---
 gptcli/assistant.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()