@@ -14,8 +14,8 @@ def test_openai_class_detection():
1414 "ft:gpt-4" : guidance .models .OpenAIChat ,
1515 "ft:gpt-4-vision-preview" : guidance .models .OpenAIChat ,
1616 "ft:gpt-3.5-turbo:my-org:custom_suffix:id" : guidance .models .OpenAIChat ,
17- "gpt-3.5-turbo-instruct" : guidance .models .OpenAIInstruct ,
18- "ft:gpt-3.5-turbo-instruct" : guidance .models .OpenAIInstruct ,
17+ "gpt-3.5-turbo-instruct" : guidance .models .OpenAICompletion ,
18+ "ft:gpt-3.5-turbo-instruct" : guidance .models .OpenAICompletion ,
1919 "text-curie-001" : guidance .models .OpenAICompletion ,
2020 "ft:text-curie-001" : guidance .models .OpenAICompletion ,
2121 "text-davinci-003" : guidance .models .OpenAICompletion ,
@@ -40,19 +40,6 @@ def test_openai_basic():
4040 lm += f"""{ gen (max_tokens = 1 , suffix = nl )} aaaaaa"""
4141 assert str (lm )[- 5 :] == "aaaaa"
4242
43- def test_openai_gpt35_instruct ():
44- from guidance import gen , instruction
45-
46- # this relies on the environment variable OPENAI_API_KEY being set
47- try :
48- lm = guidance .models .OpenAI ('gpt-3.5-turbo-instruct' )
49- except :
50- pytest .skip ("Skipping OpenAI test because we can't load the model!" )
51-
52- with instruction ():
53- lm += "What is a popular flavor?"
54- lm += gen ('flavor' , max_tokens = 2 , stop = "." )
55- assert len (lm ['flavor' ]) > 0
5643
5744def test_openai_select ():
5845 try :
0 commit comments