From 26070a3c09ef473e53ff8efb90f908b4f69a58fc Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Fri, 14 Jun 2024 17:16:52 +0800
Subject: [PATCH] fix: add condition to init encoder
---
tests/integration/encoders/test_openai_integration.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/tests/integration/encoders/test_openai_integration.py b/tests/integration/encoders/test_openai_integration.py
index 7c6cfe06..a8b59281 100644
--- a/tests/integration/encoders/test_openai_integration.py
+++ b/tests/integration/encoders/test_openai_integration.py
@@ -1,5 +1,6 @@
import os
import pytest
+from semantic_router.encoders.base import BaseEncoder
from semantic_router.encoders.openai import OpenAIEncoder
with open("tests/integration/57640.4032.txt", "r") as fp:
@@ -8,7 +9,10 @@ with open("tests/integration/57640.4032.txt", "r") as fp:
@pytest.fixture
def openai_encoder():
- return OpenAIEncoder()
+ if os.environ.get("OPENAI_API_KEY") is None:
+ return BaseEncoder()
+ else:
+ return OpenAIEncoder()
class TestOpenAIEncoder:
--
GitLab