From a4191f689584694506aa2a5b9ff9c11a42fff4f6 Mon Sep 17 00:00:00 2001 From: mudler Date: Fri, 12 May 2023 12:26:55 +0200 Subject: [PATCH] tests: adapt to new added backends --- api/api_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/api_test.go b/api/api_test.go index bd52842..9fc14eb 100644 --- a/api/api_test.go +++ b/api/api_test.go @@ -46,7 +46,7 @@ var _ = Describe("API test", func() { It("returns the models list", func() { models, err := client.ListModels(context.TODO()) Expect(err).ToNot(HaveOccurred()) - Expect(len(models.Models)).To(Equal(4)) + Expect(len(models.Models)).To(Equal(5)) Expect(models.Models[0].ID).To(Equal("testmodel")) }) It("can generate completions", func() { @@ -119,7 +119,7 @@ var _ = Describe("API test", func() { models, err := client.ListModels(context.TODO()) Expect(err).ToNot(HaveOccurred()) - Expect(len(models.Models)).To(Equal(6)) + Expect(len(models.Models)).To(Equal(7)) Expect(models.Models[0].ID).To(Equal("testmodel")) }) It("can generate chat completions from config file", func() {