From f2936780257e75ed3d974058ddbaa067ab0e40f2 Mon Sep 17 00:00:00 2001
From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local>
Date: Mon, 19 Aug 2024 16:05:03 +0500
Subject: [PATCH] fix tests
---
tests/integration/encoders/test_openai_integration.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/integration/encoders/test_openai_integration.py b/tests/integration/encoders/test_openai_integration.py
index a8b59281..47e617a5 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 openai import OpenAIError
from semantic_router.encoders.base import BaseEncoder
from semantic_router.encoders.openai import OpenAIEncoder
@@ -40,7 +41,7 @@ class TestOpenAIEncoder:
os.environ.get("OPENAI_API_KEY") is None, reason="OpenAI API key required"
)
def test_openai_encoder_call_no_truncation(self, openai_encoder):
- with pytest.raises(ValueError) as _:
+ with pytest.raises(OpenAIError) as _:
# default truncation is True
openai_encoder([long_doc], truncate=False)
--
GitLab