diff --git a/edgee/__init__.py b/edgee/__init__.py index fd930a9..ad0bd25 100644 --- a/edgee/__init__.py +++ b/edgee/__init__.py @@ -8,7 +8,7 @@ from urllib.request import Request, urlopen # API Configuration -DEFAULT_BASE_URL = "https://api.edgee.ai" +DEFAULT_BASE_URL = "https://edgee.io" API_ENDPOINT = "/v1/chat/completions" diff --git a/pyproject.toml b/pyproject.toml index 0352521..9136867 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "edgee" -version = "1.0.4" +version = "1.0.5" description = "Lightweight Python SDK for Edgee AI Gateway" readme = "README.md" license = "Apache-2.0" diff --git a/tests/test_edgee.py b/tests/test_edgee.py index 5e1fd4c..6c6e2b8 100644 --- a/tests/test_edgee.py +++ b/tests/test_edgee.py @@ -102,7 +102,7 @@ def test_send_with_string_input(self, mock_urlopen): # Verify the request call_args = mock_urlopen.call_args[0][0] - assert call_args.full_url == "https://api.edgee.ai/v1/chat/completions" + assert call_args.full_url == "https://edgee.io/v1/chat/completions" body = json.loads(call_args.data.decode("utf-8")) assert body["model"] == "gpt-4" assert body["messages"] == [{"role": "user", "content": "Hello"}] diff --git a/uv.lock b/uv.lock index b66b44d..8bfa23b 100644 --- a/uv.lock +++ b/uv.lock @@ -22,7 +22,7 @@ wheels = [ [[package]] name = "edgee" -version = "1.0.4" +version = "1.0.5" source = { editable = "." } [package.optional-dependencies]