From 61fa9a98f617debb927ff1ff8001fdd5d26fbff6 Mon Sep 17 00:00:00 2001
From: Sylvain <1552102+sgaunet@users.noreply.github.com>
Date: Tue, 7 Jan 2025 21:30:00 +0100
Subject: [PATCH] fix: remove deprecated model tests (#7)

---
 perplexity_test.go | 16 ----------------
 1 file changed, 16 deletions(-)

diff --git a/perplexity_test.go b/perplexity_test.go
index 1123561..d9d96ad 100644
--- a/perplexity_test.go
+++ b/perplexity_test.go
@@ -78,14 +78,6 @@ func TestGetCompletion(t *testing.T) {
 
 func TestSetModels(t *testing.T) {
 	r := perplexity.NewClient("apikey")
-	t.Run("set model llama-3.1-sonar-small-128k-chat", func(t *testing.T) {
-		r.SetModuleLlama31SonarSmall128kChat()
-		assert.Equal(t, r.GetModel(), "llama-3.1-sonar-small-128k-chat")
-	})
-	t.Run("set model llama-3.1-sonar-large-128k-chat", func(t *testing.T) {
-		r.SetModuleLlama31SonarLarge128kChat()
-		assert.Equal(t, r.GetModel(), "llama-3.1-sonar-large-128k-chat")
-	})
 	t.Run("set model llama-3.1-sonar-small-128k-online", func(t *testing.T) {
 		r.SetModuleLlama31SonarSmall128kOnline()
 		assert.Equal(t, r.GetModel(), "llama-3.1-sonar-small-128k-online")
@@ -94,14 +86,6 @@ func TestSetModels(t *testing.T) {
 		r.SetModuleLlama31SonarLarge128kOnline()
 		assert.Equal(t, r.GetModel(), "llama-3.1-sonar-large-128k-online")
 	})
-	t.Run("set model llama-3.1-8b-instruct", func(t *testing.T) {
-		r.SetModuleLlama318bInstruct()
-		assert.Equal(t, r.GetModel(), "llama-3.1-8b-instruct")
-	})
-	t.Run("set model llama-3.1-70b-instruct", func(t *testing.T) {
-		r.SetModuleLlama3170bInstruct()
-		assert.Equal(t, r.GetModel(), "llama-3.1-70b-instruct")
-	})
 }
 
 func TestGetLastContent(t *testing.T) {