diff --git a/docs/examples/function_calling.ipynb b/docs/examples/function_calling.ipynb index c12eed7354f020e4d9435b801bf2cc9ad0a47070..a418d05f951ff0ac6f5abbff6dbce1907291737e 100644 --- a/docs/examples/function_calling.ipynb +++ b/docs/examples/function_calling.ipynb @@ -28,6 +28,7 @@ "import openai\n", "from semantic_router.utils.logger import logger\n", "\n", + "\n", "# Docs # https://platform.openai.com/docs/guides/function-calling\n", "def llm_openai(prompt: str, model: str = \"gpt-4\") -> str:\n", " try:\n", @@ -349,7 +350,6 @@ "metadata": {}, "outputs": [], "source": [ - "\n", "def get_time(location: str) -> str:\n", " \"\"\"Useful to get the time in a specific location\"\"\"\n", " print(f\"Calling `get_time` function with location: {location}\")\n", diff --git a/semantic_router/encoders/openai.py b/semantic_router/encoders/openai.py index 920fac208deba1a69201510bad901d26f7d5e15b..0b4aa94e0ed4f4067ab5ed462bf4b0ad3aeecf35 100644 --- a/semantic_router/encoders/openai.py +++ b/semantic_router/encoders/openai.py @@ -4,6 +4,7 @@ from time import sleep import openai from openai import OpenAIError from openai.types import CreateEmbeddingResponse + from semantic_router.encoders import BaseEncoder from semantic_router.utils.logger import logger diff --git a/tests/unit/encoders/test_openai.py b/tests/unit/encoders/test_openai.py index a1f458d13ccef3b2418efd1f15cbee12c7d5b91b..501a9b04f5d12c59e014edb6bc5798d2aa9d31b5 100644 --- a/tests/unit/encoders/test_openai.py +++ b/tests/unit/encoders/test_openai.py @@ -71,7 +71,7 @@ class TestOpenAIEncoder: ) with pytest.raises(ValueError) as e: openai_encoder(["test document"]) - + assert "OpenAI API call failed. Error: Non-OpenAIError" in str(e.value) def test_openai_encoder_call_successful_retry(self, openai_encoder, mocker):